From 457d6d1c358ee25e142b8788e8654d7794a62d2e Mon Sep 17 00:00:00 2001
From: Drakulix <github@drakulix.de>
Date: Tue, 19 Sep 2017 17:26:11 +0200
Subject: [PATCH] Free framebuffers correctly

---
 src/backend/drm/backend.rs | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/src/backend/drm/backend.rs b/src/backend/drm/backend.rs
index 6d0dedb..7703d84 100644
--- a/src/backend/drm/backend.rs
+++ b/src/backend/drm/backend.rs
@@ -424,6 +424,36 @@ impl DrmBackend {
     }
 }
 
+impl Drop for DrmBackend {
+    fn drop(&mut self) {
+        // Drop framebuffers attached to the userdata of the gbm surface buffers.
+        // (They don't implement drop, as they need the device)
+        self.0.borrow_mut().graphics.rent_all_mut(|graphics| {
+            if let Some(fb) = graphics.gbm.surface.rent(|egl| {
+                if let Some(mut next) = egl.buffers.next_buffer.take() {
+                    return next.take_userdata();
+                } else {
+                    if let Ok(mut next) = graphics.gbm.surface.head().lock_front_buffer() {
+                        return next.take_userdata();
+                    }
+                }
+                None
+            }) {
+                // ignore failure at this point
+                let _ = framebuffer::destroy(&*graphics.context.devices.drm, fb.handle());
+            }
+
+            if let Some(fb) = graphics.gbm.surface.rent_mut(|egl| {
+                let first = egl.buffers.front_buffer.get_mut();
+                first.take_userdata()
+            }) {
+                // ignore failure at this point
+                let _ = framebuffer::destroy(&*graphics.context.devices.drm, fb.handle());
+            }
+        })
+    }
+}
+
 impl GraphicsBackend for DrmBackend {
     type CursorFormat = ImageBuffer<Rgba<u8>, Vec<u8>>;
     type Error = Error;