109 lines
3.4 KiB
Rust
109 lines
3.4 KiB
Rust
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
use anyhow::{anyhow, Context, Result};
|
|
use smithay::{
|
|
backend::{
|
|
allocator::format::FormatSet,
|
|
drm::{DrmNode, NodeType},
|
|
},
|
|
reexports::{
|
|
calloop::RegistrationToken,
|
|
wayland_server::{backend::GlobalId, Client, DisplayHandle},
|
|
},
|
|
wayland::{
|
|
dmabuf::{DmabufFeedbackBuilder, DmabufGlobal},
|
|
socket::ListeningSocketSource,
|
|
},
|
|
};
|
|
use std::sync::Arc;
|
|
use tracing::{info, warn};
|
|
|
|
use crate::state::{advertised_node_for_client, ClientState, State};
|
|
|
|
#[derive(Debug)]
|
|
pub struct Socket {
|
|
pub token: RegistrationToken,
|
|
pub drm_global: GlobalId,
|
|
pub dmabuf_global: DmabufGlobal,
|
|
}
|
|
|
|
impl State {
|
|
pub(super) fn create_socket(
|
|
&mut self,
|
|
dh: &DisplayHandle,
|
|
render_node: DrmNode,
|
|
formats: FormatSet,
|
|
) -> Result<Socket> {
|
|
let socket_name = format!(
|
|
"{}-{}",
|
|
&self.common.socket.to_string_lossy(),
|
|
render_node
|
|
.dev_path()
|
|
.unwrap()
|
|
.file_name()
|
|
.unwrap()
|
|
.to_string_lossy()
|
|
);
|
|
|
|
// initialize globals
|
|
let filter = move |client: &Client| advertised_node_for_client(client) == Some(render_node);
|
|
|
|
let feedback = DmabufFeedbackBuilder::new(render_node.dev_id(), formats.clone())
|
|
.build()
|
|
.with_context(|| "Failed to create drm format shared memory table")?;
|
|
|
|
let dmabuf_global = self
|
|
.common
|
|
.dmabuf_state
|
|
.create_global_with_filter_and_default_feedback::<State, _>(dh, &feedback, filter);
|
|
|
|
let drm_global_id = self
|
|
.common
|
|
.wl_drm_state
|
|
.create_global_with_filter::<State, _>(
|
|
dh,
|
|
render_node
|
|
.dev_path_with_type(NodeType::Render)
|
|
.or_else(|| render_node.dev_path())
|
|
.ok_or(anyhow!(
|
|
"Could not determine path for gpu node: {}",
|
|
render_node
|
|
))?,
|
|
formats,
|
|
&dmabuf_global,
|
|
filter,
|
|
);
|
|
|
|
// add a special socket for the gpu
|
|
let listener = ListeningSocketSource::with_name(&socket_name)
|
|
.with_context(|| format!("Failed to bind socket to {}", socket_name))?;
|
|
let socket_name_clone = socket_name.clone();
|
|
let token = self
|
|
.common
|
|
.event_loop_handle
|
|
.insert_source(listener, move |client_stream, _, state: &mut State| {
|
|
if let Err(err) = state.common.display_handle.insert_client(
|
|
client_stream,
|
|
Arc::new(ClientState {
|
|
advertised_drm_node: Some(render_node),
|
|
..state.new_client_state()
|
|
}),
|
|
) {
|
|
warn!(
|
|
socket_name = socket_name_clone,
|
|
?err,
|
|
"Error adding wayland client."
|
|
);
|
|
}
|
|
})
|
|
.context("Failed to add gpu-wayland socket to the event loop")?;
|
|
|
|
info!(socket_name, ?render_node, "Added gpu-specific socket.");
|
|
|
|
Ok(Socket {
|
|
token,
|
|
drm_global: drm_global_id,
|
|
dmabuf_global,
|
|
})
|
|
}
|
|
}
|