Fixed device unrefed

This commit is contained in:
Erik Reider 2022-03-19 10:52:05 +01:00
parent 175d82858b
commit 105f1cefe1
2 changed files with 57 additions and 30 deletions

View File

@ -2,11 +2,9 @@
#include <libupower-glib/upower.h> #include <libupower-glib/upower.h>
#include <algorithm>
#include <iostream> #include <iostream>
#include <map> #include <map>
#include <string> #include <string>
#include <vector>
#include "ALabel.hpp" #include "ALabel.hpp"
#include "gtkmm/box.h" #include "gtkmm/box.h"
@ -22,14 +20,16 @@ class UPower : public AModule {
auto update() -> void; auto update() -> void;
private: private:
typedef std::unordered_map<std::string, UpDevice *> Devices;
static void deviceAdded_cb(UpClient *client, UpDevice *device, gpointer data); static void deviceAdded_cb(UpClient *client, UpDevice *device, gpointer data);
static void deviceRemoved_cb(UpClient *client, const gchar *object_path, gpointer data); static void deviceRemoved_cb(UpClient *client, const gchar *objectPath, gpointer data);
static void deviceNotify_cb(gpointer data); static void deviceNotify_cb(UpDevice *device, GParamSpec *pspec, gpointer user_data);
static void prepareForSleep_cb(GDBusConnection *system_bus, const gchar *sender_name, static void prepareForSleep_cb(GDBusConnection *system_bus, const gchar *sender_name,
const gchar *object_path, const gchar *interface_name, const gchar *object_path, const gchar *interface_name,
const gchar *signal_name, GVariant *parameters, const gchar *signal_name, GVariant *parameters,
gpointer user_data); gpointer user_data);
void removeDevice(const std::string devicePath); void removeDevice(const gchar *objectPath);
void addDevice(UpDevice *device); void addDevice(UpDevice *device);
void setDisplayDevice(); void setDisplayDevice();
void resetDevices(); void resetDevices();
@ -42,9 +42,10 @@ class UPower : public AModule {
bool hideIfEmpty = true; bool hideIfEmpty = true;
uint iconSize = 32; uint iconSize = 32;
UpClient *client = NULL; Devices devices;
UpDevice *displayDevice = NULL; std::mutex m_Mutex;
std::map<std::string, UpDevice *> devices; UpClient *client;
UpDevice *displayDevice;
guint login1_id; guint login1_id;
GDBusConnection *login1_connection; GDBusConnection *login1_connection;
}; };

View File

