Remove some redundant out-of-memory checks
[platform/upstream/at-spi2-core.git] / atspi / atspi-misc.c
1 /*
2  * AT-SPI - Assistive Technology Service Provider Interface
3  * (Gnome Accessibility Project; http://developer.gnome.org/projects/gap)
4  *
5  * Copyright 2001, 2002 Sun Microsystems Inc.,
6  * Copyright 2001, 2002 Ximian, Inc.
7  * Copyright 2010, 2011 Novell, Inc.
8  *
9  * This library is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Library General Public
11  * License as published by the Free Software Foundation; either
12  * version 2 of the License, or (at your option) any later version.
13  *
14  * This library is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Library General Public License for more details.
18  *
19  * You should have received a copy of the GNU Library General Public
20  * License along with this library; if not, write to the
21  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
22  * Boston, MA 02111-1307, USA.
23  */
24
25 /*
26  *
27  * Basic SPI initialization and event loop function prototypes
28  *
29  */
30
31 #include "atspi-private.h"
32 #include "X11/Xlib.h"
33 #include "atspi-gmain.h"
34 #include <stdio.h>
35 #include <string.h>
36
37 static void handle_get_items (DBusPendingCall *pending, void *user_data);
38
39 static DBusConnection *bus = NULL;
40 static GHashTable *live_refs = NULL;
41
42 const char *atspi_path_dec = ATSPI_DBUS_PATH_DEC;
43 const char *atspi_path_registry = ATSPI_DBUS_PATH_REGISTRY;
44 const char *atspi_path_root = ATSPI_DBUS_PATH_ROOT;
45 const char *atspi_bus_registry = ATSPI_DBUS_NAME_REGISTRY;
46 const char *atspi_interface_accessible = ATSPI_DBUS_INTERFACE_ACCESSIBLE;
47 const char *atspi_interface_action = ATSPI_DBUS_INTERFACE_ACTION;
48 const char *atspi_interface_application = ATSPI_DBUS_INTERFACE_APPLICATION;
49 const char *atspi_interface_collection = ATSPI_DBUS_INTERFACE_COLLECTION;
50 const char *atspi_interface_component = ATSPI_DBUS_INTERFACE_COMPONENT;
51 const char *atspi_interface_dec = ATSPI_DBUS_INTERFACE_DEC;
52 const char *atspi_interface_device_event_listener = ATSPI_DBUS_INTERFACE_DEVICE_EVENT_LISTENER;
53 const char *atspi_interface_document = ATSPI_DBUS_INTERFACE_DOCUMENT;
54 const char *atspi_interface_editable_text = ATSPI_DBUS_INTERFACE_EDITABLE_TEXT;
55 const char *atspi_interface_event_object = ATSPI_DBUS_INTERFACE_EVENT_OBJECT;
56 const char *atspi_interface_hyperlink = ATSPI_DBUS_INTERFACE_HYPERLINK;
57 const char *atspi_interface_hypertext = ATSPI_DBUS_INTERFACE_HYPERTEXT;
58 const char *atspi_interface_image = ATSPI_DBUS_INTERFACE_IMAGE;
59 const char *atspi_interface_registry = ATSPI_DBUS_INTERFACE_REGISTRY;
60 const char *atspi_interface_selection = ATSPI_DBUS_INTERFACE_SELECTION;
61 const char *atspi_interface_table = ATSPI_DBUS_INTERFACE_TABLE;
62 const char *atspi_interface_text = ATSPI_DBUS_INTERFACE_TEXT;
63 const char *atspi_interface_cache = ATSPI_DBUS_INTERFACE_CACHE;
64 const char *atspi_interface_value = ATSPI_DBUS_INTERFACE_VALUE;
65
66 static const char *interfaces[] =
67 {
68   ATSPI_DBUS_INTERFACE_ACCESSIBLE,
69   ATSPI_DBUS_INTERFACE_ACTION,
70   ATSPI_DBUS_INTERFACE_APPLICATION,
71   ATSPI_DBUS_INTERFACE_COLLECTION,
72   ATSPI_DBUS_INTERFACE_COMPONENT,
73   ATSPI_DBUS_INTERFACE_DOCUMENT,
74   ATSPI_DBUS_INTERFACE_EDITABLE_TEXT,
75   ATSPI_DBUS_INTERFACE_HYPERLINK,
76   ATSPI_DBUS_INTERFACE_HYPERTEXT,
77   ATSPI_DBUS_INTERFACE_IMAGE,
78   "org.a11y.atspi.LoginHelper",
79   ATSPI_DBUS_INTERFACE_SELECTION,
80   ATSPI_DBUS_INTERFACE_TABLE,
81   ATSPI_DBUS_INTERFACE_TEXT,
82   ATSPI_DBUS_INTERFACE_VALUE,
83   NULL
84 };
85
86 gint
87 _atspi_get_iface_num (const char *iface)
88 {
89   /* TODO: Use a binary search or hash to improve performance */
90   int i;
91
92   for (i = 0; interfaces[i]; i++)
93   {
94     if (!strcmp(iface, interfaces[i])) return i;
95   }
96   return -1;
97 }
98
99 static GHashTable *
100 get_live_refs (void)
101 {
102   if (!live_refs) 
103     {
104       live_refs = g_hash_table_new (g_direct_hash, g_direct_equal);
105     }
106   return live_refs;
107 }
108
109 /* TODO: Add an application parameter */
110 DBusConnection *
111 _atspi_bus ()
112 {
113   if (!bus)
114     atspi_init ();
115   return bus;
116 }
117
118 #define APP_IS_REGISTRY(app) (!strcmp (app->bus_name, atspi_bus_registry))
119
120 static void
121 cleanup ()
122 {
123   GHashTable *refs;
124
125   refs = live_refs;
126   live_refs = NULL;
127   if (refs)
128     {
129       g_hash_table_destroy (refs);
130     }
131 }
132
133 static gboolean atspi_inited = FALSE;
134
135 static GHashTable *app_hash = NULL;
136
137 static void
138 handle_get_bus_address (DBusPendingCall *pending, void *user_data)
139 {
140   AtspiApplication *app = user_data;
141   DBusMessage *reply = dbus_pending_call_steal_reply (pending);
142   DBusMessage *message;
143   const char *address;
144   DBusPendingCall *new_pending;
145
146   if (dbus_message_get_type (reply) == DBUS_MESSAGE_TYPE_METHOD_RETURN)
147   {
148     if (dbus_message_get_args (reply, NULL, DBUS_TYPE_STRING, &address,
149                                DBUS_TYPE_INVALID))
150     {
151       DBusError error;
152       dbus_error_init (&error);
153       DBusConnection *bus = dbus_connection_open (address, &error);
154       if (bus)
155       {
156         if (app->bus)
157           dbus_connection_unref (app->bus);
158         app->bus = bus;
159       }
160     }
161   }
162   dbus_message_unref (reply);
163   dbus_pending_call_unref (pending);
164
165   if (!app->bus)
166     return; /* application has gone away / been disposed */
167
168   message = dbus_message_new_method_call (app->bus_name,
169                                           "/org/a11y/atspi/cache",
170                                           atspi_interface_cache, "GetItems");
171
172    dbus_connection_send_with_reply (app->bus, message, &new_pending, 2000);
173   dbus_pending_call_set_notify (new_pending, handle_get_items, app, NULL);
174   dbus_message_unref (message);
175 }
176
177 static AtspiApplication *
178 get_application (const char *bus_name)
179 {
180   AtspiApplication *app = NULL;
181   char *bus_name_dup;
182   DBusMessage *message;
183   DBusError error;
184   DBusPendingCall *pending = NULL;
185
186   if (!app_hash)
187   {
188     app_hash = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, (GDestroyNotify)g_object_unref);
189     if (!app_hash) return NULL;
190   }
191   app = g_hash_table_lookup (app_hash, bus_name);
192   if (app) return app;
193   bus_name_dup = g_strdup (bus_name);
194   if (!bus_name_dup) return NULL;
195   // TODO: change below to something that will send state-change:defunct notification if necessary */
196   app = _atspi_application_new (bus_name);
197   if (!app) return NULL;
198   app->hash = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, g_object_unref);
199   app->bus = dbus_connection_ref (_atspi_bus ());
200   app->cache = ATSPI_CACHE_UNDEFINED;
201   g_hash_table_insert (app_hash, bus_name_dup, app);
202   dbus_error_init (&error);
203   message = dbus_message_new_method_call (bus_name, atspi_path_root,
204                                           atspi_interface_application, "GetApplicationBusAddress");
205
206    dbus_connection_send_with_reply (app->bus, message, &pending, 2000);
207   dbus_pending_call_set_notify (pending, handle_get_bus_address, app, NULL);
208   dbus_message_unref (message);
209   return app;
210 }
211
212 static AtspiAccessible *
213 ref_accessible (const char *app_name, const char *path)
214 {
215   AtspiApplication *app;
216   AtspiAccessible *a;
217
218   if (!strcmp (path, ATSPI_DBUS_PATH_NULL))
219     return NULL;
220
221   app = get_application (app_name);
222
223   if (!strcmp (path, "/org/a11y/atspi/accessible/root"))
224   {
225     if (!app->root)
226     {
227       app->root = atspi_accessible_new (app, atspi_path_root);
228       app->root->accessible_parent = atspi_get_desktop (0);
229     }
230     return g_object_ref (app->root);
231   }
232
233   a = g_hash_table_lookup (app->hash, path);
234   if (a)
235   {
236     return g_object_ref (a);
237   }
238   a = atspi_accessible_new (app, path);
239   if (!a)
240     return NULL;
241   g_hash_table_insert (app->hash, g_strdup (a->parent.path), a);
242   g_object_ref (a);     /* for the hash */
243   return a;
244 }
245
246 static AtspiHyperlink *
247 ref_hyperlink (const char *app_name, const char *path)
248 {
249   AtspiApplication *app = get_application (app_name);
250   AtspiHyperlink *hyperlink;
251
252   if (!strcmp (path, ATSPI_DBUS_PATH_NULL))
253     return NULL;
254
255   hyperlink = g_hash_table_lookup (app->hash, path);
256   if (hyperlink)
257   {
258     return g_object_ref (hyperlink);
259   }
260   hyperlink = atspi_hyperlink_new (app, path);
261   if (!hyperlink)
262     return NULL;
263   g_hash_table_insert (app->hash, g_strdup (hyperlink->parent.path), hyperlink);
264   /* TODO: This should be a weak ref */
265   g_object_ref (hyperlink);     /* for the hash */
266   return hyperlink;
267 }
268
269 typedef struct
270 {
271   char *path;
272   char *parent;
273   GArray *children;
274   GArray *interfaces;
275   char *name;
276   dbus_uint32_t role;
277   char *description;
278   GArray *state_bitflags;
279 } CACHE_ADDITION;
280
281 static DBusHandlerResult
282 handle_remove_accessible (DBusConnection *bus, DBusMessage *message, void *user_data)
283 {
284   const char *sender = dbus_message_get_sender (message);
285   AtspiApplication *app;
286   const char *path;
287   DBusMessageIter iter, iter_struct;
288   const char *signature = dbus_message_get_signature (message);
289   AtspiAccessible *a;
290
291   if (strcmp (signature, "(so)") != 0)
292   {
293     g_warning ("AT-SPI: Unknown signature %s for RemoveAccessible", signature);
294     return DBUS_HANDLER_RESULT_HANDLED;
295   }
296
297   dbus_message_iter_init (message, &iter);
298   dbus_message_iter_recurse (&iter, &iter_struct);
299   dbus_message_iter_get_basic (&iter_struct, &sender);
300   dbus_message_iter_next (&iter_struct);
301   dbus_message_iter_get_basic (&iter_struct, &path);
302   app = get_application (sender);
303   a = ref_accessible (sender, path);
304   if (!a)
305     return DBUS_HANDLER_RESULT_HANDLED;
306   g_object_run_dispose (G_OBJECT (a));
307   g_hash_table_remove (app->hash, a->parent.path);
308   g_object_unref (a);   /* unref our own ref */
309   return DBUS_HANDLER_RESULT_HANDLED;
310 }
311
312 static gboolean
313 add_app_to_desktop (AtspiAccessible *a, const char *bus_name)
314 {
315   DBusError error;
316
317   dbus_error_init (&error);
318   AtspiAccessible *obj = ref_accessible (bus_name, atspi_path_root);
319   if (obj)
320   {
321     GList *new_list = g_list_append (a->children, obj);
322     if (new_list)
323     {
324       a->children = new_list;
325       return TRUE;
326     }
327   }
328   else
329   {
330     g_warning ("AT-SPI: Error calling getRoot for %s: %s", bus_name, error.message);
331   }
332   return FALSE;
333 }
334
335 static void
336 send_children_changed (AtspiAccessible *parent, AtspiAccessible *child, gboolean add)
337 {
338   AtspiEvent e;
339
340   memset (&e, 0, sizeof (e));
341   e.type = (add? "object:children-changed:add": "object:children-changed:remove");
342   e.source = parent;
343   e.detail1 = g_list_index (parent->children, child);
344   e.detail2 = 0;
345   _atspi_send_event (&e);
346 }
347
348 static void
349 unref_object_and_descendants (AtspiAccessible *obj)
350 {
351   GList *l;
352
353   for (l = obj->children; l; l = l->next)
354   {
355     unref_object_and_descendants (l->data);
356   }
357   g_object_unref (obj);
358 }
359
360 static gboolean
361 remove_app_from_desktop (AtspiAccessible *a, const char *bus_name)
362 {
363   GList *l;
364   AtspiAccessible *child;
365
366   for (l = a->children; l; l = l->next)
367   {
368     child = l->data;
369     if (!strcmp (bus_name, child->parent.app->bus_name)) break;
370   }
371   if (!l)
372   {
373     return FALSE;
374   }
375   send_children_changed (a, child, FALSE);
376   a->children = g_list_remove (a->children, child);
377   unref_object_and_descendants (child);
378   return TRUE;
379 }
380
381 static AtspiAccessible *desktop;
382
383 void
384 get_reference_from_iter (DBusMessageIter *iter, const char **app_name, const char **path)
385 {
386   DBusMessageIter iter_struct;
387
388   dbus_message_iter_recurse (iter, &iter_struct);
389   dbus_message_iter_get_basic (&iter_struct, app_name);
390   dbus_message_iter_next (&iter_struct);
391   dbus_message_iter_get_basic (&iter_struct, path);
392   dbus_message_iter_next (iter);
393 }
394
395 static void
396 add_accessible_from_iter (DBusMessageIter *iter)
397 {
398   GList *new_list;
399   DBusMessageIter iter_struct, iter_array;
400   const char *app_name, *path;
401   AtspiAccessible *accessible;
402   const char *name, *description;
403   dbus_uint32_t role;
404
405   dbus_message_iter_recurse (iter, &iter_struct);
406
407   /* get accessible */
408   get_reference_from_iter (&iter_struct, &app_name, &path);
409   accessible = ref_accessible (app_name, path);
410   if (!accessible)
411     return;
412
413   /* Get application: TODO */
414   dbus_message_iter_next (&iter_struct);
415
416   /* get parent */
417   get_reference_from_iter (&iter_struct, &app_name, &path);
418   if (accessible->accessible_parent)
419     g_object_unref (accessible->accessible_parent);
420   accessible->accessible_parent = ref_accessible (app_name, path);
421
422   /* Get children */
423   while (accessible->children)
424   {
425     g_object_unref (accessible->children->data);
426     accessible->children = g_list_remove (accessible->children, accessible->children->data);
427   }
428   dbus_message_iter_recurse (&iter_struct, &iter_array);
429   while (dbus_message_iter_get_arg_type (&iter_array) != DBUS_TYPE_INVALID)
430   {
431     AtspiAccessible *child;
432     get_reference_from_iter (&iter_array, &app_name, &path);
433     child = ref_accessible (app_name, path);
434     new_list = g_list_append (accessible->children, child);
435     if (new_list) accessible->children = new_list;
436   }
437
438   /* interfaces */
439   dbus_message_iter_next (&iter_struct);
440   _atspi_dbus_set_interfaces (accessible, &iter_struct);
441   dbus_message_iter_next (&iter_struct);
442
443   /* name */
444   if (accessible->name)
445     g_free (accessible->name);
446   dbus_message_iter_get_basic (&iter_struct, &name);
447   accessible->name = g_strdup (name);
448   dbus_message_iter_next (&iter_struct);
449
450   /* role */
451   dbus_message_iter_get_basic (&iter_struct, &role);
452   accessible->role = role;
453   dbus_message_iter_next (&iter_struct);
454
455   /* description */
456   if (accessible->description)
457     g_free (accessible->description);
458   dbus_message_iter_get_basic (&iter_struct, &description);
459   accessible->description = g_strdup (description);
460   dbus_message_iter_next (&iter_struct);
461
462   _atspi_dbus_set_state (accessible, &iter_struct);
463   dbus_message_iter_next (&iter_struct);
464
465   _atspi_accessible_add_cache (accessible, ATSPI_CACHE_NAME | ATSPI_CACHE_ROLE |
466                                ATSPI_CACHE_PARENT | ATSPI_CACHE_DESCRIPTION);
467   if (!atspi_state_set_contains (accessible->states,
468                                        ATSPI_STATE_MANAGES_DESCENDANTS))
469     _atspi_accessible_add_cache (accessible, ATSPI_CACHE_CHILDREN);
470
471   /* This is a bit of a hack since the cache holds a ref, so we don't need
472    * the one provided for us anymore */
473   g_object_unref (accessible);
474 }
475
476 static void
477 handle_get_items (DBusPendingCall *pending, void *user_data)
478 {
479   DBusMessage *reply = dbus_pending_call_steal_reply (pending);
480   DBusMessageIter iter, iter_array;
481
482   if (dbus_message_get_type (reply) == DBUS_MESSAGE_TYPE_ERROR)
483   {
484     const char *sender = dbus_message_get_sender (reply);
485     const char *error = NULL;
486     dbus_message_get_args (reply, NULL, DBUS_TYPE_STRING, &error,
487                            DBUS_TYPE_INVALID);
488     g_warning ("AT-SPI: Error in GetItems, sender=%s, error=%s", sender, error);
489     dbus_message_unref (reply);
490     dbus_pending_call_unref (pending);
491     return;
492   }
493
494   dbus_message_iter_init (reply, &iter);
495   dbus_message_iter_recurse (&iter, &iter_array);
496   while (dbus_message_iter_get_arg_type (&iter_array) != DBUS_TYPE_INVALID)
497   {
498     add_accessible_from_iter (&iter_array);
499     dbus_message_iter_next (&iter_array);
500   }
501   dbus_message_unref (reply);
502   dbus_pending_call_unref (pending);
503 }
504
505 /* TODO: Do we stil need this function? */
506 static AtspiAccessible *
507 ref_accessible_desktop (AtspiApplication *app)
508 {
509   DBusError error;
510   DBusMessage *message, *reply;
511   DBusMessageIter iter, iter_array;
512   gchar *bus_name_dup;
513
514   if (desktop)
515   {
516     g_object_ref (desktop);
517     return desktop;
518   }
519   desktop = atspi_accessible_new (app, atspi_path_root);
520   if (!desktop)
521   {
522     return NULL;
523   }
524   g_hash_table_insert (app->hash, desktop->parent.path, desktop);
525   g_object_ref (desktop);       /* for the hash */
526   desktop->name = g_strdup ("main");
527   dbus_error_init (&error);
528   message = dbus_message_new_method_call (atspi_bus_registry,
529         atspi_path_root,
530         atspi_interface_accessible,
531         "GetChildren");
532   if (!message)
533     return NULL;
534   reply = _atspi_dbus_send_with_reply_and_block (message, NULL);
535   if (!reply || strcmp (dbus_message_get_signature (reply), "a(so)") != 0)
536   {
537     g_warning ("Couldn't get application list: %s", error.message);
538     if (reply)
539       dbus_message_unref (reply);
540     return NULL;
541   }
542   dbus_message_iter_init (reply, &iter);
543   dbus_message_iter_recurse (&iter, &iter_array);
544   while (dbus_message_iter_get_arg_type (&iter_array) != DBUS_TYPE_INVALID)
545   {
546     const char *app_name, *path;
547     get_reference_from_iter (&iter_array, &app_name, &path);
548     add_app_to_desktop (desktop, app_name);
549   }
550   dbus_message_unref (reply);
551
552   /* Record the alternate name as an alias for org.a11y.atspi.Registry */
553   bus_name_dup = g_strdup (dbus_message_get_sender (reply));
554   if (bus_name_dup)
555     g_hash_table_insert (app_hash, bus_name_dup, app);
556
557   return desktop;
558 }
559
560 AtspiAccessible *
561 _atspi_ref_accessible (const char *app, const char *path)
562 {
563   AtspiApplication *a = get_application (app);
564   if (!a) return NULL;
565   if ( APP_IS_REGISTRY(a))
566   {
567     return a->root = ref_accessible_desktop (a);
568   }
569   return ref_accessible (app, path);
570 }
571
572 AtspiAccessible *
573 _atspi_dbus_return_accessible_from_message (DBusMessage *message)
574 {
575   DBusMessageIter iter;
576   AtspiAccessible *retval = NULL;
577   const char *signature;
578
579   if (!message)
580     return NULL;
581
582   signature = dbus_message_get_signature (message);
583   if (!strcmp (signature, "(so)"))
584   {
585     dbus_message_iter_init (message, &iter);
586     retval =  _atspi_dbus_return_accessible_from_iter (&iter);
587   }
588   else
589   {
590     g_warning ("AT-SPI: Called _atspi_dbus_return_accessible_from_message with strange signature %s", signature);
591   }
592   dbus_message_unref (message);
593   return retval;
594 }
595
596 AtspiAccessible *
597 _atspi_dbus_return_accessible_from_iter (DBusMessageIter *iter)
598 {
599   const char *app_name, *path;
600
601   get_reference_from_iter (iter, &app_name, &path);
602   return ref_accessible (app_name, path);
603 }
604
605 AtspiHyperlink *
606 _atspi_dbus_return_hyperlink_from_message (DBusMessage *message)
607 {
608   DBusMessageIter iter;
609   AtspiHyperlink *retval = NULL;
610   const char *signature = dbus_message_get_signature (message);
611    
612   if (!strcmp (signature, "(so)"))
613   {
614     dbus_message_iter_init (message, &iter);
615     retval =  _atspi_dbus_return_hyperlink_from_iter (&iter);
616   }
617   else
618   {
619     g_warning ("AT-SPI: Called _atspi_dbus_return_hyperlink_from_message with strange signature %s", signature);
620   }
621   dbus_message_unref (message);
622   return retval;
623 }
624
625 AtspiHyperlink *
626 _atspi_dbus_return_hyperlink_from_iter (DBusMessageIter *iter)
627 {
628   const char *app_name, *path;
629
630   get_reference_from_iter (iter, &app_name, &path);
631   return ref_hyperlink (app_name, path);
632 }
633
634 const char *cache_signal_type = "((so)(so)(so)a(so)assusau)";
635
636 static DBusHandlerResult
637 handle_add_accessible (DBusConnection *bus, DBusMessage *message, void *user_data)
638 {
639   DBusMessageIter iter;
640   const char *sender = dbus_message_get_sender (message);
641
642   if (strcmp (dbus_message_get_signature (message), cache_signal_type) != 0)
643   {
644     g_warning ("AT-SPI: AddAccessible with unknown signature %s\n",
645                dbus_message_get_signature (message));
646     return DBUS_HANDLER_RESULT_HANDLED;
647   }
648
649   dbus_message_iter_init (message, &iter);
650   add_accessible_from_iter (&iter);
651   return DBUS_HANDLER_RESULT_HANDLED;
652 }
653
654 typedef struct
655 {
656   DBusConnection *bus;
657   DBusMessage *message;
658   void *data;
659 } BusDataClosure;
660
661 static guint process_deferred_messages_id = -1;
662
663 static void
664 process_deferred_message (BusDataClosure *closure)
665 {
666   int type = dbus_message_get_type (closure->message);
667   const char *interface = dbus_message_get_interface (closure->message);
668
669   if (type == DBUS_MESSAGE_TYPE_SIGNAL &&
670       !strncmp (interface, "org.a11y.atspi.Event.", 21))
671   {
672     atspi_dbus_handle_event (closure->bus, closure->message, closure->data);
673   }
674   if (dbus_message_is_method_call (closure->message, atspi_interface_device_event_listener, "NotifyEvent"))
675   {
676     atspi_dbus_handle_DeviceEvent (closure->bus,
677                                    closure->message, closure->data);
678   }
679   if (dbus_message_is_signal (closure->message, atspi_interface_cache, "AddAccessible"))
680   {
681     handle_add_accessible (closure->bus, closure->message, closure->data);
682   }
683   if (dbus_message_is_signal (closure->message, atspi_interface_cache, "RemoveAccessible"))
684   {
685     handle_remove_accessible (closure->bus, closure->message, closure->data);
686   }
687 }
688
689 static GList *deferred_messages = NULL;
690
691 gboolean
692 _atspi_process_deferred_messages (gpointer data)
693 {
694   static int in_process_deferred_messages = 0;
695
696   if (in_process_deferred_messages)
697     return TRUE;
698   in_process_deferred_messages = 1;
699   while (deferred_messages != NULL)
700   {
701     BusDataClosure *closure = deferred_messages->data;
702     process_deferred_message (closure);
703     deferred_messages = g_list_remove (deferred_messages, closure);
704     dbus_message_unref (closure->message);
705     dbus_connection_unref (closure->bus);
706     g_free (closure);
707   }
708   /* If data is NULL, assume that we were called from GLib */
709   if (!data)
710     process_deferred_messages_id = -1;
711   in_process_deferred_messages = 0;
712   return FALSE;
713 }
714
715 static DBusHandlerResult
716 defer_message (DBusConnection *connection, DBusMessage *message, void *user_data)
717 {
718   BusDataClosure *closure = g_new (BusDataClosure, 1);
719   GList *new_list;
720
721   closure->bus = dbus_connection_ref (bus);
722   closure->message = dbus_message_ref (message);
723   closure->data = user_data;
724
725   new_list = g_list_append (deferred_messages, closure);
726   if (new_list)
727     deferred_messages = new_list;
728
729   if (process_deferred_messages_id == -1)
730     process_deferred_messages_id = g_idle_add (_atspi_process_deferred_messages, NULL);
731   return DBUS_HANDLER_RESULT_HANDLED;
732 }
733
734 static DBusHandlerResult
735 atspi_dbus_filter (DBusConnection *bus, DBusMessage *message, void *data)
736 {
737   int type = dbus_message_get_type (message);
738   const char *interface = dbus_message_get_interface (message);
739
740   if (type == DBUS_MESSAGE_TYPE_SIGNAL &&
741       !strncmp (interface, "org.a11y.atspi.Event.", 21))
742   {
743     return defer_message (bus, message, data);
744   }
745   if (dbus_message_is_method_call (message, atspi_interface_device_event_listener, "NotifyEvent"))
746   {
747     return defer_message (bus, message, data);
748   }
749   if (dbus_message_is_signal (message, atspi_interface_cache, "AddAccessible"))
750   {
751     return defer_message (bus, message, data);
752   }
753   if (dbus_message_is_signal (message, atspi_interface_cache, "RemoveAccessible"))
754   {
755     return defer_message (bus, message, data);
756   }
757   return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
758 }
759
760 static const char *signal_interfaces[] =
761 {
762   "org.a11y.atspi.Event.Object",
763   "org.a11y.atspi.Event.Window",
764   "org.a11y.atspi.Event.Mouse",
765   "org.a11y.atspi.Event.Terminal",
766   "org.a11y.atspi.Event.Document",
767   "org.a11y.atspi.Event.Focus",
768   NULL
769 };
770
771 /*
772  * Returns a 'canonicalized' value for DISPLAY,
773  * with the screen number stripped off if present.
774  *
775  * TODO: Avoid having duplicate functions for this here and in at-spi2-atk
776  */
777 static const gchar *
778 spi_display_name (void)
779 {
780   static const char *canonical_display_name = NULL;
781   if (!canonical_display_name)
782     {
783       const gchar *display_env = g_getenv ("AT_SPI_DISPLAY");
784       if (!display_env)
785         {
786           display_env = g_getenv ("DISPLAY");
787           if (!display_env || !display_env[0])
788             canonical_display_name = ":0";
789           else
790             {
791               gchar *display_p, *screen_p;
792               canonical_display_name = g_strdup (display_env);
793               display_p = g_utf8_strrchr (canonical_display_name, -1, ':');
794               screen_p = g_utf8_strrchr (canonical_display_name, -1, '.');
795               if (screen_p && display_p && (screen_p > display_p))
796                 {
797                   *screen_p = '\0';
798                 }
799             }
800         }
801       else
802         {
803           canonical_display_name = display_env;
804         }
805     }
806   return canonical_display_name;
807 }
808
809 /**
810  * atspi_init:
811  *
812  * Connects to the accessibility registry and initializes the SPI.
813  *
814  * Returns: 0 on success, otherwise an integer error code.  
815  **/
816 int
817 atspi_init (void)
818 {
819   DBusError error;
820   char *match;
821
822   if (atspi_inited)
823     {
824       return 1;
825     }
826
827   atspi_inited = TRUE;
828
829   g_type_init ();
830
831   get_live_refs();
832
833   dbus_error_init (&error);
834   bus = atspi_get_a11y_bus ();
835   if (!bus)
836     return 2;
837   dbus_bus_register (bus, &error);
838   atspi_dbus_connection_setup_with_g_main(bus, g_main_context_default());
839   dbus_connection_add_filter (bus, atspi_dbus_filter, NULL, NULL);
840   dbind_set_timeout (1000);
841   match = g_strdup_printf ("type='signal',interface='%s',member='AddAccessible'", atspi_interface_cache);
842   dbus_error_init (&error);
843   dbus_bus_add_match (bus, match, &error);
844   g_free (match);
845   match = g_strdup_printf ("type='signal',interface='%s',member='RemoveAccessible'", atspi_interface_cache);
846   dbus_bus_add_match (bus, match, &error);
847   g_free (match);
848   match = g_strdup_printf ("type='signal',interface='%s',member='ChildrenChanged'", atspi_interface_event_object);
849   dbus_bus_add_match (bus, match, &error);
850   g_free (match);
851   match = g_strdup_printf ("type='signal',interface='%s',member='PropertyChange'", atspi_interface_event_object);
852   dbus_bus_add_match (bus, match, &error);
853   g_free (match);
854   match = g_strdup_printf ("type='signal',interface='%s',member='StateChanged'", atspi_interface_event_object);
855   dbus_bus_add_match (bus, match, &error);
856   g_free (match);
857   return 0;
858 }
859
860   static GMainLoop *mainloop;
861
862 /**
863  * atspi_event_main:
864  *
865  * Starts/enters the main event loop for the AT-SPI services.
866  *
867  * (NOTE: This method does not return control, it is exited via a call to
868  *  atspi_event_quit () from within an event handler).
869  *
870  **/
871 void
872 atspi_event_main (void)
873 {
874   mainloop = g_main_loop_new (NULL, FALSE);
875   g_main_loop_run (mainloop);
876 }
877
878 /**
879  * atspi_event_quit:
880  *
881  * Quits the last main event loop for the SPI services,
882  * see atspi_event_main
883  **/
884 void
885 atspi_event_quit (void)
886 {
887   g_main_loop_quit (mainloop);
888 }
889
890 /**
891  * atspi_exit:
892  *
893  * Disconnects from the Accessibility Registry and releases 
894  * any floating resources. Call only once at exit.
895  *
896  * Returns: 0 if there were no leaks, otherwise non zero.
897  **/
898 int
899 atspi_exit (void)
900 {
901   int leaked;
902
903   if (!atspi_inited)
904     {
905       return 0;
906     }
907
908   atspi_inited = FALSE;
909
910   if (live_refs)
911     {
912       leaked = g_hash_table_size (live_refs);
913     }
914   else
915     {
916       leaked = 0;
917     }
918
919   cleanup ();
920
921   return leaked;
922 }
923
924 dbus_bool_t
925 _atspi_dbus_call (gpointer obj, const char *interface, const char *method, GError **error, const char *type, ...)
926 {
927   va_list args;
928   dbus_bool_t retval;
929   DBusError err;
930   AtspiObject *aobj = ATSPI_OBJECT (obj);
931
932   if (!aobj->app || !aobj->app->bus)
933   {
934     g_set_error_literal (error, ATSPI_ERROR, ATSPI_ERROR_APPLICATION_GONE,
935                           _("The application no longer exists"));
936     return FALSE;
937   }
938
939   va_start (args, type);
940   dbus_error_init (&err);
941   retval = dbind_method_call_reentrant_va (aobj->app->bus, aobj->app->bus_name,
942                                            aobj->path, interface, method, &err,
943                                            type, args);
944   va_end (args);
945   _atspi_process_deferred_messages ((gpointer)TRUE);
946   if (dbus_error_is_set (&err))
947   {
948     /* TODO: Set gerror */
949     dbus_error_free (&err);
950   }
951   return retval;
952 }
953
954 DBusMessage *
955 _atspi_dbus_call_partial (gpointer obj,
956                           const char *interface,
957                           const char *method,
958                           GError **error,
959                           const char *type, ...)
960 {
961   va_list args;
962
963   va_start (args, type);
964   return _atspi_dbus_call_partial_va (obj, interface, method, error, type, args);
965 }
966
967 DBusMessage *
968 _atspi_dbus_call_partial_va (gpointer obj,
969                           const char *interface,
970                           const char *method,
971                           GError **error,
972                           const char *type,
973                           va_list args)
974 {
975   AtspiObject *aobj = ATSPI_OBJECT (obj);
976   DBusError err;
977     DBusMessage *msg = NULL, *reply = NULL;
978     DBusMessageIter iter;
979     const char *p;
980
981   dbus_error_init (&err);
982
983   if (!aobj->app || !aobj->app->bus)
984   {
985     g_set_error_literal (error, ATSPI_ERROR, ATSPI_ERROR_APPLICATION_GONE,
986                           _("The application no longer exists"));
987     goto out;
988   }
989
990     msg = dbus_message_new_method_call (aobj->app->bus_name, aobj->path, interface, method);
991     if (!msg)
992         goto out;
993
994     p = type;
995     dbus_message_iter_init_append (msg, &iter);
996     dbind_any_marshal_va (&iter, &p, args);
997
998     reply = dbind_send_and_allow_reentry (aobj->app->bus, msg, &err);
999 out:
1000   va_end (args);
1001   if (msg)
1002     dbus_message_unref (msg);
1003   _atspi_process_deferred_messages ((gpointer)TRUE);
1004   if (dbus_error_is_set (&err))
1005   {
1006     /* TODO: Set gerror */
1007     dbus_error_free (&err);
1008   }
1009   return reply;
1010 }
1011
1012 dbus_bool_t
1013 _atspi_dbus_get_property (gpointer obj, const char *interface, const char *name, GError **error, const char *type, void *data)
1014 {
1015   DBusMessage *message, *reply;
1016   DBusMessageIter iter, iter_variant;
1017   DBusError err;
1018   dbus_bool_t retval = FALSE;
1019   AtspiObject *aobj = ATSPI_OBJECT (obj);
1020
1021   if (!aobj)
1022     return FALSE;
1023
1024   if (!aobj->app || !aobj->app->bus)
1025   {
1026     g_set_error_literal (error, ATSPI_ERROR, ATSPI_ERROR_APPLICATION_GONE,
1027                           _("The application no longer exists"));
1028     return FALSE;
1029   }
1030
1031   message = dbus_message_new_method_call (aobj->app->bus_name,
1032                                           aobj->path,
1033                                           "org.freedesktop.DBus.Properties",
1034                                           "Get");
1035   if (!message)
1036   {
1037     // TODO: throw exception
1038     return FALSE;
1039   }
1040   dbus_message_append_args (message, DBUS_TYPE_STRING, &interface, DBUS_TYPE_STRING, &name, DBUS_TYPE_INVALID);
1041   dbus_error_init (&err);
1042   reply = dbind_send_and_allow_reentry (aobj->app->bus, message, &err);
1043   dbus_message_unref (message);
1044   _atspi_process_deferred_messages ((gpointer)TRUE);
1045   if (!reply)
1046   {
1047     // TODO: throw exception
1048     goto done;
1049   }
1050
1051   if (dbus_message_get_type (reply) == DBUS_MESSAGE_TYPE_ERROR)
1052   {
1053     const char *err;
1054     dbus_message_get_args (message, NULL, DBUS_TYPE_STRING, &err, DBUS_TYPE_INVALID);
1055     if (err)
1056       g_set_error_literal (error, ATSPI_ERROR, ATSPI_ERROR_IPC, err);
1057     goto done;
1058   }
1059
1060   dbus_message_iter_init (reply, &iter);
1061   if (dbus_message_iter_get_arg_type (&iter) != 'v')
1062   {
1063     g_warning ("AT-SPI: expected a variant when fetching %s from interface %s; got %s\n", name, interface, dbus_message_get_signature (reply));
1064     goto done;
1065   }
1066   dbus_message_iter_recurse (&iter, &iter_variant);
1067   if (dbus_message_iter_get_arg_type (&iter_variant) != type[0])
1068   {
1069     g_warning ("atspi_dbus_get_property: Wrong type: expected %s, got %c\n", type, dbus_message_iter_get_arg_type (&iter_variant));
1070     goto done;
1071   }
1072   if (!strcmp (type, "(so)"))
1073   {
1074     *((AtspiAccessible **)data) = _atspi_dbus_return_accessible_from_iter (&iter_variant);
1075   }
1076   else
1077   {
1078     dbus_message_iter_get_basic (&iter_variant, data);
1079     if (type [0] == 's')
1080       *(char **)data = g_strdup (*(char **)data);
1081   }
1082   retval = TRUE;
1083 done:
1084   if (reply)
1085     dbus_message_unref (reply);
1086   return retval;
1087 }
1088
1089 DBusMessage *
1090 _atspi_dbus_send_with_reply_and_block (DBusMessage *message, GError **error)
1091 {
1092   DBusMessage *reply;
1093   DBusError err;
1094   AtspiApplication *app;
1095   DBusConnection *bus;
1096
1097   app = get_application (dbus_message_get_destination (message));
1098
1099   if (app && !app->bus)
1100     return NULL;        /* will fail anyway; app has been disposed */
1101
1102   bus = (app ? app->bus : _atspi_bus());
1103   dbus_error_init (&err);
1104   reply = dbind_send_and_allow_reentry (bus, message, &err);
1105   _atspi_process_deferred_messages ((gpointer)TRUE);
1106   dbus_message_unref (message);
1107   if (err.message)
1108   {
1109     if (error)
1110       g_set_error_literal (error, ATSPI_ERROR, ATSPI_ERROR_IPC, err.message);
1111     dbus_error_free (&err);
1112   }
1113   return reply;
1114 }
1115
1116 GHashTable *
1117 _atspi_dbus_return_hash_from_message (DBusMessage *message)
1118 {
1119   DBusMessageIter iter;
1120   GHashTable *ret;
1121
1122   if (!message)
1123     return NULL;
1124
1125   _ATSPI_DBUS_CHECK_SIG (message, "a{ss}", NULL, NULL);
1126
1127   dbus_message_iter_init (message, &iter);
1128   ret = _atspi_dbus_hash_from_iter (&iter);
1129   dbus_message_unref (message);
1130   return ret;
1131 }
1132
1133 GHashTable *
1134 _atspi_dbus_hash_from_iter (DBusMessageIter *iter)
1135 {
1136   GHashTable *hash = g_hash_table_new (g_str_hash, g_str_equal);
1137   DBusMessageIter iter_array, iter_dict;
1138
1139   dbus_message_iter_recurse (iter, &iter_array);
1140   while (dbus_message_iter_get_arg_type (&iter_array) != DBUS_TYPE_INVALID)
1141   {
1142     const char *name, *value;
1143     dbus_message_iter_recurse (&iter_array, &iter_dict);
1144     dbus_message_iter_get_basic (&iter_dict, &name);
1145     dbus_message_iter_next (&iter_dict);
1146     dbus_message_iter_get_basic (&iter_dict, &value);
1147     g_hash_table_insert (hash, g_strdup (name), g_strdup (value));
1148     dbus_message_iter_next (&iter_array);
1149   }
1150   return hash;
1151 }
1152
1153 GArray *
1154 _atspi_dbus_return_attribute_array_from_message (DBusMessage *message)
1155 {
1156   DBusMessageIter iter;
1157   GArray *ret;
1158
1159   if (!message)
1160     return NULL;
1161
1162   _ATSPI_DBUS_CHECK_SIG (message, "a{ss}", NULL, NULL);
1163
1164   dbus_message_iter_init (message, &iter);
1165
1166   ret = _atspi_dbus_attribute_array_from_iter (&iter);
1167   dbus_message_unref (message);
1168   return ret;
1169 }
1170
1171 GArray *
1172 _atspi_dbus_attribute_array_from_iter (DBusMessageIter *iter)
1173 {
1174   DBusMessageIter iter_array, iter_dict;
1175   GArray *array = g_array_new (TRUE, TRUE, sizeof (gchar *));
1176
1177   dbus_message_iter_recurse (iter, &iter_array);
1178   while (dbus_message_iter_get_arg_type (&iter_array) != DBUS_TYPE_INVALID)
1179   {
1180     const char *name, *value;
1181     gchar *str;
1182     GArray *new_array;
1183     dbus_message_iter_recurse (&iter_array, &iter_dict);
1184     dbus_message_iter_get_basic (&iter_dict, &name);
1185     dbus_message_iter_next (&iter_dict);
1186     dbus_message_iter_get_basic (&iter_dict, &value);
1187     str = g_strdup_printf ("%s:%s", name, value);
1188     new_array = g_array_append_val (array, str);
1189     if (new_array)
1190       array = new_array;
1191     dbus_message_iter_next (&iter_array);;
1192   }
1193   return array;
1194 }
1195
1196 void
1197 _atspi_dbus_set_interfaces (AtspiAccessible *accessible, DBusMessageIter *iter)
1198 {
1199   DBusMessageIter iter_array;
1200
1201   accessible->interfaces = 0;
1202   dbus_message_iter_recurse (iter, &iter_array);
1203   while (dbus_message_iter_get_arg_type (&iter_array) != DBUS_TYPE_INVALID)
1204   {
1205     const char *iface;
1206     gint n;
1207     dbus_message_iter_get_basic (&iter_array, &iface);
1208     if (!strcmp (iface, "org.freedesktop.DBus.Introspectable")) continue;
1209     n = _atspi_get_iface_num (iface);
1210     if (n == -1)
1211     {
1212       g_warning ("AT-SPI: Unknown interface %s", iface);
1213     }
1214     else
1215       accessible->interfaces |= (1 << n);
1216     dbus_message_iter_next (&iter_array);
1217   }
1218   _atspi_accessible_add_cache (accessible, ATSPI_CACHE_INTERFACES);
1219 }
1220
1221 void
1222 _atspi_dbus_set_state (AtspiAccessible *accessible, DBusMessageIter *iter)
1223 {
1224   DBusMessageIter iter_array;
1225   gint count;
1226   dbus_uint32_t *states;
1227
1228   dbus_message_iter_recurse (iter, &iter_array);
1229   dbus_message_iter_get_fixed_array (&iter_array, &states, &count);
1230   if (count != 2)
1231   {
1232     g_warning ("AT-SPI: expected 2 values in states array; got %d\n", count);
1233     if (!accessible->states)
1234       accessible->states = _atspi_state_set_new_internal (accessible, 0);
1235   }
1236   else
1237   {
1238     guint64 val = ((guint64)states [1]) << 32;
1239     val += states [0];
1240     if (!accessible->states)
1241       accessible->states = _atspi_state_set_new_internal (accessible, val);
1242     else
1243       accessible->states->states = val;
1244   }
1245   _atspi_accessible_add_cache (accessible, ATSPI_CACHE_STATES);
1246 }
1247
1248 GQuark
1249 atspi_error_quark (void)
1250 {
1251   return g_quark_from_static_string ("atspi_error");
1252 }
1253
1254 /*
1255  * Gets the IOR from the XDisplay.
1256  */
1257 static char *
1258 get_accessibility_bus_address_x11 (void)
1259 {
1260   Atom AT_SPI_BUS;
1261   Atom actual_type;
1262   Display *bridge_display;
1263   int actual_format;
1264   unsigned char *data = NULL;
1265   unsigned long nitems;
1266   unsigned long leftover;
1267
1268   bridge_display = XOpenDisplay (spi_display_name ());
1269   if (!bridge_display)
1270     {
1271       g_warning ("Could not open X display");
1272       return NULL;
1273     }
1274       
1275   AT_SPI_BUS = XInternAtom (bridge_display, "AT_SPI_BUS", False);
1276   XGetWindowProperty (bridge_display,
1277                       XDefaultRootWindow (bridge_display),
1278                       AT_SPI_BUS, 0L,
1279                       (long) BUFSIZ, False,
1280                       (Atom) 31, &actual_type, &actual_format,
1281                       &nitems, &leftover, &data);
1282   XCloseDisplay (bridge_display);
1283
1284   return g_strdup (data);
1285 }
1286
1287 static char *
1288 get_accessibility_bus_address_dbus (void)
1289 {
1290   DBusConnection *session_bus = NULL;
1291   DBusMessage *message;
1292   DBusMessage *reply;
1293   char *address = NULL;
1294
1295   session_bus = dbus_bus_get (DBUS_BUS_SESSION, NULL);
1296   if (!session_bus)
1297     return NULL;
1298
1299   message = dbus_message_new_method_call ("org.a11y.Bus",
1300                                           "/org/a11y/bus",
1301                                           "org.a11y.Bus",
1302                                           "GetAddress");
1303
1304   reply = dbus_connection_send_with_reply_and_block (session_bus,
1305                                                      message,
1306                                                      -1,
1307                                                      NULL);
1308   dbus_message_unref (message);
1309
1310   if (!reply)
1311     return NULL;
1312   
1313   {
1314     const char *tmp_address;
1315     if (!dbus_message_get_args (reply,
1316                                 NULL,
1317                                 DBUS_TYPE_STRING,
1318                                 &tmp_address,
1319                                 DBUS_TYPE_INVALID))
1320       {
1321         dbus_message_unref (reply);
1322         return NULL;
1323       }
1324     address = g_strdup (tmp_address);
1325     dbus_message_unref (reply);
1326   }
1327   
1328   return address;
1329 }
1330
1331 DBusConnection *
1332 atspi_get_a11y_bus (void)
1333 {
1334   DBusConnection *bus = NULL;
1335   DBusError error;
1336   char *address;
1337
1338   address = get_accessibility_bus_address_x11 ();
1339   if (!address)
1340     address = get_accessibility_bus_address_dbus ();
1341   if (!address)
1342     return NULL;
1343
1344   dbus_error_init (&error);
1345   bus = dbus_connection_open (address, &error);
1346   g_free (address);
1347
1348   if (!bus)
1349     {
1350       g_warning ("Couldn't connect to accessibility bus: %s", error.message);
1351       return NULL;
1352     }
1353   else
1354     {
1355       if (!dbus_bus_register (bus, &error))
1356         {
1357           g_warning ("Couldn't register with accessibility bus: %s", error.message);
1358           return NULL;
1359         }
1360     }
1361   
1362   return bus;
1363 }