From a6315b0af4e417cf396649d9c5792d3c5420b713 Mon Sep 17 00:00:00 2001
From: Ikalco <73481042+ikalco@users.noreply.github.com>
Date: Sat, 31 Aug 2024 14:55:08 -0500
Subject: [PATCH] core: fix crash on monitor removed with gammaControl (#7601)

* fix crash on monitor removed with gammaControl

* Update GammaControl.cpp
---
 src/protocols/GammaControl.cpp | 14 +++++++-------
 src/protocols/GammaControl.hpp |  2 +-
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/src/protocols/GammaControl.cpp b/src/protocols/GammaControl.cpp
index cb75a2020d2..d7992981307 100644
--- a/src/protocols/GammaControl.cpp
+++ b/src/protocols/GammaControl.cpp
@@ -17,7 +17,7 @@ CGammaControl::CGammaControl(SP<CZwlrGammaControlV1> resource_, wl_resource* out
         return;
     }
 
-    pMonitor = OUTPUTRES->monitor.get();
+    pMonitor = OUTPUTRES->monitor;
 
     if (!pMonitor) {
         LOGM(ERR, "No CMonitor");
@@ -103,7 +103,7 @@ CGammaControl::CGammaControl(SP<CZwlrGammaControlV1> resource_, wl_resource* out
     resource->sendGammaSize(gammaSize);
 
     listeners.monitorDestroy    = pMonitor->events.destroy.registerListener([this](std::any) { this->onMonitorDestroy(); });
-    listeners.monitorDisconnect = pMonitor->events.destroy.registerListener([this](std::any) { this->onMonitorDestroy(); });
+    listeners.monitorDisconnect = pMonitor->events.disconnect.registerListener([this](std::any) { this->onMonitorDestroy(); });
 }
 
 CGammaControl::~CGammaControl() {
@@ -119,7 +119,7 @@ bool CGammaControl::good() {
 }
 
 void CGammaControl::applyToMonitor() {
-    if (!pMonitor)
+    if (!pMonitor || !pMonitor->output)
         return; // ??
 
     LOGM(LOG, "setting to monitor {}", pMonitor->szName);
@@ -136,16 +136,16 @@ void CGammaControl::applyToMonitor() {
         pMonitor->output->state->setGammaLut({});
     }
 
-    g_pHyprRenderer->damageMonitor(pMonitor);
+    g_pHyprRenderer->damageMonitor(pMonitor.get());
 }
 
 CMonitor* CGammaControl::getMonitor() {
-    return pMonitor;
+    return pMonitor ? pMonitor.get() : nullptr;
 }
 
 void CGammaControl::onMonitorDestroy() {
+    LOGM(LOG, "Destroying gamma control for {}", pMonitor->szName);
     resource->sendFailed();
-    pMonitor = nullptr;
 }
 
 CGammaControlProtocol::CGammaControlProtocol(const wl_interface* iface, const int& ver, const std::string& name) : IWaylandProtocol(iface, ver, name) {
@@ -187,4 +187,4 @@ void CGammaControlProtocol::applyGammaToState(CMonitor* pMonitor) {
         g->applyToMonitor();
         break;
     }
-}
\ No newline at end of file
+}
diff --git a/src/protocols/GammaControl.hpp b/src/protocols/GammaControl.hpp
index 963eea5c192..bbdc0139126 100644
--- a/src/protocols/GammaControl.hpp
+++ b/src/protocols/GammaControl.hpp
@@ -20,7 +20,7 @@ class CGammaControl {
 
   private:
     SP<CZwlrGammaControlV1> resource;
-    CMonitor*               pMonitor      = nullptr;
+    WP<CMonitor>            pMonitor;
     size_t                  gammaSize     = 0;
     bool                    gammaTableSet = false;
     std::vector<uint16_t>   gammaTable; // [r,g,b]+