From 4ca125d05a148204eccf2eaf9d359801f3579261 Mon Sep 17 00:00:00 2001
From: Vaxry <vaxry@vaxry.net>
Date: Tue, 2 Jul 2024 23:05:50 +0200
Subject: [PATCH] drm: set fbs on vt restore

thanks @ikalco
---
 src/backend/drm/DRM.cpp | 28 ++++++++++++++++++++++++++--
 1 file changed, 26 insertions(+), 2 deletions(-)

diff --git a/src/backend/drm/DRM.cpp b/src/backend/drm/DRM.cpp
index 2296316..c7c3528 100644
--- a/src/backend/drm/DRM.cpp
+++ b/src/backend/drm/DRM.cpp
@@ -271,11 +271,35 @@ void Aquamarine::CDRMBackend::restoreAfterVT() {
             .test     = false,
         };
 
-        if (c->output->state->state().mode && c->output->state->state().mode->modeInfo.has_value())
-            data.modeInfo = *c->output->state->state().mode->modeInfo;
+        auto& STATE = c->output->state->state();
+
+        if (STATE.mode && STATE.mode->modeInfo.has_value())
+            data.modeInfo = *STATE.mode->modeInfo;
         else
             data.calculateMode(c);
 
+        if (STATE.buffer) {
+            SP<CDRMFB> drmFB;
+            auto       buf   = STATE.buffer;
+            bool       isNew = false;
+
+            drmFB = CDRMFB::create(buf, self, &isNew);
+
+            if (!drmFB)
+                backend->log(AQ_LOG_ERROR, "drm: Buffer failed to import to KMS");
+
+            if (!isNew && primary && drmFB)
+                drmFB->reimport();
+
+            data.mainFB = drmFB;
+        }
+
+        if (c->crtc->pendingCursor)
+            data.cursorFB = c->crtc->pendingCursor;
+
+        if (data.cursorFB && data.cursorFB->buffer->dmabuf().modifier == DRM_FORMAT_MOD_INVALID)
+            data.cursorFB = nullptr;
+
         backend->log(AQ_LOG_DEBUG,
                      std::format("drm: Restoring crtc {} with clock {} hdisplay {} vdisplay {} vrefresh {}", c->crtc->id, data.modeInfo.clock, data.modeInfo.hdisplay,
                                  data.modeInfo.vdisplay, data.modeInfo.vrefresh));