session_lock: Verify BufferAssignment is NewBuffer in attached bu…

…ffer check
Smithay · May 17, 2024 · 900b938 · 900b938
1 parent 271fbc0
commit 900b938
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/wayland/session_lock/lock.rs b/src/wayland/session_lock/lock.rs
@@ -67,8 +67,14 @@ where
                 // Ensure surface has no existing buffers attached.
                 let has_buffer = compositor::with_states(&surface, |states| {
                     let cached = &states.cached_state;
-                    let pending = cached.pending::<SurfaceAttributes>().buffer.is_some();
-                    let current = cached.current::<SurfaceAttributes>().buffer.is_some();
+                    let pending = matches!(
+                        cached.pending::<SurfaceAttributes>().buffer,
+                        Some(BufferAssignment::NewBuffer(_))
+                    );
+                    let current = matches!(
+                        cached.current::<SurfaceAttributes>().buffer,
+                        Some(BufferAssignment::NewBuffer(_))
+                    );
                     pending || current
                 });
                 if has_buffer {