From 9c9f744412edfeb4086c70fdc74000837431e7f2 Mon Sep 17 00:00:00 2001
From: Aleksander <aleksander@oo8.dev>
Date: Thu, 17 Apr 2025 21:14:00 +0200
Subject: [PATCH] WayVR: Automatically fallback to software blitting in case if
 dmabuf initialization has failed, fix `cargo fmt` errors

---
 src/backend/wayvr/client.rs   |  3 ++-
 src/backend/wayvr/comp.rs     |  4 +---
 src/backend/wayvr/display.rs  | 10 +++++++---
 src/backend/wayvr/egl_data.rs | 27 ++++++++++++++++++++-------
 src/graphics/mod.rs           |  7 ++++---
 5 files changed, 34 insertions(+), 17 deletions(-)

diff --git a/src/backend/wayvr/client.rs b/src/backend/wayvr/client.rs
index 1bd88aa..1ebeb83 100644
--- a/src/backend/wayvr/client.rs
+++ b/src/backend/wayvr/client.rs
@@ -211,7 +211,8 @@ impl WayVRCompositor {
 const STARTING_WAYLAND_ADDR_IDX: u32 = 20;
 
 fn export_display_number(display_num: u32) -> anyhow::Result<()> {
-    let mut path = std::env::var("XDG_RUNTIME_DIR").map_or_else(|_| PathBuf::from("/tmp"), PathBuf::from);
+    let mut path =
+        std::env::var("XDG_RUNTIME_DIR").map_or_else(|_| PathBuf::from("/tmp"), PathBuf::from);
     path.push("wayvr.disp");
     std::fs::write(path, format!("{display_num}\n"))?;
     Ok(())
diff --git a/src/backend/wayvr/comp.rs b/src/backend/wayvr/comp.rs
index de83a65..c6100f3 100644
--- a/src/backend/wayvr/comp.rs
+++ b/src/backend/wayvr/comp.rs
@@ -127,9 +127,7 @@ impl ClientData for ClientState {
 
     fn disconnected(&self, client_id: ClientId, reason: DisconnectReason) {
         *self.disconnected.lock().unwrap() = true;
-        log::debug!(
-            "Client ID {client_id:?} disconnected. Reason: {reason:?}"
-        );
+        log::debug!("Client ID {client_id:?} disconnected. Reason: {reason:?}");
     }
 }
 
diff --git a/src/backend/wayvr/display.rs b/src/backend/wayvr/display.rs
index dabdac0..7cbbb6d 100644
--- a/src/backend/wayvr/display.rs
+++ b/src/backend/wayvr/display.rs
@@ -132,9 +132,13 @@ impl Display {
         let egl_image = params.egl_data.create_egl_image(tex_id)?;
 
         let render_data = match params.config.blit_method {
-            BlitMethod::Dmabuf => {
-                egl_data::RenderData::Dmabuf(params.egl_data.create_dmabuf_data(&egl_image)?)
-            }
+            BlitMethod::Dmabuf => match params.egl_data.create_dmabuf_data(&egl_image) {
+                Ok(dmabuf_data) => egl_data::RenderData::Dmabuf(dmabuf_data),
+                Err(e) => {
+                    log::error!("create_dmabuf_data failed: {e:?}. Using software blitting (This will be slow!)");
+                    egl_data::RenderData::Software(None)
+                }
+            },
             BlitMethod::Software => egl_data::RenderData::Software(None),
         };
 
diff --git a/src/backend/wayvr/egl_data.rs b/src/backend/wayvr/egl_data.rs
index aa29ce0..771f16b 100644
--- a/src/backend/wayvr/egl_data.rs
+++ b/src/backend/wayvr/egl_data.rs
@@ -235,21 +235,34 @@ impl EGLData {
             let mut strides: [i32; 3] = [0; 3];
             let mut offsets: [i32; 3] = [0; 3];
 
-            if egl_export_dmabuf_image_mesa(
+            let ret = egl_export_dmabuf_image_mesa(
                 self.display.as_ptr(),
                 egl_image.as_ptr(),
                 fds.as_mut_ptr(),
                 strides.as_mut_ptr(),
                 offsets.as_mut_ptr(),
-            ) != khronos_egl::TRUE
-            {
-                anyhow::bail!("eglExportDMABUFImageMESA failed");
+            );
+
+            if ret != khronos_egl::TRUE {
+                anyhow::bail!("eglExportDMABUFImageMESA failed with return code {ret}");
+            }
+
+            if fds[0] <= 0 {
+                anyhow::bail!("fd is <=0 (got {})", fds[0]);
             }
 
             // many planes in RGB data?
-            debug_assert!(fds[1] == 0);
-            debug_assert!(strides[1] == 0);
-            debug_assert!(offsets[1] == 0);
+            if fds[1] != 0 || strides[1] != 0 || offsets[1] != 0 {
+                anyhow::bail!("multi-planar data received, packed RGB expected");
+            }
+
+            if strides[0] < 0 {
+                anyhow::bail!("strides is < 0");
+            }
+
+            if offsets[0] < 0 {
+                anyhow::bail!("offsets is < 0");
+            }
 
             let mod_info = self.query_dmabuf_mod_info()?;
 
diff --git a/src/graphics/mod.rs b/src/graphics/mod.rs
index 84f2ea9..f6c85ea 100644
--- a/src/graphics/mod.rs
+++ b/src/graphics/mod.rs
@@ -37,8 +37,9 @@ use vulkano::{
         allocator::StandardDescriptorSetAllocator, DescriptorSet, WriteDescriptorSet,
     },
     device::{
-        physical::{PhysicalDevice, PhysicalDeviceType}, Device, DeviceCreateInfo, DeviceExtensions, DeviceFeatures,
-        Queue, QueueCreateInfo, QueueFlags,
+        physical::{PhysicalDevice, PhysicalDeviceType},
+        Device, DeviceCreateInfo, DeviceExtensions, DeviceFeatures, Queue, QueueCreateInfo,
+        QueueFlags,
     },
     format::Format,
     image::{
@@ -600,7 +601,7 @@ impl WlxGraphics {
                     None
                 }
             })
-            .filter_map(|(p, my_extensions)| 
+            .filter_map(|(p, my_extensions)|
                 try_all_queue_families(p.as_ref()).map(|families| (p, my_extensions, families))
             )
             .min_by_key(|(p, _, _)| prio_from_device_type(p)