GdkDisplay: Call the correct push|pop_error_trap()

It is wrong to assume all Displays are of the same class as the default.

https://bugzilla.gnome.org/show_bug.cgi?id=784016

diff --git a/gdk/gdkdisplay.c b/gdk/gdkdisplay.c
index f07d188d4e1bdae6d37337489e1c610908055cd7..40225e92ac597532b3c03291e059cbac39e708b1 100644 (file)
--- a/gdk/gdkdisplay.c
+++ b/gdk/gdkdisplay.c
@@ -2301,41 +2301,38 @@ void
 gdk_error_trap_push (void)
 {
   GdkDisplayManager *manager;
-  GdkDisplayClass *class;
   GdkGlobalErrorTrap *trap;
+  GSList *displays;
   GSList *l;
 
   manager = gdk_display_manager_get ();
-  class = GDK_DISPLAY_GET_CLASS (gdk_display_manager_get_default_display (manager));
+  displays = gdk_display_manager_list_displays (manager);
 
-  if (class->push_error_trap == NULL)
-    return;
-
-  trap = g_slice_new (GdkGlobalErrorTrap);
-  trap->displays = gdk_display_manager_list_displays (manager);
+  trap = g_slice_new0 (GdkGlobalErrorTrap);
+  for (l = displays; l != NULL; l = l->next)
+    {
+      GdkDisplay *display = l->data;
+      GdkDisplayClass *class = GDK_DISPLAY_GET_CLASS (display);
 
-  g_slist_foreach (trap->displays, (GFunc) g_object_ref, NULL);
-  for (l = trap->displays; l != NULL; l = l->next)
-    class->push_error_trap (l->data);
+      if (class->push_error_trap != NULL)
+        {
+          class->push_error_trap (display);
+          trap->displays = g_slist_prepend (trap->displays, g_object_ref (display));
+        }
+    }
 
   g_queue_push_head (&gdk_error_traps, trap);
+
+  g_slist_free (displays);
 }
 
 static gint
 gdk_error_trap_pop_internal (gboolean need_code)
 {
-  GdkDisplayManager *manager;
-  GdkDisplayClass *class;
   GdkGlobalErrorTrap *trap;
   gint result;
   GSList *l;
 
-  manager = gdk_display_manager_get ();
-  class = GDK_DISPLAY_GET_CLASS (gdk_display_manager_get_default_display (manager));
-
-  if (class->pop_error_trap == NULL)
-    return 0;
-
   trap = g_queue_pop_head (&gdk_error_traps);
 
   g_return_val_if_fail (trap != NULL, 0);
@@ -2343,7 +2340,9 @@ gdk_error_trap_pop_internal (gboolean need_code)
   result = 0;
   for (l = trap->displays; l != NULL; l = l->next)
     {
-      gint code = class->pop_error_trap (l->data, !need_code);
+      GdkDisplay *display = l->data;
+      GdkDisplayClass *class = GDK_DISPLAY_GET_CLASS (display);
+      gint code = class->pop_error_trap (display, !need_code);
 
       /* we use the error on the last display listed, why not. */
       if (code != 0)