66 lines
2.5 KiB
Rust
66 lines
2.5 KiB
Rust
use crate::state::{ClientState, State};
|
|
use smithay::{
|
|
backend::drm::DrmNode,
|
|
delegate_security_context,
|
|
wayland::security_context::{
|
|
SecurityContext, SecurityContextHandler, SecurityContextListenerSource,
|
|
},
|
|
xwayland::XWaylandClientData,
|
|
};
|
|
use std::sync::Arc;
|
|
use tracing::warn;
|
|
|
|
impl SecurityContextHandler for State {
|
|
fn context_created(
|
|
&mut self,
|
|
source: SecurityContextListenerSource,
|
|
security_context: SecurityContext,
|
|
) {
|
|
self.common
|
|
.event_loop_handle
|
|
.insert_source(source, move |client_stream, _, state| {
|
|
let client_data = state
|
|
.common
|
|
.display_handle
|
|
.backend_handle()
|
|
.get_client_data(security_context.creator_client_id.clone())
|
|
.ok();
|
|
|
|
let privileged = client_data
|
|
.as_ref()
|
|
.and_then(|data| data.downcast_ref::<ClientState>())
|
|
.map(|data| data.privileged)
|
|
.unwrap_or(false);
|
|
|
|
let new_state = state.new_client_state();
|
|
|
|
let drm_node = client_data
|
|
.as_ref()
|
|
.and_then(|data| data.downcast_ref::<ClientState>())
|
|
.and_then(|data| data.advertised_drm_node.clone())
|
|
.or_else(|| {
|
|
client_data
|
|
.as_ref()
|
|
.and_then(|data| data.downcast_ref::<XWaylandClientData>())
|
|
.and_then(|data| data.user_data().get::<DrmNode>().cloned())
|
|
})
|
|
.or_else(|| new_state.advertised_drm_node.clone());
|
|
|
|
if let Err(err) = state.common.display_handle.insert_client(
|
|
client_stream,
|
|
Arc::new(ClientState {
|
|
security_context: Some(security_context.clone()),
|
|
privileged: privileged
|
|
&& security_context.sandbox_engine.as_deref()
|
|
== Some("com.system76.CosmicPanel"),
|
|
advertised_drm_node: drm_node,
|
|
..new_state
|
|
}),
|
|
) {
|
|
warn!(?err, "Error adding wayland client");
|
|
};
|
|
})
|
|
.expect("Failed to init the wayland socket source.");
|
|
}
|
|
}
|
|
delegate_security_context!(State);
|