mirror of
https://github.com/rustdesk/rustdesk.git
synced 2026-09-19 02:41:10 +03:00
revert(cursor): return to f5b98b32 before splitting plugin changes
This commit is contained in:
fufesou
@@ -1,248 +0,0 @@
|
||||
use super::wrap_hresult;
|
||||
use std::{
|
||||
collections::{hash_map::DefaultHasher, HashMap},
|
||||
hash::{Hash, Hasher},
|
||||
io,
|
||||
sync::{Arc, Mutex, Weak},
|
||||
time::Instant,
|
||||
};
|
||||
use winapi::shared::dxgi1_2::{
|
||||
IDXGIOutputDuplication, DXGI_OUTDUPL_FRAME_INFO, DXGI_OUTDUPL_POINTER_SHAPE_INFO,
|
||||
DXGI_OUTDUPL_POINTER_SHAPE_TYPE_COLOR, DXGI_OUTDUPL_POINTER_SHAPE_TYPE_MASKED_COLOR,
|
||||
DXGI_OUTDUPL_POINTER_SHAPE_TYPE_MONOCHROME,
|
||||
};
|
||||
|
||||
pub const CURSOR_ID_FLAG: u64 = 1 << 63;
|
||||
const CHANNELS: u32 = 4;
|
||||
const BITS_PER_BYTE: u32 = 8;
|
||||
|
||||
pub struct Shape {
|
||||
pub id: u64,
|
||||
pub kind: u32,
|
||||
pub width: u32,
|
||||
pub height: u32,
|
||||
pub pitch: u32,
|
||||
pub hotspot: (i32, i32),
|
||||
pub pixels: Vec<u8>,
|
||||
}
|
||||
|
||||
impl Shape {
|
||||
fn new(info: DXGI_OUTDUPL_POINTER_SHAPE_INFO, pixels: Vec<u8>) -> io::Result<Self> {
|
||||
let (height, minimum_pitch) = match info.Type {
|
||||
DXGI_OUTDUPL_POINTER_SHAPE_TYPE_MONOCHROME if info.Height % 2 == 0 => {
|
||||
(info.Height / 2, info.Width.div_ceil(BITS_PER_BYTE))
|
||||
}
|
||||
DXGI_OUTDUPL_POINTER_SHAPE_TYPE_COLOR
|
||||
| DXGI_OUTDUPL_POINTER_SHAPE_TYPE_MASKED_COLOR => (
|
||||
info.Height,
|
||||
info.Width.checked_mul(CHANNELS).ok_or_else(invalid_shape)?,
|
||||
),
|
||||
_ => return Err(invalid_shape()),
|
||||
};
|
||||
let length = (info.Pitch as usize).checked_mul(info.Height as usize);
|
||||
if info.Width == 0
|
||||
|| height == 0
|
||||
|| info.Width > i32::MAX as u32
|
||||
|| info.Height > i32::MAX as u32
|
||||
|| info.Pitch > i32::MAX as u32
|
||||
|| info.Pitch < minimum_pitch
|
||||
|| length != Some(pixels.len())
|
||||
|| info.HotSpot.x < 0
|
||||
|| info.HotSpot.x as u32 >= info.Width
|
||||
|| info.HotSpot.y < 0
|
||||
|| info.HotSpot.y as u32 >= height
|
||||
{
|
||||
return Err(invalid_shape());
|
||||
}
|
||||
let hotspot = (info.HotSpot.x, info.HotSpot.y);
|
||||
let mut hash = DefaultHasher::new();
|
||||
(info.Type, info.Width, height, info.Pitch, hotspot, &pixels).hash(&mut hash);
|
||||
Ok(Self {
|
||||
id: hash.finish() | CURSOR_ID_FLAG,
|
||||
kind: info.Type,
|
||||
width: info.Width,
|
||||
height,
|
||||
pitch: info.Pitch,
|
||||
hotspot,
|
||||
pixels,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
fn invalid_shape() -> io::Error {
|
||||
io::Error::new(io::ErrorKind::InvalidData, "Invalid DXGI cursor shape")
|
||||
}
|
||||
|
||||
#[derive(Clone)]
|
||||
pub enum Snapshot {
|
||||
Unavailable,
|
||||
Pending,
|
||||
Ready(Arc<Shape>),
|
||||
Failed(String),
|
||||
}
|
||||
|
||||
struct State {
|
||||
updated: Instant,
|
||||
snapshot: Snapshot,
|
||||
}
|
||||
|
||||
type SharedState = Arc<Mutex<State>>;
|
||||
|
||||
lazy_static::lazy_static! {
|
||||
static ref CAPTURES: Mutex<HashMap<usize, Vec<Weak<Mutex<State>>>>> =
|
||||
Mutex::new(HashMap::new());
|
||||
}
|
||||
|
||||
pub fn snapshot(monitor: usize) -> Snapshot {
|
||||
let captures = CAPTURES.lock().unwrap();
|
||||
captures
|
||||
.get(&monitor)
|
||||
.into_iter()
|
||||
.flatten()
|
||||
.filter_map(Weak::upgrade)
|
||||
.map(|state| {
|
||||
let state = state.lock().unwrap();
|
||||
(state.updated, state.snapshot.clone())
|
||||
})
|
||||
.max_by_key(|(updated, _)| *updated)
|
||||
.map(|(_, snapshot)| snapshot)
|
||||
.unwrap_or(Snapshot::Unavailable)
|
||||
}
|
||||
|
||||
pub fn shape(id: u64) -> Option<Arc<Shape>> {
|
||||
CAPTURES
|
||||
.lock()
|
||||
.unwrap()
|
||||
.values()
|
||||
.flatten()
|
||||
.filter_map(Weak::upgrade)
|
||||
.find_map(|state| match &state.lock().unwrap().snapshot {
|
||||
Snapshot::Ready(shape) if shape.id == id => Some(shape.clone()),
|
||||
_ => None,
|
||||
})
|
||||
}
|
||||
|
||||
pub(super) struct Capture {
|
||||
monitor: usize,
|
||||
state: SharedState,
|
||||
}
|
||||
|
||||
impl Capture {
|
||||
pub fn new(monitor: usize) -> Self {
|
||||
let state = Arc::new(Mutex::new(State {
|
||||
updated: Instant::now(),
|
||||
snapshot: Snapshot::Pending,
|
||||
}));
|
||||
let capture = Self { monitor, state };
|
||||
capture.activate();
|
||||
capture
|
||||
}
|
||||
|
||||
pub fn activate(&self) {
|
||||
let mut captures = CAPTURES.lock().unwrap();
|
||||
let states = captures.entry(self.monitor).or_default();
|
||||
let own = Arc::downgrade(&self.state);
|
||||
if !states.iter().any(|state| state.ptr_eq(&own)) {
|
||||
states.push(own);
|
||||
}
|
||||
}
|
||||
|
||||
pub fn deactivate(&self) {
|
||||
let mut captures = CAPTURES.lock().unwrap();
|
||||
if let Some(states) = captures.get_mut(&self.monitor) {
|
||||
let own = Arc::downgrade(&self.state);
|
||||
states.retain(|state| !state.ptr_eq(&own));
|
||||
if states.is_empty() {
|
||||
captures.remove(&self.monitor);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub unsafe fn update(
|
||||
&self,
|
||||
duplication: *mut IDXGIOutputDuplication,
|
||||
frame: &DXGI_OUTDUPL_FRAME_INFO,
|
||||
) {
|
||||
if frame.PointerShapeBufferSize == 0 {
|
||||
return;
|
||||
}
|
||||
let snapshot = match read(duplication, frame.PointerShapeBufferSize) {
|
||||
Ok(shape) => Snapshot::Ready(Arc::new(shape)),
|
||||
Err(error) => {
|
||||
hbb_common::log::error!("DXGI cursor capture failed: {error}");
|
||||
Snapshot::Failed(error.to_string())
|
||||
}
|
||||
};
|
||||
*self.state.lock().unwrap() = State {
|
||||
updated: Instant::now(),
|
||||
snapshot,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
unsafe fn read(duplication: *mut IDXGIOutputDuplication, size: u32) -> io::Result<Shape> {
|
||||
let mut pixels = vec![0; size as usize];
|
||||
let mut required = 0;
|
||||
let mut info = std::mem::zeroed();
|
||||
wrap_hresult((*duplication).GetFramePointerShape(
|
||||
size,
|
||||
pixels.as_mut_ptr().cast(),
|
||||
&mut required,
|
||||
&mut info,
|
||||
))?;
|
||||
if required > size {
|
||||
return Err(invalid_shape());
|
||||
}
|
||||
pixels.truncate(required as usize);
|
||||
Shape::new(info, pixels)
|
||||
}
|
||||
|
||||
impl Drop for Capture {
|
||||
fn drop(&mut self) {
|
||||
self.deactivate();
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use winapi::shared::windef::POINT;
|
||||
|
||||
#[test]
|
||||
fn cursor_state_follows_capture_lifetime_and_gdi_switches() {
|
||||
const MONITOR: usize = usize::MAX;
|
||||
let first = Capture::new(MONITOR);
|
||||
let second = Capture::new(MONITOR);
|
||||
first.deactivate();
|
||||
assert!(matches!(snapshot(MONITOR), Snapshot::Pending));
|
||||
drop(second);
|
||||
assert!(matches!(snapshot(MONITOR), Snapshot::Unavailable));
|
||||
first.activate();
|
||||
first.activate();
|
||||
assert!(matches!(snapshot(MONITOR), Snapshot::Pending));
|
||||
drop(first);
|
||||
assert!(matches!(snapshot(MONITOR), Snapshot::Unavailable));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn cursor_keeps_physical_hotspot_and_both_monochrome_planes() {
|
||||
let info = DXGI_OUTDUPL_POINTER_SHAPE_INFO {
|
||||
Type: DXGI_OUTDUPL_POINTER_SHAPE_TYPE_MONOCHROME,
|
||||
Width: 64,
|
||||
Height: 128,
|
||||
Pitch: 8,
|
||||
HotSpot: POINT { x: 31, y: 29 },
|
||||
};
|
||||
let pixels = vec![0xa5; 1024];
|
||||
let shape = Shape::new(info, pixels.clone()).unwrap();
|
||||
assert_eq!(
|
||||
(shape.width, shape.height, shape.hotspot),
|
||||
(64, 64, (31, 29))
|
||||
);
|
||||
assert_eq!(shape.pixels, pixels);
|
||||
assert!(Shape::new(info, vec![0; 512]).is_err());
|
||||
let mut changed = info;
|
||||
changed.HotSpot.y += 1;
|
||||
assert_ne!(shape.id, Shape::new(changed, pixels).unwrap().id);
|
||||
}
|
||||
}
|
||||
@@ -2,8 +2,6 @@ use std::{io, mem, ptr, slice};
|
||||
pub mod gdi;
|
||||
pub use gdi::CapturerGDI;
|
||||
pub mod mag;
|
||||
#[cfg(feature = "cursor")]
|
||||
pub mod cursor;
|
||||
|
||||
use winapi::{
|
||||
shared::{
|
||||
@@ -44,8 +42,6 @@ impl<T> Drop for ComPtr<T> {
|
||||
}
|
||||
|
||||
pub struct Capturer {
|
||||
#[cfg(feature = "cursor")]
|
||||
cursor: Option<cursor::Capture>,
|
||||
device: ComPtr<ID3D11Device>,
|
||||
display: Display,
|
||||
context: ComPtr<ID3D11DeviceContext>,
|
||||
@@ -162,9 +158,6 @@ impl Capturer {
|
||||
let rotate = Self::create_rotations(device.0, context.0, &display);
|
||||
|
||||
Ok(Capturer {
|
||||
#[cfg(feature = "cursor")]
|
||||
cursor: (!duplication.is_null())
|
||||
.then(|| cursor::Capture::new(display.hmonitor() as usize)),
|
||||
device,
|
||||
context,
|
||||
duplication: ComPtr(duplication),
|
||||
@@ -323,25 +316,12 @@ impl Capturer {
|
||||
|
||||
pub fn set_gdi(&mut self) -> bool {
|
||||
self.gdi_capturer = self.display.create_gdi();
|
||||
#[cfg(feature = "cursor")]
|
||||
if self.is_gdi() {
|
||||
if let Some(cursor) = &self.cursor {
|
||||
cursor.deactivate();
|
||||
}
|
||||
}
|
||||
self.is_gdi()
|
||||
}
|
||||
|
||||
pub fn cancel_gdi(&mut self) {
|
||||
self.gdi_buffer = Vec::new();
|
||||
self.gdi_capturer.take();
|
||||
#[cfg(feature = "cursor")]
|
||||
if !self.duplication.is_null() {
|
||||
let monitor = self.display.hmonitor() as usize;
|
||||
self.cursor
|
||||
.get_or_insert_with(|| cursor::Capture::new(monitor))
|
||||
.activate();
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(feature = "vram")]
|
||||
@@ -356,10 +336,6 @@ impl Capturer {
|
||||
|
||||
wrap_hresult((*self.duplication.0).AcquireNextFrame(timeout, &mut info, &mut frame))?;
|
||||
let frame = ComPtr(frame);
|
||||
#[cfg(feature = "cursor")]
|
||||
if let Some(cursor) = &self.cursor {
|
||||
cursor.update(self.duplication.0, &info);
|
||||
}
|
||||
|
||||
if *info.LastPresentTime.QuadPart() == 0 {
|
||||
return Err(std::io::ErrorKind::WouldBlock.into());
|
||||
@@ -503,10 +479,6 @@ impl Capturer {
|
||||
|
||||
wrap_hresult((*self.duplication.0).AcquireNextFrame(timeout, &mut info, &mut frame))?;
|
||||
let frame = ComPtr(frame);
|
||||
#[cfg(feature = "cursor")]
|
||||
if let Some(cursor) = &self.cursor {
|
||||
cursor.update(self.duplication.0, &info);
|
||||
}
|
||||
|
||||
if info.AccumulatedFrames == 0 || *info.LastPresentTime.QuadPart() == 0 {
|
||||
return Err(std::io::ErrorKind::WouldBlock.into());
|
||||
|
||||
Reference in New Issue
Block a user