@ -1,12 +1,5 @@
#include "modules/upower.hpp" #include "modules/upower.hpp"
#include <gio/gio.h>
#include <iostream>
#include <map>
#include <string>
#include "gtkmm/enums.h"
#include "gtkmm/icontheme.h" #include "gtkmm/icontheme.h"
namespace waybar::modules { namespace waybar::modules {
@ -15,7 +8,9 @@ UPower::UPower(const std::string& id, const Json::Value& config)
box_(Gtk::ORIENTATION_HORIZONTAL, 0), box_(Gtk::ORIENTATION_HORIZONTAL, 0),
icon_(), icon_(),
label_(), label_(),
devices() { devices(),
m_Mutex(),
client() {
box_.pack_start(icon_); box_.pack_start(icon_);
box_.pack_start(label_); box_.pack_start(label_);
event_box_.add(box_); event_box_.add(box_);
@ -56,11 +51,8 @@ UPower::UPower(const std::string& id, const Json::Value& config)
g_signal_connect(client, "device-added", G_CALLBACK(deviceAdded_cb), this); g_signal_connect(client, "device-added", G_CALLBACK(deviceAdded_cb), this);
g_signal_connect(client, "device-removed", G_CALLBACK(deviceRemoved_cb), this); g_signal_connect(client, "device-removed", G_CALLBACK(deviceRemoved_cb), this);
g_signal_connect(client, "notify", G_CALLBACK(deviceNotify_cb), this);
resetDevices(); resetDevices();
dp.emit();
} }
UPower::~UPower() { UPower::~UPower() {
@ -78,14 +70,14 @@ void UPower::deviceAdded_cb(UpClient* client, UpDevice* device, gpointer data) {
// Update the widget // Update the widget
up->dp.emit(); up->dp.emit();
} }
void UPower::deviceRemoved_cb(UpClient* client, const gchar* object_path, gpointer data) { void UPower::deviceRemoved_cb(UpClient* client, const gchar* objectPath, gpointer data) {
UPower* up = static_cast<UPower*>(data); UPower* up = static_cast<UPower*>(data);
up->removeDevice(object_path); up->removeDevice(objectPath);
up->setDisplayDevice(); up->setDisplayDevice();
// Update the widget // Update the widget
up->dp.emit(); up->dp.emit();
} }
void UPower::deviceNotify_cb(gpointer data) { void UPower::deviceNotify_cb(UpDevice* device, GParamSpec* pspec, gpointer data) {
UPower* up = static_cast<UPower*>(data); UPower* up = static_cast<UPower*>(data);
// Update the widget // Update the widget
up->dp.emit(); up->dp.emit();
@ -104,17 +96,47 @@ void UPower::prepareForSleep_cb(GDBusConnection* system_bus, const gchar* sender
} }
} }
void UPower::removeDevice(const std::string devicePath) { devices.erase(devicePath); } void UPower::removeDevice(const gchar* objectPath) {
std::lock_guard<std::mutex> guard(m_Mutex);
if (devices.find(objectPath) != devices.end()) {
UpDevice* device = devices[objectPath];
if (G_IS_OBJECT(device)) {
g_object_unref(device);
}
devices.erase(objectPath);
}
}
void UPower::addDevice(UpDevice* device) { void UPower::addDevice(UpDevice* device) {
if (device) { if (G_IS_OBJECT(device)) {
const gchar* objectPath = up_device_get_object_path(device); const gchar* objectPath = up_device_get_object_path(device);
devices[objectPath] = device;
// Due to the device getting cleared after this event is fired, we
// create a new object pointing to its objectPath
gboolean ret;
device = up_device_new();
ret = up_device_set_object_path_sync(device, objectPath, NULL, NULL);
if (!ret) {
g_object_unref(G_OBJECT(device));
return;
}
std::lock_guard<std::mutex> guard(m_Mutex);
if (devices.find(objectPath) != devices.end()) {
UpDevice* device = devices[objectPath];
if (G_IS_OBJECT(device)) {
g_object_unref(device);
}
devices.erase(objectPath);
}
g_signal_connect(device, "notify", G_CALLBACK(deviceNotify_cb), this); g_signal_connect(device, "notify", G_CALLBACK(deviceNotify_cb), this);
devices.emplace(Devices::value_type(objectPath, device));
} }
} }
void UPower::setDisplayDevice() { void UPower::setDisplayDevice() {
std::lock_guard<std::mutex> guard(m_Mutex);
displayDevice = up_client_get_display_device(client); displayDevice = up_client_get_display_device(client);
g_signal_connect(displayDevice, "notify", G_CALLBACK(deviceNotify_cb), this); g_signal_connect(displayDevice, "notify", G_CALLBACK(deviceNotify_cb), this);
} }
@ -122,9 +144,12 @@ void UPower::setDisplayDevice() {
/** Removes all devices and adds the current devices */ /** Removes all devices and adds the current devices */
void UPower::resetDevices() { void UPower::resetDevices() {
// Removes all devices // Removes all devices
if (devices.size() > 0) { if (!devices.empty()) {
auto it = devices.cbegin(); auto it = devices.cbegin();
while (it != devices.cend()) { while (it != devices.cend()) {
if (G_IS_OBJECT(it->second)) {
g_object_unref(it->second);
}
devices.erase(it++); devices.erase(it++);
} }
} }
@ -143,10 +168,11 @@ void UPower::resetDevices() {
} }
auto UPower::update() -> void { auto UPower::update() -> void {
std::lock_guard<std::mutex> guard(m_Mutex);
if (devices.size() == 0 && hideIfEmpty) { if (devices.size() == 0 && hideIfEmpty) {
box_.set_visible(false); event_box_.set_visible(false);
} else { } else {
box_.set_visible(true); event_box_.set_visible(true);
UpDeviceKind kind; UpDeviceKind kind;
UpDeviceState state; UpDeviceState state;