Fix some ref count issues
[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 static gint method_call_timeout = 800;
42 static gint app_startup_time = 15000;
43
44 GMainLoop *atspi_main_loop;
45 gboolean atspi_no_cache;
46
47 const char *atspi_path_dec = ATSPI_DBUS_PATH_DEC;
48 const char *atspi_path_registry = ATSPI_DBUS_PATH_REGISTRY;
49 const char *atspi_path_root = ATSPI_DBUS_PATH_ROOT;
50 const char *atspi_bus_registry = ATSPI_DBUS_NAME_REGISTRY;
51 const char *atspi_interface_accessible = ATSPI_DBUS_INTERFACE_ACCESSIBLE;
52 const char *atspi_interface_action = ATSPI_DBUS_INTERFACE_ACTION;
53 const char *atspi_interface_application = ATSPI_DBUS_INTERFACE_APPLICATION;
54 const char *atspi_interface_collection = ATSPI_DBUS_INTERFACE_COLLECTION;
55 const char *atspi_interface_component = ATSPI_DBUS_INTERFACE_COMPONENT;
56 const char *atspi_interface_dec = ATSPI_DBUS_INTERFACE_DEC;
57 const char *atspi_interface_device_event_listener = ATSPI_DBUS_INTERFACE_DEVICE_EVENT_LISTENER;
58 const char *atspi_interface_document = ATSPI_DBUS_INTERFACE_DOCUMENT;
59 const char *atspi_interface_editable_text = ATSPI_DBUS_INTERFACE_EDITABLE_TEXT;
60 const char *atspi_interface_event_object = ATSPI_DBUS_INTERFACE_EVENT_OBJECT;
61 const char *atspi_interface_hyperlink = ATSPI_DBUS_INTERFACE_HYPERLINK;
62 const char *atspi_interface_hypertext = ATSPI_DBUS_INTERFACE_HYPERTEXT;
63 const char *atspi_interface_image = ATSPI_DBUS_INTERFACE_IMAGE;
64 const char *atspi_interface_registry = ATSPI_DBUS_INTERFACE_REGISTRY;
65 const char *atspi_interface_selection = ATSPI_DBUS_INTERFACE_SELECTION;
66 const char *atspi_interface_table = ATSPI_DBUS_INTERFACE_TABLE;
67 const char *atspi_interface_text = ATSPI_DBUS_INTERFACE_TEXT;
68 const char *atspi_interface_cache = ATSPI_DBUS_INTERFACE_CACHE;
69 const char *atspi_interface_value = ATSPI_DBUS_INTERFACE_VALUE;
70
71 static const char *interfaces[] =
72 {
73   ATSPI_DBUS_INTERFACE_ACCESSIBLE,
74   ATSPI_DBUS_INTERFACE_ACTION,
75   ATSPI_DBUS_INTERFACE_APPLICATION,
76   ATSPI_DBUS_INTERFACE_COLLECTION,
77   ATSPI_DBUS_INTERFACE_COMPONENT,
78   ATSPI_DBUS_INTERFACE_DOCUMENT,
79   ATSPI_DBUS_INTERFACE_EDITABLE_TEXT,
80   ATSPI_DBUS_INTERFACE_HYPERLINK,
81   ATSPI_DBUS_INTERFACE_HYPERTEXT,
82   ATSPI_DBUS_INTERFACE_IMAGE,
83   "org.a11y.atspi.LoginHelper",
84   ATSPI_DBUS_INTERFACE_SELECTION,
85   ATSPI_DBUS_INTERFACE_TABLE,
86   ATSPI_DBUS_INTERFACE_TEXT,
87   ATSPI_DBUS_INTERFACE_VALUE,
88   NULL
89 };
90
91 gint
92 _atspi_get_iface_num (const char *iface)
93 {
94   /* TODO: Use a binary search or hash to improve performance */
95   int i;
96
97   for (i = 0; interfaces[i]; i++)
98   {
99     if (!strcmp(iface, interfaces[i])) return i;
100   }
101   return -1;
102 }
103
104 static GHashTable *
105 get_live_refs (void)
106 {
107   if (!live_refs) 
108     {
109       live_refs = g_hash_table_new (g_direct_hash, g_direct_equal);
110     }
111   return live_refs;
112 }
113
114 /* TODO: Add an application parameter */
115 DBusConnection *
116 _atspi_bus ()
117 {
118   if (!bus)
119     atspi_init ();
120   if (!bus)
121     g_error ("AT-SPI: COuldn't connect to accessibility bus. Is at-spi-bus-launcher running?");
122   return bus;
123 }
124
125 #define APP_IS_REGISTRY(app) (!strcmp (app->bus_name, atspi_bus_registry))
126
127 static AtspiAccessible *desktop;
128
129 static void
130 cleanup ()
131 {
132   GHashTable *refs;
133   GList *l;
134
135   refs = live_refs;
136   live_refs = NULL;
137   if (refs)
138     {
139       g_hash_table_destroy (refs);
140     }
141
142   if (bus)
143     {
144       dbus_connection_close (bus);
145       dbus_connection_unref (bus);
146       bus = NULL;
147     }
148
149   if (!desktop)
150     return;
151   for (l = desktop->children; l;)
152   {
153     GList *next = l->next;
154     AtspiAccessible *child = l->data;
155     g_object_run_dispose (G_OBJECT (child->parent.app));
156     g_object_run_dispose (G_OBJECT (child));
157     l = next;
158   }
159   g_object_run_dispose (G_OBJECT (desktop->parent.app));
160   g_object_unref (desktop);
161   desktop = NULL;
162 }
163
164 static gboolean atspi_inited = FALSE;
165
166 static GHashTable *app_hash = NULL;
167
168 static void
169 handle_get_bus_address (DBusPendingCall *pending, void *user_data)
170 {
171   AtspiApplication *app = user_data;
172   DBusMessage *reply = dbus_pending_call_steal_reply (pending);
173   DBusMessage *message;
174   const char *address;
175   DBusPendingCall *new_pending;
176
177   if (dbus_message_get_type (reply) == DBUS_MESSAGE_TYPE_METHOD_RETURN)
178   {
179     if (dbus_message_get_args (reply, NULL, DBUS_TYPE_STRING, &address,
180                                DBUS_TYPE_INVALID) && address [0])
181     {
182       DBusError error;
183       DBusConnection *bus;
184
185       dbus_error_init (&error);
186       bus = dbus_connection_open_private (address, &error);
187       if (bus)
188       {
189         if (app->bus)
190           {
191             dbus_connection_unref (app->bus);
192           }
193         app->bus = bus;
194       }
195       else
196       {
197         g_warning ("Unable to open bus connection: %s", error.message);
198         dbus_error_free (&error);
199       }
200     }
201   }
202   dbus_message_unref (reply);
203   dbus_pending_call_unref (pending);
204
205   if (!app->bus)
206     return; /* application has gone away / been disposed */
207
208   message = dbus_message_new_method_call (app->bus_name,
209                                           "/org/a11y/atspi/cache",
210                                           atspi_interface_cache, "GetItems");
211
212    dbus_connection_send_with_reply (app->bus, message, &new_pending, 2000);
213   dbus_pending_call_set_notify (new_pending, handle_get_items, app, NULL);
214   dbus_message_unref (message);
215 }
216
217 static AtspiApplication *
218 get_application (const char *bus_name)
219 {
220   AtspiApplication *app = NULL;
221   char *bus_name_dup;
222   DBusMessage *message;
223   DBusPendingCall *pending = NULL;
224
225   if (!app_hash)
226   {
227     app_hash = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, (GDestroyNotify)g_object_unref);
228     if (!app_hash) return NULL;
229   }
230   app = g_hash_table_lookup (app_hash, bus_name);
231   if (app) return app;
232   bus_name_dup = g_strdup (bus_name);
233   if (!bus_name_dup) return NULL;
234   // TODO: change below to something that will send state-change:defunct notification if necessary */
235   app = _atspi_application_new (bus_name);
236   app->hash = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, g_object_unref);
237   app->bus = dbus_connection_ref (_atspi_bus ());
238   gettimeofday (&app->time_added, NULL);
239   app->cache = ATSPI_CACHE_UNDEFINED;
240   g_hash_table_insert (app_hash, bus_name_dup, app);
241   message = dbus_message_new_method_call (bus_name, atspi_path_root,
242                                           atspi_interface_application, "GetApplicationBusAddress");
243
244    dbus_connection_send_with_reply (app->bus, message, &pending, 2000);
245   dbus_pending_call_set_notify (pending, handle_get_bus_address, app, NULL);
246   dbus_message_unref (message);
247   return app;
248 }
249
250 static AtspiAccessible *
251 ref_accessible (const char *app_name, const char *path)
252 {
253   AtspiApplication *app;
254   AtspiAccessible *a;
255
256   if (!strcmp (path, ATSPI_DBUS_PATH_NULL))
257     return NULL;
258
259   app = get_application (app_name);
260
261   if (!strcmp (path, "/org/a11y/atspi/accessible/root"))
262   {
263     if (!app->root)
264     {
265       app->root = _atspi_accessible_new (app, atspi_path_root);
266       app->root->accessible_parent = atspi_get_desktop (0);
267       app->root->accessible_parent->children = g_list_append (app->root->accessible_parent->children, g_object_ref (app->root));
268     }
269     return g_object_ref (app->root);
270   }
271
272   a = g_hash_table_lookup (app->hash, path);
273   if (a)
274   {
275     return g_object_ref (a);
276   }
277   a = _atspi_accessible_new (app, path);
278   if (!a)
279     return NULL;
280   g_hash_table_insert (app->hash, g_strdup (a->parent.path), g_object_ref (a));
281   return a;
282 }
283
284 static AtspiHyperlink *
285 ref_hyperlink (const char *app_name, const char *path)
286 {
287   AtspiApplication *app = get_application (app_name);
288   AtspiHyperlink *hyperlink;
289
290   if (!strcmp (path, ATSPI_DBUS_PATH_NULL))
291     return NULL;
292
293   hyperlink = g_hash_table_lookup (app->hash, path);
294   if (hyperlink)
295   {
296     return g_object_ref (hyperlink);
297   }
298   hyperlink = _atspi_hyperlink_new (app, path);
299   g_hash_table_insert (app->hash, g_strdup (hyperlink->parent.path), hyperlink);
300   /* TODO: This should be a weak ref */
301   g_object_ref (hyperlink);     /* for the hash */
302   return hyperlink;
303 }
304
305 typedef struct
306 {
307   char *path;
308   char *parent;
309   GArray *children;
310   GArray *interfaces;
311   char *name;
312   dbus_uint32_t role;
313   char *description;
314   GArray *state_bitflags;
315 } CACHE_ADDITION;
316
317 static DBusHandlerResult
318 handle_remove_accessible (DBusConnection *bus, DBusMessage *message, void *user_data)
319 {
320   const char *sender = dbus_message_get_sender (message);
321   AtspiApplication *app;
322   const char *path;
323   DBusMessageIter iter, iter_struct;
324   const char *signature = dbus_message_get_signature (message);
325   AtspiAccessible *a;
326
327   if (strcmp (signature, "(so)") != 0)
328   {
329     g_warning ("AT-SPI: Unknown signature %s for RemoveAccessible", signature);
330     return DBUS_HANDLER_RESULT_HANDLED;
331   }
332
333   dbus_message_iter_init (message, &iter);
334   dbus_message_iter_recurse (&iter, &iter_struct);
335   dbus_message_iter_get_basic (&iter_struct, &sender);
336   dbus_message_iter_next (&iter_struct);
337   dbus_message_iter_get_basic (&iter_struct, &path);
338   app = get_application (sender);
339   a = ref_accessible (sender, path);
340   if (!a)
341     return DBUS_HANDLER_RESULT_HANDLED;
342   g_object_run_dispose (G_OBJECT (a));
343   g_hash_table_remove (app->hash, a->parent.path);
344   g_object_unref (a);   /* unref our own ref */
345   return DBUS_HANDLER_RESULT_HANDLED;
346 }
347
348 static DBusHandlerResult
349 handle_name_owner_changed (DBusConnection *bus, DBusMessage *message, void *user_data)
350 {
351   const char *name, *new, *old;
352   static gboolean registry_lost = FALSE;
353
354   if (!dbus_message_get_args (message, NULL,
355                               DBUS_TYPE_STRING, &name,
356                               DBUS_TYPE_STRING, &old,
357                               DBUS_TYPE_STRING, &new,
358                               DBUS_TYPE_INVALID))
359   {
360     return;
361   }
362
363   if (!strcmp (name, "org.a11y.atspi.Registry"))
364   {
365     if (registry_lost && !old[0])
366     {
367       _atspi_reregister_event_listeners ();
368       _atspi_reregister_device_listeners ();
369       registry_lost = FALSE;
370     }
371     else if (!new[0])
372       registry_lost = TRUE;
373   }
374   else
375   {
376     AtspiAccessible *desktop = atspi_get_desktop (0);
377     GList *l;
378     for (l = desktop->children; l; l = l->next)
379     {
380       AtspiAccessible *child = l->data;
381       if (!strcmp (child->parent.app->bus_name, old))
382         g_object_run_dispose (G_OBJECT (child->parent.app));
383     }
384     g_object_unref (desktop);
385   }
386   return DBUS_HANDLER_RESULT_HANDLED;
387 }
388
389 static gboolean
390 add_app_to_desktop (AtspiAccessible *a, const char *bus_name)
391 {
392   AtspiAccessible *obj = ref_accessible (bus_name, atspi_path_root);
393   /* The app will be added to the desktop as a side-effect of calling
394    * ref_accessible */
395   g_object_unref (obj);
396   return (obj != NULL);
397 }
398
399 static void
400 send_children_changed (AtspiAccessible *parent, AtspiAccessible *child, gboolean add)
401 {
402   AtspiEvent e;
403
404   memset (&e, 0, sizeof (e));
405   e.type = (add? "object:children-changed:add": "object:children-changed:remove");
406   e.source = parent;
407   e.detail1 = g_list_index (parent->children, child);
408   e.detail2 = 0;
409   _atspi_send_event (&e);
410 }
411
412 static void
413 unref_object_and_descendants (AtspiAccessible *obj)
414 {
415   GList *l;
416
417   for (l = obj->children; l; l = l->next)
418   {
419     unref_object_and_descendants (l->data);
420   }
421   g_object_unref (obj);
422 }
423
424 static gboolean
425 remove_app_from_desktop (AtspiAccessible *a, const char *bus_name)
426 {
427   GList *l;
428   AtspiAccessible *child;
429
430   for (l = a->children; l; l = l->next)
431   {
432     child = l->data;
433     if (!strcmp (bus_name, child->parent.app->bus_name)) break;
434   }
435   if (!l)
436   {
437     return FALSE;
438   }
439   send_children_changed (a, child, FALSE);
440   a->children = g_list_remove (a->children, child);
441   unref_object_and_descendants (child);
442   return TRUE;
443 }
444
445 void
446 get_reference_from_iter (DBusMessageIter *iter, const char **app_name, const char **path)
447 {
448   DBusMessageIter iter_struct;
449
450   dbus_message_iter_recurse (iter, &iter_struct);
451   dbus_message_iter_get_basic (&iter_struct, app_name);
452   dbus_message_iter_next (&iter_struct);
453   dbus_message_iter_get_basic (&iter_struct, path);
454   dbus_message_iter_next (iter);
455 }
456
457 static void
458 add_accessible_from_iter (DBusMessageIter *iter)
459 {
460   DBusMessageIter iter_struct, iter_array;
461   const char *app_name, *path;
462   AtspiAccessible *accessible;
463   const char *name, *description;
464   dbus_uint32_t role;
465
466   dbus_message_iter_recurse (iter, &iter_struct);
467
468   /* get accessible */
469   get_reference_from_iter (&iter_struct, &app_name, &path);
470   accessible = ref_accessible (app_name, path);
471   if (!accessible)
472     return;
473
474   /* Get application: TODO */
475   dbus_message_iter_next (&iter_struct);
476
477   /* get parent */
478   get_reference_from_iter (&iter_struct, &app_name, &path);
479   if (accessible->accessible_parent)
480     g_object_unref (accessible->accessible_parent);
481   accessible->accessible_parent = ref_accessible (app_name, path);
482
483   /* Get children */
484   while (accessible->children)
485   {
486     g_object_unref (accessible->children->data);
487     accessible->children = g_list_remove (accessible->children, accessible->children->data);
488   }
489   dbus_message_iter_recurse (&iter_struct, &iter_array);
490   while (dbus_message_iter_get_arg_type (&iter_array) != DBUS_TYPE_INVALID)
491   {
492     AtspiAccessible *child;
493     get_reference_from_iter (&iter_array, &app_name, &path);
494     child = ref_accessible (app_name, path);
495     accessible->children = g_list_append (accessible->children, child);
496   }
497
498   /* interfaces */
499   dbus_message_iter_next (&iter_struct);
500   _atspi_dbus_set_interfaces (accessible, &iter_struct);
501   dbus_message_iter_next (&iter_struct);
502
503   /* name */
504   if (accessible->name)
505     g_free (accessible->name);
506   dbus_message_iter_get_basic (&iter_struct, &name);
507   accessible->name = g_strdup (name);
508   dbus_message_iter_next (&iter_struct);
509
510   /* role */
511   dbus_message_iter_get_basic (&iter_struct, &role);
512   accessible->role = role;
513   dbus_message_iter_next (&iter_struct);
514
515   /* description */
516   if (accessible->description)
517     g_free (accessible->description);
518   dbus_message_iter_get_basic (&iter_struct, &description);
519   accessible->description = g_strdup (description);
520   dbus_message_iter_next (&iter_struct);
521
522   _atspi_dbus_set_state (accessible, &iter_struct);
523   dbus_message_iter_next (&iter_struct);
524
525   _atspi_accessible_add_cache (accessible, ATSPI_CACHE_NAME | ATSPI_CACHE_ROLE |
526                                ATSPI_CACHE_PARENT | ATSPI_CACHE_DESCRIPTION);
527   if (!atspi_state_set_contains (accessible->states,
528                                        ATSPI_STATE_MANAGES_DESCENDANTS))
529     _atspi_accessible_add_cache (accessible, ATSPI_CACHE_CHILDREN);
530
531   /* This is a bit of a hack since the cache holds a ref, so we don't need
532    * the one provided for us anymore */
533   g_object_unref (accessible);
534 }
535
536 static void
537 handle_get_items (DBusPendingCall *pending, void *user_data)
538 {
539   DBusMessage *reply = dbus_pending_call_steal_reply (pending);
540   DBusMessageIter iter, iter_array;
541
542   if (dbus_message_get_type (reply) == DBUS_MESSAGE_TYPE_ERROR)
543   {
544     const char *sender = dbus_message_get_sender (reply);
545     const char *error = NULL;
546     dbus_message_get_args (reply, NULL, DBUS_TYPE_STRING, &error,
547                            DBUS_TYPE_INVALID);
548     g_warning ("AT-SPI: Error in GetItems, sender=%s, error=%s", sender, error);
549     dbus_message_unref (reply);
550     dbus_pending_call_unref (pending);
551     return;
552   }
553
554   dbus_message_iter_init (reply, &iter);
555   dbus_message_iter_recurse (&iter, &iter_array);
556   while (dbus_message_iter_get_arg_type (&iter_array) != DBUS_TYPE_INVALID)
557   {
558     add_accessible_from_iter (&iter_array);
559     dbus_message_iter_next (&iter_array);
560   }
561   dbus_message_unref (reply);
562   dbus_pending_call_unref (pending);
563 }
564
565 /* TODO: Do we stil need this function? */
566 static AtspiAccessible *
567 ref_accessible_desktop (AtspiApplication *app)
568 {
569   GError *error;
570   DBusMessage *message, *reply;
571   DBusMessageIter iter, iter_array;
572   gchar *bus_name_dup;
573
574   if (desktop)
575   {
576     g_object_ref (desktop);
577     return desktop;
578   }
579   desktop = _atspi_accessible_new (app, atspi_path_root);
580   if (!desktop)
581   {
582     return NULL;
583   }
584   g_hash_table_insert (app->hash, g_strdup (desktop->parent.path),
585                        g_object_ref (desktop));
586   app->root = g_object_ref (desktop);
587   desktop->name = g_strdup ("main");
588   message = dbus_message_new_method_call (atspi_bus_registry,
589         atspi_path_root,
590         atspi_interface_accessible,
591         "GetChildren");
592   if (!message)
593     return NULL;
594   error = NULL;
595   reply = _atspi_dbus_send_with_reply_and_block (message, &error);
596   if (!reply || strcmp (dbus_message_get_signature (reply), "a(so)") != 0)
597   {
598     if (error != NULL)
599     {
600       g_warning ("Couldn't get application list: %s", error->message);
601       g_clear_error (&error);
602     }
603     if (reply)
604       dbus_message_unref (reply);
605     return NULL;
606   }
607   dbus_message_iter_init (reply, &iter);
608   dbus_message_iter_recurse (&iter, &iter_array);
609   while (dbus_message_iter_get_arg_type (&iter_array) != DBUS_TYPE_INVALID)
610   {
611     const char *app_name, *path;
612     get_reference_from_iter (&iter_array, &app_name, &path);
613     add_app_to_desktop (desktop, app_name);
614   }
615   dbus_message_unref (reply);
616
617   /* Record the alternate name as an alias for org.a11y.atspi.Registry */
618   bus_name_dup = g_strdup (dbus_message_get_sender (reply));
619   if (bus_name_dup)
620     g_hash_table_insert (app_hash, bus_name_dup, app);
621
622   return g_object_ref (desktop);
623 }
624
625 AtspiAccessible *
626 _atspi_ref_accessible (const char *app, const char *path)
627 {
628   AtspiApplication *a = get_application (app);
629   if (!a)
630     return NULL;
631   if ( APP_IS_REGISTRY(a))
632   {
633     if (!a->root)
634       g_object_unref (ref_accessible_desktop (a));      /* sets a->root */
635     return g_object_ref (a->root);
636   }
637   return ref_accessible (app, path);
638 }
639
640 AtspiAccessible *
641 _atspi_dbus_return_accessible_from_message (DBusMessage *message)
642 {
643   DBusMessageIter iter;
644   AtspiAccessible *retval = NULL;
645   const char *signature;
646
647   if (!message)
648     return NULL;
649
650   signature = dbus_message_get_signature (message);
651   if (!strcmp (signature, "(so)"))
652   {
653     dbus_message_iter_init (message, &iter);
654     retval =  _atspi_dbus_return_accessible_from_iter (&iter);
655   }
656   else
657   {
658     g_warning ("AT-SPI: Called _atspi_dbus_return_accessible_from_message with strange signature %s", signature);
659   }
660   dbus_message_unref (message);
661   return retval;
662 }
663
664 AtspiAccessible *
665 _atspi_dbus_return_accessible_from_iter (DBusMessageIter *iter)
666 {
667   const char *app_name, *path;
668
669   get_reference_from_iter (iter, &app_name, &path);
670   return ref_accessible (app_name, path);
671 }
672
673 AtspiHyperlink *
674 _atspi_dbus_return_hyperlink_from_message (DBusMessage *message)
675 {
676   DBusMessageIter iter;
677   AtspiHyperlink *retval = NULL;
678   const char *signature;
679    
680   if (!message)
681     return NULL;
682
683   signature = dbus_message_get_signature (message);
684   if (!strcmp (signature, "(so)"))
685   {
686     dbus_message_iter_init (message, &iter);
687     retval =  _atspi_dbus_return_hyperlink_from_iter (&iter);
688   }
689   else
690   {
691     g_warning ("AT-SPI: Called _atspi_dbus_return_hyperlink_from_message with strange signature %s", signature);
692   }
693   dbus_message_unref (message);
694   return retval;
695 }
696
697 AtspiHyperlink *
698 _atspi_dbus_return_hyperlink_from_iter (DBusMessageIter *iter)
699 {
700   const char *app_name, *path;
701
702   get_reference_from_iter (iter, &app_name, &path);
703   return ref_hyperlink (app_name, path);
704 }
705
706 const char *cache_signal_type = "((so)(so)(so)a(so)assusau)";
707
708 static DBusHandlerResult
709 handle_add_accessible (DBusConnection *bus, DBusMessage *message, void *user_data)
710 {
711   DBusMessageIter iter;
712   const char *sender = dbus_message_get_sender (message);
713
714   if (strcmp (dbus_message_get_signature (message), cache_signal_type) != 0)
715   {
716     g_warning ("AT-SPI: AddAccessible with unknown signature %s\n",
717                dbus_message_get_signature (message));
718     return DBUS_HANDLER_RESULT_HANDLED;
719   }
720
721   dbus_message_iter_init (message, &iter);
722   add_accessible_from_iter (&iter);
723   return DBUS_HANDLER_RESULT_HANDLED;
724 }
725
726 typedef struct
727 {
728   DBusConnection *bus;
729   DBusMessage *message;
730   void *data;
731 } BusDataClosure;
732
733 static guint process_deferred_messages_id = -1;
734
735 static void
736 process_deferred_message (BusDataClosure *closure)
737 {
738   int type = dbus_message_get_type (closure->message);
739   const char *interface = dbus_message_get_interface (closure->message);
740
741   if (type == DBUS_MESSAGE_TYPE_SIGNAL &&
742       !strncmp (interface, "org.a11y.atspi.Event.", 21))
743   {
744     _atspi_dbus_handle_event (closure->bus, closure->message, closure->data);
745   }
746   if (dbus_message_is_method_call (closure->message, atspi_interface_device_event_listener, "NotifyEvent"))
747   {
748     _atspi_dbus_handle_DeviceEvent (closure->bus,
749                                    closure->message, closure->data);
750   }
751   if (dbus_message_is_signal (closure->message, atspi_interface_cache, "AddAccessible"))
752   {
753     handle_add_accessible (closure->bus, closure->message, closure->data);
754   }
755   if (dbus_message_is_signal (closure->message, atspi_interface_cache, "RemoveAccessible"))
756   {
757     handle_remove_accessible (closure->bus, closure->message, closure->data);
758   }
759   if (dbus_message_is_signal (closure->message, "org.freedesktop.DBus", "NameOwnerChanged"))
760   {
761     handle_name_owner_changed (closure->bus, closure->message, closure->data);
762   }
763 }
764
765 static GQueue *deferred_messages = NULL;
766
767 gboolean
768 _atspi_process_deferred_messages (gpointer data)
769 {
770   static int in_process_deferred_messages = 0;
771   BusDataClosure *closure;
772
773   if (in_process_deferred_messages)
774     return TRUE;
775   in_process_deferred_messages = 1;
776   while (closure = g_queue_pop_head (deferred_messages))
777   {
778     process_deferred_message (closure);
779     dbus_message_unref (closure->message);
780     dbus_connection_unref (closure->bus);
781     g_free (closure);
782   }
783   /* If data is NULL, assume that we were called from GLib */
784   if (!data)
785     process_deferred_messages_id = -1;
786   in_process_deferred_messages = 0;
787   return FALSE;
788 }
789
790 static DBusHandlerResult
791 defer_message (DBusConnection *connection, DBusMessage *message, void *user_data)
792 {
793   BusDataClosure *closure = g_new (BusDataClosure, 1);
794
795   closure->bus = dbus_connection_ref (bus);
796   closure->message = dbus_message_ref (message);
797   closure->data = user_data;
798
799   g_queue_push_tail (deferred_messages, closure);
800
801   if (process_deferred_messages_id == -1)
802     process_deferred_messages_id = g_idle_add (_atspi_process_deferred_messages, NULL);
803   return DBUS_HANDLER_RESULT_HANDLED;
804 }
805
806 static DBusHandlerResult
807 atspi_dbus_filter (DBusConnection *bus, DBusMessage *message, void *data)
808 {
809   int type = dbus_message_get_type (message);
810   const char *interface = dbus_message_get_interface (message);
811
812   if (type == DBUS_MESSAGE_TYPE_SIGNAL &&
813       !strncmp (interface, "org.a11y.atspi.Event.", 21))
814   {
815     return defer_message (bus, message, data);
816   }
817   if (dbus_message_is_method_call (message, atspi_interface_device_event_listener, "NotifyEvent"))
818   {
819     return defer_message (bus, message, data);
820   }
821   if (dbus_message_is_signal (message, atspi_interface_cache, "AddAccessible"))
822   {
823     return defer_message (bus, message, data);
824   }
825   if (dbus_message_is_signal (message, atspi_interface_cache, "RemoveAccessible"))
826   {
827     return defer_message (bus, message, data);
828   }
829   if (dbus_message_is_signal (message, "org.freedesktop.DBus", "NameOwnerChanged"))
830   {
831     defer_message (bus, message, data);
832     return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
833   }
834   return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
835 }
836
837 static const char *signal_interfaces[] =
838 {
839   "org.a11y.atspi.Event.Object",
840   "org.a11y.atspi.Event.Window",
841   "org.a11y.atspi.Event.Mouse",
842   "org.a11y.atspi.Event.Terminal",
843   "org.a11y.atspi.Event.Document",
844   "org.a11y.atspi.Event.Focus",
845   NULL
846 };
847
848 /*
849  * Returns a 'canonicalized' value for DISPLAY,
850  * with the screen number stripped off if present.
851  *
852  * TODO: Avoid having duplicate functions for this here and in at-spi2-atk
853  */
854 static gchar *
855 spi_display_name (void)
856 {
857   char *canonical_display_name = NULL;
858   const gchar *display_env = g_getenv ("AT_SPI_DISPLAY");
859
860   if (!display_env)
861     {
862       display_env = g_getenv ("DISPLAY");
863       if (!display_env || !display_env[0])
864         canonical_display_name = g_strdup (":0");
865       else
866         {
867           gchar *display_p, *screen_p;
868           canonical_display_name = g_strdup (display_env);
869           display_p = g_utf8_strrchr (canonical_display_name, -1, ':');
870           screen_p = g_utf8_strrchr (canonical_display_name, -1, '.');
871           if (screen_p && display_p && (screen_p > display_p))
872             {
873               *screen_p = '\0';
874             }
875         }
876     }
877   else
878     {
879       canonical_display_name = g_strdup (display_env);
880     }
881
882   return canonical_display_name;
883 }
884
885 /**
886  * atspi_init:
887  *
888  * Connects to the accessibility registry and initializes the SPI.
889  *
890  * Returns: 0 on success, 1 if already initialized, or an integer error code.  
891  **/
892 int
893 atspi_init (void)
894 {
895   char *match;
896   const gchar *no_cache;
897
898   if (atspi_inited)
899     {
900       return 1;
901     }
902
903   atspi_inited = TRUE;
904
905   g_type_init ();
906
907   get_live_refs();
908
909   bus = atspi_get_a11y_bus ();
910   if (!bus)
911     return 2;
912   dbus_bus_register (bus, NULL);
913   atspi_dbus_connection_setup_with_g_main(bus, g_main_context_default());
914   dbus_connection_add_filter (bus, atspi_dbus_filter, NULL, NULL);
915   match = g_strdup_printf ("type='signal',interface='%s',member='AddAccessible'", atspi_interface_cache);
916   dbus_bus_add_match (bus, match, NULL);
917   g_free (match);
918   match = g_strdup_printf ("type='signal',interface='%s',member='RemoveAccessible'", atspi_interface_cache);
919   dbus_bus_add_match (bus, match, NULL);
920   g_free (match);
921   match = g_strdup_printf ("type='signal',interface='%s',member='ChildrenChanged'", atspi_interface_event_object);
922   dbus_bus_add_match (bus, match, NULL);
923   g_free (match);
924   match = g_strdup_printf ("type='signal',interface='%s',member='PropertyChange'", atspi_interface_event_object);
925   dbus_bus_add_match (bus, match, NULL);
926   g_free (match);
927   match = g_strdup_printf ("type='signal',interface='%s',member='StateChanged'", atspi_interface_event_object);
928   dbus_bus_add_match (bus, match, NULL);
929   g_free (match);
930
931   dbus_bus_add_match (bus,
932                       "type='signal', interface='org.freedesktop.DBus', member='NameOwnerChanged'",
933                       NULL);
934
935   no_cache = g_getenv ("ATSPI_NO_CACHE");
936   if (no_cache && g_strcmp0 (no_cache, "0") != 0)
937     atspi_no_cache = TRUE;
938
939   deferred_messages = g_queue_new ();
940
941   return 0;
942 }
943
944 /**
945  * atspi_event_main:
946  *
947  * Starts/enters the main event loop for the AT-SPI services.
948  *
949  * NOTE: This method does not return control; it is exited via a call to
950  * #atspi_event_quit from within an event handler.
951  *
952  **/
953 void
954 atspi_event_main (void)
955 {
956   atspi_main_loop = g_main_loop_new (NULL, FALSE);
957   g_main_loop_run (atspi_main_loop);
958   atspi_main_loop = NULL;
959 }
960
961 /**
962  * atspi_event_quit:
963  *
964  * Quits the last main event loop for the AT-SPI services,
965  * See: #atspi_event_main
966  **/
967 void
968 atspi_event_quit (void)
969 {
970   g_main_loop_quit (atspi_main_loop);
971 }
972
973 /**
974  * atspi_exit:
975  *
976  * Disconnects from #AtspiRegistry instances and releases 
977  * any floating resources. Call only once at exit.
978  *
979  * Returns: 0 if there were no leaks, otherwise other integer values.
980  **/
981 int
982 atspi_exit (void)
983 {
984   int leaked;
985
986   if (!atspi_inited)
987     {
988       return 0;
989     }
990
991   atspi_inited = FALSE;
992
993   if (live_refs)
994     {
995       leaked = g_hash_table_size (live_refs);
996     }
997   else
998     {
999       leaked = 0;
1000     }
1001
1002   cleanup ();
1003
1004   return leaked;
1005 }
1006
1007 static GSList *hung_processes;
1008
1009 static void
1010 remove_hung_process (DBusPendingCall *pending, void *data)
1011 {
1012   gchar *bus_name = data;
1013
1014   hung_processes = g_slist_remove (hung_processes, data);
1015   g_free (data);
1016   dbus_pending_call_unref (pending);
1017 }
1018
1019 static void
1020 check_for_hang (DBusMessage *message, DBusError *error, DBusConnection *bus, const char *bus_name)
1021 {
1022   if (!message && error->name &&
1023       !strcmp (error->name, "org.freedesktop.DBus.Error.NoReply"))
1024   {
1025     GSList *l;
1026     DBusMessage *message;
1027     gchar *bus_name_dup;
1028     DBusPendingCall *pending = NULL;
1029     for (l = hung_processes; l; l = l->next)
1030       if (!strcmp (l->data, bus_name))
1031         return;
1032     message = dbus_message_new_method_call (bus_name, "/",
1033                                             "org.freedesktop.DBus.Peer",
1034                                             "Ping");
1035     if (!message)
1036       return;
1037     dbus_connection_send_with_reply (bus, message, &pending, -1);
1038     dbus_message_unref (message);
1039     if (!pending)
1040       return;
1041     bus_name_dup = g_strdup (bus_name);
1042     hung_processes = g_slist_append (hung_processes, bus_name_dup);
1043     dbus_pending_call_set_notify (pending, remove_hung_process, bus_name_dup, NULL);
1044   }
1045 }
1046
1047 static gboolean
1048 connection_is_hung (const char *bus_name)
1049 {
1050   GSList *l;
1051
1052   for (l = hung_processes; l; l = l->next)
1053     if (!strcmp (l->data, bus_name))
1054       return TRUE;
1055   return FALSE;
1056 }
1057
1058 static gboolean
1059 check_app (AtspiApplication *app, GError **error)
1060 {
1061   if (!app || !app->bus)
1062   {
1063     g_set_error_literal (error, ATSPI_ERROR, ATSPI_ERROR_APPLICATION_GONE,
1064                           _("The application no longer exists"));
1065     return FALSE;
1066   }
1067
1068   if (atspi_main_loop && connection_is_hung (app->bus_name))
1069   {
1070       g_set_error_literal (error, ATSPI_ERROR, ATSPI_ERROR_IPC,
1071                            "The process appears to be hung.");
1072     return FALSE;
1073   }
1074
1075   return TRUE;
1076 }
1077
1078 static void
1079 set_timeout (AtspiApplication *app)
1080 {
1081   struct timeval tv;
1082   int diff;
1083
1084   if (app && app_startup_time > 0)
1085   {
1086     gettimeofday (&tv, NULL);
1087     diff = (tv.tv_sec - app->time_added.tv_sec) * 1000 + (tv.tv_usec - app->time_added.tv_usec) / 1000;
1088     dbind_set_timeout (MAX(method_call_timeout, app_startup_time - diff));
1089   }
1090   else
1091     dbind_set_timeout (method_call_timeout);
1092 }
1093
1094 dbus_bool_t
1095 _atspi_dbus_call (gpointer obj, const char *interface, const char *method, GError **error, const char *type, ...)
1096 {
1097   va_list args;
1098   dbus_bool_t retval;
1099   DBusError err;
1100   AtspiObject *aobj = ATSPI_OBJECT (obj);
1101
1102   if (!check_app (aobj->app, error))
1103     return FALSE;
1104
1105   va_start (args, type);
1106   dbus_error_init (&err);
1107   set_timeout (aobj->app);
1108   retval = dbind_method_call_reentrant_va (aobj->app->bus, aobj->app->bus_name,
1109                                            aobj->path, interface, method, &err,
1110                                            type, args);
1111   va_end (args);
1112   check_for_hang (NULL, &err, aobj->app->bus, aobj->app->bus_name);
1113   _atspi_process_deferred_messages ((gpointer)TRUE);
1114   if (dbus_error_is_set (&err))
1115   {
1116     g_set_error(error, ATSPI_ERROR, ATSPI_ERROR_IPC, "%s", err.message);
1117     dbus_error_free (&err);
1118   }
1119   return retval;
1120 }
1121
1122 DBusMessage *
1123 _atspi_dbus_call_partial (gpointer obj,
1124                           const char *interface,
1125                           const char *method,
1126                           GError **error,
1127                           const char *type, ...)
1128 {
1129   va_list args;
1130
1131   va_start (args, type);
1132   return _atspi_dbus_call_partial_va (obj, interface, method, error, type, args);
1133 }
1134
1135
1136 DBusMessage *
1137 _atspi_dbus_call_partial_va (gpointer obj,
1138                           const char *interface,
1139                           const char *method,
1140                           GError **error,
1141                           const char *type,
1142                           va_list args)
1143 {
1144   AtspiObject *aobj = ATSPI_OBJECT (obj);
1145   DBusError err;
1146     DBusMessage *msg = NULL, *reply = NULL;
1147     DBusMessageIter iter;
1148     const char *p;
1149
1150   dbus_error_init (&err);
1151
1152   if (!check_app (aobj->app, error))
1153     goto out;
1154
1155   msg = dbus_message_new_method_call (aobj->app->bus_name, aobj->path, interface, method);
1156   if (!msg)
1157     goto out;
1158
1159   p = type;
1160   dbus_message_iter_init_append (msg, &iter);
1161   dbind_any_marshal_va (&iter, &p, args);
1162
1163   set_timeout (aobj->app);
1164   reply = dbind_send_and_allow_reentry (aobj->app->bus, msg, &err);
1165   check_for_hang (reply, &err, aobj->app->bus, aobj->app->bus_name);
1166 out:
1167   va_end (args);
1168   if (msg)
1169     dbus_message_unref (msg);
1170   _atspi_process_deferred_messages ((gpointer)TRUE);
1171   if (dbus_error_is_set (&err))
1172   {
1173     /* TODO: Set gerror */
1174     dbus_error_free (&err);
1175   }
1176   return reply;
1177 }
1178
1179 dbus_bool_t
1180 _atspi_dbus_get_property (gpointer obj, const char *interface, const char *name, GError **error, const char *type, void *data)
1181 {
1182   DBusMessage *message, *reply;
1183   DBusMessageIter iter, iter_variant;
1184   DBusError err;
1185   dbus_bool_t retval = FALSE;
1186   AtspiObject *aobj = ATSPI_OBJECT (obj);
1187   char expected_type = (type [0] == '(' ? 'r' : type [0]);
1188
1189   if (!aobj)
1190     return FALSE;
1191
1192   if (!check_app (aobj->app, error))
1193     return FALSE;
1194
1195   message = dbus_message_new_method_call (aobj->app->bus_name,
1196                                           aobj->path,
1197                                           "org.freedesktop.DBus.Properties",
1198                                           "Get");
1199   if (!message)
1200   {
1201     // TODO: throw exception
1202     return FALSE;
1203   }
1204   dbus_message_append_args (message, DBUS_TYPE_STRING, &interface, DBUS_TYPE_STRING, &name, DBUS_TYPE_INVALID);
1205   dbus_error_init (&err);
1206   set_timeout (aobj->app);
1207   reply = dbind_send_and_allow_reentry (aobj->app->bus, message, &err);
1208   check_for_hang (reply, &err, aobj->app->bus, aobj->app->bus_name);
1209   dbus_message_unref (message);
1210   _atspi_process_deferred_messages ((gpointer)TRUE);
1211   if (!reply)
1212   {
1213     // TODO: throw exception
1214     goto done;
1215   }
1216
1217   if (dbus_message_get_type (reply) == DBUS_MESSAGE_TYPE_ERROR)
1218   {
1219     const char *err_str = NULL;
1220     dbus_message_get_args (reply, NULL, DBUS_TYPE_STRING, &err_str, DBUS_TYPE_INVALID);
1221     if (err_str)
1222       g_set_error_literal (error, ATSPI_ERROR, ATSPI_ERROR_IPC, err_str);
1223     goto done;
1224   }
1225
1226   dbus_message_iter_init (reply, &iter);
1227   if (dbus_message_iter_get_arg_type (&iter) != 'v')
1228   {
1229     g_warning ("AT-SPI: expected a variant when fetching %s from interface %s; got %s\n", name, interface, dbus_message_get_signature (reply));
1230     goto done;
1231   }
1232   dbus_message_iter_recurse (&iter, &iter_variant);
1233   if (dbus_message_iter_get_arg_type (&iter_variant) != expected_type)
1234   {
1235     g_warning ("atspi_dbus_get_property: Wrong type: expected %s, got %c\n", type, dbus_message_iter_get_arg_type (&iter_variant));
1236     goto done;
1237   }
1238   if (!strcmp (type, "(so)"))
1239   {
1240     *((AtspiAccessible **)data) = _atspi_dbus_return_accessible_from_iter (&iter_variant);
1241   }
1242   else
1243   {
1244     dbus_message_iter_get_basic (&iter_variant, data);
1245     if (type [0] == 's')
1246       *(char **)data = g_strdup (*(char **)data);
1247   }
1248   retval = TRUE;
1249 done:
1250   dbus_error_free (&err);
1251   if (reply)
1252     dbus_message_unref (reply);
1253   return retval;
1254 }
1255
1256 DBusMessage *
1257 _atspi_dbus_send_with_reply_and_block (DBusMessage *message, GError **error)
1258 {
1259   DBusMessage *reply;
1260   DBusError err;
1261   AtspiApplication *app;
1262   DBusConnection *bus;
1263
1264   app = get_application (dbus_message_get_destination (message));
1265
1266   if (app && !app->bus)
1267     return NULL;        /* will fail anyway; app has been disposed */
1268
1269   bus = (app ? app->bus : _atspi_bus());
1270   dbus_error_init (&err);
1271   set_timeout (app);
1272   reply = dbind_send_and_allow_reentry (bus, message, &err);
1273   _atspi_process_deferred_messages ((gpointer)TRUE);
1274   dbus_message_unref (message);
1275   if (dbus_error_is_set (&err))
1276   {
1277     if (error)
1278       g_set_error_literal (error, ATSPI_ERROR, ATSPI_ERROR_IPC, err.message);
1279     dbus_error_free (&err);
1280   }
1281   return reply;
1282 }
1283
1284 GHashTable *
1285 _atspi_dbus_return_hash_from_message (DBusMessage *message)
1286 {
1287   DBusMessageIter iter;
1288   GHashTable *ret;
1289
1290   if (!message)
1291     return NULL;
1292
1293   _ATSPI_DBUS_CHECK_SIG (message, "a{ss}", NULL, NULL);
1294
1295   dbus_message_iter_init (message, &iter);
1296   ret = _atspi_dbus_hash_from_iter (&iter);
1297   dbus_message_unref (message);
1298   return ret;
1299 }
1300
1301 GHashTable *
1302 _atspi_dbus_hash_from_iter (DBusMessageIter *iter)
1303 {
1304   GHashTable *hash = g_hash_table_new_full (g_str_hash, g_str_equal,
1305                                             (GDestroyNotify) g_free,
1306                                             (GDestroyNotify) g_free);
1307   DBusMessageIter iter_array, iter_dict;
1308
1309   dbus_message_iter_recurse (iter, &iter_array);
1310   while (dbus_message_iter_get_arg_type (&iter_array) != DBUS_TYPE_INVALID)
1311   {
1312     const char *name, *value;
1313     dbus_message_iter_recurse (&iter_array, &iter_dict);
1314     dbus_message_iter_get_basic (&iter_dict, &name);
1315     dbus_message_iter_next (&iter_dict);
1316     dbus_message_iter_get_basic (&iter_dict, &value);
1317     g_hash_table_insert (hash, g_strdup (name), g_strdup (value));
1318     dbus_message_iter_next (&iter_array);
1319   }
1320   return hash;
1321 }
1322
1323 GArray *
1324 _atspi_dbus_return_attribute_array_from_message (DBusMessage *message)
1325 {
1326   DBusMessageIter iter;
1327   GArray *ret;
1328
1329   if (!message)
1330     return NULL;
1331
1332   _ATSPI_DBUS_CHECK_SIG (message, "a{ss}", NULL, NULL);
1333
1334   dbus_message_iter_init (message, &iter);
1335
1336   ret = _atspi_dbus_attribute_array_from_iter (&iter);
1337   dbus_message_unref (message);
1338   return ret;
1339 }
1340
1341 GArray *
1342 _atspi_dbus_attribute_array_from_iter (DBusMessageIter *iter)
1343 {
1344   DBusMessageIter iter_array, iter_dict;
1345   GArray *array = g_array_new (TRUE, TRUE, sizeof (gchar *));
1346
1347   dbus_message_iter_recurse (iter, &iter_array);
1348   while (dbus_message_iter_get_arg_type (&iter_array) != DBUS_TYPE_INVALID)
1349   {
1350     const char *name, *value;
1351     gchar *str;
1352     dbus_message_iter_recurse (&iter_array, &iter_dict);
1353     dbus_message_iter_get_basic (&iter_dict, &name);
1354     dbus_message_iter_next (&iter_dict);
1355     dbus_message_iter_get_basic (&iter_dict, &value);
1356     str = g_strdup_printf ("%s:%s", name, value);
1357     array = g_array_append_val (array, str);
1358     dbus_message_iter_next (&iter_array);;
1359   }
1360   return array;
1361 }
1362
1363 void
1364 _atspi_dbus_set_interfaces (AtspiAccessible *accessible, DBusMessageIter *iter)
1365 {
1366   DBusMessageIter iter_array;
1367
1368   accessible->interfaces = 0;
1369   dbus_message_iter_recurse (iter, &iter_array);
1370   while (dbus_message_iter_get_arg_type (&iter_array) != DBUS_TYPE_INVALID)
1371   {
1372     const char *iface;
1373     gint n;
1374     dbus_message_iter_get_basic (&iter_array, &iface);
1375     if (!strcmp (iface, "org.freedesktop.DBus.Introspectable")) continue;
1376     n = _atspi_get_iface_num (iface);
1377     if (n == -1)
1378     {
1379       g_warning ("AT-SPI: Unknown interface %s", iface);
1380     }
1381     else
1382       accessible->interfaces |= (1 << n);
1383     dbus_message_iter_next (&iter_array);
1384   }
1385   _atspi_accessible_add_cache (accessible, ATSPI_CACHE_INTERFACES);
1386 }
1387
1388 void
1389 _atspi_dbus_set_state (AtspiAccessible *accessible, DBusMessageIter *iter)
1390 {
1391   DBusMessageIter iter_array;
1392   gint count;
1393   dbus_uint32_t *states;
1394
1395   dbus_message_iter_recurse (iter, &iter_array);
1396   dbus_message_iter_get_fixed_array (&iter_array, &states, &count);
1397   if (count != 2)
1398   {
1399     g_warning ("AT-SPI: expected 2 values in states array; got %d\n", count);
1400     if (!accessible->states)
1401       accessible->states = _atspi_state_set_new_internal (accessible, 0);
1402   }
1403   else
1404   {
1405     guint64 val = ((guint64)states [1]) << 32;
1406     val += states [0];
1407     if (!accessible->states)
1408       accessible->states = _atspi_state_set_new_internal (accessible, val);
1409     else
1410       accessible->states->states = val;
1411   }
1412   _atspi_accessible_add_cache (accessible, ATSPI_CACHE_STATES);
1413 }
1414
1415 GQuark
1416 _atspi_error_quark (void)
1417 {
1418   return g_quark_from_static_string ("atspi_error");
1419 }
1420
1421 /*
1422  * Gets the IOR from the XDisplay.
1423  */
1424 static char *
1425 get_accessibility_bus_address_x11 (void)
1426 {
1427   Atom AT_SPI_BUS;
1428   Atom actual_type;
1429   Display *bridge_display = NULL;
1430   int actual_format;
1431   char *data;
1432   unsigned char *data_x11 = NULL;
1433   unsigned long nitems;
1434   unsigned long leftover;
1435   char *display_name;
1436
1437   display_name = spi_display_name ();
1438   if (display_name != NULL)
1439     {
1440       bridge_display = XOpenDisplay (display_name);
1441       g_free (display_name);
1442     }
1443   if (!bridge_display)
1444     {
1445       g_warning ("Could not open X display");
1446       return NULL;
1447     }
1448       
1449   AT_SPI_BUS = XInternAtom (bridge_display, "AT_SPI_BUS", False);
1450   XGetWindowProperty (bridge_display,
1451                       XDefaultRootWindow (bridge_display),
1452                       AT_SPI_BUS, 0L,
1453                       (long) BUFSIZ, False,
1454                       (Atom) 31, &actual_type, &actual_format,
1455                       &nitems, &leftover, &data_x11);
1456   XCloseDisplay (bridge_display);
1457
1458   data = g_strdup (data_x11);
1459   XFree (data_x11);
1460   return data;
1461 }
1462
1463 static char *
1464 get_accessibility_bus_address_dbus (void)
1465 {
1466   DBusConnection *session_bus = NULL;
1467   DBusMessage *message;
1468   DBusMessage *reply;
1469   DBusError error;
1470   char *address = NULL;
1471
1472   session_bus = dbus_bus_get (DBUS_BUS_SESSION, NULL);
1473   if (!session_bus)
1474     return NULL;
1475
1476   message = dbus_message_new_method_call ("org.a11y.Bus",
1477                                           "/org/a11y/bus",
1478                                           "org.a11y.Bus",
1479                                           "GetAddress");
1480
1481   dbus_error_init (&error);
1482   reply = dbus_connection_send_with_reply_and_block (session_bus,
1483                                                      message,
1484                                                      -1,
1485                                                      &error);
1486   dbus_message_unref (message);
1487
1488   if (!reply)
1489   {
1490     g_warning ("Error retrieving accessibility bus address: %s: %s",
1491                error.name, error.message);
1492     dbus_error_free (&error);
1493     return NULL;
1494   }
1495   
1496   {
1497     const char *tmp_address;
1498     if (!dbus_message_get_args (reply,
1499                                 NULL,
1500                                 DBUS_TYPE_STRING,
1501                                 &tmp_address,
1502                                 DBUS_TYPE_INVALID))
1503       {
1504         dbus_message_unref (reply);
1505         return NULL;
1506       }
1507     address = g_strdup (tmp_address);
1508     dbus_message_unref (reply);
1509   }
1510   
1511   return address;
1512 }
1513
1514 static DBusConnection *a11y_bus;
1515 static dbus_int32_t a11y_dbus_slot = -1;
1516
1517 static void
1518 a11y_bus_free (void *data)
1519 {
1520   if (data == a11y_bus)
1521     {
1522       a11y_bus = NULL;
1523       dbus_connection_free_data_slot (&a11y_dbus_slot);
1524     }
1525 }
1526
1527 DBusConnection *
1528 atspi_get_a11y_bus (void)
1529 {
1530   DBusConnection *bus = NULL;
1531   DBusError error;
1532   char *address;
1533
1534   if (a11y_bus && dbus_connection_get_is_connected (a11y_bus))
1535     return a11y_bus;
1536
1537   if (a11y_dbus_slot == -1)
1538     if (!dbus_connection_allocate_data_slot (&a11y_dbus_slot))
1539       g_warning ("at-spi: Unable to allocate D-Bus slot");
1540
1541   address = get_accessibility_bus_address_x11 ();
1542   if (!address)
1543     address = get_accessibility_bus_address_dbus ();
1544   if (!address)
1545     return NULL;
1546
1547   dbus_error_init (&error);
1548   a11y_bus = dbus_connection_open_private (address, &error);
1549   g_free (address);
1550
1551   if (!a11y_bus)
1552     {
1553       g_warning ("Couldn't connect to accessibility bus: %s", error.message);
1554       dbus_error_free (&error);
1555       return NULL;
1556     }
1557   else
1558     {
1559       if (!dbus_bus_register (a11y_bus, &error))
1560         {
1561           g_warning ("Couldn't register with accessibility bus: %s", error.message);
1562           dbus_error_free (&error);
1563           dbus_connection_close (a11y_bus);
1564           dbus_connection_unref (a11y_bus);
1565           a11y_bus = NULL;
1566           return NULL;
1567         }
1568     }
1569   
1570   /* Simulate a weak ref on the bus */
1571   dbus_connection_set_data (a11y_bus, a11y_dbus_slot, a11y_bus, a11y_bus_free);
1572
1573   return a11y_bus;
1574 }
1575
1576 /**
1577  *  Set the timeout used for method calls. If this is not set explicitly,
1578  *  a default of 0.8 ms is used.
1579  *  Note that at-spi2-registryd currently uses a timeout of 3 seconds when
1580  *  sending a keyboard event notification. This means that, if an AT makes
1581  *  a call in response to the keyboard notification and the application
1582  *  being called does not respond before the timeout is reached,
1583  *  at-spi2-registryd will time out on the keyboard event notification and
1584  *  pass the key onto the application (ie, reply to indicate that the key
1585  *  was not consumed), so this may make it undesirable to set a timeout
1586  *  larger than 3 seconds.
1587  *
1588  *  @val: The timeout value, in milliseconds, or -1 to disable the timeout.
1589  *  @startup_time: The amount of time, in milliseconds, to allow to pass
1590  *  before enforcing timeouts on an application. Can be used to prevent
1591  *  timeout exceptions if an application is likely to block for an extended
1592  *  period of time on initialization. -1 can be passed to disable this
1593  *  behavior.
1594  *
1595  * By default, the normal timeout is set to 800 ms, and the application startup
1596  * timeout is set to 15 seconds.
1597  */
1598 void
1599 atspi_set_timeout (gint val, gint startup_time)
1600 {
1601   method_call_timeout = val;
1602   app_startup_time = startup_time;
1603 }