mirror of
https://github.com/feschber/lan-mouse.git
synced 2026-03-11 07:10:54 +03:00
Compare commits
13 Commits
cleanup-ca
...
fix-key-re
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
196d01fe9e | ||
|
|
b071201dcb | ||
|
|
f52f19d2e3 | ||
|
|
39fed0344c | ||
|
|
6cd190191e | ||
|
|
be677d4c81 | ||
|
|
12bc0d86ca | ||
|
|
1f7a7309eb | ||
|
|
8926d8f803 | ||
|
|
9248007986 | ||
|
|
e7a1d72149 | ||
|
|
19c2c4327f | ||
|
|
fe06ca1fae |
96
Cargo.lock
generated
96
Cargo.lock
generated
@@ -1196,10 +1196,15 @@ version = "0.2.0"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"ashpd",
|
"ashpd",
|
||||||
"async-trait",
|
"async-trait",
|
||||||
|
"bitflags 2.6.0",
|
||||||
|
"core-foundation",
|
||||||
|
"core-foundation-sys",
|
||||||
"core-graphics",
|
"core-graphics",
|
||||||
"futures",
|
"futures",
|
||||||
"futures-core",
|
"futures-core",
|
||||||
"input-event",
|
"input-event",
|
||||||
|
"keycode",
|
||||||
|
"libc",
|
||||||
"log",
|
"log",
|
||||||
"memmap",
|
"memmap",
|
||||||
"once_cell",
|
"once_cell",
|
||||||
@@ -1308,21 +1313,19 @@ dependencies = [
|
|||||||
name = "lan-mouse"
|
name = "lan-mouse"
|
||||||
version = "0.9.1"
|
version = "0.9.1"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
|
||||||
"async-channel",
|
|
||||||
"clap",
|
"clap",
|
||||||
"endi",
|
|
||||||
"env_logger",
|
"env_logger",
|
||||||
"futures",
|
"futures",
|
||||||
"glib-build-tools",
|
|
||||||
"gtk4",
|
|
||||||
"hickory-resolver",
|
"hickory-resolver",
|
||||||
"hostname 0.4.0",
|
|
||||||
"input-capture",
|
"input-capture",
|
||||||
"input-emulation",
|
"input-emulation",
|
||||||
"input-event",
|
"input-event",
|
||||||
"libadwaita",
|
"lan-mouse-cli",
|
||||||
|
"lan-mouse-gtk",
|
||||||
|
"lan-mouse-ipc",
|
||||||
|
"lan-mouse-proto",
|
||||||
"libc",
|
"libc",
|
||||||
|
"local-channel",
|
||||||
"log",
|
"log",
|
||||||
"serde",
|
"serde",
|
||||||
"serde_json",
|
"serde_json",
|
||||||
@@ -1333,6 +1336,51 @@ dependencies = [
|
|||||||
"toml",
|
"toml",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "lan-mouse-cli"
|
||||||
|
version = "0.1.0"
|
||||||
|
dependencies = [
|
||||||
|
"futures",
|
||||||
|
"lan-mouse-ipc",
|
||||||
|
"tokio",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "lan-mouse-gtk"
|
||||||
|
version = "0.1.0"
|
||||||
|
dependencies = [
|
||||||
|
"async-channel",
|
||||||
|
"glib-build-tools",
|
||||||
|
"gtk4",
|
||||||
|
"hostname 0.4.0",
|
||||||
|
"lan-mouse-ipc",
|
||||||
|
"libadwaita",
|
||||||
|
"log",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "lan-mouse-ipc"
|
||||||
|
version = "0.1.0"
|
||||||
|
dependencies = [
|
||||||
|
"futures",
|
||||||
|
"log",
|
||||||
|
"serde",
|
||||||
|
"serde_json",
|
||||||
|
"thiserror",
|
||||||
|
"tokio",
|
||||||
|
"tokio-stream",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "lan-mouse-proto"
|
||||||
|
version = "0.1.0"
|
||||||
|
dependencies = [
|
||||||
|
"input-event",
|
||||||
|
"num_enum",
|
||||||
|
"paste",
|
||||||
|
"thiserror",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "libadwaita"
|
name = "libadwaita"
|
||||||
version = "0.7.0"
|
version = "0.7.0"
|
||||||
@@ -1392,6 +1440,23 @@ version = "0.4.14"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "78b3ae25bc7c8c38cec158d1f2757ee79e9b3740fbc7ccf0e59e4b08d793fa89"
|
checksum = "78b3ae25bc7c8c38cec158d1f2757ee79e9b3740fbc7ccf0e59e4b08d793fa89"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "local-channel"
|
||||||
|
version = "0.1.5"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "b6cbc85e69b8df4b8bb8b89ec634e7189099cea8927a276b7384ce5488e53ec8"
|
||||||
|
dependencies = [
|
||||||
|
"futures-core",
|
||||||
|
"futures-sink",
|
||||||
|
"local-waker",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "local-waker"
|
||||||
|
version = "0.1.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "4d873d7c67ce09b42110d801813efbc9364414e356be9935700d368351657487"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "lock_api"
|
name = "lock_api"
|
||||||
version = "0.4.12"
|
version = "0.4.12"
|
||||||
@@ -1596,6 +1661,12 @@ dependencies = [
|
|||||||
"windows-targets 0.52.6",
|
"windows-targets 0.52.6",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "paste"
|
||||||
|
version = "1.0.15"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "57c0d7b74b563b49d38dae00a0c37d4d6de9b432382b2892f0574ddcae73fd0a"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "percent-encoding"
|
name = "percent-encoding"
|
||||||
version = "2.3.1"
|
version = "2.3.1"
|
||||||
@@ -2050,6 +2121,17 @@ dependencies = [
|
|||||||
"syn",
|
"syn",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "tokio-stream"
|
||||||
|
version = "0.1.15"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "267ac89e0bec6e691e5813911606935d77c476ff49024f98abcea3e7b15e37af"
|
||||||
|
dependencies = [
|
||||||
|
"futures-core",
|
||||||
|
"pin-project-lite",
|
||||||
|
"tokio",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "tokio-util"
|
name = "tokio-util"
|
||||||
version = "0.7.11"
|
version = "0.7.11"
|
||||||
|
|||||||
30
Cargo.toml
30
Cargo.toml
@@ -1,5 +1,13 @@
|
|||||||
[workspace]
|
[workspace]
|
||||||
members = ["input-capture", "input-emulation", "input-event"]
|
members = [
|
||||||
|
"input-capture",
|
||||||
|
"input-emulation",
|
||||||
|
"input-event",
|
||||||
|
"lan-mouse-ipc",
|
||||||
|
"lan-mouse-cli",
|
||||||
|
"lan-mouse-gtk",
|
||||||
|
"lan-mouse-proto",
|
||||||
|
]
|
||||||
|
|
||||||
[package]
|
[package]
|
||||||
name = "lan-mouse"
|
name = "lan-mouse"
|
||||||
@@ -17,11 +25,14 @@ lto = "fat"
|
|||||||
input-event = { path = "input-event", version = "0.2.1" }
|
input-event = { path = "input-event", version = "0.2.1" }
|
||||||
input-emulation = { path = "input-emulation", version = "0.2.1", default-features = false }
|
input-emulation = { path = "input-emulation", version = "0.2.1", default-features = false }
|
||||||
input-capture = { path = "input-capture", version = "0.2.0", default-features = false }
|
input-capture = { path = "input-capture", version = "0.2.0", default-features = false }
|
||||||
|
lan-mouse-cli = { path = "lan-mouse-cli", version = "0.1.0" }
|
||||||
|
lan-mouse-gtk = { path = "lan-mouse-gtk", version = "0.1.0", optional = true }
|
||||||
|
lan-mouse-ipc = { path = "lan-mouse-ipc", version = "0.1.0" }
|
||||||
|
lan-mouse-proto = { path = "lan-mouse-proto", version = "0.1.0" }
|
||||||
|
|
||||||
hickory-resolver = "0.24.1"
|
hickory-resolver = "0.24.1"
|
||||||
toml = "0.8"
|
toml = "0.8"
|
||||||
serde = { version = "1.0", features = ["derive"] }
|
serde = { version = "1.0", features = ["derive"] }
|
||||||
anyhow = "1.0.71"
|
|
||||||
log = "0.4.20"
|
log = "0.4.20"
|
||||||
env_logger = "0.11.3"
|
env_logger = "0.11.3"
|
||||||
serde_json = "1.0.107"
|
serde_json = "1.0.107"
|
||||||
@@ -37,29 +48,18 @@ tokio = { version = "1.32.0", features = [
|
|||||||
] }
|
] }
|
||||||
futures = "0.3.28"
|
futures = "0.3.28"
|
||||||
clap = { version = "4.4.11", features = ["derive"] }
|
clap = { version = "4.4.11", features = ["derive"] }
|
||||||
gtk = { package = "gtk4", version = "0.9.0", features = [
|
|
||||||
"v4_2",
|
|
||||||
], optional = true }
|
|
||||||
adw = { package = "libadwaita", version = "0.7.0", features = [
|
|
||||||
"v1_1",
|
|
||||||
], optional = true }
|
|
||||||
async-channel = { version = "2.1.1", optional = true }
|
|
||||||
hostname = "0.4.0"
|
|
||||||
slab = "0.4.9"
|
slab = "0.4.9"
|
||||||
endi = "1.1.0"
|
|
||||||
thiserror = "1.0.61"
|
thiserror = "1.0.61"
|
||||||
tokio-util = "0.7.11"
|
tokio-util = "0.7.11"
|
||||||
|
local-channel = "0.1.5"
|
||||||
|
|
||||||
[target.'cfg(unix)'.dependencies]
|
[target.'cfg(unix)'.dependencies]
|
||||||
libc = "0.2.148"
|
libc = "0.2.148"
|
||||||
|
|
||||||
[build-dependencies]
|
|
||||||
glib-build-tools = { version = "0.20.0", optional = true }
|
|
||||||
|
|
||||||
[features]
|
[features]
|
||||||
default = ["wayland", "x11", "xdg_desktop_portal", "libei", "gtk"]
|
default = ["wayland", "x11", "xdg_desktop_portal", "libei", "gtk"]
|
||||||
wayland = ["input-capture/wayland", "input-emulation/wayland"]
|
wayland = ["input-capture/wayland", "input-emulation/wayland"]
|
||||||
x11 = ["input-capture/x11", "input-emulation/x11"]
|
x11 = ["input-capture/x11", "input-emulation/x11"]
|
||||||
xdg_desktop_portal = ["input-emulation/xdg_desktop_portal"]
|
xdg_desktop_portal = ["input-emulation/xdg_desktop_portal"]
|
||||||
libei = ["input-event/libei", "input-capture/libei", "input-emulation/libei"]
|
libei = ["input-event/libei", "input-capture/libei", "input-emulation/libei"]
|
||||||
gtk = ["dep:gtk", "dep:adw", "dep:async-channel", "dep:glib-build-tools"]
|
gtk = ["dep:lan-mouse-gtk"]
|
||||||
|
|||||||
@@ -20,7 +20,7 @@ Focus lies on performance and a clean, manageable implementation that can easily
|
|||||||
|
|
||||||
***blazingly fast™*** because it's written in rust.
|
***blazingly fast™*** because it's written in rust.
|
||||||
|
|
||||||
For an alternative (with slightly different goals) you may check out [Input Leap](https://github.com/input-leap).
|
For an alternative (with slightly different goals) you may check out [Synergy 1 Community Edition](https://github.com/symless/synergy) or [Input Leap](https://github.com/input-leap) (Synergy fork).
|
||||||
|
|
||||||
|
|
||||||
> [!WARNING]
|
> [!WARNING]
|
||||||
@@ -116,7 +116,7 @@ sudo cp target/release/lan-mouse /usr/local/bin/
|
|||||||
|
|
||||||
# install app icon
|
# install app icon
|
||||||
sudo mkdir -p /usr/local/share/icons/hicolor/scalable/apps
|
sudo mkdir -p /usr/local/share/icons/hicolor/scalable/apps
|
||||||
sudo cp resources/de.feschber.LanMouse.svg /usr/local/share/icons/hicolor/scalable/apps
|
sudo cp lan-mouse-gtk/resources/de.feschber.LanMouse.svg /usr/local/share/icons/hicolor/scalable/apps
|
||||||
|
|
||||||
# update icon cache
|
# update icon cache
|
||||||
gtk-update-icon-cache /usr/local/share/icons/hicolor/
|
gtk-update-icon-cache /usr/local/share/icons/hicolor/
|
||||||
|
|||||||
20
build.rs
20
build.rs
@@ -1,9 +1,15 @@
|
|||||||
|
use std::process::Command;
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
// composite_templates
|
// commit hash
|
||||||
#[cfg(feature = "gtk")]
|
let git_describe = Command::new("git")
|
||||||
glib_build_tools::compile_resources(
|
.arg("describe")
|
||||||
&["resources"],
|
.arg("--always")
|
||||||
"resources/resources.gresource.xml",
|
.arg("--dirty")
|
||||||
"lan-mouse.gresource",
|
.arg("--tags")
|
||||||
);
|
.output()
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let git_describe = String::from_utf8(git_describe.stdout).unwrap();
|
||||||
|
println!("cargo::rustc-env=GIT_DESCRIBE={git_describe}");
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -53,9 +53,11 @@
|
|||||||
libadwaita
|
libadwaita
|
||||||
librsvg
|
librsvg
|
||||||
xorg.libXtst
|
xorg.libXtst
|
||||||
] ++ lib.optionals stdenv.isDarwin [
|
] ++ lib.optionals stdenv.isDarwin
|
||||||
darwin.apple_sdk_11_0.frameworks.CoreGraphics
|
(with darwin.apple_sdk_11_0.frameworks; [
|
||||||
];
|
CoreGraphics
|
||||||
|
ApplicationServices
|
||||||
|
]);
|
||||||
|
|
||||||
RUST_SRC_PATH = "${rust}/lib/rustlib/src/rust/library";
|
RUST_SRC_PATH = "${rust}/lib/rustlib/src/rust/library";
|
||||||
};
|
};
|
||||||
|
|||||||
@@ -47,6 +47,11 @@ reis = { version = "0.2", features = ["tokio"], optional = true }
|
|||||||
|
|
||||||
[target.'cfg(target_os="macos")'.dependencies]
|
[target.'cfg(target_os="macos")'.dependencies]
|
||||||
core-graphics = { version = "0.23", features = ["highsierra"] }
|
core-graphics = { version = "0.23", features = ["highsierra"] }
|
||||||
|
core-foundation = "0.9.4"
|
||||||
|
core-foundation-sys = "0.8.6"
|
||||||
|
libc = "0.2.155"
|
||||||
|
keycode = "0.4.0"
|
||||||
|
bitflags = "2.5.0"
|
||||||
|
|
||||||
[target.'cfg(windows)'.dependencies]
|
[target.'cfg(windows)'.dependencies]
|
||||||
windows = { version = "0.58.0", features = [
|
windows = { version = "0.58.0", features = [
|
||||||
|
|||||||
@@ -1,20 +1,28 @@
|
|||||||
|
use std::f64::consts::PI;
|
||||||
use std::pin::Pin;
|
use std::pin::Pin;
|
||||||
use std::task::{Context, Poll};
|
use std::task::{ready, Context, Poll};
|
||||||
|
use std::time::Duration;
|
||||||
|
|
||||||
use async_trait::async_trait;
|
use async_trait::async_trait;
|
||||||
use futures_core::Stream;
|
use futures_core::Stream;
|
||||||
|
use input_event::PointerEvent;
|
||||||
|
use tokio::time::{self, Instant, Interval};
|
||||||
|
|
||||||
use input_event::Event;
|
use super::{Capture, CaptureError, CaptureEvent, CaptureHandle, Position};
|
||||||
|
|
||||||
use crate::CaptureError;
|
pub struct DummyInputCapture {
|
||||||
|
start: Option<Instant>,
|
||||||
use super::{Capture, CaptureHandle, Position};
|
interval: Interval,
|
||||||
|
offset: (i32, i32),
|
||||||
pub struct DummyInputCapture {}
|
}
|
||||||
|
|
||||||
impl DummyInputCapture {
|
impl DummyInputCapture {
|
||||||
pub fn new() -> Self {
|
pub fn new() -> Self {
|
||||||
Self {}
|
Self {
|
||||||
|
start: None,
|
||||||
|
interval: time::interval(Duration::from_millis(1)),
|
||||||
|
offset: (0, 0),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -43,10 +51,36 @@ impl Capture for DummyInputCapture {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Stream for DummyInputCapture {
|
const FREQUENCY_HZ: f64 = 1.0;
|
||||||
type Item = Result<(CaptureHandle, Event), CaptureError>;
|
const RADIUS: f64 = 100.0;
|
||||||
|
|
||||||
fn poll_next(self: Pin<&mut Self>, _cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
|
impl Stream for DummyInputCapture {
|
||||||
Poll::Pending
|
type Item = Result<(CaptureHandle, CaptureEvent), CaptureError>;
|
||||||
|
|
||||||
|
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
|
||||||
|
let current = ready!(self.interval.poll_tick(cx));
|
||||||
|
let event = match self.start {
|
||||||
|
None => {
|
||||||
|
self.start.replace(current);
|
||||||
|
CaptureEvent::Begin
|
||||||
|
}
|
||||||
|
Some(start) => {
|
||||||
|
let elapsed = start.elapsed();
|
||||||
|
let elapsed_sec_f64 = elapsed.as_secs_f64();
|
||||||
|
let second_fraction = elapsed_sec_f64 - elapsed_sec_f64 as u64 as f64;
|
||||||
|
let radians = second_fraction * 2. * PI * FREQUENCY_HZ;
|
||||||
|
let offset = (radians.cos() * RADIUS * 2., (radians * 2.).sin() * RADIUS);
|
||||||
|
let offset = (offset.0 as i32, offset.1 as i32);
|
||||||
|
let relative_motion = (offset.0 - self.offset.0, offset.1 - self.offset.1);
|
||||||
|
self.offset = offset;
|
||||||
|
let (dx, dy) = (relative_motion.0 as f64, relative_motion.1 as f64);
|
||||||
|
CaptureEvent::Input(input_event::Event::Pointer(PointerEvent::Motion {
|
||||||
|
time: 0,
|
||||||
|
dx,
|
||||||
|
dy,
|
||||||
|
}))
|
||||||
|
}
|
||||||
|
};
|
||||||
|
Poll::Ready(Some(Ok((0, event))))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -22,6 +22,9 @@ use ashpd::desktop::ResponseError;
|
|||||||
#[cfg(all(unix, feature = "libei", not(target_os = "macos")))]
|
#[cfg(all(unix, feature = "libei", not(target_os = "macos")))]
|
||||||
use reis::tokio::{EiConvertEventStreamError, HandshakeError};
|
use reis::tokio::{EiConvertEventStreamError, HandshakeError};
|
||||||
|
|
||||||
|
#[cfg(target_os = "macos")]
|
||||||
|
use core_graphics::base::CGError;
|
||||||
|
|
||||||
#[cfg(all(unix, feature = "libei", not(target_os = "macos")))]
|
#[cfg(all(unix, feature = "libei", not(target_os = "macos")))]
|
||||||
#[derive(Debug, Error)]
|
#[derive(Debug, Error)]
|
||||||
#[error("error in libei stream: {inner:?}")]
|
#[error("error in libei stream: {inner:?}")]
|
||||||
@@ -56,6 +59,21 @@ pub enum CaptureError {
|
|||||||
#[cfg(all(unix, feature = "libei", not(target_os = "macos")))]
|
#[cfg(all(unix, feature = "libei", not(target_os = "macos")))]
|
||||||
#[error("libei disconnected - reason: `{0}`")]
|
#[error("libei disconnected - reason: `{0}`")]
|
||||||
Disconnected(String),
|
Disconnected(String),
|
||||||
|
#[cfg(target_os = "macos")]
|
||||||
|
#[error("failed to warp mouse cursor: `{0}`")]
|
||||||
|
WarpCursor(CGError),
|
||||||
|
#[cfg(target_os = "macos")]
|
||||||
|
#[error("reset_mouse_position called without a connected client")]
|
||||||
|
ResetMouseWithoutClient,
|
||||||
|
#[cfg(target_os = "macos")]
|
||||||
|
#[error("core-graphics error: {0}")]
|
||||||
|
CoreGraphics(CGError),
|
||||||
|
#[cfg(target_os = "macos")]
|
||||||
|
#[error("unable to map key event: {0}")]
|
||||||
|
KeyMapError(i64),
|
||||||
|
#[cfg(target_os = "macos")]
|
||||||
|
#[error("Event tap disabled")]
|
||||||
|
EventTapDisabled,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Error)]
|
#[derive(Debug, Error)]
|
||||||
@@ -71,12 +89,12 @@ pub enum CaptureCreationError {
|
|||||||
#[cfg(all(unix, feature = "x11", not(target_os = "macos")))]
|
#[cfg(all(unix, feature = "x11", not(target_os = "macos")))]
|
||||||
#[error("error creating x11 capture backend: `{0}`")]
|
#[error("error creating x11 capture backend: `{0}`")]
|
||||||
X11(#[from] X11InputCaptureCreationError),
|
X11(#[from] X11InputCaptureCreationError),
|
||||||
#[cfg(target_os = "macos")]
|
|
||||||
#[error("error creating macos capture backend: `{0}`")]
|
|
||||||
Macos(#[from] MacOSInputCaptureCreationError),
|
|
||||||
#[cfg(windows)]
|
#[cfg(windows)]
|
||||||
#[error("error creating windows capture backend")]
|
#[error("error creating windows capture backend")]
|
||||||
Windows,
|
Windows,
|
||||||
|
#[cfg(target_os = "macos")]
|
||||||
|
#[error("error creating macos capture backend")]
|
||||||
|
MacOS(#[from] MacosCaptureCreationError),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl CaptureCreationError {
|
impl CaptureCreationError {
|
||||||
@@ -144,7 +162,12 @@ pub enum X11InputCaptureCreationError {
|
|||||||
|
|
||||||
#[cfg(target_os = "macos")]
|
#[cfg(target_os = "macos")]
|
||||||
#[derive(Debug, Error)]
|
#[derive(Debug, Error)]
|
||||||
pub enum MacOSInputCaptureCreationError {
|
pub enum MacosCaptureCreationError {
|
||||||
#[error("MacOS input capture is not yet implemented :(")]
|
#[error("event source creation failed!")]
|
||||||
NotImplemented,
|
EventSourceCreation,
|
||||||
|
#[error("failed to set CG Cursor property")]
|
||||||
|
CGCursorProperty,
|
||||||
|
#[cfg(target_os = "macos")]
|
||||||
|
#[error("failed to get display ids: {0}")]
|
||||||
|
ActiveDisplays(CGError),
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -30,6 +30,23 @@ mod dummy;
|
|||||||
|
|
||||||
pub type CaptureHandle = u64;
|
pub type CaptureHandle = u64;
|
||||||
|
|
||||||
|
#[derive(Copy, Clone, Debug, PartialEq)]
|
||||||
|
pub enum CaptureEvent {
|
||||||
|
/// capture on this capture handle is now active
|
||||||
|
Begin,
|
||||||
|
/// input event coming from capture handle
|
||||||
|
Input(Event),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Display for CaptureEvent {
|
||||||
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
|
match self {
|
||||||
|
CaptureEvent::Begin => write!(f, "begin capture"),
|
||||||
|
CaptureEvent::Input(e) => write!(f, "{e}"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Copy, Eq, Hash, PartialEq)]
|
#[derive(Debug, Clone, Copy, Eq, Hash, PartialEq)]
|
||||||
pub enum Position {
|
pub enum Position {
|
||||||
Left,
|
Left,
|
||||||
@@ -147,7 +164,7 @@ impl InputCapture {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl Stream for InputCapture {
|
impl Stream for InputCapture {
|
||||||
type Item = Result<(CaptureHandle, Event), CaptureError>;
|
type Item = Result<(CaptureHandle, CaptureEvent), CaptureError>;
|
||||||
|
|
||||||
fn poll_next(
|
fn poll_next(
|
||||||
mut self: std::pin::Pin<&mut Self>,
|
mut self: std::pin::Pin<&mut Self>,
|
||||||
@@ -155,7 +172,11 @@ impl Stream for InputCapture {
|
|||||||
) -> Poll<Option<Self::Item>> {
|
) -> Poll<Option<Self::Item>> {
|
||||||
match self.capture.poll_next_unpin(cx) {
|
match self.capture.poll_next_unpin(cx) {
|
||||||
Poll::Ready(e) => {
|
Poll::Ready(e) => {
|
||||||
if let Some(Ok((_, Event::Keyboard(KeyboardEvent::Key { key, state, .. })))) = e {
|
if let Some(Ok((
|
||||||
|
_,
|
||||||
|
CaptureEvent::Input(Event::Keyboard(KeyboardEvent::Key { key, state, .. })),
|
||||||
|
))) = e
|
||||||
|
{
|
||||||
self.update_pressed_keys(key, state);
|
self.update_pressed_keys(key, state);
|
||||||
}
|
}
|
||||||
Poll::Ready(e)
|
Poll::Ready(e)
|
||||||
@@ -166,7 +187,7 @@ impl Stream for InputCapture {
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[async_trait]
|
#[async_trait]
|
||||||
trait Capture: Stream<Item = Result<(CaptureHandle, Event), CaptureError>> + Unpin {
|
trait Capture: Stream<Item = Result<(CaptureHandle, CaptureEvent), CaptureError>> + Unpin {
|
||||||
/// create a new client with the given id
|
/// create a new client with the given id
|
||||||
async fn create(&mut self, id: CaptureHandle, pos: Position) -> Result<(), CaptureError>;
|
async fn create(&mut self, id: CaptureHandle, pos: Position) -> Result<(), CaptureError>;
|
||||||
|
|
||||||
@@ -183,7 +204,7 @@ trait Capture: Stream<Item = Result<(CaptureHandle, Event), CaptureError>> + Unp
|
|||||||
async fn create_backend(
|
async fn create_backend(
|
||||||
backend: Backend,
|
backend: Backend,
|
||||||
) -> Result<
|
) -> Result<
|
||||||
Box<dyn Capture<Item = Result<(CaptureHandle, Event), CaptureError>>>,
|
Box<dyn Capture<Item = Result<(CaptureHandle, CaptureEvent), CaptureError>>>,
|
||||||
CaptureCreationError,
|
CaptureCreationError,
|
||||||
> {
|
> {
|
||||||
match backend {
|
match backend {
|
||||||
@@ -196,7 +217,7 @@ async fn create_backend(
|
|||||||
#[cfg(windows)]
|
#[cfg(windows)]
|
||||||
Backend::Windows => Ok(Box::new(windows::WindowsInputCapture::new())),
|
Backend::Windows => Ok(Box::new(windows::WindowsInputCapture::new())),
|
||||||
#[cfg(target_os = "macos")]
|
#[cfg(target_os = "macos")]
|
||||||
Backend::MacOs => Ok(Box::new(macos::MacOSInputCapture::new()?)),
|
Backend::MacOs => Ok(Box::new(macos::MacOSInputCapture::new().await?)),
|
||||||
Backend::Dummy => Ok(Box::new(dummy::DummyInputCapture::new())),
|
Backend::Dummy => Ok(Box::new(dummy::DummyInputCapture::new())),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -204,7 +225,7 @@ async fn create_backend(
|
|||||||
async fn create(
|
async fn create(
|
||||||
backend: Option<Backend>,
|
backend: Option<Backend>,
|
||||||
) -> Result<
|
) -> Result<
|
||||||
Box<dyn Capture<Item = Result<(CaptureHandle, Event), CaptureError>>>,
|
Box<dyn Capture<Item = Result<(CaptureHandle, CaptureEvent), CaptureError>>>,
|
||||||
CaptureCreationError,
|
CaptureCreationError,
|
||||||
> {
|
> {
|
||||||
if let Some(backend) = backend {
|
if let Some(backend) = backend {
|
||||||
|
|||||||
@@ -36,6 +36,8 @@ use once_cell::sync::Lazy;
|
|||||||
|
|
||||||
use input_event::Event;
|
use input_event::Event;
|
||||||
|
|
||||||
|
use crate::CaptureEvent;
|
||||||
|
|
||||||
use super::{
|
use super::{
|
||||||
error::{CaptureError, LibeiCaptureCreationError, ReisConvertEventStreamError},
|
error::{CaptureError, LibeiCaptureCreationError, ReisConvertEventStreamError},
|
||||||
Capture as LanMouseInputCapture, CaptureHandle, Position,
|
Capture as LanMouseInputCapture, CaptureHandle, Position,
|
||||||
@@ -56,7 +58,7 @@ enum LibeiNotifyEvent {
|
|||||||
pub struct LibeiInputCapture<'a> {
|
pub struct LibeiInputCapture<'a> {
|
||||||
input_capture: Pin<Box<InputCapture<'a>>>,
|
input_capture: Pin<Box<InputCapture<'a>>>,
|
||||||
capture_task: JoinHandle<Result<(), CaptureError>>,
|
capture_task: JoinHandle<Result<(), CaptureError>>,
|
||||||
event_rx: Receiver<(CaptureHandle, Event)>,
|
event_rx: Receiver<(CaptureHandle, CaptureEvent)>,
|
||||||
notify_capture: Sender<LibeiNotifyEvent>,
|
notify_capture: Sender<LibeiNotifyEvent>,
|
||||||
notify_release: Arc<Notify>,
|
notify_release: Arc<Notify>,
|
||||||
cancellation_token: CancellationToken,
|
cancellation_token: CancellationToken,
|
||||||
@@ -201,7 +203,7 @@ async fn connect_to_eis(
|
|||||||
async fn libei_event_handler(
|
async fn libei_event_handler(
|
||||||
mut ei_event_stream: EiConvertEventStream,
|
mut ei_event_stream: EiConvertEventStream,
|
||||||
context: ei::Context,
|
context: ei::Context,
|
||||||
event_tx: Sender<(CaptureHandle, Event)>,
|
event_tx: Sender<(CaptureHandle, CaptureEvent)>,
|
||||||
release_session: Arc<Notify>,
|
release_session: Arc<Notify>,
|
||||||
current_client: Rc<Cell<Option<CaptureHandle>>>,
|
current_client: Rc<Cell<Option<CaptureHandle>>>,
|
||||||
) -> Result<(), CaptureError> {
|
) -> Result<(), CaptureError> {
|
||||||
@@ -258,7 +260,7 @@ async fn do_capture(
|
|||||||
mut capture_event: Receiver<LibeiNotifyEvent>,
|
mut capture_event: Receiver<LibeiNotifyEvent>,
|
||||||
notify_release: Arc<Notify>,
|
notify_release: Arc<Notify>,
|
||||||
session: Option<(Session<'_, InputCapture<'_>>, BitFlags<Capabilities>)>,
|
session: Option<(Session<'_, InputCapture<'_>>, BitFlags<Capabilities>)>,
|
||||||
event_tx: Sender<(CaptureHandle, Event)>,
|
event_tx: Sender<(CaptureHandle, CaptureEvent)>,
|
||||||
cancellation_token: CancellationToken,
|
cancellation_token: CancellationToken,
|
||||||
) -> Result<(), CaptureError> {
|
) -> Result<(), CaptureError> {
|
||||||
let mut session = session.map(|s| s.0);
|
let mut session = session.map(|s| s.0);
|
||||||
@@ -354,7 +356,7 @@ async fn do_capture(
|
|||||||
async fn do_capture_session(
|
async fn do_capture_session(
|
||||||
input_capture: &InputCapture<'_>,
|
input_capture: &InputCapture<'_>,
|
||||||
session: &mut Session<'_, InputCapture<'_>>,
|
session: &mut Session<'_, InputCapture<'_>>,
|
||||||
event_tx: &Sender<(CaptureHandle, Event)>,
|
event_tx: &Sender<(CaptureHandle, CaptureEvent)>,
|
||||||
active_clients: &[(CaptureHandle, Position)],
|
active_clients: &[(CaptureHandle, Position)],
|
||||||
next_barrier_id: &mut u32,
|
next_barrier_id: &mut u32,
|
||||||
notify_release: &Notify,
|
notify_release: &Notify,
|
||||||
@@ -423,7 +425,7 @@ async fn do_capture_session(
|
|||||||
current_client.replace(Some(client));
|
current_client.replace(Some(client));
|
||||||
|
|
||||||
// client entered => send event
|
// client entered => send event
|
||||||
event_tx.send((client, Event::Enter())).await.expect("no channel");
|
event_tx.send((client, CaptureEvent::Begin)).await.expect("no channel");
|
||||||
|
|
||||||
tokio::select! {
|
tokio::select! {
|
||||||
_ = notify_release.notified() => { /* capture release */
|
_ = notify_release.notified() => { /* capture release */
|
||||||
@@ -554,7 +556,7 @@ async fn handle_ei_event(
|
|||||||
ei_event: EiEvent,
|
ei_event: EiEvent,
|
||||||
current_client: Option<CaptureHandle>,
|
current_client: Option<CaptureHandle>,
|
||||||
context: &ei::Context,
|
context: &ei::Context,
|
||||||
event_tx: &Sender<(CaptureHandle, Event)>,
|
event_tx: &Sender<(CaptureHandle, CaptureEvent)>,
|
||||||
release_session: &Notify,
|
release_session: &Notify,
|
||||||
) -> Result<(), CaptureError> {
|
) -> Result<(), CaptureError> {
|
||||||
match ei_event {
|
match ei_event {
|
||||||
@@ -575,7 +577,7 @@ async fn handle_ei_event(
|
|||||||
_ => {
|
_ => {
|
||||||
if let Some(handle) = current_client {
|
if let Some(handle) = current_client {
|
||||||
for event in Event::from_ei_event(ei_event) {
|
for event in Event::from_ei_event(ei_event) {
|
||||||
event_tx.send((handle, event)).await.expect("no channel");
|
event_tx.send((handle, CaptureEvent::Input(event))).await.expect("no channel");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -627,7 +629,7 @@ impl<'a> Drop for LibeiInputCapture<'a> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl<'a> Stream for LibeiInputCapture<'a> {
|
impl<'a> Stream for LibeiInputCapture<'a> {
|
||||||
type Item = Result<(CaptureHandle, Event), CaptureError>;
|
type Item = Result<(CaptureHandle, CaptureEvent), CaptureError>;
|
||||||
|
|
||||||
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
|
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
|
||||||
match self.capture_task.poll_unpin(cx) {
|
match self.capture_task.poll_unpin(cx) {
|
||||||
|
|||||||
@@ -1,39 +1,522 @@
|
|||||||
use crate::{
|
use super::{
|
||||||
error::MacOSInputCaptureCreationError, Capture, CaptureError, CaptureHandle, Position,
|
error::MacosCaptureCreationError, Capture, CaptureError, CaptureEvent, CaptureHandle, Position,
|
||||||
};
|
};
|
||||||
use async_trait::async_trait;
|
use async_trait::async_trait;
|
||||||
|
use bitflags::bitflags;
|
||||||
|
use core_foundation::base::{kCFAllocatorDefault, CFRelease};
|
||||||
|
use core_foundation::date::CFTimeInterval;
|
||||||
|
use core_foundation::number::{kCFBooleanTrue, CFBooleanRef};
|
||||||
|
use core_foundation::runloop::{kCFRunLoopCommonModes, CFRunLoop, CFRunLoopSource};
|
||||||
|
use core_foundation::string::{kCFStringEncodingUTF8, CFStringCreateWithCString, CFStringRef};
|
||||||
|
use core_graphics::base::{kCGErrorSuccess, CGError};
|
||||||
|
use core_graphics::display::{CGDisplay, CGPoint};
|
||||||
|
use core_graphics::event::{
|
||||||
|
CGEvent, CGEventFlags, CGEventTap, CGEventTapLocation, CGEventTapOptions, CGEventTapPlacement,
|
||||||
|
CGEventTapProxy, CGEventType, EventField,
|
||||||
|
};
|
||||||
|
use core_graphics::event_source::{CGEventSource, CGEventSourceStateID};
|
||||||
use futures_core::Stream;
|
use futures_core::Stream;
|
||||||
use input_event::Event;
|
use input_event::{Event, KeyboardEvent, PointerEvent, BTN_LEFT, BTN_MIDDLE, BTN_RIGHT};
|
||||||
|
use keycode::{KeyMap, KeyMapping};
|
||||||
|
use libc::c_void;
|
||||||
|
use once_cell::unsync::Lazy;
|
||||||
|
use std::collections::HashMap;
|
||||||
|
use std::ffi::{c_char, CString};
|
||||||
use std::pin::Pin;
|
use std::pin::Pin;
|
||||||
use std::task::{Context, Poll};
|
use std::sync::Arc;
|
||||||
|
use std::task::{ready, Context, Poll};
|
||||||
|
use std::thread::{self};
|
||||||
|
use tokio::sync::mpsc::{Receiver, Sender};
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
pub struct MacOSInputCapture;
|
#[derive(Debug, Default)]
|
||||||
|
struct Bounds {
|
||||||
|
xmin: f64,
|
||||||
|
xmax: f64,
|
||||||
|
ymin: f64,
|
||||||
|
ymax: f64,
|
||||||
|
}
|
||||||
|
|
||||||
impl MacOSInputCapture {
|
#[derive(Debug)]
|
||||||
pub fn new() -> std::result::Result<Self, MacOSInputCaptureCreationError> {
|
struct InputCaptureState {
|
||||||
Err(MacOSInputCaptureCreationError::NotImplemented)
|
client_for_pos: Lazy<HashMap<Position, CaptureHandle>>,
|
||||||
|
current_client: Option<(CaptureHandle, Position)>,
|
||||||
|
bounds: Bounds,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
enum ProducerEvent {
|
||||||
|
Release,
|
||||||
|
Create(CaptureHandle, Position),
|
||||||
|
Destroy(CaptureHandle),
|
||||||
|
Grab((CaptureHandle, Position)),
|
||||||
|
EventTapDisabled,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl InputCaptureState {
|
||||||
|
fn new() -> Result<Self, MacosCaptureCreationError> {
|
||||||
|
let mut res = Self {
|
||||||
|
client_for_pos: Lazy::new(HashMap::new),
|
||||||
|
current_client: None,
|
||||||
|
bounds: Bounds::default(),
|
||||||
|
};
|
||||||
|
res.update_bounds()?;
|
||||||
|
Ok(res)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn crossed(&mut self, event: &CGEvent) -> Option<(CaptureHandle, Position)> {
|
||||||
|
let location = event.location();
|
||||||
|
let relative_x = event.get_double_value_field(EventField::MOUSE_EVENT_DELTA_X);
|
||||||
|
let relative_y = event.get_double_value_field(EventField::MOUSE_EVENT_DELTA_Y);
|
||||||
|
|
||||||
|
for (position, client) in self.client_for_pos.iter() {
|
||||||
|
if (position == &Position::Left && (location.x + relative_x) <= self.bounds.xmin)
|
||||||
|
|| (position == &Position::Right && (location.x + relative_x) >= self.bounds.xmax)
|
||||||
|
|| (position == &Position::Top && (location.y + relative_y) <= self.bounds.ymin)
|
||||||
|
|| (position == &Position::Bottom && (location.y + relative_y) >= self.bounds.ymax)
|
||||||
|
{
|
||||||
|
log::debug!("Crossed barrier into client: {client}, {position:?}");
|
||||||
|
return Some((*client, *position));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get the max bounds of all displays
|
||||||
|
fn update_bounds(&mut self) -> Result<(), MacosCaptureCreationError> {
|
||||||
|
let active_ids =
|
||||||
|
CGDisplay::active_displays().map_err(MacosCaptureCreationError::ActiveDisplays)?;
|
||||||
|
active_ids.iter().for_each(|d| {
|
||||||
|
let bounds = CGDisplay::new(*d).bounds();
|
||||||
|
self.bounds.xmin = self.bounds.xmin.min(bounds.origin.x);
|
||||||
|
self.bounds.xmax = self.bounds.xmax.max(bounds.origin.x + bounds.size.width);
|
||||||
|
self.bounds.ymin = self.bounds.ymin.min(bounds.origin.y);
|
||||||
|
self.bounds.ymax = self.bounds.ymax.max(bounds.origin.y + bounds.size.height);
|
||||||
|
});
|
||||||
|
|
||||||
|
log::debug!("Updated displays bounds: {0:?}", self.bounds);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
// We can't disable mouse movement when in a client so we need to reset the cursor position
|
||||||
|
// to the edge of the screen, the cursor will be hidden but we dont want it to appear in a
|
||||||
|
// random location when we exit the client
|
||||||
|
fn reset_mouse_position(&self, event: &CGEvent) -> Result<(), CaptureError> {
|
||||||
|
if let Some((_, pos)) = self.current_client {
|
||||||
|
let location = event.location();
|
||||||
|
let edge_offset = 1.0;
|
||||||
|
|
||||||
|
// After the cursor is warped no event is produced but the next event
|
||||||
|
// will carry the delta from the warp so only half the delta is needed to move the cursor
|
||||||
|
let delta_y = event.get_double_value_field(EventField::MOUSE_EVENT_DELTA_Y) / 2.0;
|
||||||
|
let delta_x = event.get_double_value_field(EventField::MOUSE_EVENT_DELTA_X) / 2.0;
|
||||||
|
|
||||||
|
let mut new_x = location.x + delta_x;
|
||||||
|
let mut new_y = location.y + delta_y;
|
||||||
|
|
||||||
|
match pos {
|
||||||
|
Position::Left => {
|
||||||
|
new_x = self.bounds.xmin + edge_offset;
|
||||||
|
}
|
||||||
|
Position::Right => {
|
||||||
|
new_x = self.bounds.xmax - edge_offset;
|
||||||
|
}
|
||||||
|
Position::Top => {
|
||||||
|
new_y = self.bounds.ymin + edge_offset;
|
||||||
|
}
|
||||||
|
Position::Bottom => {
|
||||||
|
new_y = self.bounds.ymax - edge_offset;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let new_pos = CGPoint::new(new_x, new_y);
|
||||||
|
|
||||||
|
log::trace!("Resetting cursor position to: {new_x}, {new_y}");
|
||||||
|
|
||||||
|
return CGDisplay::warp_mouse_cursor_position(new_pos)
|
||||||
|
.map_err(CaptureError::WarpCursor);
|
||||||
|
}
|
||||||
|
|
||||||
|
Err(CaptureError::ResetMouseWithoutClient)
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn handle_producer_event(
|
||||||
|
&mut self,
|
||||||
|
producer_event: ProducerEvent,
|
||||||
|
) -> Result<(), CaptureError> {
|
||||||
|
log::debug!("handling event: {producer_event:?}");
|
||||||
|
match producer_event {
|
||||||
|
ProducerEvent::Release => {
|
||||||
|
if self.current_client.is_some() {
|
||||||
|
CGDisplay::show_cursor(&CGDisplay::main())
|
||||||
|
.map_err(CaptureError::CoreGraphics)?;
|
||||||
|
self.current_client = None;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ProducerEvent::Grab(client) => {
|
||||||
|
if self.current_client.is_none() {
|
||||||
|
CGDisplay::hide_cursor(&CGDisplay::main())
|
||||||
|
.map_err(CaptureError::CoreGraphics)?;
|
||||||
|
self.current_client = Some(client);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ProducerEvent::Create(c, p) => {
|
||||||
|
self.client_for_pos.insert(p, c);
|
||||||
|
}
|
||||||
|
ProducerEvent::Destroy(c) => {
|
||||||
|
for pos in [
|
||||||
|
Position::Left,
|
||||||
|
Position::Right,
|
||||||
|
Position::Top,
|
||||||
|
Position::Bottom,
|
||||||
|
] {
|
||||||
|
if let Some((current_c, _)) = self.current_client {
|
||||||
|
if current_c == c {
|
||||||
|
CGDisplay::show_cursor(&CGDisplay::main())
|
||||||
|
.map_err(CaptureError::CoreGraphics)?;
|
||||||
|
self.current_client = None;
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if self.client_for_pos.get(&pos).copied() == Some(c) {
|
||||||
|
self.client_for_pos.remove(&pos);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ProducerEvent::EventTapDisabled => return Err(CaptureError::EventTapDisabled),
|
||||||
|
};
|
||||||
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Stream for MacOSInputCapture {
|
fn get_events(
|
||||||
type Item = Result<(CaptureHandle, Event), CaptureError>;
|
ev_type: &CGEventType,
|
||||||
|
ev: &CGEvent,
|
||||||
|
result: &mut Vec<CaptureEvent>,
|
||||||
|
) -> Result<(), CaptureError> {
|
||||||
|
fn map_pointer_event(ev: &CGEvent) -> PointerEvent {
|
||||||
|
PointerEvent::Motion {
|
||||||
|
time: 0,
|
||||||
|
dx: ev.get_double_value_field(EventField::MOUSE_EVENT_DELTA_X),
|
||||||
|
dy: ev.get_double_value_field(EventField::MOUSE_EVENT_DELTA_Y),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
fn poll_next(self: Pin<&mut Self>, _cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
|
fn map_key(ev: &CGEvent) -> Result<u32, CaptureError> {
|
||||||
Poll::Pending
|
let code = ev.get_integer_value_field(EventField::KEYBOARD_EVENT_KEYCODE);
|
||||||
|
match KeyMap::from_key_mapping(KeyMapping::Mac(code as u16)) {
|
||||||
|
Ok(k) => Ok(k.evdev as u32),
|
||||||
|
Err(()) => Err(CaptureError::KeyMapError(code)),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
match ev_type {
|
||||||
|
CGEventType::KeyDown => {
|
||||||
|
let k = map_key(ev)?;
|
||||||
|
result.push(CaptureEvent::Input(Event::Keyboard(KeyboardEvent::Key {
|
||||||
|
time: 0,
|
||||||
|
key: k,
|
||||||
|
state: 1,
|
||||||
|
})));
|
||||||
|
}
|
||||||
|
CGEventType::KeyUp => {
|
||||||
|
let k = map_key(ev)?;
|
||||||
|
result.push(CaptureEvent::Input(Event::Keyboard(KeyboardEvent::Key {
|
||||||
|
time: 0,
|
||||||
|
key: k,
|
||||||
|
state: 0,
|
||||||
|
})));
|
||||||
|
}
|
||||||
|
CGEventType::FlagsChanged => {
|
||||||
|
let mut mods = XMods::empty();
|
||||||
|
let mut mods_locked = XMods::empty();
|
||||||
|
let cg_flags = ev.get_flags();
|
||||||
|
|
||||||
|
if cg_flags.contains(CGEventFlags::CGEventFlagShift) {
|
||||||
|
mods |= XMods::ShiftMask;
|
||||||
|
}
|
||||||
|
if cg_flags.contains(CGEventFlags::CGEventFlagControl) {
|
||||||
|
mods |= XMods::ControlMask;
|
||||||
|
}
|
||||||
|
if cg_flags.contains(CGEventFlags::CGEventFlagAlternate) {
|
||||||
|
mods |= XMods::Mod1Mask;
|
||||||
|
}
|
||||||
|
if cg_flags.contains(CGEventFlags::CGEventFlagCommand) {
|
||||||
|
mods |= XMods::Mod4Mask;
|
||||||
|
}
|
||||||
|
if cg_flags.contains(CGEventFlags::CGEventFlagAlphaShift) {
|
||||||
|
mods |= XMods::LockMask;
|
||||||
|
mods_locked |= XMods::LockMask;
|
||||||
|
}
|
||||||
|
|
||||||
|
let modifier_event = KeyboardEvent::Modifiers {
|
||||||
|
depressed: mods.bits(),
|
||||||
|
latched: 0,
|
||||||
|
locked: mods_locked.bits(),
|
||||||
|
group: 0,
|
||||||
|
};
|
||||||
|
|
||||||
|
result.push(CaptureEvent::Input(Event::Keyboard(modifier_event)));
|
||||||
|
}
|
||||||
|
CGEventType::MouseMoved => {
|
||||||
|
result.push(CaptureEvent::Input(Event::Pointer(map_pointer_event(ev))))
|
||||||
|
}
|
||||||
|
CGEventType::LeftMouseDragged => {
|
||||||
|
result.push(CaptureEvent::Input(Event::Pointer(map_pointer_event(ev))))
|
||||||
|
}
|
||||||
|
CGEventType::RightMouseDragged => {
|
||||||
|
result.push(CaptureEvent::Input(Event::Pointer(map_pointer_event(ev))))
|
||||||
|
}
|
||||||
|
CGEventType::OtherMouseDragged => {
|
||||||
|
result.push(CaptureEvent::Input(Event::Pointer(map_pointer_event(ev))))
|
||||||
|
}
|
||||||
|
CGEventType::LeftMouseDown => {
|
||||||
|
result.push(CaptureEvent::Input(Event::Pointer(PointerEvent::Button {
|
||||||
|
time: 0,
|
||||||
|
button: BTN_LEFT,
|
||||||
|
state: 1,
|
||||||
|
})))
|
||||||
|
}
|
||||||
|
CGEventType::LeftMouseUp => {
|
||||||
|
result.push(CaptureEvent::Input(Event::Pointer(PointerEvent::Button {
|
||||||
|
time: 0,
|
||||||
|
button: BTN_LEFT,
|
||||||
|
state: 0,
|
||||||
|
})))
|
||||||
|
}
|
||||||
|
CGEventType::RightMouseDown => {
|
||||||
|
result.push(CaptureEvent::Input(Event::Pointer(PointerEvent::Button {
|
||||||
|
time: 0,
|
||||||
|
button: BTN_RIGHT,
|
||||||
|
state: 1,
|
||||||
|
})))
|
||||||
|
}
|
||||||
|
CGEventType::RightMouseUp => {
|
||||||
|
result.push(CaptureEvent::Input(Event::Pointer(PointerEvent::Button {
|
||||||
|
time: 0,
|
||||||
|
button: BTN_RIGHT,
|
||||||
|
state: 0,
|
||||||
|
})))
|
||||||
|
}
|
||||||
|
CGEventType::OtherMouseDown => {
|
||||||
|
result.push(CaptureEvent::Input(Event::Pointer(PointerEvent::Button {
|
||||||
|
time: 0,
|
||||||
|
button: BTN_MIDDLE,
|
||||||
|
state: 1,
|
||||||
|
})))
|
||||||
|
}
|
||||||
|
CGEventType::OtherMouseUp => {
|
||||||
|
result.push(CaptureEvent::Input(Event::Pointer(PointerEvent::Button {
|
||||||
|
time: 0,
|
||||||
|
button: BTN_MIDDLE,
|
||||||
|
state: 0,
|
||||||
|
})))
|
||||||
|
}
|
||||||
|
CGEventType::ScrollWheel => {
|
||||||
|
let v = ev.get_integer_value_field(EventField::SCROLL_WHEEL_EVENT_POINT_DELTA_AXIS_1);
|
||||||
|
let h = ev.get_integer_value_field(EventField::SCROLL_WHEEL_EVENT_POINT_DELTA_AXIS_2);
|
||||||
|
if v != 0 {
|
||||||
|
result.push(CaptureEvent::Input(Event::Pointer(PointerEvent::Axis {
|
||||||
|
time: 0,
|
||||||
|
axis: 0, // Vertical
|
||||||
|
value: v as f64,
|
||||||
|
})));
|
||||||
|
}
|
||||||
|
if h != 0 {
|
||||||
|
result.push(CaptureEvent::Input(Event::Pointer(PointerEvent::Axis {
|
||||||
|
time: 0,
|
||||||
|
axis: 1, // Horizontal
|
||||||
|
value: h as f64,
|
||||||
|
})));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => (),
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn event_tap_thread(
|
||||||
|
client_state: Arc<Mutex<InputCaptureState>>,
|
||||||
|
event_tx: Sender<(CaptureHandle, CaptureEvent)>,
|
||||||
|
notify_tx: Sender<ProducerEvent>,
|
||||||
|
exit: tokio::sync::oneshot::Sender<Result<(), &'static str>>,
|
||||||
|
) {
|
||||||
|
let cg_events_of_interest: Vec<CGEventType> = vec![
|
||||||
|
CGEventType::LeftMouseDown,
|
||||||
|
CGEventType::LeftMouseUp,
|
||||||
|
CGEventType::RightMouseDown,
|
||||||
|
CGEventType::RightMouseUp,
|
||||||
|
CGEventType::OtherMouseDown,
|
||||||
|
CGEventType::OtherMouseUp,
|
||||||
|
CGEventType::MouseMoved,
|
||||||
|
CGEventType::LeftMouseDragged,
|
||||||
|
CGEventType::RightMouseDragged,
|
||||||
|
CGEventType::OtherMouseDragged,
|
||||||
|
CGEventType::ScrollWheel,
|
||||||
|
CGEventType::KeyDown,
|
||||||
|
CGEventType::KeyUp,
|
||||||
|
CGEventType::FlagsChanged,
|
||||||
|
];
|
||||||
|
|
||||||
|
let tap = CGEventTap::new(
|
||||||
|
CGEventTapLocation::Session,
|
||||||
|
CGEventTapPlacement::HeadInsertEventTap,
|
||||||
|
CGEventTapOptions::Default,
|
||||||
|
cg_events_of_interest,
|
||||||
|
|_proxy: CGEventTapProxy, event_type: CGEventType, cg_ev: &CGEvent| {
|
||||||
|
log::trace!("Got event from tap: {event_type:?}");
|
||||||
|
let mut state = client_state.blocking_lock();
|
||||||
|
let mut client = None;
|
||||||
|
let mut res_events = vec![];
|
||||||
|
|
||||||
|
if matches!(
|
||||||
|
event_type,
|
||||||
|
CGEventType::TapDisabledByTimeout | CGEventType::TapDisabledByUserInput
|
||||||
|
) {
|
||||||
|
log::error!("CGEventTap disabled");
|
||||||
|
notify_tx
|
||||||
|
.blocking_send(ProducerEvent::EventTapDisabled)
|
||||||
|
.unwrap_or_else(|e| {
|
||||||
|
log::error!("Failed to send notification: {e}");
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// Are we in a client?
|
||||||
|
if let Some((current_client, _)) = state.current_client {
|
||||||
|
client = Some(current_client);
|
||||||
|
get_events(&event_type, cg_ev, &mut res_events).unwrap_or_else(|e| {
|
||||||
|
log::error!("Failed to get events: {e}");
|
||||||
|
});
|
||||||
|
|
||||||
|
// Keep (hidden) cursor at the edge of the screen
|
||||||
|
if matches!(event_type, CGEventType::MouseMoved) {
|
||||||
|
state.reset_mouse_position(cg_ev).unwrap_or_else(|e| {
|
||||||
|
log::error!("Failed to reset mouse position: {e}");
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Did we cross a barrier?
|
||||||
|
else if matches!(event_type, CGEventType::MouseMoved) {
|
||||||
|
if let Some((new_client, pos)) = state.crossed(cg_ev) {
|
||||||
|
client = Some(new_client);
|
||||||
|
res_events.push(CaptureEvent::Begin);
|
||||||
|
notify_tx
|
||||||
|
.blocking_send(ProducerEvent::Grab((new_client, pos)))
|
||||||
|
.expect("Failed to send notification");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(client) = client {
|
||||||
|
res_events.iter().for_each(|e| {
|
||||||
|
event_tx
|
||||||
|
.blocking_send((client, *e))
|
||||||
|
.expect("Failed to send event");
|
||||||
|
});
|
||||||
|
// Returning None should stop the event from being processed
|
||||||
|
// but core fundation still returns the event
|
||||||
|
cg_ev.set_type(CGEventType::Null);
|
||||||
|
}
|
||||||
|
Some(cg_ev.to_owned())
|
||||||
|
},
|
||||||
|
)
|
||||||
|
.expect("Failed creating tap");
|
||||||
|
|
||||||
|
let tap_source: CFRunLoopSource = tap
|
||||||
|
.mach_port
|
||||||
|
.create_runloop_source(0)
|
||||||
|
.expect("Failed creating loop source");
|
||||||
|
|
||||||
|
unsafe {
|
||||||
|
CFRunLoop::get_current().add_source(&tap_source, kCFRunLoopCommonModes);
|
||||||
|
}
|
||||||
|
|
||||||
|
CFRunLoop::run_current();
|
||||||
|
|
||||||
|
let _ = exit.send(Err("tap thread exited"));
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct MacOSInputCapture {
|
||||||
|
event_rx: Receiver<(CaptureHandle, CaptureEvent)>,
|
||||||
|
notify_tx: Sender<ProducerEvent>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl MacOSInputCapture {
|
||||||
|
pub async fn new() -> Result<Self, MacosCaptureCreationError> {
|
||||||
|
let state = Arc::new(Mutex::new(InputCaptureState::new()?));
|
||||||
|
let (event_tx, event_rx) = tokio::sync::mpsc::channel(32);
|
||||||
|
let (notify_tx, mut notify_rx) = tokio::sync::mpsc::channel(32);
|
||||||
|
let (tap_exit_tx, mut tap_exit_rx) = tokio::sync::oneshot::channel();
|
||||||
|
|
||||||
|
unsafe {
|
||||||
|
configure_cf_settings()?;
|
||||||
|
}
|
||||||
|
|
||||||
|
log::info!("Enabling CGEvent tap");
|
||||||
|
let event_tap_thread_state = state.clone();
|
||||||
|
let event_tap_notify = notify_tx.clone();
|
||||||
|
thread::spawn(move || {
|
||||||
|
event_tap_thread(
|
||||||
|
event_tap_thread_state,
|
||||||
|
event_tx,
|
||||||
|
event_tap_notify,
|
||||||
|
tap_exit_tx,
|
||||||
|
)
|
||||||
|
});
|
||||||
|
|
||||||
|
let _tap_task: tokio::task::JoinHandle<()> = tokio::task::spawn_local(async move {
|
||||||
|
loop {
|
||||||
|
tokio::select! {
|
||||||
|
producer_event = notify_rx.recv() => {
|
||||||
|
let producer_event = producer_event.expect("channel closed");
|
||||||
|
let mut state = state.lock().await;
|
||||||
|
state.handle_producer_event(producer_event).await.unwrap_or_else(|e| {
|
||||||
|
log::error!("Failed to handle producer event: {e}");
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
res = &mut tap_exit_rx => {
|
||||||
|
if let Err(e) = res.expect("channel closed") {
|
||||||
|
log::error!("Tap thread failed: {:?}", e);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
Ok(Self {
|
||||||
|
event_rx,
|
||||||
|
notify_tx,
|
||||||
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[async_trait]
|
#[async_trait]
|
||||||
impl Capture for MacOSInputCapture {
|
impl Capture for MacOSInputCapture {
|
||||||
async fn create(&mut self, _id: CaptureHandle, _pos: Position) -> Result<(), CaptureError> {
|
async fn create(&mut self, id: CaptureHandle, pos: Position) -> Result<(), CaptureError> {
|
||||||
|
let notify_tx = self.notify_tx.clone();
|
||||||
|
tokio::task::spawn_local(async move {
|
||||||
|
log::debug!("creating client {id}, {pos}");
|
||||||
|
let _ = notify_tx.send(ProducerEvent::Create(id, pos)).await;
|
||||||
|
log::debug!("done !");
|
||||||
|
});
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn destroy(&mut self, _id: CaptureHandle) -> Result<(), CaptureError> {
|
async fn destroy(&mut self, id: CaptureHandle) -> Result<(), CaptureError> {
|
||||||
|
let notify_tx = self.notify_tx.clone();
|
||||||
|
tokio::task::spawn_local(async move {
|
||||||
|
log::debug!("destroying client {id}");
|
||||||
|
let _ = notify_tx.send(ProducerEvent::Destroy(id)).await;
|
||||||
|
log::debug!("done !");
|
||||||
|
});
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn release(&mut self) -> Result<(), CaptureError> {
|
async fn release(&mut self) -> Result<(), CaptureError> {
|
||||||
|
let notify_tx = self.notify_tx.clone();
|
||||||
|
tokio::task::spawn_local(async move {
|
||||||
|
log::debug!("notifying Release");
|
||||||
|
let _ = notify_tx.send(ProducerEvent::Release).await;
|
||||||
|
});
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -41,3 +524,79 @@ impl Capture for MacOSInputCapture {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl Stream for MacOSInputCapture {
|
||||||
|
type Item = Result<(CaptureHandle, CaptureEvent), CaptureError>;
|
||||||
|
|
||||||
|
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
|
||||||
|
match ready!(self.event_rx.poll_recv(cx)) {
|
||||||
|
None => Poll::Ready(None),
|
||||||
|
Some(e) => Poll::Ready(Some(Ok(e))),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type CGSConnectionID = u32;
|
||||||
|
|
||||||
|
#[link(name = "ApplicationServices", kind = "framework")]
|
||||||
|
extern "C" {
|
||||||
|
fn CGSSetConnectionProperty(
|
||||||
|
cid: CGSConnectionID,
|
||||||
|
targetCID: CGSConnectionID,
|
||||||
|
key: CFStringRef,
|
||||||
|
value: CFBooleanRef,
|
||||||
|
) -> CGError;
|
||||||
|
fn _CGSDefaultConnection() -> CGSConnectionID;
|
||||||
|
}
|
||||||
|
|
||||||
|
extern "C" {
|
||||||
|
fn CGEventSourceSetLocalEventsSuppressionInterval(
|
||||||
|
event_source: CGEventSource,
|
||||||
|
seconds: CFTimeInterval,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
unsafe fn configure_cf_settings() -> Result<(), MacosCaptureCreationError> {
|
||||||
|
// When we warp the cursor using CGWarpMouseCursorPosition local events are suppressed for a short time
|
||||||
|
// this leeds to the cursor not flowing when crossing back from a clinet, set this to to 0 stops the warp
|
||||||
|
// from working, set a low value by trial and error, 0.05s seems good. 0.25s is the default
|
||||||
|
let event_source = CGEventSource::new(CGEventSourceStateID::CombinedSessionState)
|
||||||
|
.map_err(|_| MacosCaptureCreationError::EventSourceCreation)?;
|
||||||
|
CGEventSourceSetLocalEventsSuppressionInterval(event_source, 0.05);
|
||||||
|
|
||||||
|
// This is a private settings that allows the cursor to be hidden while in the background.
|
||||||
|
// It is used by Barrier and other apps.
|
||||||
|
let key = CString::new("SetsCursorInBackground").unwrap();
|
||||||
|
let cf_key = CFStringCreateWithCString(
|
||||||
|
kCFAllocatorDefault,
|
||||||
|
key.as_ptr() as *const c_char,
|
||||||
|
kCFStringEncodingUTF8,
|
||||||
|
);
|
||||||
|
if CGSSetConnectionProperty(
|
||||||
|
_CGSDefaultConnection(),
|
||||||
|
_CGSDefaultConnection(),
|
||||||
|
cf_key,
|
||||||
|
kCFBooleanTrue,
|
||||||
|
) != kCGErrorSuccess
|
||||||
|
{
|
||||||
|
return Err(MacosCaptureCreationError::CGCursorProperty);
|
||||||
|
}
|
||||||
|
CFRelease(cf_key as *const c_void);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
// From X11/X.h
|
||||||
|
bitflags! {
|
||||||
|
#[repr(C)]
|
||||||
|
#[derive(Clone, Copy, Debug, Default, Eq, Hash, Ord, PartialEq, PartialOrd)]
|
||||||
|
struct XMods: u32 {
|
||||||
|
const ShiftMask = (1<<0);
|
||||||
|
const LockMask = (1<<1);
|
||||||
|
const ControlMask = (1<<2);
|
||||||
|
const Mod1Mask = (1<<3);
|
||||||
|
const Mod2Mask = (1<<4);
|
||||||
|
const Mod3Mask = (1<<5);
|
||||||
|
const Mod4Mask = (1<<6);
|
||||||
|
const Mod5Mask = (1<<7);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -60,7 +60,7 @@ use wayland_client::{
|
|||||||
|
|
||||||
use input_event::{Event, KeyboardEvent, PointerEvent};
|
use input_event::{Event, KeyboardEvent, PointerEvent};
|
||||||
|
|
||||||
use crate::CaptureError;
|
use crate::{CaptureError, CaptureEvent};
|
||||||
|
|
||||||
use super::{
|
use super::{
|
||||||
error::{LayerShellCaptureCreationError, WaylandBindError},
|
error::{LayerShellCaptureCreationError, WaylandBindError},
|
||||||
@@ -71,7 +71,7 @@ struct Globals {
|
|||||||
compositor: wl_compositor::WlCompositor,
|
compositor: wl_compositor::WlCompositor,
|
||||||
pointer_constraints: ZwpPointerConstraintsV1,
|
pointer_constraints: ZwpPointerConstraintsV1,
|
||||||
relative_pointer_manager: ZwpRelativePointerManagerV1,
|
relative_pointer_manager: ZwpRelativePointerManagerV1,
|
||||||
shortcut_inhibit_manager: ZwpKeyboardShortcutsInhibitManagerV1,
|
shortcut_inhibit_manager: Option<ZwpKeyboardShortcutsInhibitManagerV1>,
|
||||||
seat: wl_seat::WlSeat,
|
seat: wl_seat::WlSeat,
|
||||||
shm: wl_shm::WlShm,
|
shm: wl_shm::WlShm,
|
||||||
layer_shell: ZwlrLayerShellV1,
|
layer_shell: ZwlrLayerShellV1,
|
||||||
@@ -108,7 +108,7 @@ struct State {
|
|||||||
wayland_fd: RawFd,
|
wayland_fd: RawFd,
|
||||||
read_guard: Option<ReadEventsGuard>,
|
read_guard: Option<ReadEventsGuard>,
|
||||||
qh: QueueHandle<Self>,
|
qh: QueueHandle<Self>,
|
||||||
pending_events: VecDeque<(CaptureHandle, Event)>,
|
pending_events: VecDeque<(CaptureHandle, CaptureEvent)>,
|
||||||
output_info: Vec<(WlOutput, OutputInfo)>,
|
output_info: Vec<(WlOutput, OutputInfo)>,
|
||||||
scroll_discrete_pending: bool,
|
scroll_discrete_pending: bool,
|
||||||
}
|
}
|
||||||
@@ -285,9 +285,18 @@ impl WaylandInputCapture {
|
|||||||
let relative_pointer_manager: ZwpRelativePointerManagerV1 = g
|
let relative_pointer_manager: ZwpRelativePointerManagerV1 = g
|
||||||
.bind(&qh, 1..=1, ())
|
.bind(&qh, 1..=1, ())
|
||||||
.map_err(|e| WaylandBindError::new(e, "zwp_relative_pointer_manager_v1"))?;
|
.map_err(|e| WaylandBindError::new(e, "zwp_relative_pointer_manager_v1"))?;
|
||||||
let shortcut_inhibit_manager: ZwpKeyboardShortcutsInhibitManagerV1 = g
|
let shortcut_inhibit_manager: Result<
|
||||||
|
ZwpKeyboardShortcutsInhibitManagerV1,
|
||||||
|
WaylandBindError,
|
||||||
|
> = g
|
||||||
.bind(&qh, 1..=1, ())
|
.bind(&qh, 1..=1, ())
|
||||||
.map_err(|e| WaylandBindError::new(e, "zwp_keyboard_shortcuts_inhibit_manager_v1"))?;
|
.map_err(|e| WaylandBindError::new(e, "zwp_keyboard_shortcuts_inhibit_manager_v1"));
|
||||||
|
// layer-shell backend still works without this protocol so we make it an optional dependency
|
||||||
|
if let Err(e) = &shortcut_inhibit_manager {
|
||||||
|
log::warn!("shortcut_inhibit_manager not supported: {e}\nkeybinds handled by the compositor will not be passed
|
||||||
|
to the client");
|
||||||
|
}
|
||||||
|
let shortcut_inhibit_manager = shortcut_inhibit_manager.ok();
|
||||||
let outputs = vec![];
|
let outputs = vec![];
|
||||||
|
|
||||||
let g = Globals {
|
let g = Globals {
|
||||||
@@ -424,13 +433,11 @@ impl State {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// capture modifier keys
|
// capture modifier keys
|
||||||
if self.shortcut_inhibitor.is_none() {
|
if let Some(shortcut_inhibit_manager) = &self.g.shortcut_inhibit_manager {
|
||||||
self.shortcut_inhibitor = Some(self.g.shortcut_inhibit_manager.inhibit_shortcuts(
|
if self.shortcut_inhibitor.is_none() {
|
||||||
surface,
|
self.shortcut_inhibitor =
|
||||||
&self.g.seat,
|
Some(shortcut_inhibit_manager.inhibit_shortcuts(surface, &self.g.seat, qh, ()));
|
||||||
qh,
|
}
|
||||||
(),
|
|
||||||
));
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -585,7 +592,7 @@ impl Capture for WaylandInputCapture {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl Stream for WaylandInputCapture {
|
impl Stream for WaylandInputCapture {
|
||||||
type Item = Result<(CaptureHandle, Event), CaptureError>;
|
type Item = Result<(CaptureHandle, CaptureEvent), CaptureError>;
|
||||||
|
|
||||||
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
|
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
|
||||||
if let Some(event) = self.0.get_mut().state.pending_events.pop_front() {
|
if let Some(event) = self.0.get_mut().state.pending_events.pop_front() {
|
||||||
@@ -694,7 +701,7 @@ impl Dispatch<WlPointer, ()> for State {
|
|||||||
.iter()
|
.iter()
|
||||||
.find(|(w, _c)| w.surface == surface)
|
.find(|(w, _c)| w.surface == surface)
|
||||||
.unwrap();
|
.unwrap();
|
||||||
app.pending_events.push_back((*client, Event::Enter()));
|
app.pending_events.push_back((*client, CaptureEvent::Begin));
|
||||||
}
|
}
|
||||||
wl_pointer::Event::Leave { .. } => {
|
wl_pointer::Event::Leave { .. } => {
|
||||||
/* There are rare cases, where when a window is opened in
|
/* There are rare cases, where when a window is opened in
|
||||||
@@ -718,11 +725,11 @@ impl Dispatch<WlPointer, ()> for State {
|
|||||||
let (_, client) = app.focused.as_ref().unwrap();
|
let (_, client) = app.focused.as_ref().unwrap();
|
||||||
app.pending_events.push_back((
|
app.pending_events.push_back((
|
||||||
*client,
|
*client,
|
||||||
Event::Pointer(PointerEvent::Button {
|
CaptureEvent::Input(Event::Pointer(PointerEvent::Button {
|
||||||
time,
|
time,
|
||||||
button,
|
button,
|
||||||
state: u32::from(state),
|
state: u32::from(state),
|
||||||
}),
|
})),
|
||||||
));
|
));
|
||||||
}
|
}
|
||||||
wl_pointer::Event::Axis { time, axis, value } => {
|
wl_pointer::Event::Axis { time, axis, value } => {
|
||||||
@@ -735,11 +742,11 @@ impl Dispatch<WlPointer, ()> for State {
|
|||||||
} else {
|
} else {
|
||||||
app.pending_events.push_back((
|
app.pending_events.push_back((
|
||||||
*client,
|
*client,
|
||||||
Event::Pointer(PointerEvent::Axis {
|
CaptureEvent::Input(Event::Pointer(PointerEvent::Axis {
|
||||||
time,
|
time,
|
||||||
axis: u32::from(axis) as u8,
|
axis: u32::from(axis) as u8,
|
||||||
value,
|
value,
|
||||||
}),
|
})),
|
||||||
));
|
));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -748,10 +755,10 @@ impl Dispatch<WlPointer, ()> for State {
|
|||||||
app.scroll_discrete_pending = true;
|
app.scroll_discrete_pending = true;
|
||||||
app.pending_events.push_back((
|
app.pending_events.push_back((
|
||||||
*client,
|
*client,
|
||||||
Event::Pointer(PointerEvent::AxisDiscrete120 {
|
CaptureEvent::Input(Event::Pointer(PointerEvent::AxisDiscrete120 {
|
||||||
axis: u32::from(axis) as u8,
|
axis: u32::from(axis) as u8,
|
||||||
value: value120,
|
value: value120,
|
||||||
}),
|
})),
|
||||||
));
|
));
|
||||||
}
|
}
|
||||||
wl_pointer::Event::Frame {} => {
|
wl_pointer::Event::Frame {} => {
|
||||||
@@ -787,11 +794,11 @@ impl Dispatch<WlKeyboard, ()> for State {
|
|||||||
if let Some(client) = client {
|
if let Some(client) = client {
|
||||||
app.pending_events.push_back((
|
app.pending_events.push_back((
|
||||||
*client,
|
*client,
|
||||||
Event::Keyboard(KeyboardEvent::Key {
|
CaptureEvent::Input(Event::Keyboard(KeyboardEvent::Key {
|
||||||
time,
|
time,
|
||||||
key,
|
key,
|
||||||
state: u32::from(state) as u8,
|
state: u32::from(state) as u8,
|
||||||
}),
|
})),
|
||||||
));
|
));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -805,12 +812,12 @@ impl Dispatch<WlKeyboard, ()> for State {
|
|||||||
if let Some(client) = client {
|
if let Some(client) = client {
|
||||||
app.pending_events.push_back((
|
app.pending_events.push_back((
|
||||||
*client,
|
*client,
|
||||||
Event::Keyboard(KeyboardEvent::Modifiers {
|
CaptureEvent::Input(Event::Keyboard(KeyboardEvent::Modifiers {
|
||||||
mods_depressed,
|
depressed: mods_depressed,
|
||||||
mods_latched,
|
latched: mods_latched,
|
||||||
mods_locked,
|
locked: mods_locked,
|
||||||
group,
|
group,
|
||||||
}),
|
})),
|
||||||
));
|
));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -840,7 +847,7 @@ impl Dispatch<ZwpRelativePointerV1, ()> for State {
|
|||||||
let time = (((utime_hi as u64) << 32 | utime_lo as u64) / 1000) as u32;
|
let time = (((utime_hi as u64) << 32 | utime_lo as u64) / 1000) as u32;
|
||||||
app.pending_events.push_back((
|
app.pending_events.push_back((
|
||||||
*client,
|
*client,
|
||||||
Event::Pointer(PointerEvent::Motion { time, dx, dy }),
|
CaptureEvent::Input(Event::Pointer(PointerEvent::Motion { time, dx, dy })),
|
||||||
));
|
));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -37,7 +37,7 @@ use input_event::{
|
|||||||
Event, KeyboardEvent, PointerEvent, BTN_BACK, BTN_FORWARD, BTN_LEFT, BTN_MIDDLE, BTN_RIGHT,
|
Event, KeyboardEvent, PointerEvent, BTN_BACK, BTN_FORWARD, BTN_LEFT, BTN_MIDDLE, BTN_RIGHT,
|
||||||
};
|
};
|
||||||
|
|
||||||
use super::{Capture, CaptureError, CaptureHandle, Position};
|
use super::{Capture, CaptureError, CaptureEvent, CaptureHandle, Position};
|
||||||
|
|
||||||
enum Request {
|
enum Request {
|
||||||
Create(CaptureHandle, Position),
|
Create(CaptureHandle, Position),
|
||||||
@@ -45,7 +45,7 @@ enum Request {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub struct WindowsInputCapture {
|
pub struct WindowsInputCapture {
|
||||||
event_rx: Receiver<(CaptureHandle, Event)>,
|
event_rx: Receiver<(CaptureHandle, CaptureEvent)>,
|
||||||
msg_thread: Option<std::thread::JoinHandle<()>>,
|
msg_thread: Option<std::thread::JoinHandle<()>>,
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -100,7 +100,7 @@ impl Capture for WindowsInputCapture {
|
|||||||
static mut REQUEST_BUFFER: Mutex<Vec<Request>> = Mutex::new(Vec::new());
|
static mut REQUEST_BUFFER: Mutex<Vec<Request>> = Mutex::new(Vec::new());
|
||||||
static mut ACTIVE_CLIENT: Option<CaptureHandle> = None;
|
static mut ACTIVE_CLIENT: Option<CaptureHandle> = None;
|
||||||
static mut CLIENT_FOR_POS: Lazy<HashMap<Position, CaptureHandle>> = Lazy::new(HashMap::new);
|
static mut CLIENT_FOR_POS: Lazy<HashMap<Position, CaptureHandle>> = Lazy::new(HashMap::new);
|
||||||
static mut EVENT_TX: Option<Sender<(CaptureHandle, Event)>> = None;
|
static mut EVENT_TX: Option<Sender<(CaptureHandle, CaptureEvent)>> = None;
|
||||||
static mut EVENT_THREAD_ID: AtomicU32 = AtomicU32::new(0);
|
static mut EVENT_THREAD_ID: AtomicU32 = AtomicU32::new(0);
|
||||||
unsafe fn set_event_tid(tid: u32) {
|
unsafe fn set_event_tid(tid: u32) {
|
||||||
EVENT_THREAD_ID.store(tid, Ordering::SeqCst);
|
EVENT_THREAD_ID.store(tid, Ordering::SeqCst);
|
||||||
@@ -249,7 +249,7 @@ fn clamp_to_display_bounds(prev_point: (i32, i32), point: (i32, i32)) -> (i32, i
|
|||||||
(x.clamp(min_x, max_x), y.clamp(min_y, max_y))
|
(x.clamp(min_x, max_x), y.clamp(min_y, max_y))
|
||||||
}
|
}
|
||||||
|
|
||||||
unsafe fn send_blocking(event: Event) {
|
unsafe fn send_blocking(event: CaptureEvent) {
|
||||||
if let Some(active) = ACTIVE_CLIENT {
|
if let Some(active) = ACTIVE_CLIENT {
|
||||||
block_on(async move {
|
block_on(async move {
|
||||||
let _ = EVENT_TX.as_ref().unwrap().send((active, event)).await;
|
let _ = EVENT_TX.as_ref().unwrap().send((active, event)).await;
|
||||||
@@ -291,7 +291,7 @@ unsafe fn check_client_activation(wparam: WPARAM, lparam: LPARAM) -> bool {
|
|||||||
|
|
||||||
/* notify main thread */
|
/* notify main thread */
|
||||||
log::debug!("ENTERED @ {prev_pos:?} -> {curr_pos:?}");
|
log::debug!("ENTERED @ {prev_pos:?} -> {curr_pos:?}");
|
||||||
send_blocking(Event::Enter());
|
send_blocking(CaptureEvent::Begin);
|
||||||
|
|
||||||
ret
|
ret
|
||||||
}
|
}
|
||||||
@@ -313,7 +313,7 @@ unsafe extern "system" fn mouse_proc(ncode: i32, wparam: WPARAM, lparam: LPARAM)
|
|||||||
let Some(pointer_event) = to_mouse_event(wparam, lparam) else {
|
let Some(pointer_event) = to_mouse_event(wparam, lparam) else {
|
||||||
return LRESULT(1);
|
return LRESULT(1);
|
||||||
};
|
};
|
||||||
let event = (client, Event::Pointer(pointer_event));
|
let event = (client, CaptureEvent::Input(Event::Pointer(pointer_event)));
|
||||||
|
|
||||||
/* notify mainthread (drop events if sending too fast) */
|
/* notify mainthread (drop events if sending too fast) */
|
||||||
if let Err(e) = EVENT_TX.as_ref().unwrap().try_send(event) {
|
if let Err(e) = EVENT_TX.as_ref().unwrap().try_send(event) {
|
||||||
@@ -334,7 +334,7 @@ unsafe extern "system" fn kybrd_proc(ncode: i32, wparam: WPARAM, lparam: LPARAM)
|
|||||||
let Some(key_event) = to_key_event(wparam, lparam) else {
|
let Some(key_event) = to_key_event(wparam, lparam) else {
|
||||||
return LRESULT(1);
|
return LRESULT(1);
|
||||||
};
|
};
|
||||||
let event = (client, Event::Keyboard(key_event));
|
let event = (client, CaptureEvent::Input(Event::Keyboard(key_event)));
|
||||||
|
|
||||||
if let Err(e) = EVENT_TX.as_ref().unwrap().try_send(event) {
|
if let Err(e) = EVENT_TX.as_ref().unwrap().try_send(event) {
|
||||||
log::warn!("e: {e}");
|
log::warn!("e: {e}");
|
||||||
@@ -614,7 +614,7 @@ impl WindowsInputCapture {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl Stream for WindowsInputCapture {
|
impl Stream for WindowsInputCapture {
|
||||||
type Item = Result<(CaptureHandle, Event), CaptureError>;
|
type Item = Result<(CaptureHandle, CaptureEvent), CaptureError>;
|
||||||
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
|
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
|
||||||
match ready!(self.event_rx.poll_recv(cx)) {
|
match ready!(self.event_rx.poll_recv(cx)) {
|
||||||
None => Poll::Ready(None),
|
None => Poll::Ready(None),
|
||||||
|
|||||||
@@ -3,13 +3,10 @@ use std::task::Poll;
|
|||||||
use async_trait::async_trait;
|
use async_trait::async_trait;
|
||||||
use futures_core::Stream;
|
use futures_core::Stream;
|
||||||
|
|
||||||
use crate::CaptureError;
|
use super::{
|
||||||
|
error::X11InputCaptureCreationError, Capture, CaptureError, CaptureEvent, CaptureHandle,
|
||||||
use super::Capture;
|
Position,
|
||||||
use input_event::Event;
|
};
|
||||||
|
|
||||||
use super::error::X11InputCaptureCreationError;
|
|
||||||
use super::{CaptureHandle, Position};
|
|
||||||
|
|
||||||
pub struct X11InputCapture {}
|
pub struct X11InputCapture {}
|
||||||
|
|
||||||
@@ -39,7 +36,7 @@ impl Capture for X11InputCapture {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl Stream for X11InputCapture {
|
impl Stream for X11InputCapture {
|
||||||
type Item = Result<(CaptureHandle, Event), CaptureError>;
|
type Item = Result<(CaptureHandle, CaptureEvent), CaptureError>;
|
||||||
|
|
||||||
fn poll_next(
|
fn poll_next(
|
||||||
self: std::pin::Pin<&mut Self>,
|
self: std::pin::Pin<&mut Self>,
|
||||||
|
|||||||
@@ -195,9 +195,9 @@ impl InputEmulation {
|
|||||||
}
|
}
|
||||||
|
|
||||||
let event = Event::Keyboard(KeyboardEvent::Modifiers {
|
let event = Event::Keyboard(KeyboardEvent::Modifiers {
|
||||||
mods_depressed: 0,
|
depressed: 0,
|
||||||
mods_latched: 0,
|
latched: 0,
|
||||||
mods_locked: 0,
|
locked: 0,
|
||||||
group: 0,
|
group: 0,
|
||||||
});
|
});
|
||||||
self.emulation.consume(event, handle).await?;
|
self.emulation.consume(event, handle).await?;
|
||||||
|
|||||||
@@ -213,7 +213,6 @@ impl<'a> Emulation for LibeiEmulation<'a> {
|
|||||||
d.frame(self.serial.load(Ordering::SeqCst), now);
|
d.frame(self.serial.load(Ordering::SeqCst), now);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
PointerEvent::Frame {} => {}
|
|
||||||
},
|
},
|
||||||
Event::Keyboard(k) => match k {
|
Event::Keyboard(k) => match k {
|
||||||
KeyboardEvent::Key {
|
KeyboardEvent::Key {
|
||||||
@@ -235,7 +234,6 @@ impl<'a> Emulation for LibeiEmulation<'a> {
|
|||||||
}
|
}
|
||||||
KeyboardEvent::Modifiers { .. } => {}
|
KeyboardEvent::Modifiers { .. } => {}
|
||||||
},
|
},
|
||||||
_ => {}
|
|
||||||
}
|
}
|
||||||
self.context
|
self.context
|
||||||
.flush()
|
.flush()
|
||||||
|
|||||||
@@ -1,6 +1,9 @@
|
|||||||
use super::{error::EmulationError, Emulation, EmulationHandle};
|
use super::{error::EmulationError, Emulation, EmulationHandle};
|
||||||
use async_trait::async_trait;
|
use async_trait::async_trait;
|
||||||
use core_graphics::display::{CGDisplayBounds, CGMainDisplayID, CGPoint};
|
use core_graphics::base::CGFloat;
|
||||||
|
use core_graphics::display::{
|
||||||
|
CGDirectDisplayID, CGDisplayBounds, CGGetDisplaysWithRect, CGPoint, CGRect, CGSize,
|
||||||
|
};
|
||||||
use core_graphics::event::{
|
use core_graphics::event::{
|
||||||
CGEvent, CGEventTapLocation, CGEventType, CGKeyCode, CGMouseButton, EventField, ScrollEventUnit,
|
CGEvent, CGEventTapLocation, CGEventType, CGKeyCode, CGMouseButton, EventField, ScrollEventUnit,
|
||||||
};
|
};
|
||||||
@@ -8,8 +11,9 @@ use core_graphics::event_source::{CGEventSource, CGEventSourceStateID};
|
|||||||
use input_event::{Event, KeyboardEvent, PointerEvent};
|
use input_event::{Event, KeyboardEvent, PointerEvent};
|
||||||
use keycode::{KeyMap, KeyMapping};
|
use keycode::{KeyMap, KeyMapping};
|
||||||
use std::ops::{Index, IndexMut};
|
use std::ops::{Index, IndexMut};
|
||||||
|
use std::sync::Arc;
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
use tokio::task::AbortHandle;
|
use tokio::{sync::Notify, task::JoinHandle};
|
||||||
|
|
||||||
use super::error::MacOSEmulationCreationError;
|
use super::error::MacOSEmulationCreationError;
|
||||||
|
|
||||||
@@ -18,8 +22,9 @@ const DEFAULT_REPEAT_INTERVAL: Duration = Duration::from_millis(32);
|
|||||||
|
|
||||||
pub(crate) struct MacOSEmulation {
|
pub(crate) struct MacOSEmulation {
|
||||||
event_source: CGEventSource,
|
event_source: CGEventSource,
|
||||||
repeat_task: Option<AbortHandle>,
|
repeat_task: Option<JoinHandle<()>>,
|
||||||
button_state: ButtonState,
|
button_state: ButtonState,
|
||||||
|
notify_repeat_task: Arc<Notify>,
|
||||||
}
|
}
|
||||||
|
|
||||||
struct ButtonState {
|
struct ButtonState {
|
||||||
@@ -65,6 +70,7 @@ impl MacOSEmulation {
|
|||||||
event_source,
|
event_source,
|
||||||
button_state,
|
button_state,
|
||||||
repeat_task: None,
|
repeat_task: None,
|
||||||
|
notify_repeat_task: Arc::new(Notify::new()),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -76,20 +82,33 @@ impl MacOSEmulation {
|
|||||||
async fn spawn_repeat_task(&mut self, key: u16) {
|
async fn spawn_repeat_task(&mut self, key: u16) {
|
||||||
// there can only be one repeating key and it's
|
// there can only be one repeating key and it's
|
||||||
// always the last to be pressed
|
// always the last to be pressed
|
||||||
self.kill_repeat_task();
|
self.cancel_repeat_task().await;
|
||||||
let event_source = self.event_source.clone();
|
let event_source = self.event_source.clone();
|
||||||
|
let notify = self.notify_repeat_task.clone();
|
||||||
let repeat_task = tokio::task::spawn_local(async move {
|
let repeat_task = tokio::task::spawn_local(async move {
|
||||||
tokio::time::sleep(DEFAULT_REPEAT_DELAY).await;
|
let stop = tokio::select! {
|
||||||
loop {
|
_ = tokio::time::sleep(DEFAULT_REPEAT_DELAY) => false,
|
||||||
key_event(event_source.clone(), key, 1);
|
_ = notify.notified() => true,
|
||||||
tokio::time::sleep(DEFAULT_REPEAT_INTERVAL).await;
|
};
|
||||||
|
if !stop {
|
||||||
|
loop {
|
||||||
|
key_event(event_source.clone(), key, 1);
|
||||||
|
tokio::select! {
|
||||||
|
_ = tokio::time::sleep(DEFAULT_REPEAT_INTERVAL) => {},
|
||||||
|
_ = notify.notified() => break,
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
// release key when cancelled
|
||||||
|
key_event(event_source.clone(), key, 0);
|
||||||
});
|
});
|
||||||
self.repeat_task = Some(repeat_task.abort_handle());
|
self.repeat_task = Some(repeat_task);
|
||||||
}
|
}
|
||||||
fn kill_repeat_task(&mut self) {
|
|
||||||
|
async fn cancel_repeat_task(&mut self) {
|
||||||
if let Some(task) = self.repeat_task.take() {
|
if let Some(task) = self.repeat_task.take() {
|
||||||
task.abort();
|
self.notify_repeat_task.notify_waiters();
|
||||||
|
let _ = task.await;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -105,6 +124,77 @@ fn key_event(event_source: CGEventSource, key: u16, state: u8) {
|
|||||||
event.post(CGEventTapLocation::HID);
|
event.post(CGEventTapLocation::HID);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn get_display_at_point(x: CGFloat, y: CGFloat) -> Option<CGDirectDisplayID> {
|
||||||
|
let mut displays: [CGDirectDisplayID; 16] = [0; 16];
|
||||||
|
let mut display_count: u32 = 0;
|
||||||
|
let rect = CGRect::new(&CGPoint::new(x, y), &CGSize::new(0.0, 0.0));
|
||||||
|
|
||||||
|
let error = unsafe {
|
||||||
|
CGGetDisplaysWithRect(
|
||||||
|
rect,
|
||||||
|
1,
|
||||||
|
displays.as_mut_ptr(),
|
||||||
|
&mut display_count as *mut u32,
|
||||||
|
)
|
||||||
|
};
|
||||||
|
|
||||||
|
if error != 0 {
|
||||||
|
log::warn!("error getting displays at point ({}, {}): {}", x, y, error);
|
||||||
|
return Option::None;
|
||||||
|
}
|
||||||
|
|
||||||
|
if display_count == 0 {
|
||||||
|
log::debug!("no displays found at point ({}, {})", x, y);
|
||||||
|
return Option::None;
|
||||||
|
}
|
||||||
|
|
||||||
|
return displays.first().copied();
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_display_bounds(display: CGDirectDisplayID) -> (CGFloat, CGFloat, CGFloat, CGFloat) {
|
||||||
|
unsafe {
|
||||||
|
let bounds = CGDisplayBounds(display);
|
||||||
|
let min_x = bounds.origin.x;
|
||||||
|
let max_x = bounds.origin.x + bounds.size.width;
|
||||||
|
let min_y = bounds.origin.y;
|
||||||
|
let max_y = bounds.origin.y + bounds.size.height;
|
||||||
|
(min_x as f64, min_y as f64, max_x as f64, max_y as f64)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn clamp_to_screen_space(
|
||||||
|
current_x: CGFloat,
|
||||||
|
current_y: CGFloat,
|
||||||
|
dx: CGFloat,
|
||||||
|
dy: CGFloat,
|
||||||
|
) -> (CGFloat, CGFloat) {
|
||||||
|
// Check which display the mouse is currently on
|
||||||
|
// Determine what the location of the mouse would be after applying the move
|
||||||
|
// Get the display at the new location
|
||||||
|
// If the point is not on a display
|
||||||
|
// Clamp the mouse to the current display
|
||||||
|
// Else If the point is on a display
|
||||||
|
// Clamp the mouse to the new display
|
||||||
|
let current_display = match get_display_at_point(current_x, current_y) {
|
||||||
|
Some(display) => display,
|
||||||
|
None => {
|
||||||
|
log::warn!("could not get current display!");
|
||||||
|
return (current_x, current_y);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let new_x = current_x + dx;
|
||||||
|
let new_y = current_y + dy;
|
||||||
|
|
||||||
|
let final_display = get_display_at_point(new_x, new_y).unwrap_or(current_display);
|
||||||
|
let (min_x, min_y, max_x, max_y) = get_display_bounds(final_display);
|
||||||
|
|
||||||
|
(
|
||||||
|
new_x.clamp(min_x, max_x - 1.),
|
||||||
|
new_y.clamp(min_y, max_y - 1.),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
#[async_trait]
|
#[async_trait]
|
||||||
impl Emulation for MacOSEmulation {
|
impl Emulation for MacOSEmulation {
|
||||||
async fn consume(
|
async fn consume(
|
||||||
@@ -115,16 +205,6 @@ impl Emulation for MacOSEmulation {
|
|||||||
match event {
|
match event {
|
||||||
Event::Pointer(pointer_event) => match pointer_event {
|
Event::Pointer(pointer_event) => match pointer_event {
|
||||||
PointerEvent::Motion { time: _, dx, dy } => {
|
PointerEvent::Motion { time: _, dx, dy } => {
|
||||||
// FIXME secondary displays?
|
|
||||||
let (min_x, min_y, max_x, max_y) = unsafe {
|
|
||||||
let display = CGMainDisplayID();
|
|
||||||
let bounds = CGDisplayBounds(display);
|
|
||||||
let min_x = bounds.origin.x;
|
|
||||||
let max_x = bounds.origin.x + bounds.size.width;
|
|
||||||
let min_y = bounds.origin.y;
|
|
||||||
let max_y = bounds.origin.y + bounds.size.height;
|
|
||||||
(min_x as f64, min_y as f64, max_x as f64, max_y as f64)
|
|
||||||
};
|
|
||||||
let mut mouse_location = match self.get_mouse_location() {
|
let mut mouse_location = match self.get_mouse_location() {
|
||||||
Some(l) => l,
|
Some(l) => l,
|
||||||
None => {
|
None => {
|
||||||
@@ -133,8 +213,11 @@ impl Emulation for MacOSEmulation {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
mouse_location.x = (mouse_location.x + dx).clamp(min_x, max_x - 1.);
|
let (new_mouse_x, new_mouse_y) =
|
||||||
mouse_location.y = (mouse_location.y + dy).clamp(min_y, max_y - 1.);
|
clamp_to_screen_space(mouse_location.x, mouse_location.y, dx, dy);
|
||||||
|
|
||||||
|
mouse_location.x = new_mouse_x;
|
||||||
|
mouse_location.y = new_mouse_y;
|
||||||
|
|
||||||
let mut event_type = CGEventType::MouseMoved;
|
let mut event_type = CGEventType::MouseMoved;
|
||||||
if self.button_state.left {
|
if self.button_state.left {
|
||||||
@@ -262,7 +345,6 @@ impl Emulation for MacOSEmulation {
|
|||||||
};
|
};
|
||||||
event.post(CGEventTapLocation::HID);
|
event.post(CGEventTapLocation::HID);
|
||||||
}
|
}
|
||||||
PointerEvent::Frame { .. } => {}
|
|
||||||
},
|
},
|
||||||
Event::Keyboard(keyboard_event) => match keyboard_event {
|
Event::Keyboard(keyboard_event) => match keyboard_event {
|
||||||
KeyboardEvent::Key {
|
KeyboardEvent::Key {
|
||||||
@@ -280,13 +362,12 @@ impl Emulation for MacOSEmulation {
|
|||||||
match state {
|
match state {
|
||||||
// pressed
|
// pressed
|
||||||
1 => self.spawn_repeat_task(code).await,
|
1 => self.spawn_repeat_task(code).await,
|
||||||
_ => self.kill_repeat_task(),
|
_ => self.cancel_repeat_task().await,
|
||||||
}
|
}
|
||||||
key_event(self.event_source.clone(), code, state)
|
key_event(self.event_source.clone(), code, state)
|
||||||
}
|
}
|
||||||
KeyboardEvent::Modifiers { .. } => {}
|
KeyboardEvent::Modifiers { .. } => {}
|
||||||
},
|
},
|
||||||
_ => (),
|
|
||||||
}
|
}
|
||||||
// FIXME
|
// FIXME
|
||||||
Ok(())
|
Ok(())
|
||||||
|
|||||||
@@ -53,7 +53,6 @@ impl Emulation for WindowsEmulation {
|
|||||||
value,
|
value,
|
||||||
} => scroll(axis, value as i32),
|
} => scroll(axis, value as i32),
|
||||||
PointerEvent::AxisDiscrete120 { axis, value } => scroll(axis, value),
|
PointerEvent::AxisDiscrete120 { axis, value } => scroll(axis, value),
|
||||||
PointerEvent::Frame {} => {}
|
|
||||||
},
|
},
|
||||||
Event::Keyboard(keyboard_event) => match keyboard_event {
|
Event::Keyboard(keyboard_event) => match keyboard_event {
|
||||||
KeyboardEvent::Key {
|
KeyboardEvent::Key {
|
||||||
@@ -71,7 +70,6 @@ impl Emulation for WindowsEmulation {
|
|||||||
}
|
}
|
||||||
KeyboardEvent::Modifiers { .. } => {}
|
KeyboardEvent::Modifiers { .. } => {}
|
||||||
},
|
},
|
||||||
_ => {}
|
|
||||||
}
|
}
|
||||||
// FIXME
|
// FIXME
|
||||||
Ok(())
|
Ok(())
|
||||||
|
|||||||
@@ -200,7 +200,6 @@ impl VirtualInput {
|
|||||||
.axis_discrete(0, axis, value as f64 / 6., value / 120);
|
.axis_discrete(0, axis, value as f64 / 6., value / 120);
|
||||||
self.pointer.frame();
|
self.pointer.frame();
|
||||||
}
|
}
|
||||||
PointerEvent::Frame {} => self.pointer.frame(),
|
|
||||||
}
|
}
|
||||||
self.pointer.frame();
|
self.pointer.frame();
|
||||||
}
|
}
|
||||||
@@ -209,16 +208,15 @@ impl VirtualInput {
|
|||||||
self.keyboard.key(time, key, state as u32);
|
self.keyboard.key(time, key, state as u32);
|
||||||
}
|
}
|
||||||
KeyboardEvent::Modifiers {
|
KeyboardEvent::Modifiers {
|
||||||
mods_depressed,
|
depressed: mods_depressed,
|
||||||
mods_latched,
|
latched: mods_latched,
|
||||||
mods_locked,
|
locked: mods_locked,
|
||||||
group,
|
group,
|
||||||
} => {
|
} => {
|
||||||
self.keyboard
|
self.keyboard
|
||||||
.modifiers(mods_depressed, mods_latched, mods_locked, group);
|
.modifiers(mods_depressed, mods_latched, mods_locked, group);
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
_ => {}
|
|
||||||
}
|
}
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -123,7 +123,6 @@ impl Emulation for X11Emulation {
|
|||||||
PointerEvent::AxisDiscrete120 { axis, value } => {
|
PointerEvent::AxisDiscrete120 { axis, value } => {
|
||||||
self.emulate_scroll(axis, value as f64);
|
self.emulate_scroll(axis, value as f64);
|
||||||
}
|
}
|
||||||
PointerEvent::Frame {} => {}
|
|
||||||
},
|
},
|
||||||
Event::Keyboard(KeyboardEvent::Key {
|
Event::Keyboard(KeyboardEvent::Key {
|
||||||
time: _,
|
time: _,
|
||||||
|
|||||||
@@ -108,7 +108,6 @@ impl<'a> Emulation for DesktopPortalEmulation<'a> {
|
|||||||
.notify_pointer_axis(&self.session, dx, dy, true)
|
.notify_pointer_axis(&self.session, dx, dy, true)
|
||||||
.await?;
|
.await?;
|
||||||
}
|
}
|
||||||
PointerEvent::Frame {} => {}
|
|
||||||
},
|
},
|
||||||
Keyboard(k) => {
|
Keyboard(k) => {
|
||||||
match k {
|
match k {
|
||||||
@@ -130,7 +129,6 @@ impl<'a> Emulation for DesktopPortalEmulation<'a> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_ => {}
|
|
||||||
}
|
}
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,17 +1 @@
|
|||||||
use std::array::TryFromSliceError;
|
|
||||||
|
|
||||||
use thiserror::Error;
|
|
||||||
|
|
||||||
#[derive(Debug, Error)]
|
|
||||||
pub enum ProtocolError {
|
|
||||||
#[error(transparent)]
|
|
||||||
MissingData(#[from] TryFromSliceError),
|
|
||||||
#[error("invalid event id: `{0}`")]
|
|
||||||
InvalidEventId(u8),
|
|
||||||
#[error("invalid pointer event type: `{0}`")]
|
|
||||||
InvalidPointerEventId(u8),
|
|
||||||
#[error("invalid keyboard event type: `{0}`")]
|
|
||||||
InvalidKeyboardEventId(u8),
|
|
||||||
#[error("expected data at idx `{0:?}`")]
|
|
||||||
Data(String),
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,8 +1,6 @@
|
|||||||
pub use error::ProtocolError;
|
|
||||||
use std::fmt::{self, Display};
|
use std::fmt::{self, Display};
|
||||||
|
|
||||||
pub mod error;
|
pub mod error;
|
||||||
pub mod proto;
|
|
||||||
pub mod scancode;
|
pub mod scancode;
|
||||||
|
|
||||||
#[cfg(all(unix, feature = "libei", not(target_os = "macos")))]
|
#[cfg(all(unix, feature = "libei", not(target_os = "macos")))]
|
||||||
@@ -25,8 +23,6 @@ pub enum PointerEvent {
|
|||||||
Axis { time: u32, axis: u8, value: f64 },
|
Axis { time: u32, axis: u8, value: f64 },
|
||||||
/// discrete axis event, scroll event for mice - 120 = one scroll tick
|
/// discrete axis event, scroll event for mice - 120 = one scroll tick
|
||||||
AxisDiscrete120 { axis: u8, value: i32 },
|
AxisDiscrete120 { axis: u8, value: i32 },
|
||||||
/// frame event
|
|
||||||
Frame {},
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, PartialEq, Clone, Copy)]
|
#[derive(Debug, PartialEq, Clone, Copy)]
|
||||||
@@ -35,9 +31,9 @@ pub enum KeyboardEvent {
|
|||||||
Key { time: u32, key: u32, state: u8 },
|
Key { time: u32, key: u32, state: u8 },
|
||||||
/// modifiers changed state
|
/// modifiers changed state
|
||||||
Modifiers {
|
Modifiers {
|
||||||
mods_depressed: u32,
|
depressed: u32,
|
||||||
mods_latched: u32,
|
latched: u32,
|
||||||
mods_locked: u32,
|
locked: u32,
|
||||||
group: u32,
|
group: u32,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
@@ -48,23 +44,6 @@ pub enum Event {
|
|||||||
Pointer(PointerEvent),
|
Pointer(PointerEvent),
|
||||||
/// keyboard events (key / modifiers)
|
/// keyboard events (key / modifiers)
|
||||||
Keyboard(KeyboardEvent),
|
Keyboard(KeyboardEvent),
|
||||||
/// enter event: request to enter a client.
|
|
||||||
/// The client must release the pointer if it is grabbed
|
|
||||||
/// and reply with a leave event, as soon as its ready to
|
|
||||||
/// receive events
|
|
||||||
Enter(),
|
|
||||||
/// leave event: this client is now ready to receive events and will
|
|
||||||
/// not send any events after until it sends an enter event
|
|
||||||
Leave(),
|
|
||||||
/// ping a client, to see if it is still alive. A client that does
|
|
||||||
/// not respond with a pong event will be assumed to be offline.
|
|
||||||
Ping(),
|
|
||||||
/// response to a ping event: this event signals that a client
|
|
||||||
/// is still alive but must otherwise be ignored
|
|
||||||
Pong(),
|
|
||||||
/// explicit disconnect request. The client will no longer
|
|
||||||
/// send events until the next Enter event. All of its keys should be released.
|
|
||||||
Disconnect(),
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Display for PointerEvent {
|
impl Display for PointerEvent {
|
||||||
@@ -98,7 +77,6 @@ impl Display for PointerEvent {
|
|||||||
PointerEvent::AxisDiscrete120 { axis, value } => {
|
PointerEvent::AxisDiscrete120 { axis, value } => {
|
||||||
write!(f, "scroll-120 ({axis}, {value})")
|
write!(f, "scroll-120 ({axis}, {value})")
|
||||||
}
|
}
|
||||||
PointerEvent::Frame {} => write!(f, "frame()"),
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -119,9 +97,9 @@ impl Display for KeyboardEvent {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
KeyboardEvent::Modifiers {
|
KeyboardEvent::Modifiers {
|
||||||
mods_depressed,
|
depressed: mods_depressed,
|
||||||
mods_latched,
|
latched: mods_latched,
|
||||||
mods_locked,
|
locked: mods_locked,
|
||||||
group,
|
group,
|
||||||
} => write!(
|
} => write!(
|
||||||
f,
|
f,
|
||||||
@@ -136,11 +114,6 @@ impl Display for Event {
|
|||||||
match self {
|
match self {
|
||||||
Event::Pointer(p) => write!(f, "{}", p),
|
Event::Pointer(p) => write!(f, "{}", p),
|
||||||
Event::Keyboard(k) => write!(f, "{}", k),
|
Event::Keyboard(k) => write!(f, "{}", k),
|
||||||
Event::Enter() => write!(f, "enter"),
|
|
||||||
Event::Leave() => write!(f, "leave"),
|
|
||||||
Event::Ping() => write!(f, "ping"),
|
|
||||||
Event::Pong() => write!(f, "pong"),
|
|
||||||
Event::Disconnect() => write!(f, "disconnect"),
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -57,9 +57,9 @@ fn to_input_events(ei_event: EiEvent) -> Events {
|
|||||||
match ei_event {
|
match ei_event {
|
||||||
EiEvent::KeyboardModifiers(mods) => {
|
EiEvent::KeyboardModifiers(mods) => {
|
||||||
let modifier_event = KeyboardEvent::Modifiers {
|
let modifier_event = KeyboardEvent::Modifiers {
|
||||||
mods_depressed: mods.depressed,
|
depressed: mods.depressed,
|
||||||
mods_latched: mods.latched,
|
latched: mods.latched,
|
||||||
mods_locked: mods.locked,
|
locked: mods.locked,
|
||||||
group: mods.group,
|
group: mods.group,
|
||||||
};
|
};
|
||||||
Events::One(Event::Keyboard(modifier_event))
|
Events::One(Event::Keyboard(modifier_event))
|
||||||
|
|||||||
@@ -1,307 +0,0 @@
|
|||||||
use std::{fmt::Debug, slice::SliceIndex};
|
|
||||||
|
|
||||||
use crate::ProtocolError;
|
|
||||||
|
|
||||||
use super::{Event, KeyboardEvent, PointerEvent};
|
|
||||||
|
|
||||||
enum PointerEventType {
|
|
||||||
Motion,
|
|
||||||
Button,
|
|
||||||
Axis,
|
|
||||||
AxisDiscrete120,
|
|
||||||
Frame,
|
|
||||||
}
|
|
||||||
|
|
||||||
enum KeyboardEventType {
|
|
||||||
Key,
|
|
||||||
Modifiers,
|
|
||||||
}
|
|
||||||
|
|
||||||
enum EventType {
|
|
||||||
Pointer,
|
|
||||||
Keyboard,
|
|
||||||
Enter,
|
|
||||||
Leave,
|
|
||||||
Ping,
|
|
||||||
Pong,
|
|
||||||
Disconnect,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Event {
|
|
||||||
fn event_type(&self) -> EventType {
|
|
||||||
match self {
|
|
||||||
Self::Pointer(_) => EventType::Pointer,
|
|
||||||
Self::Keyboard(_) => EventType::Keyboard,
|
|
||||||
Self::Enter() => EventType::Enter,
|
|
||||||
Self::Leave() => EventType::Leave,
|
|
||||||
Self::Ping() => EventType::Ping,
|
|
||||||
Self::Pong() => EventType::Pong,
|
|
||||||
Self::Disconnect() => EventType::Disconnect,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl PointerEvent {
|
|
||||||
fn event_type(&self) -> PointerEventType {
|
|
||||||
match self {
|
|
||||||
Self::Motion { .. } => PointerEventType::Motion,
|
|
||||||
Self::Button { .. } => PointerEventType::Button,
|
|
||||||
Self::Axis { .. } => PointerEventType::Axis,
|
|
||||||
Self::AxisDiscrete120 { .. } => PointerEventType::AxisDiscrete120,
|
|
||||||
Self::Frame { .. } => PointerEventType::Frame,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl KeyboardEvent {
|
|
||||||
fn event_type(&self) -> KeyboardEventType {
|
|
||||||
match self {
|
|
||||||
KeyboardEvent::Key { .. } => KeyboardEventType::Key,
|
|
||||||
KeyboardEvent::Modifiers { .. } => KeyboardEventType::Modifiers,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl TryFrom<u8> for PointerEventType {
|
|
||||||
type Error = ProtocolError;
|
|
||||||
|
|
||||||
fn try_from(value: u8) -> Result<Self, ProtocolError> {
|
|
||||||
match value {
|
|
||||||
x if x == Self::Motion as u8 => Ok(Self::Motion),
|
|
||||||
x if x == Self::Button as u8 => Ok(Self::Button),
|
|
||||||
x if x == Self::Axis as u8 => Ok(Self::Axis),
|
|
||||||
x if x == Self::AxisDiscrete120 as u8 => Ok(Self::AxisDiscrete120),
|
|
||||||
x if x == Self::Frame as u8 => Ok(Self::Frame),
|
|
||||||
_ => Err(ProtocolError::InvalidPointerEventId(value)),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl TryFrom<u8> for KeyboardEventType {
|
|
||||||
type Error = ProtocolError;
|
|
||||||
|
|
||||||
fn try_from(value: u8) -> Result<Self, ProtocolError> {
|
|
||||||
match value {
|
|
||||||
x if x == Self::Key as u8 => Ok(Self::Key),
|
|
||||||
x if x == Self::Modifiers as u8 => Ok(Self::Modifiers),
|
|
||||||
_ => Err(ProtocolError::InvalidKeyboardEventId(value)),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl From<&Event> for Vec<u8> {
|
|
||||||
fn from(event: &Event) -> Self {
|
|
||||||
let event_id = vec![event.event_type() as u8];
|
|
||||||
let event_data = match event {
|
|
||||||
Event::Pointer(p) => p.into(),
|
|
||||||
Event::Keyboard(k) => k.into(),
|
|
||||||
Event::Enter() => vec![],
|
|
||||||
Event::Leave() => vec![],
|
|
||||||
Event::Ping() => vec![],
|
|
||||||
Event::Pong() => vec![],
|
|
||||||
Event::Disconnect() => vec![],
|
|
||||||
};
|
|
||||||
[event_id, event_data].concat()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl TryFrom<Vec<u8>> for Event {
|
|
||||||
type Error = ProtocolError;
|
|
||||||
|
|
||||||
fn try_from(value: Vec<u8>) -> Result<Self, ProtocolError> {
|
|
||||||
let event_id = u8::from_be_bytes(value[..1].try_into()?);
|
|
||||||
match event_id {
|
|
||||||
i if i == (EventType::Pointer as u8) => Ok(Event::Pointer(value.try_into()?)),
|
|
||||||
i if i == (EventType::Keyboard as u8) => Ok(Event::Keyboard(value.try_into()?)),
|
|
||||||
i if i == (EventType::Enter as u8) => Ok(Event::Enter()),
|
|
||||||
i if i == (EventType::Leave as u8) => Ok(Event::Leave()),
|
|
||||||
i if i == (EventType::Ping as u8) => Ok(Event::Ping()),
|
|
||||||
i if i == (EventType::Pong as u8) => Ok(Event::Pong()),
|
|
||||||
i if i == (EventType::Disconnect as u8) => Ok(Event::Disconnect()),
|
|
||||||
_ => Err(ProtocolError::InvalidEventId(event_id)),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl From<&PointerEvent> for Vec<u8> {
|
|
||||||
fn from(event: &PointerEvent) -> Self {
|
|
||||||
let id = vec![event.event_type() as u8];
|
|
||||||
let data = match event {
|
|
||||||
PointerEvent::Motion { time, dx, dy } => {
|
|
||||||
let time = time.to_be_bytes();
|
|
||||||
let dx = dx.to_be_bytes();
|
|
||||||
let dy = dy.to_be_bytes();
|
|
||||||
[&time[..], &dx[..], &dy[..]].concat()
|
|
||||||
}
|
|
||||||
PointerEvent::Button {
|
|
||||||
time,
|
|
||||||
button,
|
|
||||||
state,
|
|
||||||
} => {
|
|
||||||
let time = time.to_be_bytes();
|
|
||||||
let button = button.to_be_bytes();
|
|
||||||
let state = state.to_be_bytes();
|
|
||||||
[&time[..], &button[..], &state[..]].concat()
|
|
||||||
}
|
|
||||||
PointerEvent::Axis { time, axis, value } => {
|
|
||||||
let time = time.to_be_bytes();
|
|
||||||
let axis = axis.to_be_bytes();
|
|
||||||
let value = value.to_be_bytes();
|
|
||||||
[&time[..], &axis[..], &value[..]].concat()
|
|
||||||
}
|
|
||||||
PointerEvent::AxisDiscrete120 { axis, value } => {
|
|
||||||
let axis = axis.to_be_bytes();
|
|
||||||
let value = value.to_be_bytes();
|
|
||||||
[&axis[..], &value[..]].concat()
|
|
||||||
}
|
|
||||||
PointerEvent::Frame {} => {
|
|
||||||
vec![]
|
|
||||||
}
|
|
||||||
};
|
|
||||||
[id, data].concat()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn decode_u8<I>(data: &[u8], idx: I) -> Result<u8, ProtocolError>
|
|
||||||
where
|
|
||||||
I: SliceIndex<[u8], Output = [u8]> + Debug + Clone,
|
|
||||||
{
|
|
||||||
let data = data
|
|
||||||
.get(idx.clone())
|
|
||||||
.ok_or(ProtocolError::Data(format!("{:?}", idx)))?;
|
|
||||||
Ok(u8::from_be_bytes(data.try_into()?))
|
|
||||||
}
|
|
||||||
|
|
||||||
fn decode_u32<I>(data: &[u8], idx: I) -> Result<u32, ProtocolError>
|
|
||||||
where
|
|
||||||
I: SliceIndex<[u8], Output = [u8]> + Debug + Clone,
|
|
||||||
{
|
|
||||||
let data = data
|
|
||||||
.get(idx.clone())
|
|
||||||
.ok_or(ProtocolError::Data(format!("{:?}", idx)))?;
|
|
||||||
Ok(u32::from_be_bytes(data.try_into()?))
|
|
||||||
}
|
|
||||||
|
|
||||||
fn decode_i32<I>(data: &[u8], idx: I) -> Result<i32, ProtocolError>
|
|
||||||
where
|
|
||||||
I: SliceIndex<[u8], Output = [u8]> + Debug + Clone,
|
|
||||||
{
|
|
||||||
let data = data
|
|
||||||
.get(idx.clone())
|
|
||||||
.ok_or(ProtocolError::Data(format!("{:?}", idx)))?;
|
|
||||||
Ok(i32::from_be_bytes(data.try_into()?))
|
|
||||||
}
|
|
||||||
fn decode_f64<I>(data: &[u8], idx: I) -> Result<f64, ProtocolError>
|
|
||||||
where
|
|
||||||
I: SliceIndex<[u8], Output = [u8]> + Debug + Clone,
|
|
||||||
{
|
|
||||||
let data = data
|
|
||||||
.get(idx.clone())
|
|
||||||
.ok_or(ProtocolError::Data(format!("{:?}", idx)))?;
|
|
||||||
Ok(f64::from_be_bytes(data.try_into()?))
|
|
||||||
}
|
|
||||||
|
|
||||||
impl TryFrom<Vec<u8>> for PointerEvent {
|
|
||||||
type Error = ProtocolError;
|
|
||||||
|
|
||||||
fn try_from(data: Vec<u8>) -> Result<Self, ProtocolError> {
|
|
||||||
match data.get(1) {
|
|
||||||
Some(id) => match id.to_owned().try_into()? {
|
|
||||||
PointerEventType::Motion => {
|
|
||||||
let time = decode_u32(&data, 2..6)?;
|
|
||||||
let dx = decode_f64(&data, 6..14)?;
|
|
||||||
let dy = decode_f64(&data, 14..22)?;
|
|
||||||
|
|
||||||
Ok(Self::Motion { time, dx, dy })
|
|
||||||
}
|
|
||||||
PointerEventType::Button => {
|
|
||||||
let time = decode_u32(&data, 2..6)?;
|
|
||||||
let button = decode_u32(&data, 6..10)?;
|
|
||||||
let state = decode_u32(&data, 10..14)?;
|
|
||||||
|
|
||||||
Ok(Self::Button {
|
|
||||||
time,
|
|
||||||
button,
|
|
||||||
state,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
PointerEventType::Axis => {
|
|
||||||
let time = decode_u32(&data, 2..6)?;
|
|
||||||
let axis = decode_u8(&data, 6..7)?;
|
|
||||||
let value = decode_f64(&data, 7..15)?;
|
|
||||||
Ok(Self::Axis { time, axis, value })
|
|
||||||
}
|
|
||||||
PointerEventType::AxisDiscrete120 => {
|
|
||||||
let axis = decode_u8(&data, 2..3)?;
|
|
||||||
let value = decode_i32(&data, 3..7)?;
|
|
||||||
Ok(Self::AxisDiscrete120 { axis, value })
|
|
||||||
}
|
|
||||||
PointerEventType::Frame => Ok(Self::Frame {}),
|
|
||||||
},
|
|
||||||
None => Err(ProtocolError::Data("0".to_string())),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl From<&KeyboardEvent> for Vec<u8> {
|
|
||||||
fn from(event: &KeyboardEvent) -> Self {
|
|
||||||
let id = vec![event.event_type() as u8];
|
|
||||||
let data = match event {
|
|
||||||
KeyboardEvent::Key { time, key, state } => {
|
|
||||||
let time = time.to_be_bytes();
|
|
||||||
let key = key.to_be_bytes();
|
|
||||||
let state = state.to_be_bytes();
|
|
||||||
[&time[..], &key[..], &state[..]].concat()
|
|
||||||
}
|
|
||||||
KeyboardEvent::Modifiers {
|
|
||||||
mods_depressed,
|
|
||||||
mods_latched,
|
|
||||||
mods_locked,
|
|
||||||
group,
|
|
||||||
} => {
|
|
||||||
let mods_depressed = mods_depressed.to_be_bytes();
|
|
||||||
let mods_latched = mods_latched.to_be_bytes();
|
|
||||||
let mods_locked = mods_locked.to_be_bytes();
|
|
||||||
let group = group.to_be_bytes();
|
|
||||||
[
|
|
||||||
&mods_depressed[..],
|
|
||||||
&mods_latched[..],
|
|
||||||
&mods_locked[..],
|
|
||||||
&group[..],
|
|
||||||
]
|
|
||||||
.concat()
|
|
||||||
}
|
|
||||||
};
|
|
||||||
[id, data].concat()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl TryFrom<Vec<u8>> for KeyboardEvent {
|
|
||||||
type Error = ProtocolError;
|
|
||||||
|
|
||||||
fn try_from(data: Vec<u8>) -> Result<Self, ProtocolError> {
|
|
||||||
match data.get(1) {
|
|
||||||
Some(id) => match id.to_owned().try_into()? {
|
|
||||||
KeyboardEventType::Key => {
|
|
||||||
let time = decode_u32(&data, 2..6)?;
|
|
||||||
let key = decode_u32(&data, 6..10)?;
|
|
||||||
let state = decode_u8(&data, 10..11)?;
|
|
||||||
Ok(KeyboardEvent::Key { time, key, state })
|
|
||||||
}
|
|
||||||
KeyboardEventType::Modifiers => {
|
|
||||||
let mods_depressed = decode_u32(&data, 2..6)?;
|
|
||||||
let mods_latched = decode_u32(&data, 6..10)?;
|
|
||||||
let mods_locked = decode_u32(&data, 10..14)?;
|
|
||||||
let group = decode_u32(&data, 14..18)?;
|
|
||||||
Ok(KeyboardEvent::Modifiers {
|
|
||||||
mods_depressed,
|
|
||||||
mods_latched,
|
|
||||||
mods_locked,
|
|
||||||
group,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
},
|
|
||||||
None => Err(ProtocolError::Data("0".to_string())),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
18
lan-mouse-cli/Cargo.toml
Normal file
18
lan-mouse-cli/Cargo.toml
Normal file
@@ -0,0 +1,18 @@
|
|||||||
|
[package]
|
||||||
|
name = "lan-mouse-cli"
|
||||||
|
description = "CLI Frontend for lan-mouse"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2021"
|
||||||
|
license = "GPL-3.0-or-later"
|
||||||
|
repository = "https://github.com/feschber/lan-mouse"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
futures = "0.3.30"
|
||||||
|
lan-mouse-ipc = { path = "../lan-mouse-ipc", version = "0.1.0" }
|
||||||
|
tokio = { version = "1.32.0", features = [
|
||||||
|
"io-util",
|
||||||
|
"io-std",
|
||||||
|
"macros",
|
||||||
|
"net",
|
||||||
|
"rt",
|
||||||
|
] }
|
||||||
@@ -3,7 +3,7 @@ use std::{
|
|||||||
str::{FromStr, SplitWhitespace},
|
str::{FromStr, SplitWhitespace},
|
||||||
};
|
};
|
||||||
|
|
||||||
use crate::client::{ClientHandle, Position};
|
use lan_mouse_ipc::{ClientHandle, Position};
|
||||||
|
|
||||||
pub(super) enum CommandType {
|
pub(super) enum CommandType {
|
||||||
NoCommand,
|
NoCommand,
|
||||||
@@ -1,78 +1,65 @@
|
|||||||
use anyhow::{anyhow, Result};
|
use futures::StreamExt;
|
||||||
use tokio::{
|
use tokio::{
|
||||||
io::{AsyncBufReadExt, AsyncReadExt, AsyncWriteExt, BufReader},
|
io::{AsyncBufReadExt, BufReader},
|
||||||
task::LocalSet,
|
task::LocalSet,
|
||||||
};
|
};
|
||||||
|
|
||||||
#[cfg(windows)]
|
|
||||||
use tokio::net::tcp::{ReadHalf, WriteHalf};
|
|
||||||
#[cfg(unix)]
|
|
||||||
use tokio::net::unix::{ReadHalf, WriteHalf};
|
|
||||||
|
|
||||||
use std::io::{self, Write};
|
use std::io::{self, Write};
|
||||||
|
|
||||||
use crate::{
|
|
||||||
client::{ClientConfig, ClientHandle, ClientState},
|
|
||||||
config::DEFAULT_PORT,
|
|
||||||
};
|
|
||||||
|
|
||||||
use self::command::{Command, CommandType};
|
use self::command::{Command, CommandType};
|
||||||
|
|
||||||
use super::{FrontendEvent, FrontendRequest};
|
use lan_mouse_ipc::{
|
||||||
|
AsyncFrontendEventReader, AsyncFrontendRequestWriter, ClientConfig, ClientHandle, ClientState,
|
||||||
|
FrontendEvent, FrontendRequest, IpcError, DEFAULT_PORT,
|
||||||
|
};
|
||||||
|
|
||||||
mod command;
|
mod command;
|
||||||
|
|
||||||
pub fn run() -> Result<()> {
|
pub fn run() -> Result<(), IpcError> {
|
||||||
let Ok(stream) = super::wait_for_service() else {
|
|
||||||
return Err(anyhow!("Could not connect to lan-mouse-socket"));
|
|
||||||
};
|
|
||||||
|
|
||||||
let runtime = tokio::runtime::Builder::new_current_thread()
|
let runtime = tokio::runtime::Builder::new_current_thread()
|
||||||
.enable_io()
|
.enable_io()
|
||||||
.enable_time()
|
.enable_time()
|
||||||
.build()?;
|
.build()?;
|
||||||
runtime.block_on(LocalSet::new().run_until(async move {
|
runtime.block_on(LocalSet::new().run_until(async move {
|
||||||
stream.set_nonblocking(true)?;
|
let (rx, tx) = lan_mouse_ipc::connect_async().await?;
|
||||||
#[cfg(unix)]
|
|
||||||
let mut stream = tokio::net::UnixStream::from_std(stream)?;
|
|
||||||
#[cfg(windows)]
|
|
||||||
let mut stream = tokio::net::TcpStream::from_std(stream)?;
|
|
||||||
let (rx, tx) = stream.split();
|
|
||||||
|
|
||||||
let mut cli = Cli::new(rx, tx);
|
let mut cli = Cli::new(rx, tx);
|
||||||
cli.run().await
|
cli.run().await
|
||||||
}))?;
|
}))?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
struct Cli<'a> {
|
struct Cli {
|
||||||
clients: Vec<(ClientHandle, ClientConfig, ClientState)>,
|
clients: Vec<(ClientHandle, ClientConfig, ClientState)>,
|
||||||
rx: ReadHalf<'a>,
|
changed: Option<ClientHandle>,
|
||||||
tx: WriteHalf<'a>,
|
rx: AsyncFrontendEventReader,
|
||||||
|
tx: AsyncFrontendRequestWriter,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'a> Cli<'a> {
|
impl Cli {
|
||||||
fn new(rx: ReadHalf<'a>, tx: WriteHalf<'a>) -> Cli<'a> {
|
fn new(rx: AsyncFrontendEventReader, tx: AsyncFrontendRequestWriter) -> Cli {
|
||||||
Self {
|
Self {
|
||||||
clients: vec![],
|
clients: vec![],
|
||||||
|
changed: None,
|
||||||
rx,
|
rx,
|
||||||
tx,
|
tx,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn run(&mut self) -> Result<()> {
|
async fn run(&mut self) -> Result<(), IpcError> {
|
||||||
let stdin = tokio::io::stdin();
|
let stdin = tokio::io::stdin();
|
||||||
let stdin = BufReader::new(stdin);
|
let stdin = BufReader::new(stdin);
|
||||||
let mut stdin = stdin.lines();
|
let mut stdin = stdin.lines();
|
||||||
|
|
||||||
/* initial state sync */
|
/* initial state sync */
|
||||||
let request = FrontendRequest::Enumerate();
|
|
||||||
self.send_request(request).await?;
|
|
||||||
|
|
||||||
self.clients = loop {
|
self.clients = loop {
|
||||||
let event = self.await_event().await?;
|
match self.rx.next().await {
|
||||||
if let FrontendEvent::Enumerate(clients) = event {
|
Some(Ok(e)) => {
|
||||||
break clients;
|
if let FrontendEvent::Enumerate(clients) = e {
|
||||||
|
break clients;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Some(Err(e)) => return Err(e),
|
||||||
|
None => return Ok(()),
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -92,18 +79,23 @@ impl<'a> Cli<'a> {
|
|||||||
};
|
};
|
||||||
self.execute(cmd).await?;
|
self.execute(cmd).await?;
|
||||||
}
|
}
|
||||||
event = self.await_event() => {
|
event = self.rx.next() => {
|
||||||
let event = event?;
|
if let Some(event) = event {
|
||||||
self.handle_event(event);
|
self.handle_event(event?);
|
||||||
|
} else {
|
||||||
|
break Ok(());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
if let Some(handle) = self.changed.take() {
|
||||||
|
self.update_client(handle).await?;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn update_client(&mut self, handle: ClientHandle) -> Result<()> {
|
async fn update_client(&mut self, handle: ClientHandle) -> Result<(), IpcError> {
|
||||||
self.send_request(FrontendRequest::GetState(handle)).await?;
|
self.tx.request(FrontendRequest::GetState(handle)).await?;
|
||||||
loop {
|
while let Some(Ok(event)) = self.rx.next().await {
|
||||||
let event = self.await_event().await?;
|
|
||||||
self.handle_event(event.clone());
|
self.handle_event(event.clone());
|
||||||
if let FrontendEvent::State(_, _, _) | FrontendEvent::NoSuchClient(_) = event {
|
if let FrontendEvent::State(_, _, _) | FrontendEvent::NoSuchClient(_) = event {
|
||||||
break;
|
break;
|
||||||
@@ -112,22 +104,23 @@ impl<'a> Cli<'a> {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn execute(&mut self, cmd: Command) -> Result<()> {
|
async fn execute(&mut self, cmd: Command) -> Result<(), IpcError> {
|
||||||
match cmd {
|
match cmd {
|
||||||
Command::None => {}
|
Command::None => {}
|
||||||
Command::Connect(pos, host, port) => {
|
Command::Connect(pos, host, port) => {
|
||||||
let request = FrontendRequest::Create;
|
let request = FrontendRequest::Create;
|
||||||
self.send_request(request).await?;
|
self.tx.request(request).await?;
|
||||||
let handle = loop {
|
let handle = loop {
|
||||||
let event = self.await_event().await?;
|
if let Some(Ok(event)) = self.rx.next().await {
|
||||||
match event {
|
match event {
|
||||||
FrontendEvent::Created(h, c, s) => {
|
FrontendEvent::Created(h, c, s) => {
|
||||||
self.clients.push((h, c, s));
|
self.clients.push((h, c, s));
|
||||||
break h;
|
break h;
|
||||||
}
|
}
|
||||||
_ => {
|
_ => {
|
||||||
self.handle_event(event);
|
self.handle_event(event);
|
||||||
continue;
|
continue;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@@ -136,35 +129,36 @@ impl<'a> Cli<'a> {
|
|||||||
FrontendRequest::UpdatePort(handle, port.unwrap_or(DEFAULT_PORT)),
|
FrontendRequest::UpdatePort(handle, port.unwrap_or(DEFAULT_PORT)),
|
||||||
FrontendRequest::UpdatePosition(handle, pos),
|
FrontendRequest::UpdatePosition(handle, pos),
|
||||||
] {
|
] {
|
||||||
self.send_request(request).await?;
|
self.tx.request(request).await?;
|
||||||
}
|
}
|
||||||
self.update_client(handle).await?;
|
self.update_client(handle).await?;
|
||||||
}
|
}
|
||||||
Command::Disconnect(id) => {
|
Command::Disconnect(id) => {
|
||||||
self.send_request(FrontendRequest::Delete(id)).await?;
|
self.tx.request(FrontendRequest::Delete(id)).await?;
|
||||||
loop {
|
loop {
|
||||||
let event = self.await_event().await?;
|
if let Some(Ok(event)) = self.rx.next().await {
|
||||||
self.handle_event(event.clone());
|
self.handle_event(event.clone());
|
||||||
if let FrontendEvent::Deleted(_) = event {
|
if let FrontendEvent::Deleted(_) = event {
|
||||||
self.handle_event(event);
|
self.handle_event(event);
|
||||||
break;
|
break;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Command::Activate(id) => {
|
Command::Activate(id) => {
|
||||||
self.send_request(FrontendRequest::Activate(id, true))
|
self.tx.request(FrontendRequest::Activate(id, true)).await?;
|
||||||
.await?;
|
|
||||||
self.update_client(id).await?;
|
self.update_client(id).await?;
|
||||||
}
|
}
|
||||||
Command::Deactivate(id) => {
|
Command::Deactivate(id) => {
|
||||||
self.send_request(FrontendRequest::Activate(id, false))
|
self.tx
|
||||||
|
.request(FrontendRequest::Activate(id, false))
|
||||||
.await?;
|
.await?;
|
||||||
self.update_client(id).await?;
|
self.update_client(id).await?;
|
||||||
}
|
}
|
||||||
Command::List => {
|
Command::List => {
|
||||||
self.send_request(FrontendRequest::Enumerate()).await?;
|
self.tx.request(FrontendRequest::Enumerate()).await?;
|
||||||
loop {
|
while let Some(e) = self.rx.next().await {
|
||||||
let event = self.await_event().await?;
|
let event = e?;
|
||||||
self.handle_event(event.clone());
|
self.handle_event(event.clone());
|
||||||
if let FrontendEvent::Enumerate(_) = event {
|
if let FrontendEvent::Enumerate(_) = event {
|
||||||
break;
|
break;
|
||||||
@@ -173,12 +167,12 @@ impl<'a> Cli<'a> {
|
|||||||
}
|
}
|
||||||
Command::SetHost(handle, host) => {
|
Command::SetHost(handle, host) => {
|
||||||
let request = FrontendRequest::UpdateHostname(handle, Some(host.clone()));
|
let request = FrontendRequest::UpdateHostname(handle, Some(host.clone()));
|
||||||
self.send_request(request).await?;
|
self.tx.request(request).await?;
|
||||||
self.update_client(handle).await?;
|
self.update_client(handle).await?;
|
||||||
}
|
}
|
||||||
Command::SetPort(handle, port) => {
|
Command::SetPort(handle, port) => {
|
||||||
let request = FrontendRequest::UpdatePort(handle, port.unwrap_or(DEFAULT_PORT));
|
let request = FrontendRequest::UpdatePort(handle, port.unwrap_or(DEFAULT_PORT));
|
||||||
self.send_request(request).await?;
|
self.tx.request(request).await?;
|
||||||
self.update_client(handle).await?;
|
self.update_client(handle).await?;
|
||||||
}
|
}
|
||||||
Command::Help => {
|
Command::Help => {
|
||||||
@@ -215,6 +209,7 @@ impl<'a> Cli<'a> {
|
|||||||
|
|
||||||
fn handle_event(&mut self, event: FrontendEvent) {
|
fn handle_event(&mut self, event: FrontendEvent) {
|
||||||
match event {
|
match event {
|
||||||
|
FrontendEvent::Changed(h) => self.changed = Some(h),
|
||||||
FrontendEvent::Created(h, c, s) => {
|
FrontendEvent::Created(h, c, s) => {
|
||||||
eprint!("client added ({h}): ");
|
eprint!("client added ({h}): ");
|
||||||
print_config(&c);
|
print_config(&c);
|
||||||
@@ -291,23 +286,6 @@ impl<'a> Cli<'a> {
|
|||||||
eprintln!();
|
eprintln!();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn send_request(&mut self, request: FrontendRequest) -> io::Result<()> {
|
|
||||||
let json = serde_json::to_string(&request).unwrap();
|
|
||||||
let bytes = json.as_bytes();
|
|
||||||
let len = bytes.len();
|
|
||||||
self.tx.write_u64(len as u64).await?;
|
|
||||||
self.tx.write_all(bytes).await?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn await_event(&mut self) -> Result<FrontendEvent> {
|
|
||||||
let len = self.rx.read_u64().await?;
|
|
||||||
let mut buf = vec![0u8; len as usize];
|
|
||||||
self.rx.read_exact(&mut buf).await?;
|
|
||||||
let event: FrontendEvent = serde_json::from_slice(&buf)?;
|
|
||||||
Ok(event)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn prompt() -> io::Result<()> {
|
fn prompt() -> io::Result<()> {
|
||||||
18
lan-mouse-gtk/Cargo.toml
Normal file
18
lan-mouse-gtk/Cargo.toml
Normal file
@@ -0,0 +1,18 @@
|
|||||||
|
[package]
|
||||||
|
name = "lan-mouse-gtk"
|
||||||
|
description = "GTK4 / Libadwaita Frontend for lan-mouse"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2021"
|
||||||
|
license = "GPL-3.0-or-later"
|
||||||
|
repository = "https://github.com/feschber/lan-mouse"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
gtk = { package = "gtk4", version = "0.9.0", features = ["v4_2"] }
|
||||||
|
adw = { package = "libadwaita", version = "0.7.0", features = ["v1_1"] }
|
||||||
|
async-channel = { version = "2.1.1" }
|
||||||
|
hostname = "0.4.0"
|
||||||
|
log = "0.4.20"
|
||||||
|
lan-mouse-ipc = { path = "../lan-mouse-ipc", version = "0.1.0" }
|
||||||
|
|
||||||
|
[build-dependencies]
|
||||||
|
glib-build-tools = { version = "0.20.0" }
|
||||||
8
lan-mouse-gtk/build.rs
Normal file
8
lan-mouse-gtk/build.rs
Normal file
@@ -0,0 +1,8 @@
|
|||||||
|
fn main() {
|
||||||
|
// composite_templates
|
||||||
|
glib_build_tools::compile_resources(
|
||||||
|
&["resources"],
|
||||||
|
"resources/resources.gresource.xml",
|
||||||
|
"lan-mouse.gresource",
|
||||||
|
);
|
||||||
|
}
|
||||||
|
Before Width: | Height: | Size: 7.6 KiB After Width: | Height: | Size: 7.6 KiB |
@@ -3,7 +3,7 @@ mod imp;
|
|||||||
use adw::subclass::prelude::*;
|
use adw::subclass::prelude::*;
|
||||||
use gtk::glib::{self, Object};
|
use gtk::glib::{self, Object};
|
||||||
|
|
||||||
use crate::client::{ClientConfig, ClientHandle, ClientState};
|
use lan_mouse_ipc::{ClientConfig, ClientHandle, ClientState};
|
||||||
|
|
||||||
glib::wrapper! {
|
glib::wrapper! {
|
||||||
pub struct ClientObject(ObjectSubclass<imp::ClientObject>);
|
pub struct ClientObject(ObjectSubclass<imp::ClientObject>);
|
||||||
@@ -5,7 +5,7 @@ use gtk::glib;
|
|||||||
use gtk::prelude::*;
|
use gtk::prelude::*;
|
||||||
use gtk::subclass::prelude::*;
|
use gtk::subclass::prelude::*;
|
||||||
|
|
||||||
use crate::client::ClientHandle;
|
use lan_mouse_ipc::ClientHandle;
|
||||||
|
|
||||||
use super::ClientData;
|
use super::ClientData;
|
||||||
|
|
||||||
@@ -4,7 +4,7 @@ use adw::prelude::*;
|
|||||||
use adw::subclass::prelude::*;
|
use adw::subclass::prelude::*;
|
||||||
use gtk::glib::{self, Object};
|
use gtk::glib::{self, Object};
|
||||||
|
|
||||||
use crate::config::DEFAULT_PORT;
|
use lan_mouse_ipc::DEFAULT_PORT;
|
||||||
|
|
||||||
use super::ClientObject;
|
use super::ClientObject;
|
||||||
|
|
||||||
@@ -2,16 +2,13 @@ mod client_object;
|
|||||||
mod client_row;
|
mod client_row;
|
||||||
mod window;
|
mod window;
|
||||||
|
|
||||||
use std::{
|
use std::{env, process, str};
|
||||||
env,
|
|
||||||
io::{ErrorKind, Read},
|
|
||||||
process, str,
|
|
||||||
};
|
|
||||||
|
|
||||||
use crate::frontend::gtk::window::Window;
|
use window::Window;
|
||||||
|
|
||||||
|
use lan_mouse_ipc::{FrontendEvent, FrontendRequest};
|
||||||
|
|
||||||
use adw::Application;
|
use adw::Application;
|
||||||
use endi::{Endian, ReadBytes};
|
|
||||||
use gtk::{
|
use gtk::{
|
||||||
gdk::Display, glib::clone, prelude::*, subclass::prelude::ObjectSubclassIsExt, IconTheme,
|
gdk::Display, glib::clone, prelude::*, subclass::prelude::ObjectSubclassIsExt, IconTheme,
|
||||||
};
|
};
|
||||||
@@ -19,8 +16,6 @@ use gtk::{gio, glib, prelude::ApplicationExt};
|
|||||||
|
|
||||||
use self::client_object::ClientObject;
|
use self::client_object::ClientObject;
|
||||||
|
|
||||||
use super::FrontendEvent;
|
|
||||||
|
|
||||||
pub fn run() -> glib::ExitCode {
|
pub fn run() -> glib::ExitCode {
|
||||||
log::debug!("running gtk frontend");
|
log::debug!("running gtk frontend");
|
||||||
#[cfg(windows)]
|
#[cfg(windows)]
|
||||||
@@ -65,15 +60,8 @@ fn load_icons() {
|
|||||||
|
|
||||||
fn build_ui(app: &Application) {
|
fn build_ui(app: &Application) {
|
||||||
log::debug!("connecting to lan-mouse-socket");
|
log::debug!("connecting to lan-mouse-socket");
|
||||||
let mut rx = match super::wait_for_service() {
|
let (mut frontend_rx, frontend_tx) = match lan_mouse_ipc::connect() {
|
||||||
Ok(stream) => stream,
|
Ok(conn) => conn,
|
||||||
Err(e) => {
|
|
||||||
log::error!("could not connect to lan-mouse-socket: {e}");
|
|
||||||
process::exit(1);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
let tx = match rx.try_clone() {
|
|
||||||
Ok(sock) => sock,
|
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
log::error!("{e}");
|
log::error!("{e}");
|
||||||
process::exit(1);
|
process::exit(1);
|
||||||
@@ -84,35 +72,18 @@ fn build_ui(app: &Application) {
|
|||||||
let (sender, receiver) = async_channel::bounded(10);
|
let (sender, receiver) = async_channel::bounded(10);
|
||||||
|
|
||||||
gio::spawn_blocking(move || {
|
gio::spawn_blocking(move || {
|
||||||
match loop {
|
while let Some(e) = frontend_rx.next_event() {
|
||||||
// read length
|
match e {
|
||||||
let len = match rx.read_u64(Endian::Big) {
|
Ok(e) => sender.send_blocking(e).unwrap(),
|
||||||
Ok(l) => l,
|
Err(e) => {
|
||||||
Err(e) if e.kind() == ErrorKind::UnexpectedEof => break Ok(()),
|
log::error!("{e}");
|
||||||
Err(e) => break Err(e),
|
break;
|
||||||
};
|
}
|
||||||
|
|
||||||
// read payload
|
|
||||||
let mut buf = vec![0u8; len as usize];
|
|
||||||
match rx.read_exact(&mut buf) {
|
|
||||||
Ok(_) => (),
|
|
||||||
Err(e) if e.kind() == ErrorKind::UnexpectedEof => break Ok(()),
|
|
||||||
Err(e) => break Err(e),
|
|
||||||
};
|
|
||||||
|
|
||||||
// parse json
|
|
||||||
let json = str::from_utf8(&buf).unwrap();
|
|
||||||
match serde_json::from_str(json) {
|
|
||||||
Ok(notify) => sender.send_blocking(notify).unwrap(),
|
|
||||||
Err(e) => log::error!("{e}"),
|
|
||||||
}
|
}
|
||||||
} {
|
|
||||||
Ok(()) => {}
|
|
||||||
Err(e) => log::error!("{e}"),
|
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
let window = Window::new(app, tx);
|
let window = Window::new(app, frontend_tx);
|
||||||
|
|
||||||
glib::spawn_future_local(clone!(
|
glib::spawn_future_local(clone!(
|
||||||
#[weak]
|
#[weak]
|
||||||
@@ -121,6 +92,9 @@ fn build_ui(app: &Application) {
|
|||||||
loop {
|
loop {
|
||||||
let notify = receiver.recv().await.unwrap_or_else(|_| process::exit(1));
|
let notify = receiver.recv().await.unwrap_or_else(|_| process::exit(1));
|
||||||
match notify {
|
match notify {
|
||||||
|
FrontendEvent::Changed(handle) => {
|
||||||
|
window.request(FrontendRequest::GetState(handle));
|
||||||
|
}
|
||||||
FrontendEvent::Created(handle, client, state) => {
|
FrontendEvent::Created(handle, client, state) => {
|
||||||
window.new_client(handle, client, state);
|
window.new_client(handle, client, state);
|
||||||
}
|
}
|
||||||
@@ -1,16 +1,7 @@
|
|||||||
mod imp;
|
mod imp;
|
||||||
|
|
||||||
use std::io::Write;
|
|
||||||
|
|
||||||
#[cfg(unix)]
|
|
||||||
use std::os::unix::net::UnixStream;
|
|
||||||
|
|
||||||
#[cfg(windows)]
|
|
||||||
use std::net::TcpStream;
|
|
||||||
|
|
||||||
use adw::prelude::*;
|
use adw::prelude::*;
|
||||||
use adw::subclass::prelude::*;
|
use adw::subclass::prelude::*;
|
||||||
use endi::{Endian, WriteBytes};
|
|
||||||
use glib::{clone, Object};
|
use glib::{clone, Object};
|
||||||
use gtk::{
|
use gtk::{
|
||||||
gio,
|
gio,
|
||||||
@@ -18,13 +9,12 @@ use gtk::{
|
|||||||
ListBox, NoSelection,
|
ListBox, NoSelection,
|
||||||
};
|
};
|
||||||
|
|
||||||
use crate::{
|
use lan_mouse_ipc::{
|
||||||
client::{ClientConfig, ClientHandle, ClientState, Position},
|
ClientConfig, ClientHandle, ClientState, FrontendRequest, FrontendRequestWriter, Position,
|
||||||
config::DEFAULT_PORT,
|
DEFAULT_PORT,
|
||||||
frontend::{gtk::client_object::ClientObject, FrontendRequest},
|
|
||||||
};
|
};
|
||||||
|
|
||||||
use super::client_row::ClientRow;
|
use super::{client_object::ClientObject, client_row::ClientRow};
|
||||||
|
|
||||||
glib::wrapper! {
|
glib::wrapper! {
|
||||||
pub struct Window(ObjectSubclass<imp::Window>)
|
pub struct Window(ObjectSubclass<imp::Window>)
|
||||||
@@ -34,13 +24,13 @@ glib::wrapper! {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl Window {
|
impl Window {
|
||||||
pub(crate) fn new(
|
pub(crate) fn new(app: &adw::Application, conn: FrontendRequestWriter) -> Self {
|
||||||
app: &adw::Application,
|
|
||||||
#[cfg(unix)] tx: UnixStream,
|
|
||||||
#[cfg(windows)] tx: TcpStream,
|
|
||||||
) -> Self {
|
|
||||||
let window: Self = Object::builder().property("application", app).build();
|
let window: Self = Object::builder().property("application", app).build();
|
||||||
window.imp().stream.borrow_mut().replace(tx);
|
window
|
||||||
|
.imp()
|
||||||
|
.frontend_request_writer
|
||||||
|
.borrow_mut()
|
||||||
|
.replace(conn);
|
||||||
window
|
window
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -257,7 +247,11 @@ impl Window {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub fn request_client_state(&self, client: &ClientObject) {
|
pub fn request_client_state(&self, client: &ClientObject) {
|
||||||
self.request(FrontendRequest::GetState(client.handle()));
|
self.request_client_state_for(client.handle());
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn request_client_state_for(&self, handle: ClientHandle) {
|
||||||
|
self.request(FrontendRequest::GetState(handle));
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn request_client_create(&self) {
|
pub fn request_client_create(&self) {
|
||||||
@@ -292,16 +286,10 @@ impl Window {
|
|||||||
self.request(FrontendRequest::Delete(client.handle()));
|
self.request(FrontendRequest::Delete(client.handle()));
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn request(&self, event: FrontendRequest) {
|
pub fn request(&self, request: FrontendRequest) {
|
||||||
let json = serde_json::to_string(&event).unwrap();
|
let mut requester = self.imp().frontend_request_writer.borrow_mut();
|
||||||
log::debug!("requesting: {json}");
|
let requester = requester.as_mut().unwrap();
|
||||||
let mut stream = self.imp().stream.borrow_mut();
|
if let Err(e) = requester.request(request) {
|
||||||
let stream = stream.as_mut().unwrap();
|
|
||||||
let bytes = json.as_bytes();
|
|
||||||
if let Err(e) = stream.write_u64(Endian::Big, bytes.len() as u64) {
|
|
||||||
log::error!("error sending message: {e}");
|
|
||||||
};
|
|
||||||
if let Err(e) = stream.write(bytes) {
|
|
||||||
log::error!("error sending message: {e}");
|
log::error!("error sending message: {e}");
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
@@ -1,17 +1,12 @@
|
|||||||
use std::cell::{Cell, RefCell};
|
use std::cell::{Cell, RefCell};
|
||||||
|
|
||||||
#[cfg(windows)]
|
|
||||||
use std::net::TcpStream;
|
|
||||||
#[cfg(unix)]
|
|
||||||
use std::os::unix::net::UnixStream;
|
|
||||||
|
|
||||||
use adw::subclass::prelude::*;
|
use adw::subclass::prelude::*;
|
||||||
use adw::{prelude::*, ActionRow, PreferencesGroup, ToastOverlay};
|
use adw::{prelude::*, ActionRow, PreferencesGroup, ToastOverlay};
|
||||||
use glib::subclass::InitializingObject;
|
use glib::subclass::InitializingObject;
|
||||||
use gtk::glib::clone;
|
use gtk::glib::clone;
|
||||||
use gtk::{gdk, gio, glib, Button, CompositeTemplate, Entry, Label, ListBox};
|
use gtk::{gdk, gio, glib, Button, CompositeTemplate, Entry, Label, ListBox};
|
||||||
|
|
||||||
use crate::config::DEFAULT_PORT;
|
use lan_mouse_ipc::{FrontendRequestWriter, DEFAULT_PORT};
|
||||||
|
|
||||||
#[derive(CompositeTemplate, Default)]
|
#[derive(CompositeTemplate, Default)]
|
||||||
#[template(resource = "/de/feschber/LanMouse/window.ui")]
|
#[template(resource = "/de/feschber/LanMouse/window.ui")]
|
||||||
@@ -41,10 +36,7 @@ pub struct Window {
|
|||||||
#[template_child]
|
#[template_child]
|
||||||
pub input_capture_button: TemplateChild<Button>,
|
pub input_capture_button: TemplateChild<Button>,
|
||||||
pub clients: RefCell<Option<gio::ListStore>>,
|
pub clients: RefCell<Option<gio::ListStore>>,
|
||||||
#[cfg(unix)]
|
pub frontend_request_writer: RefCell<Option<FrontendRequestWriter>>,
|
||||||
pub stream: RefCell<Option<UnixStream>>,
|
|
||||||
#[cfg(windows)]
|
|
||||||
pub stream: RefCell<Option<TcpStream>>,
|
|
||||||
pub port: Cell<u16>,
|
pub port: Cell<u16>,
|
||||||
pub capture_active: Cell<bool>,
|
pub capture_active: Cell<bool>,
|
||||||
pub emulation_active: Cell<bool>,
|
pub emulation_active: Cell<bool>,
|
||||||
16
lan-mouse-ipc/Cargo.toml
Normal file
16
lan-mouse-ipc/Cargo.toml
Normal file
@@ -0,0 +1,16 @@
|
|||||||
|
[package]
|
||||||
|
name = "lan-mouse-ipc"
|
||||||
|
description = "library for communication between lan-mouse service and frontends"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2021"
|
||||||
|
license = "GPL-3.0-or-later"
|
||||||
|
repository = "https://github.com/feschber/lan-mouse"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
futures = "0.3.30"
|
||||||
|
log = "0.4.22"
|
||||||
|
serde = { version = "1.0", features = ["derive"] }
|
||||||
|
serde_json = "1.0.107"
|
||||||
|
thiserror = "1.0.63"
|
||||||
|
tokio = { version = "1.32.0", features = ["net", "io-util", "time"] }
|
||||||
|
tokio-stream = { version = "0.1.15", features = ["io-util"] }
|
||||||
89
lan-mouse-ipc/src/connect.rs
Normal file
89
lan-mouse-ipc/src/connect.rs
Normal file
@@ -0,0 +1,89 @@
|
|||||||
|
use crate::{ConnectionError, FrontendEvent, FrontendRequest, IpcError};
|
||||||
|
use std::{
|
||||||
|
cmp::min,
|
||||||
|
io::{self, prelude::*, BufReader, LineWriter, Lines},
|
||||||
|
thread,
|
||||||
|
time::Duration,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[cfg(unix)]
|
||||||
|
use std::os::unix::net::UnixStream;
|
||||||
|
|
||||||
|
#[cfg(windows)]
|
||||||
|
use std::net::TcpStream;
|
||||||
|
|
||||||
|
pub struct FrontendEventReader {
|
||||||
|
#[cfg(unix)]
|
||||||
|
lines: Lines<BufReader<UnixStream>>,
|
||||||
|
#[cfg(windows)]
|
||||||
|
lines: Lines<BufReader<TcpStream>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct FrontendRequestWriter {
|
||||||
|
#[cfg(unix)]
|
||||||
|
line_writer: LineWriter<UnixStream>,
|
||||||
|
#[cfg(windows)]
|
||||||
|
line_writer: LineWriter<TcpStream>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl FrontendEventReader {
|
||||||
|
pub fn next_event(&mut self) -> Option<Result<FrontendEvent, IpcError>> {
|
||||||
|
match self.lines.next()? {
|
||||||
|
Err(e) => Some(Err(e.into())),
|
||||||
|
Ok(l) => Some(serde_json::from_str(l.as_str()).map_err(|e| e.into())),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl FrontendRequestWriter {
|
||||||
|
pub fn request(&mut self, request: FrontendRequest) -> Result<(), io::Error> {
|
||||||
|
let mut json = serde_json::to_string(&request).unwrap();
|
||||||
|
log::debug!("requesting: {json}");
|
||||||
|
json.push('\n');
|
||||||
|
self.line_writer.write_all(json.as_bytes())?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn connect() -> Result<(FrontendEventReader, FrontendRequestWriter), ConnectionError> {
|
||||||
|
let rx = wait_for_service()?;
|
||||||
|
let tx = rx.try_clone()?;
|
||||||
|
let buf_reader = BufReader::new(rx);
|
||||||
|
let lines = buf_reader.lines();
|
||||||
|
let line_writer = LineWriter::new(tx);
|
||||||
|
let reader = FrontendEventReader { lines };
|
||||||
|
let writer = FrontendRequestWriter { line_writer };
|
||||||
|
Ok((reader, writer))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// wait for the lan-mouse socket to come online
|
||||||
|
#[cfg(unix)]
|
||||||
|
fn wait_for_service() -> Result<UnixStream, ConnectionError> {
|
||||||
|
let socket_path = crate::default_socket_path()?;
|
||||||
|
let mut duration = Duration::from_millis(10);
|
||||||
|
loop {
|
||||||
|
if let Ok(stream) = UnixStream::connect(&socket_path) {
|
||||||
|
break Ok(stream);
|
||||||
|
}
|
||||||
|
// a signaling mechanism or inotify could be used to
|
||||||
|
// improve this
|
||||||
|
thread::sleep(exponential_back_off(&mut duration));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(windows)]
|
||||||
|
fn wait_for_service() -> Result<TcpStream, ConnectionError> {
|
||||||
|
let mut duration = Duration::from_millis(10);
|
||||||
|
loop {
|
||||||
|
if let Ok(stream) = TcpStream::connect("127.0.0.1:5252") {
|
||||||
|
break Ok(stream);
|
||||||
|
}
|
||||||
|
thread::sleep(exponential_back_off(&mut duration));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn exponential_back_off(duration: &mut Duration) -> Duration {
|
||||||
|
let new = duration.saturating_mul(2);
|
||||||
|
*duration = min(new, Duration::from_secs(1));
|
||||||
|
*duration
|
||||||
|
}
|
||||||
104
lan-mouse-ipc/src/connect_async.rs
Normal file
104
lan-mouse-ipc/src/connect_async.rs
Normal file
@@ -0,0 +1,104 @@
|
|||||||
|
use crate::{ConnectionError, FrontendEvent, FrontendRequest, IpcError};
|
||||||
|
use std::{
|
||||||
|
cmp::min,
|
||||||
|
io,
|
||||||
|
task::{ready, Poll},
|
||||||
|
time::Duration,
|
||||||
|
};
|
||||||
|
|
||||||
|
use futures::{Stream, StreamExt};
|
||||||
|
use tokio::io::{AsyncBufReadExt, AsyncWriteExt, BufReader, ReadHalf, WriteHalf};
|
||||||
|
use tokio_stream::wrappers::LinesStream;
|
||||||
|
|
||||||
|
#[cfg(unix)]
|
||||||
|
use tokio::net::UnixStream;
|
||||||
|
|
||||||
|
#[cfg(windows)]
|
||||||
|
use tokio::net::TcpStream;
|
||||||
|
|
||||||
|
pub struct AsyncFrontendEventReader {
|
||||||
|
#[cfg(unix)]
|
||||||
|
lines_stream: LinesStream<BufReader<ReadHalf<UnixStream>>>,
|
||||||
|
#[cfg(windows)]
|
||||||
|
lines_stream: LinesStream<BufReader<ReadHalf<TcpStream>>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct AsyncFrontendRequestWriter {
|
||||||
|
#[cfg(unix)]
|
||||||
|
tx: WriteHalf<UnixStream>,
|
||||||
|
#[cfg(windows)]
|
||||||
|
tx: WriteHalf<TcpStream>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Stream for AsyncFrontendEventReader {
|
||||||
|
type Item = Result<FrontendEvent, IpcError>;
|
||||||
|
|
||||||
|
fn poll_next(
|
||||||
|
mut self: std::pin::Pin<&mut Self>,
|
||||||
|
cx: &mut std::task::Context<'_>,
|
||||||
|
) -> std::task::Poll<Option<Self::Item>> {
|
||||||
|
let line = ready!(self.lines_stream.poll_next_unpin(cx));
|
||||||
|
let event = line.map(|l| {
|
||||||
|
l.map_err(Into::<IpcError>::into)
|
||||||
|
.and_then(|l| serde_json::from_str(l.as_str()).map_err(|e| e.into()))
|
||||||
|
});
|
||||||
|
Poll::Ready(event)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl AsyncFrontendRequestWriter {
|
||||||
|
pub async fn request(&mut self, request: FrontendRequest) -> Result<(), io::Error> {
|
||||||
|
let mut json = serde_json::to_string(&request).unwrap();
|
||||||
|
log::debug!("requesting: {json}");
|
||||||
|
json.push('\n');
|
||||||
|
self.tx.write_all(json.as_bytes()).await?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn connect_async(
|
||||||
|
) -> Result<(AsyncFrontendEventReader, AsyncFrontendRequestWriter), ConnectionError> {
|
||||||
|
let stream = wait_for_service().await?;
|
||||||
|
#[cfg(unix)]
|
||||||
|
let (rx, tx): (ReadHalf<UnixStream>, WriteHalf<UnixStream>) = tokio::io::split(stream);
|
||||||
|
#[cfg(windows)]
|
||||||
|
let (rx, tx): (ReadHalf<TcpStream>, WriteHalf<TcpStream>) = tokio::io::split(stream);
|
||||||
|
let buf_reader = BufReader::new(rx);
|
||||||
|
let lines = buf_reader.lines();
|
||||||
|
let lines_stream = LinesStream::new(lines);
|
||||||
|
let reader = AsyncFrontendEventReader { lines_stream };
|
||||||
|
let writer = AsyncFrontendRequestWriter { tx };
|
||||||
|
Ok((reader, writer))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// wait for the lan-mouse socket to come online
|
||||||
|
#[cfg(unix)]
|
||||||
|
async fn wait_for_service() -> Result<UnixStream, ConnectionError> {
|
||||||
|
let socket_path = crate::default_socket_path()?;
|
||||||
|
let mut duration = Duration::from_millis(10);
|
||||||
|
loop {
|
||||||
|
if let Ok(stream) = UnixStream::connect(&socket_path).await {
|
||||||
|
break Ok(stream);
|
||||||
|
}
|
||||||
|
// a signaling mechanism or inotify could be used to
|
||||||
|
// improve this
|
||||||
|
tokio::time::sleep(exponential_back_off(&mut duration)).await;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(windows)]
|
||||||
|
async fn wait_for_service() -> Result<TcpStream, ConnectionError> {
|
||||||
|
let mut duration = Duration::from_millis(10);
|
||||||
|
loop {
|
||||||
|
if let Ok(stream) = TcpStream::connect("127.0.0.1:5252").await {
|
||||||
|
break Ok(stream);
|
||||||
|
}
|
||||||
|
tokio::time::sleep(exponential_back_off(&mut duration)).await;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn exponential_back_off(duration: &mut Duration) -> Duration {
|
||||||
|
let new = duration.saturating_mul(2);
|
||||||
|
*duration = min(new, Duration::from_secs(1));
|
||||||
|
*duration
|
||||||
|
}
|
||||||
264
lan-mouse-ipc/src/lib.rs
Normal file
264
lan-mouse-ipc/src/lib.rs
Normal file
@@ -0,0 +1,264 @@
|
|||||||
|
use std::{
|
||||||
|
collections::HashSet,
|
||||||
|
env::VarError,
|
||||||
|
fmt::Display,
|
||||||
|
io,
|
||||||
|
net::{IpAddr, SocketAddr},
|
||||||
|
str::FromStr,
|
||||||
|
};
|
||||||
|
use thiserror::Error;
|
||||||
|
|
||||||
|
#[cfg(unix)]
|
||||||
|
use std::{
|
||||||
|
env,
|
||||||
|
path::{Path, PathBuf},
|
||||||
|
};
|
||||||
|
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
|
||||||
|
mod connect;
|
||||||
|
mod connect_async;
|
||||||
|
mod listen;
|
||||||
|
|
||||||
|
pub use connect::{connect, FrontendEventReader, FrontendRequestWriter};
|
||||||
|
pub use connect_async::{connect_async, AsyncFrontendEventReader, AsyncFrontendRequestWriter};
|
||||||
|
pub use listen::AsyncFrontendListener;
|
||||||
|
|
||||||
|
#[derive(Debug, Error)]
|
||||||
|
pub enum ConnectionError {
|
||||||
|
#[error(transparent)]
|
||||||
|
SocketPath(#[from] SocketPathError),
|
||||||
|
#[error(transparent)]
|
||||||
|
Io(#[from] io::Error),
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Error)]
|
||||||
|
pub enum ListenerCreationError {
|
||||||
|
#[error("could not determine socket-path: `{0}`")]
|
||||||
|
SocketPath(#[from] SocketPathError),
|
||||||
|
#[error("service already running!")]
|
||||||
|
AlreadyRunning,
|
||||||
|
#[error("failed to bind lan-mouse socket: `{0}`")]
|
||||||
|
Bind(io::Error),
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Error)]
|
||||||
|
pub enum IpcError {
|
||||||
|
#[error("io error occured: `{0}`")]
|
||||||
|
Io(#[from] io::Error),
|
||||||
|
#[error("invalid json: `{0}`")]
|
||||||
|
Json(#[from] serde_json::Error),
|
||||||
|
#[error(transparent)]
|
||||||
|
Connection(#[from] ConnectionError),
|
||||||
|
#[error(transparent)]
|
||||||
|
Listen(#[from] ListenerCreationError),
|
||||||
|
}
|
||||||
|
|
||||||
|
pub const DEFAULT_PORT: u16 = 4242;
|
||||||
|
|
||||||
|
#[derive(Debug, Default, Eq, Hash, PartialEq, Clone, Copy, Serialize, Deserialize)]
|
||||||
|
pub enum Position {
|
||||||
|
#[default]
|
||||||
|
Left,
|
||||||
|
Right,
|
||||||
|
Top,
|
||||||
|
Bottom,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Error)]
|
||||||
|
#[error("not a valid position: {pos}")]
|
||||||
|
pub struct PositionParseError {
|
||||||
|
pos: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl FromStr for Position {
|
||||||
|
type Err = PositionParseError;
|
||||||
|
|
||||||
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
||||||
|
match s {
|
||||||
|
"left" => Ok(Self::Left),
|
||||||
|
"right" => Ok(Self::Right),
|
||||||
|
"top" => Ok(Self::Top),
|
||||||
|
"bottom" => Ok(Self::Bottom),
|
||||||
|
_ => Err(PositionParseError { pos: s.into() }),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Display for Position {
|
||||||
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
|
write!(
|
||||||
|
f,
|
||||||
|
"{}",
|
||||||
|
match self {
|
||||||
|
Position::Left => "left",
|
||||||
|
Position::Right => "right",
|
||||||
|
Position::Top => "top",
|
||||||
|
Position::Bottom => "bottom",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl TryFrom<&str> for Position {
|
||||||
|
type Error = ();
|
||||||
|
|
||||||
|
fn try_from(s: &str) -> Result<Self, Self::Error> {
|
||||||
|
match s {
|
||||||
|
"left" => Ok(Position::Left),
|
||||||
|
"right" => Ok(Position::Right),
|
||||||
|
"top" => Ok(Position::Top),
|
||||||
|
"bottom" => Ok(Position::Bottom),
|
||||||
|
_ => Err(()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Eq, PartialEq, Clone, Serialize, Deserialize)]
|
||||||
|
pub struct ClientConfig {
|
||||||
|
/// hostname of this client
|
||||||
|
pub hostname: Option<String>,
|
||||||
|
/// fix ips, determined by the user
|
||||||
|
pub fix_ips: Vec<IpAddr>,
|
||||||
|
/// both active_addr and addrs can be None / empty so port needs to be stored seperately
|
||||||
|
pub port: u16,
|
||||||
|
/// position of a client on screen
|
||||||
|
pub pos: Position,
|
||||||
|
/// enter hook
|
||||||
|
pub cmd: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for ClientConfig {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self {
|
||||||
|
port: DEFAULT_PORT,
|
||||||
|
hostname: Default::default(),
|
||||||
|
fix_ips: Default::default(),
|
||||||
|
pos: Default::default(),
|
||||||
|
cmd: None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub type ClientHandle = u64;
|
||||||
|
|
||||||
|
#[derive(Debug, Default, Clone, Serialize, Deserialize)]
|
||||||
|
pub struct ClientState {
|
||||||
|
/// events should be sent to and received from the client
|
||||||
|
pub active: bool,
|
||||||
|
/// `active` address of the client, used to send data to.
|
||||||
|
/// This should generally be the socket address where data
|
||||||
|
/// was last received from.
|
||||||
|
pub active_addr: Option<SocketAddr>,
|
||||||
|
/// tracks whether or not the client is responding to pings
|
||||||
|
pub alive: bool,
|
||||||
|
/// ips from dns
|
||||||
|
pub dns_ips: Vec<IpAddr>,
|
||||||
|
/// all ip addresses associated with a particular client
|
||||||
|
/// e.g. Laptops usually have at least an ethernet and a wifi port
|
||||||
|
/// which have different ip addresses
|
||||||
|
pub ips: HashSet<IpAddr>,
|
||||||
|
/// client has pressed keys
|
||||||
|
pub has_pressed_keys: bool,
|
||||||
|
/// dns resolving in progress
|
||||||
|
pub resolving: bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
|
pub enum FrontendEvent {
|
||||||
|
/// client state has changed, new state must be requested via [`FrontendRequest::GetState`]
|
||||||
|
Changed(ClientHandle),
|
||||||
|
/// a client was created
|
||||||
|
Created(ClientHandle, ClientConfig, ClientState),
|
||||||
|
/// no such client
|
||||||
|
NoSuchClient(ClientHandle),
|
||||||
|
/// state changed
|
||||||
|
State(ClientHandle, ClientConfig, ClientState),
|
||||||
|
/// the client was deleted
|
||||||
|
Deleted(ClientHandle),
|
||||||
|
/// new port, reason of failure (if failed)
|
||||||
|
PortChanged(u16, Option<String>),
|
||||||
|
/// list of all clients, used for initial state synchronization
|
||||||
|
Enumerate(Vec<(ClientHandle, ClientConfig, ClientState)>),
|
||||||
|
/// an error occured
|
||||||
|
Error(String),
|
||||||
|
/// capture status
|
||||||
|
CaptureStatus(Status),
|
||||||
|
/// emulation status
|
||||||
|
EmulationStatus(Status),
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Eq, PartialEq, Clone, Serialize, Deserialize)]
|
||||||
|
pub enum FrontendRequest {
|
||||||
|
/// activate/deactivate client
|
||||||
|
Activate(ClientHandle, bool),
|
||||||
|
/// add a new client
|
||||||
|
Create,
|
||||||
|
/// change the listen port (recreate udp listener)
|
||||||
|
ChangePort(u16),
|
||||||
|
/// remove a client
|
||||||
|
Delete(ClientHandle),
|
||||||
|
/// request an enumeration of all clients
|
||||||
|
Enumerate(),
|
||||||
|
/// resolve dns
|
||||||
|
ResolveDns(ClientHandle),
|
||||||
|
/// update hostname
|
||||||
|
UpdateHostname(ClientHandle, Option<String>),
|
||||||
|
/// update port
|
||||||
|
UpdatePort(ClientHandle, u16),
|
||||||
|
/// update position
|
||||||
|
UpdatePosition(ClientHandle, Position),
|
||||||
|
/// update fix-ips
|
||||||
|
UpdateFixIps(ClientHandle, Vec<IpAddr>),
|
||||||
|
/// request the state of the given client
|
||||||
|
GetState(ClientHandle),
|
||||||
|
/// request reenabling input capture
|
||||||
|
EnableCapture,
|
||||||
|
/// request reenabling input emulation
|
||||||
|
EnableEmulation,
|
||||||
|
/// synchronize all state
|
||||||
|
Sync,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Copy, Debug, Default, Serialize, Deserialize)]
|
||||||
|
pub enum Status {
|
||||||
|
#[default]
|
||||||
|
Disabled,
|
||||||
|
Enabled,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<Status> for bool {
|
||||||
|
fn from(status: Status) -> Self {
|
||||||
|
match status {
|
||||||
|
Status::Enabled => true,
|
||||||
|
Status::Disabled => false,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(unix)]
|
||||||
|
const LAN_MOUSE_SOCKET_NAME: &str = "lan-mouse-socket.sock";
|
||||||
|
|
||||||
|
#[derive(Debug, Error)]
|
||||||
|
pub enum SocketPathError {
|
||||||
|
#[error("could not determine $XDG_RUNTIME_DIR: `{0}`")]
|
||||||
|
XdgRuntimeDirNotFound(VarError),
|
||||||
|
#[error("could not determine $HOME: `{0}`")]
|
||||||
|
HomeDirNotFound(VarError),
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(all(unix, not(target_os = "macos")))]
|
||||||
|
pub fn default_socket_path() -> Result<PathBuf, SocketPathError> {
|
||||||
|
let xdg_runtime_dir =
|
||||||
|
env::var("XDG_RUNTIME_DIR").map_err(SocketPathError::XdgRuntimeDirNotFound)?;
|
||||||
|
Ok(Path::new(xdg_runtime_dir.as_str()).join(LAN_MOUSE_SOCKET_NAME))
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(all(unix, target_os = "macos"))]
|
||||||
|
pub fn default_socket_path() -> Result<PathBuf, SocketPathError> {
|
||||||
|
let home = env::var("HOME").map_err(SocketPathError::HomeDirNotFound)?;
|
||||||
|
Ok(Path::new(home.as_str())
|
||||||
|
.join("Library")
|
||||||
|
.join("Caches")
|
||||||
|
.join(LAN_MOUSE_SOCKET_NAME))
|
||||||
|
}
|
||||||
147
lan-mouse-ipc/src/listen.rs
Normal file
147
lan-mouse-ipc/src/listen.rs
Normal file
@@ -0,0 +1,147 @@
|
|||||||
|
use futures::{stream::SelectAll, Stream, StreamExt};
|
||||||
|
#[cfg(unix)]
|
||||||
|
use std::path::PathBuf;
|
||||||
|
use std::{
|
||||||
|
io::ErrorKind,
|
||||||
|
pin::Pin,
|
||||||
|
task::{Context, Poll},
|
||||||
|
};
|
||||||
|
|
||||||
|
use tokio::io::{AsyncBufReadExt, AsyncWriteExt, BufReader, ReadHalf, WriteHalf};
|
||||||
|
use tokio_stream::wrappers::LinesStream;
|
||||||
|
|
||||||
|
#[cfg(unix)]
|
||||||
|
use tokio::net::UnixListener;
|
||||||
|
#[cfg(unix)]
|
||||||
|
use tokio::net::UnixStream;
|
||||||
|
|
||||||
|
#[cfg(windows)]
|
||||||
|
use tokio::net::TcpListener;
|
||||||
|
#[cfg(windows)]
|
||||||
|
use tokio::net::TcpStream;
|
||||||
|
|
||||||
|
use crate::{FrontendEvent, FrontendRequest, IpcError, ListenerCreationError};
|
||||||
|
|
||||||
|
pub struct AsyncFrontendListener {
|
||||||
|
#[cfg(windows)]
|
||||||
|
listener: TcpListener,
|
||||||
|
#[cfg(unix)]
|
||||||
|
listener: UnixListener,
|
||||||
|
#[cfg(unix)]
|
||||||
|
socket_path: PathBuf,
|
||||||
|
#[cfg(unix)]
|
||||||
|
line_streams: SelectAll<LinesStream<BufReader<ReadHalf<UnixStream>>>>,
|
||||||
|
#[cfg(windows)]
|
||||||
|
line_streams: SelectAll<LinesStream<BufReader<ReadHalf<TcpStream>>>>,
|
||||||
|
#[cfg(unix)]
|
||||||
|
tx_streams: Vec<WriteHalf<UnixStream>>,
|
||||||
|
#[cfg(windows)]
|
||||||
|
tx_streams: Vec<WriteHalf<TcpStream>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl AsyncFrontendListener {
|
||||||
|
pub async fn new() -> Result<Self, ListenerCreationError> {
|
||||||
|
#[cfg(unix)]
|
||||||
|
let (socket_path, listener) = {
|
||||||
|
let socket_path = crate::default_socket_path()?;
|
||||||
|
|
||||||
|
log::debug!("remove socket: {:?}", socket_path);
|
||||||
|
if socket_path.exists() {
|
||||||
|
// try to connect to see if some other instance
|
||||||
|
// of lan-mouse is already running
|
||||||
|
match UnixStream::connect(&socket_path).await {
|
||||||
|
// connected -> lan-mouse is already running
|
||||||
|
Ok(_) => return Err(ListenerCreationError::AlreadyRunning),
|
||||||
|
// lan-mouse is not running but a socket was left behind
|
||||||
|
Err(e) => {
|
||||||
|
log::debug!("{socket_path:?}: {e} - removing left behind socket");
|
||||||
|
let _ = std::fs::remove_file(&socket_path);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let listener = match UnixListener::bind(&socket_path) {
|
||||||
|
Ok(ls) => ls,
|
||||||
|
// some other lan-mouse instance has bound the socket in the meantime
|
||||||
|
Err(e) if e.kind() == ErrorKind::AddrInUse => {
|
||||||
|
return Err(ListenerCreationError::AlreadyRunning)
|
||||||
|
}
|
||||||
|
Err(e) => return Err(ListenerCreationError::Bind(e)),
|
||||||
|
};
|
||||||
|
(socket_path, listener)
|
||||||
|
};
|
||||||
|
|
||||||
|
#[cfg(windows)]
|
||||||
|
let listener = match TcpListener::bind("127.0.0.1:5252").await {
|
||||||
|
Ok(ls) => ls,
|
||||||
|
// some other lan-mouse instance has bound the socket in the meantime
|
||||||
|
Err(e) if e.kind() == ErrorKind::AddrInUse => {
|
||||||
|
return Err(ListenerCreationError::AlreadyRunning)
|
||||||
|
}
|
||||||
|
Err(e) => return Err(ListenerCreationError::Bind(e)),
|
||||||
|
};
|
||||||
|
|
||||||
|
let adapter = Self {
|
||||||
|
listener,
|
||||||
|
#[cfg(unix)]
|
||||||
|
socket_path,
|
||||||
|
line_streams: SelectAll::new(),
|
||||||
|
tx_streams: vec![],
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(adapter)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn broadcast(&mut self, notify: FrontendEvent) {
|
||||||
|
// encode event
|
||||||
|
let mut json = serde_json::to_string(¬ify).unwrap();
|
||||||
|
json.push('\n');
|
||||||
|
|
||||||
|
let mut keep = vec![];
|
||||||
|
// TODO do simultaneously
|
||||||
|
for tx in self.tx_streams.iter_mut() {
|
||||||
|
// write len + payload
|
||||||
|
if tx.write(json.as_bytes()).await.is_err() {
|
||||||
|
keep.push(false);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
keep.push(true);
|
||||||
|
}
|
||||||
|
|
||||||
|
// could not find a better solution because async
|
||||||
|
let mut keep = keep.into_iter();
|
||||||
|
self.tx_streams.retain(|_| keep.next().unwrap());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(unix)]
|
||||||
|
impl Drop for AsyncFrontendListener {
|
||||||
|
fn drop(&mut self) {
|
||||||
|
log::debug!("remove socket: {:?}", self.socket_path);
|
||||||
|
let _ = std::fs::remove_file(&self.socket_path);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Stream for AsyncFrontendListener {
|
||||||
|
type Item = Result<FrontendRequest, IpcError>;
|
||||||
|
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
|
||||||
|
if let Poll::Ready(Some(Ok(l))) = self.line_streams.poll_next_unpin(cx) {
|
||||||
|
let request = serde_json::from_str(l.as_str()).map_err(|e| e.into());
|
||||||
|
return Poll::Ready(Some(request));
|
||||||
|
}
|
||||||
|
let mut sync = false;
|
||||||
|
while let Poll::Ready(Ok((stream, _))) = self.listener.poll_accept(cx) {
|
||||||
|
let (rx, tx) = tokio::io::split(stream);
|
||||||
|
let buf_reader = BufReader::new(rx);
|
||||||
|
let lines = buf_reader.lines();
|
||||||
|
let lines = LinesStream::new(lines);
|
||||||
|
self.line_streams.push(lines);
|
||||||
|
self.tx_streams.push(tx);
|
||||||
|
sync = true;
|
||||||
|
}
|
||||||
|
if sync {
|
||||||
|
Poll::Ready(Some(Ok(FrontendRequest::Sync)))
|
||||||
|
} else {
|
||||||
|
Poll::Pending
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
13
lan-mouse-proto/Cargo.toml
Normal file
13
lan-mouse-proto/Cargo.toml
Normal file
@@ -0,0 +1,13 @@
|
|||||||
|
[package]
|
||||||
|
name = "lan-mouse-proto"
|
||||||
|
description = "network protocol for lan-mouse"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2021"
|
||||||
|
license = "GPL-3.0-or-later"
|
||||||
|
repository = "https://github.com/feschber/lan-mouse"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
num_enum = "0.7.2"
|
||||||
|
thiserror = "1.0.61"
|
||||||
|
input-event = { path = "../input-event", version = "0.2.1" }
|
||||||
|
paste = "1.0"
|
||||||
251
lan-mouse-proto/src/lib.rs
Normal file
251
lan-mouse-proto/src/lib.rs
Normal file
@@ -0,0 +1,251 @@
|
|||||||
|
use input_event::{Event as InputEvent, KeyboardEvent, PointerEvent};
|
||||||
|
use num_enum::{IntoPrimitive, TryFromPrimitive, TryFromPrimitiveError};
|
||||||
|
use paste::paste;
|
||||||
|
use std::{
|
||||||
|
fmt::{Debug, Display},
|
||||||
|
mem::size_of,
|
||||||
|
};
|
||||||
|
use thiserror::Error;
|
||||||
|
|
||||||
|
/// defines the maximum size an encoded event can take up
|
||||||
|
/// this is currently the pointer motion event
|
||||||
|
/// type: u8, time: u32, dx: f64, dy: f64
|
||||||
|
pub const MAX_EVENT_SIZE: usize = size_of::<u8>() + size_of::<u32>() + 2 * size_of::<f64>();
|
||||||
|
|
||||||
|
/// error type for protocol violations
|
||||||
|
#[derive(Debug, Error)]
|
||||||
|
pub enum ProtocolError {
|
||||||
|
/// event type does not exist
|
||||||
|
#[error("invalid event id: `{0}`")]
|
||||||
|
InvalidEventId(#[from] TryFromPrimitiveError<EventType>),
|
||||||
|
}
|
||||||
|
|
||||||
|
/// main lan-mouse protocol event type
|
||||||
|
#[derive(Clone, Copy, Debug)]
|
||||||
|
pub enum ProtoEvent {
|
||||||
|
/// notify a client that the cursor entered its region
|
||||||
|
/// [`ProtoEvent::Ack`] with the same serial is used for synchronization between devices
|
||||||
|
Enter(u32),
|
||||||
|
/// notify a client that the cursor left its region
|
||||||
|
/// [`ProtoEvent::Ack`] with the same serial is used for synchronization between devices
|
||||||
|
Leave(u32),
|
||||||
|
/// acknowledge of an [`ProtoEvent::Enter`] or [`ProtoEvent::Leave`] event
|
||||||
|
Ack(u32),
|
||||||
|
/// Input event
|
||||||
|
Input(InputEvent),
|
||||||
|
/// Ping event for tracking unresponsive clients.
|
||||||
|
/// A client has to respond with [`ProtoEvent::Pong`].
|
||||||
|
Ping,
|
||||||
|
/// Response to [`ProtoEvent::Ping`]
|
||||||
|
Pong,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Display for ProtoEvent {
|
||||||
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
|
match self {
|
||||||
|
ProtoEvent::Enter(s) => write!(f, "Enter({s})"),
|
||||||
|
ProtoEvent::Leave(s) => write!(f, "Leave({s})"),
|
||||||
|
ProtoEvent::Ack(s) => write!(f, "Ack({s})"),
|
||||||
|
ProtoEvent::Input(e) => write!(f, "{e}"),
|
||||||
|
ProtoEvent::Ping => write!(f, "ping"),
|
||||||
|
ProtoEvent::Pong => write!(f, "pong"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(TryFromPrimitive, IntoPrimitive)]
|
||||||
|
#[repr(u8)]
|
||||||
|
pub enum EventType {
|
||||||
|
PointerMotion,
|
||||||
|
PointerButton,
|
||||||
|
PointerAxis,
|
||||||
|
PointerAxisValue120,
|
||||||
|
KeyboardKey,
|
||||||
|
KeyboardModifiers,
|
||||||
|
Ping,
|
||||||
|
Pong,
|
||||||
|
Enter,
|
||||||
|
Leave,
|
||||||
|
Ack,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ProtoEvent {
|
||||||
|
fn event_type(&self) -> EventType {
|
||||||
|
match self {
|
||||||
|
ProtoEvent::Input(e) => match e {
|
||||||
|
InputEvent::Pointer(p) => match p {
|
||||||
|
PointerEvent::Motion { .. } => EventType::PointerMotion,
|
||||||
|
PointerEvent::Button { .. } => EventType::PointerButton,
|
||||||
|
PointerEvent::Axis { .. } => EventType::PointerAxis,
|
||||||
|
PointerEvent::AxisDiscrete120 { .. } => EventType::PointerAxisValue120,
|
||||||
|
},
|
||||||
|
InputEvent::Keyboard(k) => match k {
|
||||||
|
KeyboardEvent::Key { .. } => EventType::KeyboardKey,
|
||||||
|
KeyboardEvent::Modifiers { .. } => EventType::KeyboardModifiers,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
ProtoEvent::Ping => EventType::Ping,
|
||||||
|
ProtoEvent::Pong => EventType::Pong,
|
||||||
|
ProtoEvent::Enter(_) => EventType::Enter,
|
||||||
|
ProtoEvent::Leave(_) => EventType::Leave,
|
||||||
|
ProtoEvent::Ack(_) => EventType::Ack,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl TryFrom<[u8; MAX_EVENT_SIZE]> for ProtoEvent {
|
||||||
|
type Error = ProtocolError;
|
||||||
|
|
||||||
|
fn try_from(buf: [u8; MAX_EVENT_SIZE]) -> Result<Self, Self::Error> {
|
||||||
|
let mut buf = &buf[..];
|
||||||
|
let event_type = decode_u8(&mut buf)?;
|
||||||
|
match EventType::try_from(event_type)? {
|
||||||
|
EventType::PointerMotion => {
|
||||||
|
Ok(Self::Input(InputEvent::Pointer(PointerEvent::Motion {
|
||||||
|
time: decode_u32(&mut buf)?,
|
||||||
|
dx: decode_f64(&mut buf)?,
|
||||||
|
dy: decode_f64(&mut buf)?,
|
||||||
|
})))
|
||||||
|
}
|
||||||
|
EventType::PointerButton => {
|
||||||
|
Ok(Self::Input(InputEvent::Pointer(PointerEvent::Button {
|
||||||
|
time: decode_u32(&mut buf)?,
|
||||||
|
button: decode_u32(&mut buf)?,
|
||||||
|
state: decode_u32(&mut buf)?,
|
||||||
|
})))
|
||||||
|
}
|
||||||
|
EventType::PointerAxis => Ok(Self::Input(InputEvent::Pointer(PointerEvent::Axis {
|
||||||
|
time: decode_u32(&mut buf)?,
|
||||||
|
axis: decode_u8(&mut buf)?,
|
||||||
|
value: decode_f64(&mut buf)?,
|
||||||
|
}))),
|
||||||
|
EventType::PointerAxisValue120 => Ok(Self::Input(InputEvent::Pointer(
|
||||||
|
PointerEvent::AxisDiscrete120 {
|
||||||
|
axis: decode_u8(&mut buf)?,
|
||||||
|
value: decode_i32(&mut buf)?,
|
||||||
|
},
|
||||||
|
))),
|
||||||
|
EventType::KeyboardKey => Ok(Self::Input(InputEvent::Keyboard(KeyboardEvent::Key {
|
||||||
|
time: decode_u32(&mut buf)?,
|
||||||
|
key: decode_u32(&mut buf)?,
|
||||||
|
state: decode_u8(&mut buf)?,
|
||||||
|
}))),
|
||||||
|
EventType::KeyboardModifiers => Ok(Self::Input(InputEvent::Keyboard(
|
||||||
|
KeyboardEvent::Modifiers {
|
||||||
|
depressed: decode_u32(&mut buf)?,
|
||||||
|
latched: decode_u32(&mut buf)?,
|
||||||
|
locked: decode_u32(&mut buf)?,
|
||||||
|
group: decode_u32(&mut buf)?,
|
||||||
|
},
|
||||||
|
))),
|
||||||
|
EventType::Ping => Ok(Self::Ping),
|
||||||
|
EventType::Pong => Ok(Self::Pong),
|
||||||
|
EventType::Enter => Ok(Self::Enter(decode_u32(&mut buf)?)),
|
||||||
|
EventType::Leave => Ok(Self::Leave(decode_u32(&mut buf)?)),
|
||||||
|
EventType::Ack => Ok(Self::Ack(decode_u32(&mut buf)?)),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<ProtoEvent> for ([u8; MAX_EVENT_SIZE], usize) {
|
||||||
|
fn from(event: ProtoEvent) -> Self {
|
||||||
|
let mut buf = [0u8; MAX_EVENT_SIZE];
|
||||||
|
let mut len = 0usize;
|
||||||
|
{
|
||||||
|
let mut buf = &mut buf[..];
|
||||||
|
let buf = &mut buf;
|
||||||
|
let len = &mut len;
|
||||||
|
encode_u8(buf, len, event.event_type() as u8);
|
||||||
|
match event {
|
||||||
|
ProtoEvent::Input(event) => match event {
|
||||||
|
InputEvent::Pointer(p) => match p {
|
||||||
|
PointerEvent::Motion { time, dx, dy } => {
|
||||||
|
encode_u32(buf, len, time);
|
||||||
|
encode_f64(buf, len, dx);
|
||||||
|
encode_f64(buf, len, dy);
|
||||||
|
}
|
||||||
|
PointerEvent::Button {
|
||||||
|
time,
|
||||||
|
button,
|
||||||
|
state,
|
||||||
|
} => {
|
||||||
|
encode_u32(buf, len, time);
|
||||||
|
encode_u32(buf, len, button);
|
||||||
|
encode_u32(buf, len, state);
|
||||||
|
}
|
||||||
|
PointerEvent::Axis { time, axis, value } => {
|
||||||
|
encode_u32(buf, len, time);
|
||||||
|
encode_u8(buf, len, axis);
|
||||||
|
encode_f64(buf, len, value);
|
||||||
|
}
|
||||||
|
PointerEvent::AxisDiscrete120 { axis, value } => {
|
||||||
|
encode_u8(buf, len, axis);
|
||||||
|
encode_i32(buf, len, value);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
InputEvent::Keyboard(k) => match k {
|
||||||
|
KeyboardEvent::Key { time, key, state } => {
|
||||||
|
encode_u32(buf, len, time);
|
||||||
|
encode_u32(buf, len, key);
|
||||||
|
encode_u8(buf, len, state);
|
||||||
|
}
|
||||||
|
KeyboardEvent::Modifiers {
|
||||||
|
depressed,
|
||||||
|
latched,
|
||||||
|
locked,
|
||||||
|
group,
|
||||||
|
} => {
|
||||||
|
encode_u32(buf, len, depressed);
|
||||||
|
encode_u32(buf, len, latched);
|
||||||
|
encode_u32(buf, len, locked);
|
||||||
|
encode_u32(buf, len, group);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
},
|
||||||
|
ProtoEvent::Ping => {}
|
||||||
|
ProtoEvent::Pong => {}
|
||||||
|
ProtoEvent::Enter(serial) => encode_u32(buf, len, serial),
|
||||||
|
ProtoEvent::Leave(serial) => encode_u32(buf, len, serial),
|
||||||
|
ProtoEvent::Ack(serial) => encode_u32(buf, len, serial),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
(buf, len)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
macro_rules! decode_impl {
|
||||||
|
($t:ty) => {
|
||||||
|
paste! {
|
||||||
|
fn [<decode_ $t>](data: &mut &[u8]) -> Result<$t, ProtocolError> {
|
||||||
|
let (int_bytes, rest) = data.split_at(size_of::<$t>());
|
||||||
|
*data = rest;
|
||||||
|
Ok($t::from_be_bytes(int_bytes.try_into().unwrap()))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
decode_impl!(u8);
|
||||||
|
decode_impl!(u32);
|
||||||
|
decode_impl!(i32);
|
||||||
|
decode_impl!(f64);
|
||||||
|
|
||||||
|
macro_rules! encode_impl {
|
||||||
|
($t:ty) => {
|
||||||
|
paste! {
|
||||||
|
fn [<encode_ $t>](buf: &mut &mut [u8], amt: &mut usize, n: $t) {
|
||||||
|
let src = n.to_be_bytes();
|
||||||
|
let data = std::mem::take(buf);
|
||||||
|
let (int_bytes, rest) = data.split_at_mut(size_of::<$t>());
|
||||||
|
int_bytes.copy_from_slice(&src);
|
||||||
|
*amt += size_of::<$t>();
|
||||||
|
*buf = rest
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
encode_impl!(u8);
|
||||||
|
encode_impl!(u32);
|
||||||
|
encode_impl!(i32);
|
||||||
|
encode_impl!(f64);
|
||||||
@@ -12,6 +12,7 @@ rustPlatform.buildRustPackage {
|
|||||||
version = version;
|
version = version;
|
||||||
|
|
||||||
nativeBuildInputs = with pkgs; [
|
nativeBuildInputs = with pkgs; [
|
||||||
|
git
|
||||||
pkg-config
|
pkg-config
|
||||||
cmake
|
cmake
|
||||||
makeWrapper
|
makeWrapper
|
||||||
@@ -23,9 +24,11 @@ rustPlatform.buildRustPackage {
|
|||||||
gtk4
|
gtk4
|
||||||
libadwaita
|
libadwaita
|
||||||
xorg.libXtst
|
xorg.libXtst
|
||||||
] ++ lib.optionals stdenv.isDarwin [
|
] ++ lib.optionals stdenv.isDarwin
|
||||||
darwin.apple_sdk_11_0.frameworks.CoreGraphics
|
(with darwin.apple_sdk_11_0.frameworks; [
|
||||||
];
|
CoreGraphics
|
||||||
|
ApplicationServices
|
||||||
|
]);
|
||||||
|
|
||||||
src = builtins.path {
|
src = builtins.path {
|
||||||
name = pname;
|
name = pname;
|
||||||
|
|||||||
@@ -1,22 +1,10 @@
|
|||||||
use crate::config::Config;
|
use crate::config::Config;
|
||||||
use futures::StreamExt;
|
use futures::StreamExt;
|
||||||
use input_capture::{self, CaptureError, InputCapture, InputCaptureError, Position};
|
use input_capture::{self, CaptureError, CaptureEvent, InputCapture, InputCaptureError, Position};
|
||||||
use input_event::{Event, KeyboardEvent};
|
use input_event::{Event, KeyboardEvent};
|
||||||
use tokio::task::LocalSet;
|
|
||||||
|
|
||||||
pub fn run() -> anyhow::Result<()> {
|
pub async fn run(config: Config) -> Result<(), InputCaptureError> {
|
||||||
log::info!("running input capture test");
|
log::info!("running input capture test");
|
||||||
let runtime = tokio::runtime::Builder::new_current_thread()
|
|
||||||
.enable_io()
|
|
||||||
.enable_time()
|
|
||||||
.build()?;
|
|
||||||
|
|
||||||
let config = Config::new()?;
|
|
||||||
|
|
||||||
Ok(runtime.block_on(LocalSet::new().run_until(input_capture_test(config)))?)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn input_capture_test(config: Config) -> Result<(), InputCaptureError> {
|
|
||||||
log::info!("creating input capture");
|
log::info!("creating input capture");
|
||||||
let backend = config.capture_backend.map(|b| b.into());
|
let backend = config.capture_backend.map(|b| b.into());
|
||||||
loop {
|
loop {
|
||||||
@@ -46,7 +34,7 @@ async fn do_capture(input_capture: &mut InputCapture) -> Result<(), CaptureError
|
|||||||
_ => Position::Bottom,
|
_ => Position::Bottom,
|
||||||
};
|
};
|
||||||
log::info!("position: {pos}, event: {event}");
|
log::info!("position: {pos}, event: {event}");
|
||||||
if let Event::Keyboard(KeyboardEvent::Key { key: 1, .. }) = event {
|
if let CaptureEvent::Input(Event::Keyboard(KeyboardEvent::Key { key: 1, .. })) = event {
|
||||||
input_capture.release().await?;
|
input_capture.release().await?;
|
||||||
break Ok(());
|
break Ok(());
|
||||||
}
|
}
|
||||||
|
|||||||
131
src/client.rs
131
src/client.rs
@@ -1,135 +1,8 @@
|
|||||||
use std::{
|
use std::net::SocketAddr;
|
||||||
collections::HashSet,
|
|
||||||
fmt::Display,
|
|
||||||
net::{IpAddr, SocketAddr},
|
|
||||||
str::FromStr,
|
|
||||||
};
|
|
||||||
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
use slab::Slab;
|
use slab::Slab;
|
||||||
use thiserror::Error;
|
|
||||||
|
|
||||||
use crate::config::DEFAULT_PORT;
|
use lan_mouse_ipc::{ClientConfig, ClientHandle, ClientState, Position};
|
||||||
use input_capture;
|
|
||||||
|
|
||||||
#[derive(Debug, Default, Eq, Hash, PartialEq, Clone, Copy, Serialize, Deserialize)]
|
|
||||||
pub enum Position {
|
|
||||||
#[default]
|
|
||||||
Left,
|
|
||||||
Right,
|
|
||||||
Top,
|
|
||||||
Bottom,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl From<Position> for input_capture::Position {
|
|
||||||
fn from(position: Position) -> input_capture::Position {
|
|
||||||
match position {
|
|
||||||
Position::Left => input_capture::Position::Left,
|
|
||||||
Position::Right => input_capture::Position::Right,
|
|
||||||
Position::Top => input_capture::Position::Top,
|
|
||||||
Position::Bottom => input_capture::Position::Bottom,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Error)]
|
|
||||||
#[error("not a valid position: {pos}")]
|
|
||||||
pub struct PositionParseError {
|
|
||||||
pos: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl FromStr for Position {
|
|
||||||
type Err = PositionParseError;
|
|
||||||
|
|
||||||
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
||||||
match s {
|
|
||||||
"left" => Ok(Self::Left),
|
|
||||||
"right" => Ok(Self::Right),
|
|
||||||
"top" => Ok(Self::Top),
|
|
||||||
"bottom" => Ok(Self::Bottom),
|
|
||||||
_ => Err(PositionParseError { pos: s.into() }),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Display for Position {
|
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
||||||
write!(
|
|
||||||
f,
|
|
||||||
"{}",
|
|
||||||
match self {
|
|
||||||
Position::Left => "left",
|
|
||||||
Position::Right => "right",
|
|
||||||
Position::Top => "top",
|
|
||||||
Position::Bottom => "bottom",
|
|
||||||
}
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl TryFrom<&str> for Position {
|
|
||||||
type Error = ();
|
|
||||||
|
|
||||||
fn try_from(s: &str) -> Result<Self, Self::Error> {
|
|
||||||
match s {
|
|
||||||
"left" => Ok(Position::Left),
|
|
||||||
"right" => Ok(Position::Right),
|
|
||||||
"top" => Ok(Position::Top),
|
|
||||||
"bottom" => Ok(Position::Bottom),
|
|
||||||
_ => Err(()),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Eq, PartialEq, Clone, Serialize, Deserialize)]
|
|
||||||
pub struct ClientConfig {
|
|
||||||
/// hostname of this client
|
|
||||||
pub hostname: Option<String>,
|
|
||||||
/// fix ips, determined by the user
|
|
||||||
pub fix_ips: Vec<IpAddr>,
|
|
||||||
/// both active_addr and addrs can be None / empty so port needs to be stored seperately
|
|
||||||
pub port: u16,
|
|
||||||
/// position of a client on screen
|
|
||||||
pub pos: Position,
|
|
||||||
/// enter hook
|
|
||||||
pub cmd: Option<String>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Default for ClientConfig {
|
|
||||||
fn default() -> Self {
|
|
||||||
Self {
|
|
||||||
port: DEFAULT_PORT,
|
|
||||||
hostname: Default::default(),
|
|
||||||
fix_ips: Default::default(),
|
|
||||||
pos: Default::default(),
|
|
||||||
cmd: None,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub type ClientHandle = u64;
|
|
||||||
|
|
||||||
#[derive(Debug, Default, Clone, Serialize, Deserialize)]
|
|
||||||
pub struct ClientState {
|
|
||||||
/// events should be sent to and received from the client
|
|
||||||
pub active: bool,
|
|
||||||
/// `active` address of the client, used to send data to.
|
|
||||||
/// This should generally be the socket address where data
|
|
||||||
/// was last received from.
|
|
||||||
pub active_addr: Option<SocketAddr>,
|
|
||||||
/// tracks whether or not the client is responding to pings
|
|
||||||
pub alive: bool,
|
|
||||||
/// ips from dns
|
|
||||||
pub dns_ips: Vec<IpAddr>,
|
|
||||||
/// all ip addresses associated with a particular client
|
|
||||||
/// e.g. Laptops usually have at least an ethernet and a wifi port
|
|
||||||
/// which have different ip addresses
|
|
||||||
pub ips: HashSet<IpAddr>,
|
|
||||||
/// client has pressed keys
|
|
||||||
pub has_pressed_keys: bool,
|
|
||||||
/// dns resolving in progress
|
|
||||||
pub resolving: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Default)]
|
#[derive(Default)]
|
||||||
pub struct ClientManager {
|
pub struct ClientManager {
|
||||||
|
|||||||
@@ -1,22 +1,20 @@
|
|||||||
use anyhow::Result;
|
|
||||||
use clap::{Parser, ValueEnum};
|
use clap::{Parser, ValueEnum};
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
use std::collections::HashSet;
|
use std::env::{self, VarError};
|
||||||
use std::env;
|
|
||||||
use std::fmt::Display;
|
use std::fmt::Display;
|
||||||
|
use std::fs;
|
||||||
use std::net::IpAddr;
|
use std::net::IpAddr;
|
||||||
use std::{error::Error, fs};
|
use std::{collections::HashSet, io};
|
||||||
|
use thiserror::Error;
|
||||||
use toml;
|
use toml;
|
||||||
|
|
||||||
use crate::client::Position;
|
use lan_mouse_ipc::{Position, DEFAULT_PORT};
|
||||||
|
|
||||||
use input_event::scancode::{
|
use input_event::scancode::{
|
||||||
self,
|
self,
|
||||||
Linux::{KeyLeftAlt, KeyLeftCtrl, KeyLeftMeta, KeyLeftShift},
|
Linux::{KeyLeftAlt, KeyLeftCtrl, KeyLeftMeta, KeyLeftShift},
|
||||||
};
|
};
|
||||||
|
|
||||||
pub const DEFAULT_PORT: u16 = 4242;
|
|
||||||
|
|
||||||
#[derive(Serialize, Deserialize, Debug)]
|
#[derive(Serialize, Deserialize, Debug)]
|
||||||
pub struct ConfigToml {
|
pub struct ConfigToml {
|
||||||
pub capture_backend: Option<CaptureBackend>,
|
pub capture_backend: Option<CaptureBackend>,
|
||||||
@@ -42,7 +40,7 @@ pub struct TomlClient {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl ConfigToml {
|
impl ConfigToml {
|
||||||
pub fn new(path: &str) -> Result<ConfigToml, Box<dyn Error>> {
|
pub fn new(path: &str) -> Result<ConfigToml, ConfigError> {
|
||||||
let config = fs::read_to_string(path)?;
|
let config = fs::read_to_string(path)?;
|
||||||
log::info!("using config: \"{path}\"");
|
log::info!("using config: \"{path}\"");
|
||||||
Ok(toml::from_str::<_>(&config)?)
|
Ok(toml::from_str::<_>(&config)?)
|
||||||
@@ -50,7 +48,7 @@ impl ConfigToml {
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Parser, Debug)]
|
#[derive(Parser, Debug)]
|
||||||
#[command(author, version, about, long_about = None)]
|
#[command(author, version=env!("GIT_DESCRIBE"), about, long_about = None)]
|
||||||
struct CliArgs {
|
struct CliArgs {
|
||||||
/// the listen port for lan-mouse
|
/// the listen port for lan-mouse
|
||||||
#[arg(short, long)]
|
#[arg(short, long)]
|
||||||
@@ -231,11 +229,21 @@ pub struct ConfigClient {
|
|||||||
pub enter_hook: Option<String>,
|
pub enter_hook: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Error)]
|
||||||
|
pub enum ConfigError {
|
||||||
|
#[error(transparent)]
|
||||||
|
Toml(#[from] toml::de::Error),
|
||||||
|
#[error(transparent)]
|
||||||
|
Io(#[from] io::Error),
|
||||||
|
#[error(transparent)]
|
||||||
|
Var(#[from] VarError),
|
||||||
|
}
|
||||||
|
|
||||||
const DEFAULT_RELEASE_KEYS: [scancode::Linux; 4] =
|
const DEFAULT_RELEASE_KEYS: [scancode::Linux; 4] =
|
||||||
[KeyLeftCtrl, KeyLeftShift, KeyLeftMeta, KeyLeftAlt];
|
[KeyLeftCtrl, KeyLeftShift, KeyLeftMeta, KeyLeftAlt];
|
||||||
|
|
||||||
impl Config {
|
impl Config {
|
||||||
pub fn new() -> Result<Self> {
|
pub fn new() -> Result<Self, ConfigError> {
|
||||||
let args = CliArgs::parse();
|
let args = CliArgs::parse();
|
||||||
let config_file = "config.toml";
|
let config_file = "config.toml";
|
||||||
#[cfg(unix)]
|
#[cfg(unix)]
|
||||||
|
|||||||
@@ -1,9 +1,10 @@
|
|||||||
|
use local_channel::mpsc::Receiver;
|
||||||
use std::net::IpAddr;
|
use std::net::IpAddr;
|
||||||
use tokio::sync::mpsc::Receiver;
|
|
||||||
|
|
||||||
use hickory_resolver::{error::ResolveError, TokioAsyncResolver};
|
use hickory_resolver::{error::ResolveError, TokioAsyncResolver};
|
||||||
|
|
||||||
use crate::{client::ClientHandle, server::Server};
|
use crate::server::Server;
|
||||||
|
use lan_mouse_ipc::ClientHandle;
|
||||||
|
|
||||||
pub(crate) struct DnsResolver {
|
pub(crate) struct DnsResolver {
|
||||||
resolver: TokioAsyncResolver,
|
resolver: TokioAsyncResolver,
|
||||||
|
|||||||
@@ -1,30 +1,19 @@
|
|||||||
use crate::config::Config;
|
use crate::config::Config;
|
||||||
use anyhow::Result;
|
use input_emulation::{InputEmulation, InputEmulationError};
|
||||||
use input_emulation::InputEmulation;
|
|
||||||
use input_event::{Event, PointerEvent};
|
use input_event::{Event, PointerEvent};
|
||||||
use std::f64::consts::PI;
|
use std::f64::consts::PI;
|
||||||
use std::time::{Duration, Instant};
|
use std::time::{Duration, Instant};
|
||||||
use tokio::task::LocalSet;
|
|
||||||
|
|
||||||
pub fn run() -> Result<()> {
|
|
||||||
log::info!("running input emulation test");
|
|
||||||
let runtime = tokio::runtime::Builder::new_current_thread()
|
|
||||||
.enable_io()
|
|
||||||
.enable_time()
|
|
||||||
.build()?;
|
|
||||||
|
|
||||||
let config = Config::new()?;
|
|
||||||
|
|
||||||
runtime.block_on(LocalSet::new().run_until(input_emulation_test(config)))
|
|
||||||
}
|
|
||||||
|
|
||||||
const FREQUENCY_HZ: f64 = 1.0;
|
const FREQUENCY_HZ: f64 = 1.0;
|
||||||
const RADIUS: f64 = 100.0;
|
const RADIUS: f64 = 100.0;
|
||||||
|
|
||||||
async fn input_emulation_test(config: Config) -> Result<()> {
|
pub async fn run(config: Config) -> Result<(), InputEmulationError> {
|
||||||
|
log::info!("running input emulation test");
|
||||||
|
|
||||||
let backend = config.emulation_backend.map(|b| b.into());
|
let backend = config.emulation_backend.map(|b| b.into());
|
||||||
let mut emulation = InputEmulation::new(backend).await?;
|
let mut emulation = InputEmulation::new(backend).await?;
|
||||||
emulation.create(0).await;
|
emulation.create(0).await;
|
||||||
|
|
||||||
let start = Instant::now();
|
let start = Instant::now();
|
||||||
let mut offset = (0, 0);
|
let mut offset = (0, 0);
|
||||||
loop {
|
loop {
|
||||||
|
|||||||
307
src/frontend.rs
307
src/frontend.rs
@@ -1,307 +0,0 @@
|
|||||||
use anyhow::{anyhow, Result};
|
|
||||||
use std::{cmp::min, io::ErrorKind, net::IpAddr, str, time::Duration};
|
|
||||||
|
|
||||||
#[cfg(unix)]
|
|
||||||
use std::{
|
|
||||||
env,
|
|
||||||
path::{Path, PathBuf},
|
|
||||||
};
|
|
||||||
|
|
||||||
use tokio::io::ReadHalf;
|
|
||||||
use tokio::io::{AsyncReadExt, AsyncWriteExt, WriteHalf};
|
|
||||||
|
|
||||||
#[cfg(unix)]
|
|
||||||
use tokio::net::UnixListener;
|
|
||||||
#[cfg(unix)]
|
|
||||||
use tokio::net::UnixStream;
|
|
||||||
|
|
||||||
#[cfg(windows)]
|
|
||||||
use tokio::net::TcpListener;
|
|
||||||
#[cfg(windows)]
|
|
||||||
use tokio::net::TcpStream;
|
|
||||||
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
|
|
||||||
use crate::{
|
|
||||||
client::{ClientConfig, ClientHandle, ClientState, Position},
|
|
||||||
config::{Config, Frontend},
|
|
||||||
};
|
|
||||||
|
|
||||||
/// cli frontend
|
|
||||||
pub mod cli;
|
|
||||||
|
|
||||||
/// gtk frontend
|
|
||||||
#[cfg(feature = "gtk")]
|
|
||||||
pub mod gtk;
|
|
||||||
|
|
||||||
pub fn run_frontend(config: &Config) -> Result<()> {
|
|
||||||
match config.frontend {
|
|
||||||
#[cfg(feature = "gtk")]
|
|
||||||
Frontend::Gtk => {
|
|
||||||
gtk::run();
|
|
||||||
}
|
|
||||||
#[cfg(not(feature = "gtk"))]
|
|
||||||
Frontend::Gtk => panic!("gtk frontend requested but feature not enabled!"),
|
|
||||||
Frontend::Cli => {
|
|
||||||
cli::run()?;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn exponential_back_off(duration: &mut Duration) -> &Duration {
|
|
||||||
let new = duration.saturating_mul(2);
|
|
||||||
*duration = min(new, Duration::from_secs(1));
|
|
||||||
duration
|
|
||||||
}
|
|
||||||
|
|
||||||
/// wait for the lan-mouse socket to come online
|
|
||||||
#[cfg(unix)]
|
|
||||||
pub fn wait_for_service() -> Result<std::os::unix::net::UnixStream> {
|
|
||||||
let socket_path = FrontendListener::socket_path()?;
|
|
||||||
let mut duration = Duration::from_millis(1);
|
|
||||||
loop {
|
|
||||||
use std::os::unix::net::UnixStream;
|
|
||||||
if let Ok(stream) = UnixStream::connect(&socket_path) {
|
|
||||||
break Ok(stream);
|
|
||||||
}
|
|
||||||
// a signaling mechanism or inotify could be used to
|
|
||||||
// improve this
|
|
||||||
std::thread::sleep(*exponential_back_off(&mut duration));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(windows)]
|
|
||||||
pub fn wait_for_service() -> Result<std::net::TcpStream> {
|
|
||||||
let mut duration = Duration::from_millis(1);
|
|
||||||
loop {
|
|
||||||
use std::net::TcpStream;
|
|
||||||
if let Ok(stream) = TcpStream::connect("127.0.0.1:5252") {
|
|
||||||
break Ok(stream);
|
|
||||||
}
|
|
||||||
std::thread::sleep(*exponential_back_off(&mut duration));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Eq, PartialEq, Clone, Serialize, Deserialize)]
|
|
||||||
pub enum FrontendRequest {
|
|
||||||
/// activate/deactivate client
|
|
||||||
Activate(ClientHandle, bool),
|
|
||||||
/// add a new client
|
|
||||||
Create,
|
|
||||||
/// change the listen port (recreate udp listener)
|
|
||||||
ChangePort(u16),
|
|
||||||
/// remove a client
|
|
||||||
Delete(ClientHandle),
|
|
||||||
/// request an enumeration of all clients
|
|
||||||
Enumerate(),
|
|
||||||
/// resolve dns
|
|
||||||
ResolveDns(ClientHandle),
|
|
||||||
/// update hostname
|
|
||||||
UpdateHostname(ClientHandle, Option<String>),
|
|
||||||
/// update port
|
|
||||||
UpdatePort(ClientHandle, u16),
|
|
||||||
/// update position
|
|
||||||
UpdatePosition(ClientHandle, Position),
|
|
||||||
/// update fix-ips
|
|
||||||
UpdateFixIps(ClientHandle, Vec<IpAddr>),
|
|
||||||
/// request the state of the given client
|
|
||||||
GetState(ClientHandle),
|
|
||||||
/// request reenabling input capture
|
|
||||||
EnableCapture,
|
|
||||||
/// request reenabling input emulation
|
|
||||||
EnableEmulation,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone, Copy, Debug, Default, Serialize, Deserialize)]
|
|
||||||
pub enum Status {
|
|
||||||
#[default]
|
|
||||||
Disabled,
|
|
||||||
Enabled,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl From<Status> for bool {
|
|
||||||
fn from(status: Status) -> Self {
|
|
||||||
match status {
|
|
||||||
Status::Enabled => true,
|
|
||||||
Status::Disabled => false,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
||||||
pub enum FrontendEvent {
|
|
||||||
/// a client was created
|
|
||||||
Created(ClientHandle, ClientConfig, ClientState),
|
|
||||||
/// no such client
|
|
||||||
NoSuchClient(ClientHandle),
|
|
||||||
/// state changed
|
|
||||||
State(ClientHandle, ClientConfig, ClientState),
|
|
||||||
/// the client was deleted
|
|
||||||
Deleted(ClientHandle),
|
|
||||||
/// new port, reason of failure (if failed)
|
|
||||||
PortChanged(u16, Option<String>),
|
|
||||||
/// list of all clients, used for initial state synchronization
|
|
||||||
Enumerate(Vec<(ClientHandle, ClientConfig, ClientState)>),
|
|
||||||
/// an error occured
|
|
||||||
Error(String),
|
|
||||||
/// capture status
|
|
||||||
CaptureStatus(Status),
|
|
||||||
/// emulation status
|
|
||||||
EmulationStatus(Status),
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct FrontendListener {
|
|
||||||
#[cfg(windows)]
|
|
||||||
listener: TcpListener,
|
|
||||||
#[cfg(unix)]
|
|
||||||
listener: UnixListener,
|
|
||||||
#[cfg(unix)]
|
|
||||||
socket_path: PathBuf,
|
|
||||||
#[cfg(unix)]
|
|
||||||
tx_streams: Vec<WriteHalf<UnixStream>>,
|
|
||||||
#[cfg(windows)]
|
|
||||||
tx_streams: Vec<WriteHalf<TcpStream>>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl FrontendListener {
|
|
||||||
#[cfg(all(unix, not(target_os = "macos")))]
|
|
||||||
pub fn socket_path() -> Result<PathBuf> {
|
|
||||||
let xdg_runtime_dir = match env::var("XDG_RUNTIME_DIR") {
|
|
||||||
Ok(d) => d,
|
|
||||||
Err(e) => return Err(anyhow!("could not find XDG_RUNTIME_DIR: {e}")),
|
|
||||||
};
|
|
||||||
let xdg_runtime_dir = Path::new(xdg_runtime_dir.as_str());
|
|
||||||
Ok(xdg_runtime_dir.join("lan-mouse-socket.sock"))
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(all(unix, target_os = "macos"))]
|
|
||||||
pub fn socket_path() -> Result<PathBuf> {
|
|
||||||
let home = match env::var("HOME") {
|
|
||||||
Ok(d) => d,
|
|
||||||
Err(e) => return Err(anyhow!("could not find HOME: {e}")),
|
|
||||||
};
|
|
||||||
let home = Path::new(home.as_str());
|
|
||||||
let path = home
|
|
||||||
.join("Library")
|
|
||||||
.join("Caches")
|
|
||||||
.join("lan-mouse-socket.sock");
|
|
||||||
Ok(path)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn new() -> Option<Result<Self>> {
|
|
||||||
#[cfg(unix)]
|
|
||||||
let (socket_path, listener) = {
|
|
||||||
let socket_path = match Self::socket_path() {
|
|
||||||
Ok(path) => path,
|
|
||||||
Err(e) => return Some(Err(e)),
|
|
||||||
};
|
|
||||||
|
|
||||||
log::debug!("remove socket: {:?}", socket_path);
|
|
||||||
if socket_path.exists() {
|
|
||||||
// try to connect to see if some other instance
|
|
||||||
// of lan-mouse is already running
|
|
||||||
match UnixStream::connect(&socket_path).await {
|
|
||||||
// connected -> lan-mouse is already running
|
|
||||||
Ok(_) => return None,
|
|
||||||
// lan-mouse is not running but a socket was left behind
|
|
||||||
Err(e) => {
|
|
||||||
log::debug!("{socket_path:?}: {e} - removing left behind socket");
|
|
||||||
let _ = std::fs::remove_file(&socket_path);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
let listener = match UnixListener::bind(&socket_path) {
|
|
||||||
Ok(ls) => ls,
|
|
||||||
// some other lan-mouse instance has bound the socket in the meantime
|
|
||||||
Err(e) if e.kind() == ErrorKind::AddrInUse => return None,
|
|
||||||
Err(e) => return Some(Err(anyhow!("failed to bind lan-mouse-socket: {e}"))),
|
|
||||||
};
|
|
||||||
(socket_path, listener)
|
|
||||||
};
|
|
||||||
|
|
||||||
#[cfg(windows)]
|
|
||||||
let listener = match TcpListener::bind("127.0.0.1:5252").await {
|
|
||||||
Ok(ls) => ls,
|
|
||||||
// some other lan-mouse instance has bound the socket in the meantime
|
|
||||||
Err(e) if e.kind() == ErrorKind::AddrInUse => return None,
|
|
||||||
Err(e) => return Some(Err(anyhow!("failed to bind lan-mouse-socket: {e}"))),
|
|
||||||
};
|
|
||||||
|
|
||||||
let adapter = Self {
|
|
||||||
listener,
|
|
||||||
#[cfg(unix)]
|
|
||||||
socket_path,
|
|
||||||
tx_streams: vec![],
|
|
||||||
};
|
|
||||||
|
|
||||||
Some(Ok(adapter))
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(unix)]
|
|
||||||
pub async fn accept(&mut self) -> Result<ReadHalf<UnixStream>> {
|
|
||||||
let stream = self.listener.accept().await?.0;
|
|
||||||
let (rx, tx) = tokio::io::split(stream);
|
|
||||||
self.tx_streams.push(tx);
|
|
||||||
Ok(rx)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(windows)]
|
|
||||||
pub async fn accept(&mut self) -> Result<ReadHalf<TcpStream>> {
|
|
||||||
let stream = self.listener.accept().await?.0;
|
|
||||||
let (rx, tx) = tokio::io::split(stream);
|
|
||||||
self.tx_streams.push(tx);
|
|
||||||
Ok(rx)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) async fn broadcast(&mut self, notify: FrontendEvent) {
|
|
||||||
// encode event
|
|
||||||
let json = serde_json::to_string(¬ify).unwrap();
|
|
||||||
let payload = json.as_bytes();
|
|
||||||
let len = payload.len().to_be_bytes();
|
|
||||||
log::debug!("broadcasting event to streams: {json}");
|
|
||||||
let mut keep = vec![];
|
|
||||||
// TODO do simultaneously
|
|
||||||
for tx in self.tx_streams.iter_mut() {
|
|
||||||
// write len + payload
|
|
||||||
if tx.write(&len).await.is_err() {
|
|
||||||
keep.push(false);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if tx.write(payload).await.is_err() {
|
|
||||||
keep.push(false);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
keep.push(true);
|
|
||||||
}
|
|
||||||
|
|
||||||
// could not find a better solution because async
|
|
||||||
let mut keep = keep.into_iter();
|
|
||||||
self.tx_streams.retain(|_| keep.next().unwrap());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(unix)]
|
|
||||||
impl Drop for FrontendListener {
|
|
||||||
fn drop(&mut self) {
|
|
||||||
log::debug!("remove socket: {:?}", self.socket_path);
|
|
||||||
let _ = std::fs::remove_file(&self.socket_path);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(unix)]
|
|
||||||
pub async fn wait_for_request(stream: &mut ReadHalf<UnixStream>) -> Result<FrontendRequest> {
|
|
||||||
let len = stream.read_u64().await?;
|
|
||||||
assert!(len <= 256);
|
|
||||||
let mut buf = [0u8; 256];
|
|
||||||
stream.read_exact(&mut buf[..len as usize]).await?;
|
|
||||||
Ok(serde_json::from_slice(&buf[..len as usize])?)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(windows)]
|
|
||||||
pub async fn wait_for_request(stream: &mut ReadHalf<TcpStream>) -> Result<FrontendRequest> {
|
|
||||||
let len = stream.read_u64().await?;
|
|
||||||
let mut buf = [0u8; 256];
|
|
||||||
stream.read_exact(&mut buf[..len as usize]).await?;
|
|
||||||
Ok(serde_json::from_slice(&buf[..len as usize])?)
|
|
||||||
}
|
|
||||||
@@ -5,4 +5,3 @@ pub mod server;
|
|||||||
|
|
||||||
pub mod capture_test;
|
pub mod capture_test;
|
||||||
pub mod emulation_test;
|
pub mod emulation_test;
|
||||||
pub mod frontend;
|
|
||||||
|
|||||||
98
src/main.rs
98
src/main.rs
@@ -1,11 +1,37 @@
|
|||||||
use anyhow::Result;
|
|
||||||
use std::process::{self, Child, Command};
|
|
||||||
|
|
||||||
use env_logger::Env;
|
use env_logger::Env;
|
||||||
use lan_mouse::{capture_test, config::Config, emulation_test, frontend, server::Server};
|
use input_capture::InputCaptureError;
|
||||||
|
use input_emulation::InputEmulationError;
|
||||||
|
use lan_mouse::{
|
||||||
|
capture_test,
|
||||||
|
config::{Config, ConfigError, Frontend},
|
||||||
|
emulation_test,
|
||||||
|
server::{Server, ServiceError},
|
||||||
|
};
|
||||||
|
use lan_mouse_ipc::IpcError;
|
||||||
|
use std::{
|
||||||
|
future::Future,
|
||||||
|
io,
|
||||||
|
process::{self, Child, Command},
|
||||||
|
};
|
||||||
|
use thiserror::Error;
|
||||||
use tokio::task::LocalSet;
|
use tokio::task::LocalSet;
|
||||||
|
|
||||||
|
#[derive(Debug, Error)]
|
||||||
|
enum LanMouseError {
|
||||||
|
#[error(transparent)]
|
||||||
|
Service(#[from] ServiceError),
|
||||||
|
#[error(transparent)]
|
||||||
|
IpcError(#[from] IpcError),
|
||||||
|
#[error(transparent)]
|
||||||
|
Config(#[from] ConfigError),
|
||||||
|
#[error(transparent)]
|
||||||
|
Io(#[from] io::Error),
|
||||||
|
#[error(transparent)]
|
||||||
|
Capture(#[from] InputCaptureError),
|
||||||
|
#[error(transparent)]
|
||||||
|
Emulation(#[from] InputEmulationError),
|
||||||
|
}
|
||||||
|
|
||||||
pub fn main() {
|
pub fn main() {
|
||||||
// init logging
|
// init logging
|
||||||
let env = Env::default().filter_or("LAN_MOUSE_LOG_LEVEL", "info");
|
let env = Env::default().filter_or("LAN_MOUSE_LOG_LEVEL", "info");
|
||||||
@@ -17,32 +43,24 @@ pub fn main() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn start_service() -> Result<Child> {
|
fn run() -> Result<(), LanMouseError> {
|
||||||
let child = Command::new(std::env::current_exe()?)
|
|
||||||
.args(std::env::args().skip(1))
|
|
||||||
.arg("--daemon")
|
|
||||||
.spawn()?;
|
|
||||||
Ok(child)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn run() -> Result<()> {
|
|
||||||
// parse config file + cli args
|
// parse config file + cli args
|
||||||
let config = Config::new()?;
|
let config = Config::new()?;
|
||||||
log::debug!("{config:?}");
|
log::debug!("{config:?}");
|
||||||
log::info!("release bind: {:?}", config.release_bind);
|
log::info!("release bind: {:?}", config.release_bind);
|
||||||
|
|
||||||
if config.test_capture {
|
if config.test_capture {
|
||||||
capture_test::run()?;
|
run_async(capture_test::run(config))?;
|
||||||
} else if config.test_emulation {
|
} else if config.test_emulation {
|
||||||
emulation_test::run()?;
|
run_async(emulation_test::run(config))?;
|
||||||
} else if config.daemon {
|
} else if config.daemon {
|
||||||
// if daemon is specified we run the service
|
// if daemon is specified we run the service
|
||||||
run_service(config)?;
|
run_async(run_service(config))?;
|
||||||
} else {
|
} else {
|
||||||
// otherwise start the service as a child process and
|
// otherwise start the service as a child process and
|
||||||
// run a frontend
|
// run a frontend
|
||||||
let mut service = start_service()?;
|
let mut service = start_service()?;
|
||||||
frontend::run_frontend(&config)?;
|
run_frontend(&config)?;
|
||||||
#[cfg(unix)]
|
#[cfg(unix)]
|
||||||
{
|
{
|
||||||
// on unix we give the service a chance to terminate gracefully
|
// on unix we give the service a chance to terminate gracefully
|
||||||
@@ -55,10 +73,14 @@ pub fn run() -> Result<()> {
|
|||||||
service.kill()?;
|
service.kill()?;
|
||||||
}
|
}
|
||||||
|
|
||||||
anyhow::Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn run_service(config: Config) -> Result<()> {
|
fn run_async<F, E>(f: F) -> Result<(), LanMouseError>
|
||||||
|
where
|
||||||
|
F: Future<Output = Result<(), E>>,
|
||||||
|
LanMouseError: From<E>,
|
||||||
|
{
|
||||||
// create single threaded tokio runtime
|
// create single threaded tokio runtime
|
||||||
let runtime = tokio::runtime::Builder::new_current_thread()
|
let runtime = tokio::runtime::Builder::new_current_thread()
|
||||||
.enable_io()
|
.enable_io()
|
||||||
@@ -66,15 +88,35 @@ fn run_service(config: Config) -> Result<()> {
|
|||||||
.build()?;
|
.build()?;
|
||||||
|
|
||||||
// run async event loop
|
// run async event loop
|
||||||
runtime.block_on(LocalSet::new().run_until(async {
|
Ok(runtime.block_on(LocalSet::new().run_until(f))?)
|
||||||
// run main loop
|
}
|
||||||
log::info!("Press {:?} to release the mouse", config.release_bind);
|
|
||||||
|
|
||||||
let mut server = Server::new(config);
|
fn start_service() -> Result<Child, io::Error> {
|
||||||
server.run().await?;
|
let child = Command::new(std::env::current_exe()?)
|
||||||
|
.args(std::env::args().skip(1))
|
||||||
|
.arg("--daemon")
|
||||||
|
.spawn()?;
|
||||||
|
Ok(child)
|
||||||
|
}
|
||||||
|
|
||||||
log::debug!("service exiting");
|
async fn run_service(config: Config) -> Result<(), ServiceError> {
|
||||||
anyhow::Ok(())
|
log::info!("Press {:?} to release the mouse", config.release_bind);
|
||||||
}))?;
|
Server::new(config).run().await?;
|
||||||
|
log::info!("service exited!");
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn run_frontend(config: &Config) -> Result<(), IpcError> {
|
||||||
|
match config.frontend {
|
||||||
|
#[cfg(feature = "gtk")]
|
||||||
|
Frontend::Gtk => {
|
||||||
|
lan_mouse_gtk::run();
|
||||||
|
}
|
||||||
|
#[cfg(not(feature = "gtk"))]
|
||||||
|
Frontend::Gtk => panic!("gtk frontend requested but feature not enabled!"),
|
||||||
|
Frontend::Cli => {
|
||||||
|
lan_mouse_cli::run()?;
|
||||||
|
}
|
||||||
|
};
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
300
src/server.rs
300
src/server.rs
@@ -1,37 +1,27 @@
|
|||||||
use capture_task::CaptureRequest;
|
use capture_task::CaptureRequest;
|
||||||
use emulation_task::EmulationRequest;
|
use emulation_task::EmulationRequest;
|
||||||
|
use futures::StreamExt;
|
||||||
|
use hickory_resolver::error::ResolveError;
|
||||||
|
use local_channel::mpsc::{channel, Sender};
|
||||||
use log;
|
use log;
|
||||||
use std::{
|
use std::{
|
||||||
cell::{Cell, RefCell},
|
cell::{Cell, RefCell},
|
||||||
collections::{HashSet, VecDeque},
|
collections::{HashSet, VecDeque},
|
||||||
io::ErrorKind,
|
io,
|
||||||
net::{IpAddr, SocketAddr},
|
net::{IpAddr, SocketAddr},
|
||||||
rc::Rc,
|
rc::Rc,
|
||||||
};
|
};
|
||||||
use tokio::{
|
use thiserror::Error;
|
||||||
io::ReadHalf,
|
use tokio::{join, signal, sync::Notify};
|
||||||
join, signal,
|
|
||||||
sync::{
|
|
||||||
mpsc::{channel, Sender},
|
|
||||||
Notify,
|
|
||||||
},
|
|
||||||
task::JoinHandle,
|
|
||||||
};
|
|
||||||
use tokio_util::sync::CancellationToken;
|
use tokio_util::sync::CancellationToken;
|
||||||
|
|
||||||
use crate::{
|
use crate::{client::ClientManager, config::Config, dns::DnsResolver};
|
||||||
client::{ClientConfig, ClientHandle, ClientManager, ClientState, Position},
|
|
||||||
config::Config,
|
use lan_mouse_ipc::{
|
||||||
dns::DnsResolver,
|
AsyncFrontendListener, ClientConfig, ClientHandle, ClientState, FrontendEvent, FrontendRequest,
|
||||||
frontend::{self, FrontendEvent, FrontendListener, FrontendRequest, Status},
|
ListenerCreationError, Position, Status,
|
||||||
};
|
};
|
||||||
|
|
||||||
#[cfg(unix)]
|
|
||||||
use tokio::net::UnixStream;
|
|
||||||
|
|
||||||
#[cfg(windows)]
|
|
||||||
use tokio::net::TcpStream;
|
|
||||||
|
|
||||||
mod capture_task;
|
mod capture_task;
|
||||||
mod emulation_task;
|
mod emulation_task;
|
||||||
mod network_task;
|
mod network_task;
|
||||||
@@ -45,7 +35,17 @@ enum State {
|
|||||||
Receiving,
|
Receiving,
|
||||||
/// Entered the deadzone of another device but waiting
|
/// Entered the deadzone of another device but waiting
|
||||||
/// for acknowledgement (Leave event) from the device
|
/// for acknowledgement (Leave event) from the device
|
||||||
AwaitingLeave,
|
AwaitAck,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Error)]
|
||||||
|
pub enum ServiceError {
|
||||||
|
#[error(transparent)]
|
||||||
|
Dns(#[from] ResolveError),
|
||||||
|
#[error(transparent)]
|
||||||
|
Listen(#[from] ListenerCreationError),
|
||||||
|
#[error(transparent)]
|
||||||
|
Io(#[from] io::Error),
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
@@ -58,7 +58,6 @@ pub struct Server {
|
|||||||
notifies: Rc<Notifies>,
|
notifies: Rc<Notifies>,
|
||||||
config: Rc<Config>,
|
config: Rc<Config>,
|
||||||
pending_frontend_events: Rc<RefCell<VecDeque<FrontendEvent>>>,
|
pending_frontend_events: Rc<RefCell<VecDeque<FrontendEvent>>>,
|
||||||
pending_dns_requests: Rc<RefCell<VecDeque<ClientHandle>>>,
|
|
||||||
capture_status: Rc<Cell<Status>>,
|
capture_status: Rc<Cell<Status>>,
|
||||||
emulation_status: Rc<Cell<Status>>,
|
emulation_status: Rc<Cell<Status>>,
|
||||||
}
|
}
|
||||||
@@ -70,7 +69,6 @@ struct Notifies {
|
|||||||
ping: Notify,
|
ping: Notify,
|
||||||
port_changed: Notify,
|
port_changed: Notify,
|
||||||
frontend_event_pending: Notify,
|
frontend_event_pending: Notify,
|
||||||
dns_request_pending: Notify,
|
|
||||||
cancel: CancellationToken,
|
cancel: CancellationToken,
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -114,45 +112,32 @@ impl Server {
|
|||||||
release_bind,
|
release_bind,
|
||||||
notifies,
|
notifies,
|
||||||
pending_frontend_events: Rc::new(RefCell::new(VecDeque::new())),
|
pending_frontend_events: Rc::new(RefCell::new(VecDeque::new())),
|
||||||
pending_dns_requests: Rc::new(RefCell::new(VecDeque::new())),
|
|
||||||
capture_status: Default::default(),
|
capture_status: Default::default(),
|
||||||
emulation_status: Default::default(),
|
emulation_status: Default::default(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn run(&mut self) -> anyhow::Result<()> {
|
pub async fn run(&mut self) -> Result<(), ServiceError> {
|
||||||
// create frontend communication adapter, exit if already running
|
// create frontend communication adapter, exit if already running
|
||||||
let mut frontend = match FrontendListener::new().await {
|
let mut frontend = match AsyncFrontendListener::new().await {
|
||||||
Some(f) => f?,
|
Ok(f) => f,
|
||||||
None => {
|
Err(ListenerCreationError::AlreadyRunning) => {
|
||||||
log::info!("service already running, exiting");
|
log::info!("service already running, exiting");
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
|
e => e?,
|
||||||
};
|
};
|
||||||
|
|
||||||
let (capture_tx, capture_rx) = channel(1); /* requests for input capture */
|
let (capture_tx, capture_rx) = channel(); /* requests for input capture */
|
||||||
let (emulation_tx, emulation_rx) = channel(1); /* emulation requests */
|
let (emulation_tx, emulation_rx) = channel(); /* emulation requests */
|
||||||
let (udp_recv_tx, udp_recv_rx) = channel(1); /* udp receiver */
|
let (udp_recv_tx, udp_recv_rx) = channel(); /* udp receiver */
|
||||||
let (udp_send_tx, udp_send_rx) = channel(1); /* udp sender */
|
let (udp_send_tx, udp_send_rx) = channel(); /* udp sender */
|
||||||
let (request_tx, mut request_rx) = channel(1); /* frontend requests */
|
let (dns_tx, dns_rx) = channel(); /* dns requests */
|
||||||
let (dns_tx, dns_rx) = channel(1); /* dns requests */
|
|
||||||
|
|
||||||
// udp task
|
|
||||||
let network = network_task::new(self.clone(), udp_recv_tx.clone(), udp_send_rx).await?;
|
let network = network_task::new(self.clone(), udp_recv_tx.clone(), udp_send_rx).await?;
|
||||||
|
|
||||||
// input capture
|
|
||||||
let capture = capture_task::new(self.clone(), capture_rx, udp_send_tx.clone());
|
let capture = capture_task::new(self.clone(), capture_rx, udp_send_tx.clone());
|
||||||
|
let emulation =
|
||||||
// input emulation
|
emulation_task::new(self.clone(), emulation_rx, udp_recv_rx, udp_send_tx.clone());
|
||||||
let emulation = emulation_task::new(
|
|
||||||
self.clone(),
|
|
||||||
emulation_rx,
|
|
||||||
udp_recv_rx,
|
|
||||||
udp_send_tx.clone(),
|
|
||||||
capture_tx.clone(),
|
|
||||||
);
|
|
||||||
|
|
||||||
// create dns resolver
|
|
||||||
let resolver = DnsResolver::new(dns_rx)?;
|
let resolver = DnsResolver::new(dns_rx)?;
|
||||||
let dns_task = tokio::task::spawn_local(resolver.run(self.clone()));
|
let dns_task = tokio::task::spawn_local(resolver.run(self.clone()));
|
||||||
|
|
||||||
@@ -165,30 +150,22 @@ impl Server {
|
|||||||
);
|
);
|
||||||
|
|
||||||
for handle in self.active_clients() {
|
for handle in self.active_clients() {
|
||||||
self.request_dns(handle);
|
dns_tx.send(handle).expect("channel closed");
|
||||||
}
|
}
|
||||||
|
|
||||||
log::info!("running service");
|
|
||||||
|
|
||||||
let mut join_handles = vec![];
|
|
||||||
|
|
||||||
loop {
|
loop {
|
||||||
tokio::select! {
|
tokio::select! {
|
||||||
stream = frontend.accept() => {
|
request = frontend.next() => {
|
||||||
match stream {
|
let request = match request {
|
||||||
Ok(s) => join_handles.push(handle_frontend_stream(self.notifies.cancel.clone(), s, request_tx.clone())),
|
Some(Ok(r)) => r,
|
||||||
Err(e) => log::warn!("error accepting frontend connection: {e}"),
|
Some(Err(e)) => {
|
||||||
|
log::error!("error receiving request: {e}");
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
None => break,
|
||||||
};
|
};
|
||||||
self.enumerate();
|
log::debug!("handle frontend request: {request:?}");
|
||||||
self.notify_frontend(FrontendEvent::EmulationStatus(self.emulation_status.get()));
|
self.handle_request(&capture_tx.clone(), &emulation_tx.clone(), request, &dns_tx);
|
||||||
self.notify_frontend(FrontendEvent::CaptureStatus(self.capture_status.get()));
|
|
||||||
self.notify_frontend(FrontendEvent::PortChanged(self.port.get(), None));
|
|
||||||
}
|
|
||||||
request = request_rx.recv() => {
|
|
||||||
let request = request.expect("channel closed");
|
|
||||||
log::debug!("received frontend request: {request:?}");
|
|
||||||
self.handle_request(&capture_tx.clone(), &emulation_tx.clone(), request).await;
|
|
||||||
log::debug!("handled frontend request");
|
|
||||||
}
|
}
|
||||||
_ = self.notifies.frontend_event_pending.notified() => {
|
_ = self.notifies.frontend_event_pending.notified() => {
|
||||||
while let Some(event) = {
|
while let Some(event) = {
|
||||||
@@ -199,15 +176,6 @@ impl Server {
|
|||||||
frontend.broadcast(event).await;
|
frontend.broadcast(event).await;
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
_ = self.notifies.dns_request_pending.notified() => {
|
|
||||||
while let Some(request) = {
|
|
||||||
/* need to drop borrow before next iteration! */
|
|
||||||
let request = self.pending_dns_requests.borrow_mut().pop_front();
|
|
||||||
request
|
|
||||||
} {
|
|
||||||
dns_tx.send(request).await.expect("channel closed");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ = self.cancelled() => break,
|
_ = self.cancelled() => break,
|
||||||
r = signal::ctrl_c() => {
|
r = signal::ctrl_c() => {
|
||||||
r.expect("failed to wait for CTRL+C");
|
r.expect("failed to wait for CTRL+C");
|
||||||
@@ -218,15 +186,7 @@ impl Server {
|
|||||||
|
|
||||||
log::info!("terminating service");
|
log::info!("terminating service");
|
||||||
|
|
||||||
assert!(!capture_tx.is_closed());
|
|
||||||
assert!(!emulation_tx.is_closed());
|
|
||||||
assert!(!udp_recv_tx.is_closed());
|
|
||||||
assert!(!udp_send_tx.is_closed());
|
|
||||||
assert!(!request_tx.is_closed());
|
|
||||||
assert!(!dns_tx.is_closed());
|
|
||||||
|
|
||||||
self.cancel();
|
self.cancel();
|
||||||
futures::future::join_all(join_handles).await;
|
|
||||||
let _ = join!(capture, dns_task, emulation, network, ping);
|
let _ = join!(capture, dns_task, emulation, network, ping);
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
@@ -250,6 +210,7 @@ impl Server {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn notify_capture(&self) {
|
fn notify_capture(&self) {
|
||||||
|
log::info!("received capture enable request");
|
||||||
self.notifies.capture.notify_waiters()
|
self.notifies.capture.notify_waiters()
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -258,6 +219,7 @@ impl Server {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn notify_emulation(&self) {
|
fn notify_emulation(&self) {
|
||||||
|
log::info!("received emulation enable request");
|
||||||
self.notifies.emulation.notify_waiters()
|
self.notifies.emulation.notify_waiters()
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -284,10 +246,7 @@ impl Server {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn client_updated(&self, handle: ClientHandle) {
|
pub(crate) fn client_updated(&self, handle: ClientHandle) {
|
||||||
let state = self.client_manager.borrow().get(handle).cloned();
|
self.notify_frontend(FrontendEvent::Changed(handle));
|
||||||
if let Some((config, state)) = state {
|
|
||||||
self.notify_frontend(FrontendEvent::State(handle, config, state));
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn active_clients(&self) -> Vec<ClientHandle> {
|
fn active_clients(&self) -> Vec<ClientHandle> {
|
||||||
@@ -299,55 +258,49 @@ impl Server {
|
|||||||
.collect()
|
.collect()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn request_dns(&self, handle: ClientHandle) {
|
fn handle_request(
|
||||||
self.pending_dns_requests.borrow_mut().push_back(handle);
|
|
||||||
self.notifies.dns_request_pending.notify_one();
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn handle_request(
|
|
||||||
&self,
|
&self,
|
||||||
capture: &Sender<CaptureRequest>,
|
capture: &Sender<CaptureRequest>,
|
||||||
emulate: &Sender<EmulationRequest>,
|
emulate: &Sender<EmulationRequest>,
|
||||||
event: FrontendRequest,
|
event: FrontendRequest,
|
||||||
|
dns: &Sender<ClientHandle>,
|
||||||
) -> bool {
|
) -> bool {
|
||||||
log::debug!("frontend: {event:?}");
|
log::debug!("frontend: {event:?}");
|
||||||
match event {
|
match event {
|
||||||
FrontendRequest::EnableCapture => {
|
FrontendRequest::EnableCapture => self.notify_capture(),
|
||||||
log::info!("received capture enable request");
|
FrontendRequest::EnableEmulation => self.notify_emulation(),
|
||||||
self.notify_capture();
|
|
||||||
}
|
|
||||||
FrontendRequest::EnableEmulation => {
|
|
||||||
log::info!("received emulation enable request");
|
|
||||||
self.notify_emulation();
|
|
||||||
}
|
|
||||||
FrontendRequest::Create => {
|
FrontendRequest::Create => {
|
||||||
let handle = self.add_client().await;
|
self.add_client();
|
||||||
self.request_dns(handle);
|
|
||||||
}
|
}
|
||||||
FrontendRequest::Activate(handle, active) => {
|
FrontendRequest::Activate(handle, active) => {
|
||||||
if active {
|
if active {
|
||||||
self.activate_client(capture, emulate, handle).await;
|
self.activate_client(capture, emulate, handle);
|
||||||
} else {
|
} else {
|
||||||
self.deactivate_client(capture, emulate, handle).await;
|
self.deactivate_client(capture, emulate, handle);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
FrontendRequest::ChangePort(port) => self.request_port_change(port),
|
FrontendRequest::ChangePort(port) => self.request_port_change(port),
|
||||||
FrontendRequest::Delete(handle) => {
|
FrontendRequest::Delete(handle) => {
|
||||||
self.remove_client(capture, emulate, handle).await;
|
self.remove_client(capture, emulate, handle);
|
||||||
self.notify_frontend(FrontendEvent::Deleted(handle));
|
self.notify_frontend(FrontendEvent::Deleted(handle));
|
||||||
}
|
}
|
||||||
FrontendRequest::Enumerate() => self.enumerate(),
|
FrontendRequest::Enumerate() => self.enumerate(),
|
||||||
FrontendRequest::GetState(handle) => self.broadcast_client(handle),
|
FrontendRequest::GetState(handle) => self.broadcast_client(handle),
|
||||||
FrontendRequest::UpdateFixIps(handle, fix_ips) => {
|
FrontendRequest::UpdateFixIps(handle, fix_ips) => self.update_fix_ips(handle, fix_ips),
|
||||||
self.update_fix_ips(handle, fix_ips);
|
FrontendRequest::UpdateHostname(handle, host) => {
|
||||||
self.request_dns(handle);
|
self.update_hostname(handle, host, dns)
|
||||||
}
|
}
|
||||||
FrontendRequest::UpdateHostname(handle, host) => self.update_hostname(handle, host),
|
|
||||||
FrontendRequest::UpdatePort(handle, port) => self.update_port(handle, port),
|
FrontendRequest::UpdatePort(handle, port) => self.update_port(handle, port),
|
||||||
FrontendRequest::UpdatePosition(handle, pos) => {
|
FrontendRequest::UpdatePosition(handle, pos) => {
|
||||||
self.update_pos(handle, capture, emulate, pos).await;
|
self.update_pos(handle, capture, emulate, pos)
|
||||||
|
}
|
||||||
|
FrontendRequest::ResolveDns(handle) => dns.send(handle).expect("channel closed"),
|
||||||
|
FrontendRequest::Sync => {
|
||||||
|
self.enumerate();
|
||||||
|
self.notify_frontend(FrontendEvent::EmulationStatus(self.emulation_status.get()));
|
||||||
|
self.notify_frontend(FrontendEvent::CaptureStatus(self.capture_status.get()));
|
||||||
|
self.notify_frontend(FrontendEvent::PortChanged(self.port.get(), None));
|
||||||
}
|
}
|
||||||
FrontendRequest::ResolveDns(handle) => self.request_dns(handle),
|
|
||||||
};
|
};
|
||||||
false
|
false
|
||||||
}
|
}
|
||||||
@@ -362,7 +315,7 @@ impl Server {
|
|||||||
self.notify_frontend(FrontendEvent::Enumerate(clients));
|
self.notify_frontend(FrontendEvent::Enumerate(clients));
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn add_client(&self) -> ClientHandle {
|
fn add_client(&self) -> ClientHandle {
|
||||||
let handle = self.client_manager.borrow_mut().add_client();
|
let handle = self.client_manager.borrow_mut().add_client();
|
||||||
log::info!("added client {handle}");
|
log::info!("added client {handle}");
|
||||||
let (c, s) = self.client_manager.borrow().get(handle).unwrap().clone();
|
let (c, s) = self.client_manager.borrow().get(handle).unwrap().clone();
|
||||||
@@ -370,41 +323,40 @@ impl Server {
|
|||||||
handle
|
handle
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn deactivate_client(
|
fn deactivate_client(
|
||||||
&self,
|
&self,
|
||||||
capture: &Sender<CaptureRequest>,
|
capture: &Sender<CaptureRequest>,
|
||||||
emulate: &Sender<EmulationRequest>,
|
emulate: &Sender<EmulationRequest>,
|
||||||
handle: ClientHandle,
|
handle: ClientHandle,
|
||||||
) {
|
) {
|
||||||
log::debug!("deactivating client {handle}");
|
|
||||||
match self.client_manager.borrow_mut().get_mut(handle) {
|
match self.client_manager.borrow_mut().get_mut(handle) {
|
||||||
Some((_, s)) => s.active = false,
|
|
||||||
None => return,
|
None => return,
|
||||||
|
Some((_, s)) if !s.active => return,
|
||||||
|
Some((_, s)) => s.active = false,
|
||||||
};
|
};
|
||||||
|
|
||||||
let _ = capture.send(CaptureRequest::Destroy(handle)).await;
|
let _ = capture.send(CaptureRequest::Destroy(handle));
|
||||||
let _ = emulate.send(EmulationRequest::Destroy(handle)).await;
|
let _ = emulate.send(EmulationRequest::Destroy(handle));
|
||||||
log::debug!("deactivating client {handle} done");
|
self.client_updated(handle);
|
||||||
|
log::info!("deactivated client {handle}");
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn activate_client(
|
fn activate_client(
|
||||||
&self,
|
&self,
|
||||||
capture: &Sender<CaptureRequest>,
|
capture: &Sender<CaptureRequest>,
|
||||||
emulate: &Sender<EmulationRequest>,
|
emulate: &Sender<EmulationRequest>,
|
||||||
handle: ClientHandle,
|
handle: ClientHandle,
|
||||||
) {
|
) {
|
||||||
log::debug!("activating client");
|
|
||||||
/* deactivate potential other client at this position */
|
/* deactivate potential other client at this position */
|
||||||
let pos = match self.client_manager.borrow().get(handle) {
|
let pos = match self.client_manager.borrow().get(handle) {
|
||||||
Some((client, _)) => client.pos,
|
|
||||||
None => return,
|
None => return,
|
||||||
|
Some((_, s)) if s.active => return,
|
||||||
|
Some((client, _)) => client.pos,
|
||||||
};
|
};
|
||||||
|
|
||||||
let other = self.client_manager.borrow_mut().find_client(pos);
|
let other = self.client_manager.borrow_mut().find_client(pos);
|
||||||
if let Some(other) = other {
|
if let Some(other) = other {
|
||||||
if other != handle {
|
self.deactivate_client(capture, emulate, other);
|
||||||
self.deactivate_client(capture, emulate, other).await;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/* activate the client */
|
/* activate the client */
|
||||||
@@ -415,14 +367,15 @@ impl Server {
|
|||||||
};
|
};
|
||||||
|
|
||||||
/* notify emulation, capture and frontends */
|
/* notify emulation, capture and frontends */
|
||||||
let _ = capture
|
let _ = capture.send(CaptureRequest::Create(handle, to_capture_pos(pos)));
|
||||||
.send(CaptureRequest::Create(handle, pos.into()))
|
let _ = emulate.send(EmulationRequest::Create(handle));
|
||||||
.await;
|
|
||||||
let _ = emulate.send(EmulationRequest::Create(handle)).await;
|
self.client_updated(handle);
|
||||||
log::debug!("activating client {handle} done");
|
|
||||||
|
log::info!("activated client {handle} ({pos})");
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn remove_client(
|
fn remove_client(
|
||||||
&self,
|
&self,
|
||||||
capture: &Sender<CaptureRequest>,
|
capture: &Sender<CaptureRequest>,
|
||||||
emulate: &Sender<EmulationRequest>,
|
emulate: &Sender<EmulationRequest>,
|
||||||
@@ -438,8 +391,8 @@ impl Server {
|
|||||||
};
|
};
|
||||||
|
|
||||||
if active {
|
if active {
|
||||||
let _ = capture.send(CaptureRequest::Destroy(handle)).await;
|
let _ = capture.send(CaptureRequest::Destroy(handle));
|
||||||
let _ = emulate.send(EmulationRequest::Destroy(handle)).await;
|
let _ = emulate.send(EmulationRequest::Destroy(handle));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -454,6 +407,7 @@ impl Server {
|
|||||||
c.fix_ips = fix_ips;
|
c.fix_ips = fix_ips;
|
||||||
};
|
};
|
||||||
self.update_ips(handle);
|
self.update_ips(handle);
|
||||||
|
self.client_updated(handle);
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn update_dns_ips(&self, handle: ClientHandle, dns_ips: Vec<IpAddr>) {
|
pub(crate) fn update_dns_ips(&self, handle: ClientHandle, dns_ips: Vec<IpAddr>) {
|
||||||
@@ -461,6 +415,7 @@ impl Server {
|
|||||||
s.dns_ips = dns_ips;
|
s.dns_ips = dns_ips;
|
||||||
};
|
};
|
||||||
self.update_ips(handle);
|
self.update_ips(handle);
|
||||||
|
self.client_updated(handle);
|
||||||
}
|
}
|
||||||
|
|
||||||
fn update_ips(&self, handle: ClientHandle) {
|
fn update_ips(&self, handle: ClientHandle) {
|
||||||
@@ -474,7 +429,12 @@ impl Server {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn update_hostname(&self, handle: ClientHandle, hostname: Option<String>) {
|
fn update_hostname(
|
||||||
|
&self,
|
||||||
|
handle: ClientHandle,
|
||||||
|
hostname: Option<String>,
|
||||||
|
dns: &Sender<ClientHandle>,
|
||||||
|
) {
|
||||||
let mut client_manager = self.client_manager.borrow_mut();
|
let mut client_manager = self.client_manager.borrow_mut();
|
||||||
let Some((c, s)) = client_manager.get_mut(handle) else {
|
let Some((c, s)) = client_manager.get_mut(handle) else {
|
||||||
return;
|
return;
|
||||||
@@ -483,10 +443,13 @@ impl Server {
|
|||||||
// hostname changed
|
// hostname changed
|
||||||
if c.hostname != hostname {
|
if c.hostname != hostname {
|
||||||
c.hostname = hostname;
|
c.hostname = hostname;
|
||||||
s.ips = HashSet::from_iter(c.fix_ips.iter().cloned());
|
|
||||||
s.active_addr = None;
|
s.active_addr = None;
|
||||||
self.request_dns(handle);
|
s.dns_ips.clear();
|
||||||
|
drop(client_manager);
|
||||||
|
self.update_ips(handle);
|
||||||
|
dns.send(handle).expect("channel closed");
|
||||||
}
|
}
|
||||||
|
self.client_updated(handle);
|
||||||
}
|
}
|
||||||
|
|
||||||
fn update_port(&self, handle: ClientHandle, port: u16) {
|
fn update_port(&self, handle: ClientHandle, port: u16) {
|
||||||
@@ -501,7 +464,7 @@ impl Server {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn update_pos(
|
fn update_pos(
|
||||||
&self,
|
&self,
|
||||||
handle: ClientHandle,
|
handle: ClientHandle,
|
||||||
capture: &Sender<CaptureRequest>,
|
capture: &Sender<CaptureRequest>,
|
||||||
@@ -515,20 +478,19 @@ impl Server {
|
|||||||
};
|
};
|
||||||
|
|
||||||
let changed = c.pos != pos;
|
let changed = c.pos != pos;
|
||||||
|
if changed {
|
||||||
|
log::info!("update pos {handle} {} -> {}", c.pos, pos);
|
||||||
|
}
|
||||||
c.pos = pos;
|
c.pos = pos;
|
||||||
(changed, s.active)
|
(changed, s.active)
|
||||||
};
|
};
|
||||||
|
|
||||||
// update state in event input emulator & input capture
|
// update state in event input emulator & input capture
|
||||||
if changed {
|
if changed {
|
||||||
|
self.deactivate_client(capture, emulate, handle);
|
||||||
if active {
|
if active {
|
||||||
let _ = capture.send(CaptureRequest::Destroy(handle)).await;
|
self.activate_client(capture, emulate, handle);
|
||||||
let _ = emulate.send(EmulationRequest::Destroy(handle)).await;
|
|
||||||
}
|
}
|
||||||
let _ = capture
|
|
||||||
.send(CaptureRequest::Create(handle, pos.into()))
|
|
||||||
.await;
|
|
||||||
let _ = emulate.send(EmulationRequest::Create(handle)).await;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -561,7 +523,7 @@ impl Server {
|
|||||||
self.client_updated(handle);
|
self.client_updated(handle);
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn get_hostname(&self, handle: u64) -> Option<String> {
|
pub(crate) fn get_hostname(&self, handle: ClientHandle) -> Option<String> {
|
||||||
self.client_manager
|
self.client_manager
|
||||||
.borrow_mut()
|
.borrow_mut()
|
||||||
.get_mut(handle)
|
.get_mut(handle)
|
||||||
@@ -577,12 +539,12 @@ impl Server {
|
|||||||
self.state.replace(state);
|
self.state.replace(state);
|
||||||
}
|
}
|
||||||
|
|
||||||
fn set_active(&self, handle: Option<u64>) {
|
fn set_active(&self, handle: Option<ClientHandle>) {
|
||||||
log::debug!("active client => {handle:?}");
|
log::debug!("active client => {handle:?}");
|
||||||
self.active_client.replace(handle);
|
self.active_client.replace(handle);
|
||||||
}
|
}
|
||||||
|
|
||||||
fn active_addr(&self, handle: u64) -> Option<SocketAddr> {
|
fn active_addr(&self, handle: ClientHandle) -> Option<SocketAddr> {
|
||||||
self.client_manager
|
self.client_manager
|
||||||
.borrow()
|
.borrow()
|
||||||
.get(handle)
|
.get(handle)
|
||||||
@@ -590,41 +552,11 @@ impl Server {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn listen_frontend(
|
fn to_capture_pos(pos: Position) -> input_capture::Position {
|
||||||
request_tx: Sender<FrontendRequest>,
|
match pos {
|
||||||
#[cfg(unix)] mut stream: ReadHalf<UnixStream>,
|
Position::Left => input_capture::Position::Left,
|
||||||
#[cfg(windows)] mut stream: ReadHalf<TcpStream>,
|
Position::Right => input_capture::Position::Right,
|
||||||
) {
|
Position::Top => input_capture::Position::Top,
|
||||||
use std::io;
|
Position::Bottom => input_capture::Position::Bottom,
|
||||||
loop {
|
|
||||||
let request = frontend::wait_for_request(&mut stream).await;
|
|
||||||
match request {
|
|
||||||
Ok(request) => {
|
|
||||||
let _ = request_tx.send(request).await;
|
|
||||||
}
|
|
||||||
Err(e) => {
|
|
||||||
if let Some(e) = e.downcast_ref::<io::Error>() {
|
|
||||||
if e.kind() == ErrorKind::UnexpectedEof {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
log::error!("error reading frontend event: {e}");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn handle_frontend_stream(
|
|
||||||
cancel: CancellationToken,
|
|
||||||
#[cfg(unix)] stream: ReadHalf<UnixStream>,
|
|
||||||
#[cfg(windows)] stream: ReadHalf<TcpStream>,
|
|
||||||
request_tx: Sender<FrontendRequest>,
|
|
||||||
) -> JoinHandle<()> {
|
|
||||||
tokio::task::spawn_local(async move {
|
|
||||||
tokio::select! {
|
|
||||||
_ = listen_frontend(request_tx, stream) => {},
|
|
||||||
_ = cancel.cancelled() => {},
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,17 +1,16 @@
|
|||||||
use futures::StreamExt;
|
use futures::StreamExt;
|
||||||
|
use lan_mouse_proto::ProtoEvent;
|
||||||
|
use local_channel::mpsc::{Receiver, Sender};
|
||||||
use std::net::SocketAddr;
|
use std::net::SocketAddr;
|
||||||
|
|
||||||
use tokio::{
|
use tokio::{process::Command, task::JoinHandle};
|
||||||
process::Command,
|
|
||||||
sync::mpsc::{Receiver, Sender},
|
use input_capture::{
|
||||||
task::JoinHandle,
|
self, CaptureError, CaptureEvent, CaptureHandle, InputCapture, InputCaptureError, Position,
|
||||||
};
|
};
|
||||||
|
|
||||||
use input_capture::{self, CaptureError, CaptureHandle, InputCapture, InputCaptureError, Position};
|
use crate::server::State;
|
||||||
|
use lan_mouse_ipc::{ClientHandle, Status};
|
||||||
use input_event::Event;
|
|
||||||
|
|
||||||
use crate::{client::ClientHandle, frontend::Status, server::State};
|
|
||||||
|
|
||||||
use super::Server;
|
use super::Server;
|
||||||
|
|
||||||
@@ -28,7 +27,7 @@ pub(crate) enum CaptureRequest {
|
|||||||
pub(crate) fn new(
|
pub(crate) fn new(
|
||||||
server: Server,
|
server: Server,
|
||||||
capture_rx: Receiver<CaptureRequest>,
|
capture_rx: Receiver<CaptureRequest>,
|
||||||
udp_send: Sender<(Event, SocketAddr)>,
|
udp_send: Sender<(ProtoEvent, SocketAddr)>,
|
||||||
) -> JoinHandle<()> {
|
) -> JoinHandle<()> {
|
||||||
let backend = server.config.capture_backend.map(|b| b.into());
|
let backend = server.config.capture_backend.map(|b| b.into());
|
||||||
tokio::task::spawn_local(capture_task(server, backend, udp_send, capture_rx))
|
tokio::task::spawn_local(capture_task(server, backend, udp_send, capture_rx))
|
||||||
@@ -37,7 +36,7 @@ pub(crate) fn new(
|
|||||||
async fn capture_task(
|
async fn capture_task(
|
||||||
server: Server,
|
server: Server,
|
||||||
backend: Option<input_capture::Backend>,
|
backend: Option<input_capture::Backend>,
|
||||||
sender_tx: Sender<(Event, SocketAddr)>,
|
sender_tx: Sender<(ProtoEvent, SocketAddr)>,
|
||||||
mut notify_rx: Receiver<CaptureRequest>,
|
mut notify_rx: Receiver<CaptureRequest>,
|
||||||
) {
|
) {
|
||||||
loop {
|
loop {
|
||||||
@@ -63,7 +62,7 @@ async fn capture_task(
|
|||||||
async fn do_capture(
|
async fn do_capture(
|
||||||
backend: Option<input_capture::Backend>,
|
backend: Option<input_capture::Backend>,
|
||||||
server: &Server,
|
server: &Server,
|
||||||
sender_tx: &Sender<(Event, SocketAddr)>,
|
sender_tx: &Sender<(ProtoEvent, SocketAddr)>,
|
||||||
notify_rx: &mut Receiver<CaptureRequest>,
|
notify_rx: &mut Receiver<CaptureRequest>,
|
||||||
) -> Result<(), InputCaptureError> {
|
) -> Result<(), InputCaptureError> {
|
||||||
/* allow cancelling capture request */
|
/* allow cancelling capture request */
|
||||||
@@ -87,7 +86,7 @@ async fn do_capture(
|
|||||||
)
|
)
|
||||||
});
|
});
|
||||||
for (handle, pos) in clients {
|
for (handle, pos) in clients {
|
||||||
capture.create(handle, pos.into()).await?;
|
capture.create(handle, to_capture_pos(pos)).await?;
|
||||||
}
|
}
|
||||||
|
|
||||||
loop {
|
loop {
|
||||||
@@ -120,17 +119,20 @@ async fn do_capture(
|
|||||||
async fn handle_capture_event(
|
async fn handle_capture_event(
|
||||||
server: &Server,
|
server: &Server,
|
||||||
capture: &mut InputCapture,
|
capture: &mut InputCapture,
|
||||||
sender_tx: &Sender<(Event, SocketAddr)>,
|
sender_tx: &Sender<(ProtoEvent, SocketAddr)>,
|
||||||
event: (CaptureHandle, Event),
|
event: (CaptureHandle, CaptureEvent),
|
||||||
) -> Result<(), CaptureError> {
|
) -> Result<(), CaptureError> {
|
||||||
let (handle, event) = event;
|
let (handle, event) = event;
|
||||||
log::trace!("({handle}) {event:?}");
|
log::trace!("({handle}) {event:?}");
|
||||||
|
|
||||||
// capture started
|
// capture started
|
||||||
if event == Event::Enter() {
|
if event == CaptureEvent::Begin {
|
||||||
server.set_state(State::AwaitingLeave);
|
// wait for remote to acknowlegde enter
|
||||||
|
server.set_state(State::AwaitAck);
|
||||||
server.set_active(Some(handle));
|
server.set_active(Some(handle));
|
||||||
|
// restart ping timer to release capture if unreachable
|
||||||
server.restart_ping_timer();
|
server.restart_ping_timer();
|
||||||
|
// spawn enter hook cmd
|
||||||
spawn_hook_command(server, handle);
|
spawn_hook_command(server, handle);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -148,14 +150,18 @@ async fn handle_capture_event(
|
|||||||
|
|
||||||
if let Some(addr) = server.active_addr(handle) {
|
if let Some(addr) = server.active_addr(handle) {
|
||||||
let event = match server.get_state() {
|
let event = match server.get_state() {
|
||||||
State::Sending => event,
|
State::Sending => match event {
|
||||||
|
CaptureEvent::Begin => ProtoEvent::Enter(0),
|
||||||
|
CaptureEvent::Input(e) => ProtoEvent::Input(e),
|
||||||
|
},
|
||||||
/* send additional enter events until acknowleged */
|
/* send additional enter events until acknowleged */
|
||||||
State::AwaitingLeave => Event::Enter(),
|
State::AwaitAck => ProtoEvent::Enter(0),
|
||||||
/* released capture */
|
/* released capture */
|
||||||
State::Receiving => Event::Disconnect(),
|
State::Receiving => ProtoEvent::Leave(0),
|
||||||
};
|
};
|
||||||
sender_tx.send((event, addr)).await.expect("sender closed");
|
sender_tx.send((event, addr)).expect("sender closed");
|
||||||
}
|
};
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -189,3 +195,12 @@ fn spawn_hook_command(server: &Server, handle: ClientHandle) {
|
|||||||
}
|
}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn to_capture_pos(pos: lan_mouse_ipc::Position) -> input_capture::Position {
|
||||||
|
match pos {
|
||||||
|
lan_mouse_ipc::Position::Left => input_capture::Position::Left,
|
||||||
|
lan_mouse_ipc::Position::Right => input_capture::Position::Right,
|
||||||
|
lan_mouse_ipc::Position::Top => input_capture::Position::Top,
|
||||||
|
lan_mouse_ipc::Position::Bottom => input_capture::Position::Bottom,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,19 +1,16 @@
|
|||||||
|
use local_channel::mpsc::{Receiver, Sender};
|
||||||
use std::net::SocketAddr;
|
use std::net::SocketAddr;
|
||||||
|
|
||||||
use tokio::{
|
use lan_mouse_proto::ProtoEvent;
|
||||||
sync::mpsc::{Receiver, Sender},
|
use tokio::task::JoinHandle;
|
||||||
task::JoinHandle,
|
|
||||||
};
|
|
||||||
|
|
||||||
use crate::{
|
use lan_mouse_ipc::ClientHandle;
|
||||||
client::{ClientHandle, ClientManager},
|
|
||||||
frontend::Status,
|
use crate::{client::ClientManager, server::State};
|
||||||
server::State,
|
|
||||||
};
|
|
||||||
use input_emulation::{self, EmulationError, EmulationHandle, InputEmulation, InputEmulationError};
|
use input_emulation::{self, EmulationError, EmulationHandle, InputEmulation, InputEmulationError};
|
||||||
use input_event::Event;
|
use lan_mouse_ipc::Status;
|
||||||
|
|
||||||
use super::{network_task::NetworkError, CaptureRequest, Server};
|
use super::{network_task::NetworkError, Server};
|
||||||
|
|
||||||
#[derive(Clone, Debug)]
|
#[derive(Clone, Debug)]
|
||||||
pub(crate) enum EmulationRequest {
|
pub(crate) enum EmulationRequest {
|
||||||
@@ -28,23 +25,21 @@ pub(crate) enum EmulationRequest {
|
|||||||
pub(crate) fn new(
|
pub(crate) fn new(
|
||||||
server: Server,
|
server: Server,
|
||||||
emulation_rx: Receiver<EmulationRequest>,
|
emulation_rx: Receiver<EmulationRequest>,
|
||||||
udp_rx: Receiver<Result<(Event, SocketAddr), NetworkError>>,
|
udp_rx: Receiver<Result<(ProtoEvent, SocketAddr), NetworkError>>,
|
||||||
sender_tx: Sender<(Event, SocketAddr)>,
|
sender_tx: Sender<(ProtoEvent, SocketAddr)>,
|
||||||
capture_tx: Sender<CaptureRequest>,
|
|
||||||
) -> JoinHandle<()> {
|
) -> JoinHandle<()> {
|
||||||
let emulation_task = emulation_task(server, emulation_rx, udp_rx, sender_tx, capture_tx);
|
let emulation_task = emulation_task(server, emulation_rx, udp_rx, sender_tx);
|
||||||
tokio::task::spawn_local(emulation_task)
|
tokio::task::spawn_local(emulation_task)
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn emulation_task(
|
async fn emulation_task(
|
||||||
server: Server,
|
server: Server,
|
||||||
mut rx: Receiver<EmulationRequest>,
|
mut rx: Receiver<EmulationRequest>,
|
||||||
mut udp_rx: Receiver<Result<(Event, SocketAddr), NetworkError>>,
|
mut udp_rx: Receiver<Result<(ProtoEvent, SocketAddr), NetworkError>>,
|
||||||
sender_tx: Sender<(Event, SocketAddr)>,
|
sender_tx: Sender<(ProtoEvent, SocketAddr)>,
|
||||||
capture_tx: Sender<CaptureRequest>,
|
|
||||||
) {
|
) {
|
||||||
loop {
|
loop {
|
||||||
if let Err(e) = do_emulation(&server, &mut rx, &mut udp_rx, &sender_tx, &capture_tx).await {
|
if let Err(e) = do_emulation(&server, &mut rx, &mut udp_rx, &sender_tx).await {
|
||||||
log::warn!("input emulation exited: {e}");
|
log::warn!("input emulation exited: {e}");
|
||||||
}
|
}
|
||||||
server.set_emulation_status(Status::Disabled);
|
server.set_emulation_status(Status::Disabled);
|
||||||
@@ -66,9 +61,8 @@ async fn emulation_task(
|
|||||||
async fn do_emulation(
|
async fn do_emulation(
|
||||||
server: &Server,
|
server: &Server,
|
||||||
rx: &mut Receiver<EmulationRequest>,
|
rx: &mut Receiver<EmulationRequest>,
|
||||||
udp_rx: &mut Receiver<Result<(Event, SocketAddr), NetworkError>>,
|
udp_rx: &mut Receiver<Result<(ProtoEvent, SocketAddr), NetworkError>>,
|
||||||
sender_tx: &Sender<(Event, SocketAddr)>,
|
sender_tx: &Sender<(ProtoEvent, SocketAddr)>,
|
||||||
capture_tx: &Sender<CaptureRequest>,
|
|
||||||
) -> Result<(), InputEmulationError> {
|
) -> Result<(), InputEmulationError> {
|
||||||
let backend = server.config.emulation_backend.map(|b| b.into());
|
let backend = server.config.emulation_backend.map(|b| b.into());
|
||||||
log::info!("creating input emulation...");
|
log::info!("creating input emulation...");
|
||||||
@@ -84,7 +78,7 @@ async fn do_emulation(
|
|||||||
emulation.create(handle).await;
|
emulation.create(handle).await;
|
||||||
}
|
}
|
||||||
|
|
||||||
let res = do_emulation_session(server, &mut emulation, rx, udp_rx, sender_tx, capture_tx).await;
|
let res = do_emulation_session(server, &mut emulation, rx, udp_rx, sender_tx).await;
|
||||||
emulation.terminate().await; // manual drop
|
emulation.terminate().await; // manual drop
|
||||||
res
|
res
|
||||||
}
|
}
|
||||||
@@ -93,9 +87,8 @@ async fn do_emulation_session(
|
|||||||
server: &Server,
|
server: &Server,
|
||||||
emulation: &mut InputEmulation,
|
emulation: &mut InputEmulation,
|
||||||
rx: &mut Receiver<EmulationRequest>,
|
rx: &mut Receiver<EmulationRequest>,
|
||||||
udp_rx: &mut Receiver<Result<(Event, SocketAddr), NetworkError>>,
|
udp_rx: &mut Receiver<Result<(ProtoEvent, SocketAddr), NetworkError>>,
|
||||||
sender_tx: &Sender<(Event, SocketAddr)>,
|
sender_tx: &Sender<(ProtoEvent, SocketAddr)>,
|
||||||
capture_tx: &Sender<CaptureRequest>,
|
|
||||||
) -> Result<(), InputEmulationError> {
|
) -> Result<(), InputEmulationError> {
|
||||||
let mut last_ignored = None;
|
let mut last_ignored = None;
|
||||||
|
|
||||||
@@ -109,7 +102,7 @@ async fn do_emulation_session(
|
|||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
handle_udp_rx(server, capture_tx, emulation, sender_tx, &mut last_ignored, udp_event).await?;
|
handle_incoming_event(server, emulation, sender_tx, &mut last_ignored, udp_event).await?;
|
||||||
}
|
}
|
||||||
emulate_event = rx.recv() => {
|
emulate_event = rx.recv() => {
|
||||||
match emulate_event.expect("channel closed") {
|
match emulate_event.expect("channel closed") {
|
||||||
@@ -123,13 +116,12 @@ async fn do_emulation_session(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle_udp_rx(
|
async fn handle_incoming_event(
|
||||||
server: &Server,
|
server: &Server,
|
||||||
capture_tx: &Sender<CaptureRequest>,
|
|
||||||
emulate: &mut InputEmulation,
|
emulate: &mut InputEmulation,
|
||||||
sender_tx: &Sender<(Event, SocketAddr)>,
|
sender_tx: &Sender<(ProtoEvent, SocketAddr)>,
|
||||||
last_ignored: &mut Option<SocketAddr>,
|
last_ignored: &mut Option<SocketAddr>,
|
||||||
event: (Event, SocketAddr),
|
event: (ProtoEvent, SocketAddr),
|
||||||
) -> Result<(), EmulationError> {
|
) -> Result<(), EmulationError> {
|
||||||
let (event, addr) = event;
|
let (event, addr) = event;
|
||||||
|
|
||||||
@@ -143,55 +135,26 @@ async fn handle_udp_rx(
|
|||||||
};
|
};
|
||||||
|
|
||||||
match (event, addr) {
|
match (event, addr) {
|
||||||
(Event::Pong(), _) => { /* ignore pong events */ }
|
(ProtoEvent::Pong, _) => { /* ignore pong events */ }
|
||||||
(Event::Ping(), addr) => {
|
(ProtoEvent::Ping, addr) => {
|
||||||
let _ = sender_tx.send((Event::Pong(), addr)).await;
|
let _ = sender_tx.send((ProtoEvent::Pong, addr));
|
||||||
}
|
}
|
||||||
(Event::Disconnect(), _) => emulate.release_keys(handle).await?,
|
(ProtoEvent::Leave(_), _) => emulate.release_keys(handle).await?,
|
||||||
(event, addr) => {
|
(ProtoEvent::Ack(_), _) => server.set_state(State::Sending),
|
||||||
// tell clients that we are ready to receive events
|
(ProtoEvent::Enter(_), _) => {
|
||||||
if let Event::Enter() = event {
|
server.set_state(State::Receiving);
|
||||||
let _ = sender_tx.send((Event::Leave(), addr)).await;
|
sender_tx
|
||||||
}
|
.send((ProtoEvent::Ack(0), addr))
|
||||||
|
.expect("no channel")
|
||||||
match server.state.get() {
|
}
|
||||||
State::Sending => {
|
(ProtoEvent::Input(e), _) => {
|
||||||
if let Event::Leave() = event {
|
if let State::Receiving = server.get_state() {
|
||||||
// ignore additional leave events that may
|
log::trace!("{event} => emulate");
|
||||||
// have been sent for redundancy
|
emulate.consume(e, handle).await?;
|
||||||
} else {
|
let has_pressed_keys = emulate.has_pressed_keys(handle);
|
||||||
// upon receiving any event, we go back to receiving mode
|
server.update_pressed_keys(handle, has_pressed_keys);
|
||||||
server.state.replace(State::Receiving);
|
if has_pressed_keys {
|
||||||
let _ = capture_tx.send(CaptureRequest::Release).await;
|
server.restart_ping_timer();
|
||||||
log::trace!("STATE ===> Receiving");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
State::Receiving => {
|
|
||||||
log::trace!("{event} => emulate");
|
|
||||||
emulate.consume(event, handle).await?;
|
|
||||||
let has_pressed_keys = emulate.has_pressed_keys(handle);
|
|
||||||
server.update_pressed_keys(handle, has_pressed_keys);
|
|
||||||
if has_pressed_keys {
|
|
||||||
server.restart_ping_timer();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
State::AwaitingLeave => {
|
|
||||||
// we just entered the deadzone of a client, so
|
|
||||||
// we need to ignore events that may still
|
|
||||||
// be on the way until a leave event occurs
|
|
||||||
// telling us the client registered the enter
|
|
||||||
if let Event::Leave() = event {
|
|
||||||
server.state.replace(State::Sending);
|
|
||||||
log::trace!("STATE ===> Sending");
|
|
||||||
}
|
|
||||||
|
|
||||||
// entering a client that is waiting for a leave
|
|
||||||
// event should still be possible
|
|
||||||
if let Event::Enter() = event {
|
|
||||||
server.state.replace(State::Receiving);
|
|
||||||
let _ = capture_tx.send(CaptureRequest::Release).await;
|
|
||||||
log::trace!("STATE ===> Receiving");
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,20 +1,16 @@
|
|||||||
|
use local_channel::mpsc::{Receiver, Sender};
|
||||||
use std::{io, net::SocketAddr};
|
use std::{io, net::SocketAddr};
|
||||||
|
|
||||||
use thiserror::Error;
|
use thiserror::Error;
|
||||||
use tokio::{
|
use tokio::{net::UdpSocket, task::JoinHandle};
|
||||||
net::UdpSocket,
|
|
||||||
sync::mpsc::{Receiver, Sender},
|
|
||||||
task::JoinHandle,
|
|
||||||
};
|
|
||||||
|
|
||||||
use input_event::{Event, ProtocolError};
|
|
||||||
|
|
||||||
use super::Server;
|
use super::Server;
|
||||||
|
use lan_mouse_proto::{ProtoEvent, ProtocolError};
|
||||||
|
|
||||||
pub(crate) async fn new(
|
pub(crate) async fn new(
|
||||||
server: Server,
|
server: Server,
|
||||||
udp_recv_tx: Sender<Result<(Event, SocketAddr), NetworkError>>,
|
udp_recv_tx: Sender<Result<(ProtoEvent, SocketAddr), NetworkError>>,
|
||||||
udp_send_rx: Receiver<(Event, SocketAddr)>,
|
udp_send_rx: Receiver<(ProtoEvent, SocketAddr)>,
|
||||||
) -> io::Result<JoinHandle<()>> {
|
) -> io::Result<JoinHandle<()>> {
|
||||||
// bind the udp socket
|
// bind the udp socket
|
||||||
let listen_addr = SocketAddr::new("0.0.0.0".parse().unwrap(), server.port.get());
|
let listen_addr = SocketAddr::new("0.0.0.0".parse().unwrap(), server.port.get());
|
||||||
@@ -62,15 +58,15 @@ async fn update_port(server: &Server, socket: &mut UdpSocket) {
|
|||||||
|
|
||||||
async fn udp_receiver(
|
async fn udp_receiver(
|
||||||
socket: &UdpSocket,
|
socket: &UdpSocket,
|
||||||
receiver_tx: &Sender<Result<(Event, SocketAddr), NetworkError>>,
|
receiver_tx: &Sender<Result<(ProtoEvent, SocketAddr), NetworkError>>,
|
||||||
) {
|
) {
|
||||||
loop {
|
loop {
|
||||||
let event = receive_event(socket).await;
|
let event = receive_event(socket).await;
|
||||||
receiver_tx.send(event).await.expect("channel closed");
|
receiver_tx.send(event).expect("channel closed");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn udp_sender(socket: &UdpSocket, rx: &mut Receiver<(Event, SocketAddr)>) {
|
async fn udp_sender(socket: &UdpSocket, rx: &mut Receiver<(ProtoEvent, SocketAddr)>) {
|
||||||
loop {
|
loop {
|
||||||
let (event, addr) = rx.recv().await.expect("channel closed");
|
let (event, addr) = rx.recv().await.expect("channel closed");
|
||||||
if let Err(e) = send_event(socket, event, addr) {
|
if let Err(e) = send_event(socket, event, addr) {
|
||||||
@@ -87,16 +83,17 @@ pub(crate) enum NetworkError {
|
|||||||
Io(#[from] io::Error),
|
Io(#[from] io::Error),
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn receive_event(socket: &UdpSocket) -> Result<(Event, SocketAddr), NetworkError> {
|
async fn receive_event(socket: &UdpSocket) -> Result<(ProtoEvent, SocketAddr), NetworkError> {
|
||||||
let mut buf = vec![0u8; 22];
|
let mut buf = [0u8; lan_mouse_proto::MAX_EVENT_SIZE];
|
||||||
let (_amt, src) = socket.recv_from(&mut buf).await?;
|
let (_len, src) = socket.recv_from(&mut buf).await?;
|
||||||
Ok((Event::try_from(buf)?, src))
|
let event = ProtoEvent::try_from(buf)?;
|
||||||
|
Ok((event, src))
|
||||||
}
|
}
|
||||||
|
|
||||||
fn send_event(sock: &UdpSocket, e: Event, addr: SocketAddr) -> Result<usize, NetworkError> {
|
fn send_event(sock: &UdpSocket, e: ProtoEvent, addr: SocketAddr) -> Result<usize, NetworkError> {
|
||||||
log::trace!("{:20} ------>->->-> {addr}", e.to_string());
|
log::trace!("{:20} ------>->->-> {addr}", e.to_string());
|
||||||
let data: Vec<u8> = (&e).into();
|
let (data, len): ([u8; lan_mouse_proto::MAX_EVENT_SIZE], usize) = e.into();
|
||||||
// When udp blocks, we dont want to block the event loop.
|
// When udp blocks, we dont want to block the event loop.
|
||||||
// Dropping events is better than potentially crashing the input capture.
|
// Dropping events is better than potentially crashing the input capture.
|
||||||
Ok(sock.try_send_to(&data, addr)?)
|
Ok(sock.try_send_to(&data[..len], addr)?)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,10 +1,10 @@
|
|||||||
use std::{net::SocketAddr, time::Duration};
|
use std::{net::SocketAddr, time::Duration};
|
||||||
|
|
||||||
use tokio::{sync::mpsc::Sender, task::JoinHandle};
|
use lan_mouse_proto::ProtoEvent;
|
||||||
|
use local_channel::mpsc::Sender;
|
||||||
|
use tokio::task::JoinHandle;
|
||||||
|
|
||||||
use input_event::Event;
|
use lan_mouse_ipc::ClientHandle;
|
||||||
|
|
||||||
use crate::client::ClientHandle;
|
|
||||||
|
|
||||||
use super::{capture_task::CaptureRequest, emulation_task::EmulationRequest, Server, State};
|
use super::{capture_task::CaptureRequest, emulation_task::EmulationRequest, Server, State};
|
||||||
|
|
||||||
@@ -12,7 +12,7 @@ const MAX_RESPONSE_TIME: Duration = Duration::from_millis(500);
|
|||||||
|
|
||||||
pub(crate) fn new(
|
pub(crate) fn new(
|
||||||
server: Server,
|
server: Server,
|
||||||
sender_ch: Sender<(Event, SocketAddr)>,
|
sender_ch: Sender<(ProtoEvent, SocketAddr)>,
|
||||||
emulate_notify: Sender<EmulationRequest>,
|
emulate_notify: Sender<EmulationRequest>,
|
||||||
capture_notify: Sender<CaptureRequest>,
|
capture_notify: Sender<CaptureRequest>,
|
||||||
) -> JoinHandle<()> {
|
) -> JoinHandle<()> {
|
||||||
@@ -27,7 +27,7 @@ pub(crate) fn new(
|
|||||||
|
|
||||||
async fn ping_task(
|
async fn ping_task(
|
||||||
server: &Server,
|
server: &Server,
|
||||||
sender_ch: Sender<(Event, SocketAddr)>,
|
sender_ch: Sender<(ProtoEvent, SocketAddr)>,
|
||||||
emulate_notify: Sender<EmulationRequest>,
|
emulate_notify: Sender<EmulationRequest>,
|
||||||
capture_notify: Sender<CaptureRequest>,
|
capture_notify: Sender<CaptureRequest>,
|
||||||
) {
|
) {
|
||||||
@@ -86,7 +86,7 @@ async fn ping_task(
|
|||||||
|
|
||||||
// ping clients
|
// ping clients
|
||||||
for addr in ping_addrs {
|
for addr in ping_addrs {
|
||||||
if sender_ch.send((Event::Ping(), addr)).await.is_err() {
|
if sender_ch.send((ProtoEvent::Ping, addr)).is_err() {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -123,14 +123,14 @@ async fn ping_task(
|
|||||||
if receiving {
|
if receiving {
|
||||||
for h in unresponsive_clients {
|
for h in unresponsive_clients {
|
||||||
log::warn!("device not responding, releasing keys!");
|
log::warn!("device not responding, releasing keys!");
|
||||||
let _ = emulate_notify.send(EmulationRequest::ReleaseKeys(h)).await;
|
let _ = emulate_notify.send(EmulationRequest::ReleaseKeys(h));
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
// release pointer if the active client has not responded
|
// release pointer if the active client has not responded
|
||||||
if !unresponsive_clients.is_empty() {
|
if !unresponsive_clients.is_empty() {
|
||||||
log::warn!("client not responding, releasing pointer!");
|
log::warn!("client not responding, releasing pointer!");
|
||||||
server.state.replace(State::Receiving);
|
server.state.replace(State::Receiving);
|
||||||
let _ = capture_notify.send(CaptureRequest::Release).await;
|
let _ = capture_notify.send(CaptureRequest::Release);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user