Fix undefined behavior due to recursive mutex locking, and properly clear sensors when re-enumerating

AdaptiveBrightness/AdaptiveBrightness.ui

@@ -60,6 +60,9 @@
              <layout class="QGridLayout" name="gridLayout_9">
               <item row="1" column="1">
                <widget class="QSlider" name="monitorMaxBrightnessSlider">
+                <property name="value">
+                 <number>50</number>
+                </property>
                 <property name="orientation">
                  <enum>Qt::Horizontal</enum>
                 </property>
@@ -74,6 +77,9 @@
               </item>
               <item row="0" column="1">
                <widget class="QSlider" name="monitorMinBrightnessSlider">
+                <property name="value">
+                 <number>50</number>
+                </property>
                 <property name="orientation">
                  <enum>Qt::Horizontal</enum>
                 </property>
@@ -117,6 +123,9 @@
              <layout class="QGridLayout" name="gridLayout_2">
               <item row="0" column="0">
                <widget class="QSlider" name="monitorBrightnessSlider">
+                <property name="value">
+                 <number>50</number>
+                </property>
                 <property name="orientation">
                  <enum>Qt::Horizontal</enum>
                 </property>
@@ -160,6 +169,9 @@
            </item>
            <item row="0" column="1">
             <widget class="QSlider" name="monitorOverallMinBrightnessSlider">
+             <property name="value">
+              <number>50</number>
+             </property>
              <property name="orientation">
               <enum>Qt::Horizontal</enum>
              </property>
@@ -184,6 +196,9 @@
            </item>
            <item row="1" column="1">
             <widget class="QSlider" name="monitorOverallMaxBrightnessSlider">
+             <property name="value">
+              <number>50</number>
+             </property>
              <property name="orientation">
               <enum>Qt::Horizontal</enum>
              </property>
@@ -243,6 +258,9 @@
           <layout class="QGridLayout" name="gridLayout">
            <item row="0" column="0">
             <widget class="QSlider" name="monitorOverallBrightnessSlider">
+             <property name="value">
+              <number>50</number>
+             </property>
              <property name="orientation">
               <enum>Qt::Horizontal</enum>
              </property>

AdaptiveBrightness/main_window.cpp

@@ -5,6 +5,7 @@
 
 #include <cmath>
 
+#include <QAbstractButton>
 #include <QCloseEvent>
 #include <QGroupBox>
 #include <QSlider>
@@ -18,7 +19,7 @@
 #include "sensor_driver.hpp"
 #include "utils.hpp"
 
-MainWindow::MainWindow(QWidget* parent) : QMainWindow(parent)
+MainWindow::MainWindow(QWidget* parent) : QMainWindow(parent), m_mtx(QMutex::RecursionMode::Recursive)
 {
     qDebug(ltr("Creating main window"));
 
@@ -28,6 +29,7 @@ MainWindow::MainWindow(QWidget* parent) : QMainWindow(parent)
     enumMonitors();
 
     setupMonitorsTab();
+    setupSensorsTab();
 
     setupTrayIcon();
 
@@ -52,6 +54,8 @@ void MainWindow::closeEvent(QCloseEvent* event)
 
 void MainWindow::enumSensors()
 {
+    m_sensors.clear();
+
     auto sensors = enumerateSensors();
 
     for(auto& sensor: sensors) {
@@ -61,15 +65,20 @@ void MainWindow::enumSensors()
 
 void MainWindow::enumMonitors()
 {
+    m_mtx.lock();
+
     m_ui.monitorDropdown->clear();
+    m_monitors.clear();
 
     auto monitors = enumerateMonitors();
 
     for(auto& monitor: monitors) {
         const auto currentBrightness = monitor.getBrightness();
-        m_ui.monitorDropdown->addItem(monitor.getName());
         m_monitors.push_back({std::move(monitor), 0.f, 1.f, currentBrightness, false});
+        m_ui.monitorDropdown->addItem(m_monitors.back().driver.getName());
     }
+
+    m_mtx.unlock();
 }
 
 void MainWindow::loadIcons()
@@ -110,12 +119,16 @@ void MainWindow::loadIcons()
 void MainWindow::setupMonitorsTab()
 {
     connect(m_ui.monitorDropdown, QOverload<int>::of(&QComboBox::currentIndexChanged), this, &MainWindow::updateCurrentMonitorGUI);
+    connect(m_ui.monitorEnumerateButton, &QAbstractButton::clicked, [this](bool) { enumMonitors(); });
+    connect(m_ui.monitorResetButton, &QAbstractButton::clicked, [this](bool) { resetAllMonitorSettings(); });
 
     setupLimitsSlider();
     setupOverallLimitsSlider();
 
     setupBrightness();
     setupOverallBrightness();
+
+    resetAllMonitorSettings();
 }
 
 void MainWindow::setupLimitsSlider()
@@ -208,6 +221,11 @@ void MainWindow::setupOverallBrightness()
     });
 }
 
+void MainWindow::setupSensorsTab()
+{
+    m_ui.tabs->removeTab(1);
+}
+
 void MainWindow::setupTrayIcon()
 {
     loadIcons();
@@ -259,8 +277,19 @@ void MainWindow::setupCallbackTimer()
     qDebug(ltr("Callback timer started"));
 }
 
