bus/at-spi-bus-launcher.c

   1 /* -*- mode: c; c-basic-offset: 2; indent-tabs-mode: nil; -*-
   2  *
   3  * at-spi-bus-launcher: Manage the a11y bus as a child process
   4  *
   5  * Copyright 2011 Red Hat, Inc.
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Library General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Library General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Library General Public
  18  * License along with this library; if not, write to the
  19  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  20  * Boston, MA 02111-1307, USA.
  21  */
  22
  23 #include "config.h"
  24
  25 #include <unistd.h>
  26 #include <string.h>
  27 #include <signal.h>
  28 #include <sys/wait.h>
  29 #include <errno.h>
  30 #include <stdio.h>
  31
  32 #include <gio/gio.h>
  33 #include <X11/Xlib.h>
  34 #include <X11/Xatom.h>
  35
  36 typedef enum {
  37   A11Y_BUS_STATE_IDLE = 0,
  38   A11Y_BUS_STATE_READING_ADDRESS,
  39   A11Y_BUS_STATE_RUNNING,
  40   A11Y_BUS_STATE_ERROR
  41 } A11yBusState;
  42
  43 typedef struct {
  44   GMainLoop *loop;
  45   gboolean launch_immediately;
  46   gboolean a11y_enabled;
  47   GDBusConnection *session_bus;
  48   GSettings *desktop_schema;
  49
  50   A11yBusState state;
  51   /* -1 == error, 0 == pending, > 0 == running */
  52   int a11y_bus_pid;
  53   char *a11y_bus_address;
  54   int pipefd[2];
  55   char *a11y_launch_error_message;
  56 } A11yBusLauncher;
  57
  58 static A11yBusLauncher *_global_app = NULL;
  59
  60 static const gchar introspection_xml[] =
  61   "<node>"
  62   "  <interface name='org.a11y.Bus'>"
  63   "    <method name='GetAddress'>"
  64   "      <arg type='s' name='address' direction='out'/>"
  65   "    </method>"
  66   "  </interface>"
  67   "<interface name='org.a11y.Status'>"
  68   "<property name='IsEnabled' type='b' access='readwrite'/>"
  69   "</interface>"
  70   "</node>";
  71 static GDBusNodeInfo *introspection_data = NULL;
  72
  73 static void
  74 setup_bus_child (gpointer data)
  75 {
  76   A11yBusLauncher *app = data;
  77   (void) app;
  78
  79   close (app->pipefd[0]);
  80   dup2 (app->pipefd[1], 3);
  81   close (app->pipefd[1]);
  82
  83   /* On Linux, tell the bus process to exit if this process goes away */
  84 #ifdef __linux
  85 #include <sys/prctl.h>
  86   prctl (PR_SET_PDEATHSIG, 15);
  87 #endif
  88 }
  89
  90 /**
  91  * unix_read_all_fd_to_string:
  92  *
  93  * Read all data from a file descriptor to a C string buffer.
  94  */
  95 static gboolean
  96 unix_read_all_fd_to_string (int      fd,
  97                             char    *buf,
  98                             ssize_t  max_bytes)
  99 {
 100   ssize_t bytes_read;
 101
 102   while (max_bytes > 1 && (bytes_read = read (fd, buf, MAX (4096, max_bytes - 1))))
 103     {
 104       if (bytes_read < 0)
 105         return FALSE;
 106       buf += bytes_read;
 107       max_bytes -= bytes_read;
 108     }
 109   *buf = '\0';
 110   return TRUE;
 111 }
 112
 113 static void
 114 on_bus_exited (GPid     pid,
 115                gint     status,
 116                gpointer data)
 117 {
 118   A11yBusLauncher *app = data;
 119
 120   app->a11y_bus_pid = -1;
 121   app->state = A11Y_BUS_STATE_ERROR;
 122   if (app->a11y_launch_error_message == NULL)
 123     {
 124       if (WIFEXITED (status))
 125         app->a11y_launch_error_message = g_strdup_printf ("Bus exited with code %d", WEXITSTATUS (status));
 126       else if (WIFSIGNALED (status))
 127         app->a11y_launch_error_message = g_strdup_printf ("Bus killed by signal %d", WTERMSIG (status));
 128       else if (WIFSTOPPED (status))
 129         app->a11y_launch_error_message = g_strdup_printf ("Bus stopped by signal %d", WSTOPSIG (status));
 130     }
 131   g_main_loop_quit (app->loop);
 132 }
 133
 134 static gboolean
 135 ensure_a11y_bus (A11yBusLauncher *app)
 136 {
 137   GPid pid;
 138   char *argv[] = { DBUS_DAEMON, NULL, "--nofork", "--print-address", "3", NULL };
 139   char addr_buf[2048];
 140   GError *error = NULL;
 141
 142   if (app->a11y_bus_pid != 0)
 143     return FALSE;
 144
 145   argv[1] = g_strdup_printf ("--config-file=%s/at-spi2/accessibility.conf", SYSCONFDIR);
 146
 147   if (pipe (app->pipefd) < 0)
 148     g_error ("Failed to create pipe: %s", strerror (errno));
 149
 150   if (!g_spawn_async (NULL,
 151                       argv,
 152                       NULL,
 153                       G_SPAWN_SEARCH_PATH | G_SPAWN_DO_NOT_REAP_CHILD,
 154                       setup_bus_child,
 155                       app,
 156                       &pid,
 157                       &error))
 158     {
 159       app->a11y_bus_pid = -1;
 160       app->a11y_launch_error_message = g_strdup (error->message);
 161       g_clear_error (&error);
 162       goto error;
 163     }
 164
 165   close (app->pipefd[1]);
 166   app->pipefd[1] = -1;
 167
 168   g_child_watch_add (pid, on_bus_exited, app);
 169
 170   app->state = A11Y_BUS_STATE_READING_ADDRESS;
 171   app->a11y_bus_pid = pid;
 172   g_debug ("Launched a11y bus, child is %ld", (long) pid);
 173   if (!unix_read_all_fd_to_string (app->pipefd[0], addr_buf, sizeof (addr_buf)))
 174     {
 175       app->a11y_launch_error_message = g_strdup_printf ("Failed to read address: %s", strerror (errno));
 176       kill (app->a11y_bus_pid, SIGTERM);
 177       goto error;
 178     }
 179   close (app->pipefd[0]);
 180   app->pipefd[0] = -1;
 181   app->state = A11Y_BUS_STATE_RUNNING;
 182
 183   /* Trim the trailing newline */
 184   app->a11y_bus_address = g_strchomp (g_strdup (addr_buf));
 185   g_debug ("a11y bus address: %s", app->a11y_bus_address);
 186
 187   {
 188     Display *display = XOpenDisplay (NULL);
 189     if (display)
 190       {
 191         Atom bus_address_atom = XInternAtom (display, "AT_SPI_BUS", False);
 192         XChangeProperty (display,
 193                          XDefaultRootWindow (display),
 194                          bus_address_atom,
 195                          XA_STRING, 8, PropModeReplace,
 196                          (guchar *) app->a11y_bus_address, strlen (app->a11y_bus_address));
 197         XFlush (display);
 198         XCloseDisplay (display);
 199       }
 200   }
 201
 202   return TRUE;
 203
 204  error:
 205   close (app->pipefd[0]);
 206   close (app->pipefd[1]);
 207   app->state = A11Y_BUS_STATE_ERROR;
 208
 209   return FALSE;
 210 }
 211
 212 static void
 213 handle_method_call (GDBusConnection       *connection,
 214                     const gchar           *sender,
 215                     const gchar           *object_path,
 216                     const gchar           *interface_name,
 217                     const gchar           *method_name,
 218                     GVariant              *parameters,
 219                     GDBusMethodInvocation *invocation,
 220                     gpointer               user_data)
 221 {
 222   A11yBusLauncher *app = user_data;
 223
 224   if (g_strcmp0 (method_name, "GetAddress") == 0)
 225     {
 226       ensure_a11y_bus (app);
 227       if (app->a11y_bus_pid > 0)
 228         g_dbus_method_invocation_return_value (invocation,
 229                                                g_variant_new ("(s)", app->a11y_bus_address));
 230       else
 231         g_dbus_method_invocation_return_dbus_error (invocation,
 232                                                     "org.a11y.Bus.Error",
 233                                                     app->a11y_launch_error_message);
 234     }
 235 }
 236
 237 static GVariant *
 238 handle_get_property  (GDBusConnection       *connection,
 239                       const gchar           *sender,
 240                       const gchar           *object_path,
 241                       const gchar           *interface_name,
 242                       const gchar           *property_name,
 243                     GError **error,
 244                     gpointer               user_data)
 245 {
 246   A11yBusLauncher *app = user_data;
 247
 248   if (g_strcmp0 (property_name, "IsEnabled") == 0)
 249     {
 250       return g_variant_new ("b", app->a11y_enabled);
 251     }
 252   else
 253     return NULL;
 254 }
 255
 256 static void
 257 handle_a11y_enabled_change (A11yBusLauncher *app, gboolean enabled,
 258                                gboolean notify_gsettings)
 259 {
 260   GVariantBuilder *builder;
 261   GVariantBuilder *invalidated_builder;
 262
 263   if (enabled == app->a11y_enabled)
 264     return;
 265
 266   app->a11y_enabled = enabled;
 267
 268   if (notify_gsettings && app->desktop_schema)
 269     g_settings_set_boolean (app->desktop_schema, "toolkit-accessibility",
 270                             enabled);
 271
 272   builder = g_variant_builder_new (G_VARIANT_TYPE_ARRAY);
 273   invalidated_builder = g_variant_builder_new (G_VARIANT_TYPE ("as"));
 274   g_variant_builder_add (builder, "{sv}", "IsEnabled",
 275                          g_variant_new_boolean (enabled));
 276
 277   g_dbus_connection_emit_signal (app->session_bus, NULL, "/org/a11y/bus",
 278                                  "org.freedesktop.DBus", "PropertiesChanged",
 279                                  g_variant_new ("(sa{sv}as)", "org.a11y.Status",
 280                                                 builder,
 281                                                 invalidated_builder),
 282                                  NULL);
 283 }
 284
 285 static gboolean
 286 handle_set_property  (GDBusConnection       *connection,
 287                       const gchar           *sender,
 288                       const gchar           *object_path,
 289                       const gchar           *interface_name,
 290                       const gchar           *property_name,
 291                       GVariant *value,
 292                     GError **error,
 293                     gpointer               user_data)
 294 {
 295   A11yBusLauncher *app = user_data;
 296
 297   if (g_strcmp0 (property_name, "IsEnabled") == 0)
 298     {
 299       const gchar *type = g_variant_get_type_string (value);
 300       gboolean enabled;
 301       if (g_strcmp0 (type, "b") != 0)
 302         {
 303           g_set_error (error, G_DBUS_ERROR, G_DBUS_ERROR_INVALID_ARGS,
 304                        "org.a11y.Status.IsEnabled expects a boolean but got %s", type);
 305           return FALSE;
 306         }
 307       enabled = g_variant_get_boolean (value);
 308       handle_a11y_enabled_change (app, enabled, TRUE);
 309       return TRUE;
 310     }
 311   else
 312     {
 313       g_set_error (error, G_DBUS_ERROR, G_DBUS_ERROR_INVALID_ARGS,
 314                        "Unknown property '%s'", property_name);
 315       return FALSE;
 316     }
 317 }
 318
 319 static const GDBusInterfaceVTable bus_vtable =
 320 {
 321   handle_method_call,
 322   NULL, /* handle_get_property, */
 323   NULL  /* handle_set_property */
 324 };
 325
 326 static const GDBusInterfaceVTable status_vtable =
 327 {
 328   NULL, /* handle_method_call */
 329   handle_get_property,
 330   handle_set_property
 331 };
 332
 333 static void
 334 on_bus_acquired (GDBusConnection *connection,
 335                  const gchar     *name,
 336                  gpointer         user_data)
 337 {
 338   A11yBusLauncher *app = user_data;
 339   GError *error;
 340   guint registration_id;
 341
 342   if (connection == NULL)
 343     {
 344       g_main_loop_quit (app->loop);
 345       return;
 346     }
 347   app->session_bus = connection;
 348
 349   if (app->launch_immediately)
 350     {
 351       ensure_a11y_bus (app);
 352       if (app->state == A11Y_BUS_STATE_ERROR)
 353         {
 354           g_main_loop_quit (app->loop);
 355           return;
 356         }
 357     }
 358
 359   error = NULL;
 360   registration_id = g_dbus_connection_register_object (connection,
 361                                                        "/org/a11y/bus",
 362                                                        introspection_data->interfaces[0],
 363                                                        &bus_vtable,
 364                                                        _global_app,
 365                                                        NULL,
 366                                                        &error);
 367   if (registration_id == 0)
 368     g_error ("%s", error->message);
 369
 370   g_dbus_connection_register_object (connection,
 371                                                        "/org/a11y/bus",
 372                                                        introspection_data->interfaces[1],
 373                                                        &status_vtable,
 374                                                        _global_app,
 375                                                        NULL,
 376                                                        &error);
 377 }
 378
 379 static void
 380 on_name_lost (GDBusConnection *connection,
 381               const gchar     *name,
 382               gpointer         user_data)
 383 {
 384   A11yBusLauncher *app = user_data;
 385   if (app->session_bus == NULL
 386       && connection == NULL
 387       && app->a11y_launch_error_message == NULL)
 388     app->a11y_launch_error_message = g_strdup ("Failed to connect to session bus");
 389   g_main_loop_quit (app->loop);
 390 }
 391
 392 static void
 393 on_name_acquired (GDBusConnection *connection,
 394                   const gchar     *name,
 395                   gpointer         user_data)
 396 {
 397   A11yBusLauncher *app = user_data;
 398   (void) app;
 399 }
 400
 401 static int sigterm_pipefd[2];
 402
 403 static void
 404 sigterm_handler (int signum)
 405 {
 406   write (sigterm_pipefd[1], "X", 1);
 407 }
 408
 409 static gboolean
 410 on_sigterm_pipe (GIOChannel  *channel,
 411                  GIOCondition condition,
 412                  gpointer     data)
 413 {
 414   A11yBusLauncher *app = data;
 415
 416   g_main_loop_quit (app->loop);
 417
 418   return FALSE;
 419 }
 420
 421 static void
 422 init_sigterm_handling (A11yBusLauncher *app)
 423 {
 424   GIOChannel *sigterm_channel;
 425
 426   if (pipe (sigterm_pipefd) < 0)
 427     g_error ("Failed to create pipe: %s", strerror (errno));
 428   signal (SIGTERM, sigterm_handler);
 429
 430   sigterm_channel = g_io_channel_unix_new (sigterm_pipefd[0]);
 431   g_io_add_watch (sigterm_channel,
 432                   G_IO_IN | G_IO_ERR | G_IO_HUP,
 433                   on_sigterm_pipe,
 434                   app);
 435 }
 436
 437 static gboolean
 438 already_running ()
 439 {
 440   Atom AT_SPI_BUS;
 441   Atom actual_type;
 442   Display *bridge_display;
 443   int actual_format;
 444   unsigned char *data = NULL;
 445   unsigned long nitems;
 446   unsigned long leftover;
 447   gboolean result = FALSE;
 448
 449   bridge_display = XOpenDisplay (NULL);
 450   if (!bridge_display)
 451               return FALSE;
 452
 453   AT_SPI_BUS = XInternAtom (bridge_display, "AT_SPI_BUS", False);
 454   XGetWindowProperty (bridge_display,
 455                       XDefaultRootWindow (bridge_display),
 456                       AT_SPI_BUS, 0L,
 457                       (long) BUFSIZ, False,
 458                       (Atom) 31, &actual_type, &actual_format,
 459                       &nitems, &leftover, &data);
 460
 461   if (data)
 462   {
 463     GDBusConnection *bus;
 464     GError *error = NULL;
 465     const gchar *old_session = g_getenv ("DBUS_SESSION_BUS_ADDRESS");
 466     /* TODO: Is there a better way to connect? This is really hacky */
 467     g_setenv ("DBUS_SESSION_BUS_ADDRESS", data, TRUE);
 468     bus = g_bus_get_sync (G_BUS_TYPE_SESSION, NULL, &error);
 469     g_setenv ("DBUS_SESSION_BUS_ADDRESS", old_session, TRUE);
 470     if (bus != NULL)
 471       {
 472         result = TRUE;
 473         g_object_unref (bus);
 474       }
 475   }
 476
 477   XCloseDisplay (bridge_display);
 478   return result;
 479 }
 480
 481 static GSettings *
 482 get_desktop_schema ()
 483 {
 484   const char * const *schemas = NULL;
 485   gint i;
 486
 487   schemas = g_settings_list_schemas ();
 488   for (i = 0; schemas[i]; i++)
 489   {
 490     if (!strcmp (schemas[i], "org.gnome.desktop.interface"))
 491       return g_settings_new (schemas[i]);
 492   }
 493
 494   return NULL;
 495 }
 496
 497 static void
 498 gsettings_key_changed (GSettings *gsettings, const gchar *key, void *user_data)
 499 {
 500   gboolean new_val = g_settings_get_boolean (gsettings, key);
 501   A11yBusLauncher *app = user_data;
 502
 503   if (strcmp (key, "toolkit-accessibility") != 0)
 504     return;
 505
 506   handle_a11y_enabled_change (_global_app, new_val, FALSE);
 507 }
 508
 509 int
 510 main (int    argc,
 511       char **argv)
 512 {
 513   GError *error = NULL;
 514   GMainLoop *loop;
 515   GDBusConnection *session_bus;
 516   int name_owner_id;
 517   gboolean a11y_set = FALSE;
 518   gint i;
 519
 520   g_type_init ();
 521
 522   if (already_running ())
 523     return 0;
 524
 525   _global_app = g_slice_new0 (A11yBusLauncher);
 526   _global_app->loop = g_main_loop_new (NULL, FALSE);
 527
 528   for (i = 1; i < argc; i++)
 529     {
 530       if (!strcmp (argv[i], "--launch-immediately"))
 531         _global_app->launch_immediately = TRUE;
 532       else if (sscanf (argv[i], "--a11y=%d", &_global_app->a11y_enabled) == 2)
 533         a11y_set = TRUE;
 534     else
 535       g_error ("usage: %s [--launch-immediately] [--a11y=0|1]", argv[0]);
 536     }
 537
 538   _global_app->desktop_schema = get_desktop_schema ();
 539
 540   if (!a11y_set)
 541     {
 542       _global_app->a11y_enabled = _global_app->desktop_schema
 543                                   ? g_settings_get_boolean (_global_app->desktop_schema, "toolkit-accessibility")
 544                                   : _global_app->launch_immediately;
 545     }
 546
 547   if (_global_app->desktop_schema)
 548     g_signal_connect (_global_app->desktop_schema,
 549                       "changed::toolkit-accessibility",
 550                       G_CALLBACK (gsettings_key_changed), _global_app);
 551
 552   init_sigterm_handling (_global_app);
 553
 554   introspection_data = g_dbus_node_info_new_for_xml (introspection_xml, NULL);
 555   g_assert (introspection_data != NULL);
 556
 557   name_owner_id = g_bus_own_name (G_BUS_TYPE_SESSION,
 558                                   "org.a11y.Bus",
 559                                   G_BUS_NAME_OWNER_FLAGS_ALLOW_REPLACEMENT,
 560                                   on_bus_acquired,
 561                                   on_name_acquired,
 562                                   on_name_lost,
 563                                   _global_app,
 564                                   NULL);
 565
 566   g_main_loop_run (_global_app->loop);
 567
 568   if (_global_app->a11y_bus_pid > 0)
 569     kill (_global_app->a11y_bus_pid, SIGTERM);
 570
 571   /* Clear the X property if our bus is gone; in the case where e.g.
 572    * GDM is launching a login on an X server it was using before,
 573    * we don't want early login processes to pick up the stale address.
 574    */
 575   {
 576     Display *display = XOpenDisplay (NULL);
 577     if (display)
 578       {
 579         Atom bus_address_atom = XInternAtom (display, "AT_SPI_BUS", False);
 580         XDeleteProperty (display,
 581                          XDefaultRootWindow (display),
 582                          bus_address_atom);
 583
 584         XFlush (display);
 585         XCloseDisplay (display);
 586       }
 587   }
 588
 589   if (_global_app->a11y_launch_error_message)
 590     {
 591       g_printerr ("Failed to launch bus: %s", _global_app->a11y_launch_error_message);
 592       return 1;
 593     }
 594   return 0;
 595 }