[PATCH v40 6/9] win32u: Handle StatusNotifierWatcher owner changing and registering objects to a new watcher

[Sergei Chernyadyev]

From: Sergei Chernyadyev

---
 dlls/win32u/snidrv/dbus.c | 75 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 75 insertions(+)

diff --git a/dlls/win32u/snidrv/dbus.c b/dlls/win32u/snidrv/dbus.c
index a1690e85cc4..77dce76b47f 100644
--- a/dlls/win32u/snidrv/dbus.c
+++ b/dlls/win32u/snidrv/dbus.c
@@ -311,8 +311,81 @@ static void toggle_watch(DBusWatch *w, void *data)
         remove_watch(w, data);
 }

+static const char* dbus_name_owning_match = "type='signal',"
+    "interface='org.freedesktop.DBus',"
+    "sender='org.freedesktop.DBus',"
+    "member='NameOwnerChanged'";
+
 static const char* const object_path = "/StatusNotifierItem";

+static BOOL register_notification_item(DBusConnection* ctx);
+
+static void restore_items(DBusConnection *ctx)
+{
+    struct tray_icon *icon;
+
+    LIST_FOR_EACH_ENTRY( icon, &sni_list, struct tray_icon, entry )
+        register_notification_item(icon->connection);
+}
+
+static DBusHandlerResult name_owner_filter( DBusConnection *ctx, DBusMessage *msg, void *user_data )
+{
+    char *interface_name, *old_path, *new_path;
+    DBusError error;
+
+    p_dbus_error_init( &error );
+
+    if (p_dbus_message_is_signal( msg, "org.freedesktop.DBus", "NameOwnerChanged" ) &&
+        p_dbus_message_get_args( msg, &error, DBUS_TYPE_STRING, &interface_name, DBUS_TYPE_STRING, &old_path,
+                                 DBUS_TYPE_STRING, &new_path, DBUS_TYPE_INVALID ))
+    {
+        bool is_kde_interface = strcmp(interface_name, kde_watcher_interface_name) == 0;
+        bool is_freedesktop_interface = strcmp(interface_name, freedesktop_watcher_interface_name) == 0;
+        /* check if watcher is disabled first*/
+        if (is_kde_interface || is_freedesktop_interface)
+        {
+            /* TODO: lock the mutex */
+            if (status_notifier_dst_path == NULL || status_notifier_dst_path[0] == '\0')
+            {
+                pthread_mutex_lock(&list_mutex);
+                /* switch between KDE and freedesktop interfaces, despite most implementations rely on KDE interface names */
+                if (is_kde_interface)
+                {
+                    watcher_interface_name = kde_watcher_interface_name;
+                    item_interface_name = kde_item_interface_name;
+                }
+                else if (is_freedesktop_interface)
+                {
+                    watcher_interface_name = freedesktop_watcher_interface_name;
+                    item_interface_name = freedesktop_item_interface_name;
+                }
+
+                old_path = status_notifier_dst_path;
+                status_notifier_dst_path = strdup(new_path);
+                free(old_path);
+
+                if (status_notifier_dst_path != NULL && status_notifier_dst_path[0] != '\0')
+                    restore_items(ctx);
+
+                pthread_mutex_unlock(&list_mutex);
+            }
+            else if (status_notifier_dst_path != NULL &&
+                     status_notifier_dst_path[0] != '\0' &&
+                     strcmp(interface_name, watcher_interface_name) == 0)
+            {
+                pthread_mutex_lock(&list_mutex);
+                old_path = status_notifier_dst_path;
+                status_notifier_dst_path = strdup(new_path);
+                free(old_path);
+                pthread_mutex_lock(&list_mutex);
+            }
+        }
+    }
+
+    p_dbus_error_free( &error );
+    return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
+}
+
 static BOOL get_owner_for_interface(DBusConnection* connection, const char* interface_name, char** owner_path)
 {
     DBusMessage* msg = NULL;
@@ -388,6 +461,8 @@ static BOOL get_notifier_watcher_owner(void)
         goto err;
     }

+    p_dbus_connection_add_filter( global_connection, name_owner_filter, NULL, NULL );
+    p_dbus_bus_add_match( global_connection, dbus_name_owning_match, &error );
     if (p_dbus_error_is_set(&error))
     {
         WARN("failed to register matcher %s: %s\n", error.name, error.message);