+void MainWindow::resetAllMonitorSettings()
+{
+    m_ui.monitorOverallMinBrightnessSlider->setValue(0);
+    m_ui.monitorOverallMaxBrightnessSlider->setValue(100);
+    m_ui.monitorOverallOverrideCheck->setCheckState(Qt::CheckState::Unchecked);
+}
+
 void MainWindow::updateCurrentMonitorGUI(int index)
 {
+    if(index == -1) {
+        return;
+    }
+
     m_ui.monitorMinBrightnessSlider->setValue(utils::map(m_monitors[index].minBrightness, 0.f, 1.f, 0, 100));
     m_ui.monitorMaxBrightnessSlider->setValue(utils::map(m_monitors[index].maxBrightness, 0.f, 1.f, 0, 100));
     m_ui.monitorBrightnessSlider->setValue(utils::map(m_monitors[index].brightness, 0.f, 1.f, 0, 100));
@@ -269,6 +298,8 @@ void MainWindow::updateCurrentMonitorGUI(int index)
 
 void MainWindow::updateState()
 {
+    m_mtx.lock();
+
     qDebug(ltr("Updating state"));
 
     bool sensorError = false;
@@ -301,14 +332,14 @@ void MainWindow::updateState()
         sensorError = true;
     }
 
-    const auto iconBrightness = utils::map(brightness, 0.f, 1.f, size_t{0}, size_t{8});
-
     if(!setMonitorBrightness(brightness)) {
         enumMonitors();
         qInfo(ltr("At least one monitor became invalid, re-enumerating monitors found %1 valid monitor(s)").arg(m_monitors.size()));
         monitorError = true;
     }
 
+    const auto iconBrightness = utils::map(brightness, 0.f, 1.f, size_t{0}, size_t{8});
+
     if(!sensorError && !monitorError && m_trayIcon) {
         m_trayIcon->setIcon(*m_sensorStateIcons[iconBrightness]);
     }
@@ -322,15 +353,18 @@ void MainWindow::updateState()
     if(m_trayIcon) {
         m_trayIcon->setToolTip(tr("AdaptiveBrightness - %1%").arg(brightness * 100, 0, 'f', 0));
     }
+
+    m_mtx.unlock();
 }
 
-bool MainWindow::setMonitorBrightness(float brightness)
+bool MainWindow::setMonitorBrightness(float& brightness)
 {
     if(m_monitors.empty()) {
         return false;
     }
 
     bool errorOccurred = false;
+    bool allMonitorsSameBrightness = true;
 
     for(size_t i = 0; i < m_monitors.size(); ++i) {
         if(!std::isnan(brightness) && !m_monitors[i].manualOverride) {
@@ -339,14 +373,27 @@ bool MainWindow::setMonitorBrightness(float brightness)
 
         m_monitors[i].brightness = std::clamp(m_monitors[i].brightness, m_monitors[i].minBrightness, m_monitors[i].maxBrightness);
 
+        if(i > 0 && m_monitors[i - 1].brightness != m_monitors[i].brightness) {
+            allMonitorsSameBrightness = false;
+        }
+
         if(!m_monitors[i].driver.setBrightness(m_monitors[i].brightness)) {
             errorOccurred = true;
         }
-        else {
+        else if(m_ui.monitorDropdown->currentIndex() == static_cast<int>(i)) {
-            if(m_ui.monitorDropdown->currentIndex() == i) {
+            updateCurrentMonitorGUI(static_cast<int>(i));
-                updateCurrentMonitorGUI(i);
         }
     }
+
+    if(!errorOccurred) {
+        auto avgBrightness = 0.f;
+        std::for_each(m_monitors.begin(), m_monitors.end(), [&avgBrightness](const auto& mon) { avgBrightness += mon.brightness; });
+        avgBrightness /= m_monitors.size();
+        brightness = avgBrightness;
+    }
+
+    if(allMonitorsSameBrightness) {
+        m_ui.monitorOverallBrightnessSlider->setValue(utils::map(brightness, 0.f, 1.f, 0, 100));
     }
 
     return !errorOccurred;

AdaptiveBrightness/main_window.hpp

@@ -8,6 +8,7 @@
 #include <QIcon>
 #include <QMainWindow>
 #include <QMenu>
+#include <QMutex>
 #include <QSlider>
 #include <QSystemTrayIcon>
 #include <QThread>
@@ -39,14 +40,18 @@ class MainWindow : public QMainWindow {
     void setupBrightness();
     void setupOverallBrightness();
 
+    void setupSensorsTab();
+
     void setupTrayIcon();
     void setupCallbackTimer();
 
+    void resetAllMonitorSettings();
+
     void updateCurrentMonitorGUI(int index);
 
     void updateState();
 
-    bool setMonitorBrightness(float brightness);
+    bool setMonitorBrightness(float& brightness);
 
     void trayIconClicked(QSystemTrayIcon::ActivationReason reason);
 
@@ -63,6 +68,7 @@ class MainWindow : public QMainWindow {
 
     QThread* m_timerThread = nullptr;
     QTimer* m_timer = nullptr;
+    QMutex m_mtx;
 
     struct SensorData {
         Sensor driver;

firmware/README.md

@@ -56,7 +56,7 @@ Found DFU: [0483:df11] ver=2200, devnum=57, cfg=1, intf=0, path="2-13.2.2", alt=
 Found DFU: [0483:df11] ver=2200, devnum=57, cfg=1, intf=0, path="2-13.2.2", alt=0, name="@Internal Flash  /0x08000000/032*0001Kg", serial="FFFFFFFEFFFF"
 ```
 
-The firmware can then be flashing using:
+The firmware can then be flashed using:
 
 ```console
 $ dfu-util -a 0 -s 0x08000000 -D AdaptiveBrightnessFirmware.bin