autocommit 2022-01-02 15:13:33 CET

src/ifs/wl_shm_pool/mod.rs

@@ -0,0 +1,94 @@
+mod types;
+
+use crate::clientmem::ClientMem;
+use crate::objects::{Interface, Object, ObjectId};
+use crate::utils::buffd::WlParser;
+use crate::wl_client::WlClientData;
+use std::cell::RefCell;
+use std::rc::Rc;
+pub use types::*;
+use uapi::OwnedFd;
+
+const CREATE_BUFFER: u32 = 0;
+const DESTROY: u32 = 1;
+const RESIZE: u32 = 2;
+
+pub struct WlShmPool {
+    id: ObjectId,
+    client: Rc<WlClientData>,
+    fd: OwnedFd,
+    mem: RefCell<Rc<ClientMem>>,
+}
+
+impl WlShmPool {
+    pub fn new(
+        id: ObjectId,
+        client: &Rc<WlClientData>,
+        fd: OwnedFd,
+        len: usize,
+    ) -> Result<Self, WlShmPoolError> {
+        Ok(Self {
+            id,
+            client: client.clone(),
+            mem: RefCell::new(Rc::new(ClientMem::new(fd.raw(), len)?)),
+            fd,
+        })
+    }
+
+    async fn create_buffer(&self, parser: WlParser<'_, '_>) -> Result<(), CreateBufferError> {
+        let create: CreateBuffer = self.client.parse(self, parser)?;
+        Ok(())
+    }
+
+    async fn destroy(&self, parser: WlParser<'_, '_>) -> Result<(), DestroyError> {
+        let _destroy: Destroy = self.client.parse(self, parser)?;
+        self.client
+            .objects
+            .remove_obj(&self.client, self.id)
+            .await?;
+        Ok(())
+    }
+
+    async fn resize(&self, parser: WlParser<'_, '_>) -> Result<(), ResizeError> {
+        let resize: Resize = self.client.parse(self, parser)?;
+        let mut mem = self.mem.borrow_mut();
+        if resize.size < 0 {
+            return Err(ResizeError::NegativeSize);
+        }
+        if (resize.size as usize) < mem.len() {
+            return Err(ResizeError::CannotShrink);
+        }
+        *mem = Rc::new(ClientMem::new(self.fd.raw(), resize.size as usize)?);
+        Ok(())
+    }
+
+    async fn handle_request_(
+        &self,
+        request: u32,
+        parser: WlParser<'_, '_>,
+    ) -> Result<(), WlShmPoolError> {
+        match request {
+            CREATE_BUFFER => self.create_buffer(parser).await?,
+            DESTROY => self.destroy(parser).await?,
+            RESIZE => self.resize(parser).await?,
+            _ => unreachable!(),
+        }
+        Ok(())
+    }
+}
+
+handle_request!(WlShmPool);
+
+impl Object for WlShmPool {
+    fn id(&self) -> ObjectId {
+        self.id
+    }
+
+    fn interface(&self) -> Interface {
+        Interface::WlShmPool
+    }
+
+    fn num_requests(&self) -> u32 {
+        RESIZE + 1
+    }
+}