TheLanc3/lan-mouse
1
0
Fork 0
mirror of https://github.com/feschber/lan-mouse.git synced 2026-03-29 08:00:55 +03:00
Code Issues Packages Projects Releases Wiki Activity

release capture only if no active capture at pos

Browse Source
This commit is contained in:
Ferdinand Schober
2024-10-08 14:26:29 +02:00
parent 06d4e8d836
commit 4a64a97273
2 changed files with 28 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/capture.rs
View File

@@ -228,6 +228,19 @@ async fn handle_capture_event(
enter_tx.send(handle).expect("channel closed");
}
// incoming connection
if handle >= Service::ENTER_HANDLE_BEGIN {
// if there is no active outgoing connection at the current capture,
// we release the capture
if let Some(pos) = server.get_incoming_pos(handle) {
if server.client_manager.client_at(pos).is_none() {
capture.release().await?;
}
}
// we dont care about events from incoming handles except for releasing the capture
return Ok(());
}
// activated a new client
if event == CaptureEvent::Begin && Some(handle) != server.get_active() {
*state = State::WaitingForAck;

22
src/service.rs
View File

@@ -70,7 +70,7 @@ pub struct Service {
capture_status: Rc<Cell<Status>>,
pub(crate) emulation_status: Rc<Cell<Status>>,
pub(crate) should_release: Rc<RefCell<Option<ReleaseToken>>>,
incoming_conns: Rc<RefCell<HashMap<ClientHandle, (String, SocketAddr)>>>,
incoming_conns: Rc<RefCell<HashMap<ClientHandle, (String, SocketAddr, Position)>>>,
cert: Certificate,
next_trigger_handle: u64,
}
@@ -203,7 +203,7 @@ impl Service {
}
},
handle = capture.entered() => {
if let Some((_fp, addr)) = self.incoming_conns.borrow().get(&handle) {
if let Some((_fp, addr, _pos)) = self.incoming_conns.borrow().get(&handle) {
emulation.notify_release(*addr);
}
},
@@ -225,6 +225,8 @@ impl Service {
Ok(())
}
pub(crate) const ENTER_HANDLE_BEGIN: u64 = u64::MAX / 2 + 1;
fn add_incoming(
&mut self,
addr: SocketAddr,
@@ -232,13 +234,12 @@ impl Service {
fingerprint: String,
capture: &Capture,
) {
const ENTER_HANDLE_BEGIN: u64 = u64::MAX / 2 + 1;
let handle = ENTER_HANDLE_BEGIN + self.next_trigger_handle;
let handle = Self::ENTER_HANDLE_BEGIN + self.next_trigger_handle;
self.next_trigger_handle += 1;
capture.create(handle, pos);
self.incoming_conns
.borrow_mut()
.insert(handle, (fingerprint, addr));
.insert(handle, (fingerprint, addr, pos));
}
fn remove_incoming(&mut self, addr: SocketAddr, capture: &Capture) -> Option<String> {
@@ -246,13 +247,20 @@ impl Service {
.incoming_conns
.borrow()
.iter()
.find(|(_, (_, a))| *a == addr)
.find(|(_, (_, a, _))| *a == addr)
.map(|(k, _)| *k)?;
capture.destroy(handle);
self.incoming_conns
.borrow_mut()
.remove(&handle)
.map(|(f, _)| f)
.map(|(f, _, _)| f)
}
pub(crate) fn get_incoming_pos(&self, handle: ClientHandle) -> Option<Position> {
self.incoming_conns
.borrow()
.get(&handle)
.map(|&(_, _, p)| p)
}
fn notify_frontend(&self, event: FrontendEvent) {