src/modules/module-console-kit.c

   1 /***
   2     This file is part of PulseAudio.
   3
   4     Copyright 2008 Lennart Poettering
   5
   6     PulseAudio is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU Lesser General Public License as published
   8     by the Free Software Foundation; either version 2 of the License,
   9     or (at your option) any later version.
  10
  11     PulseAudio is distributed in the hope that it will be useful, but
  12     WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14     General Public License for more details.
  15
  16     You should have received a copy of the GNU Lesser General Public License
  17     along with PulseAudio; if not, write to the Free Software
  18     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  19     USA.
  20 ***/
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include <config.h>
  24 #endif
  25
  26 #include <stdio.h>
  27 #include <unistd.h>
  28 #include <string.h>
  29 #include <stdlib.h>
  30 #include <errno.h>
  31 #include <stdlib.h>
  32 #include <sys/types.h>
  33 #include <sys/stat.h>
  34
  35 #include <pulse/xmalloc.h>
  36 #include <pulse/timeval.h>
  37
  38 #include <pulsecore/core-error.h>
  39 #include <pulsecore/module.h>
  40 #include <pulsecore/log.h>
  41 #include <pulsecore/hashmap.h>
  42 #include <pulsecore/idxset.h>
  43 #include <pulsecore/core-util.h>
  44 #include <pulsecore/namereg.h>
  45 #include <pulsecore/core-scache.h>
  46 #include <pulsecore/modargs.h>
  47
  48 #include "dbus-util.h"
  49 #include "module-console-kit-symdef.h"
  50
  51 PA_MODULE_AUTHOR("Lennart Poettering");
  52 PA_MODULE_DESCRIPTION("Create a client for each ConsoleKit session of this user");
  53 PA_MODULE_VERSION(PACKAGE_VERSION);
  54 PA_MODULE_LOAD_ONCE(TRUE);
  55
  56 static const char* const valid_modargs[] = {
  57     NULL
  58 };
  59
  60 struct session {
  61     char *id;
  62     pa_client *client;
  63 };
  64
  65 struct userdata {
  66     pa_core *core;
  67     pa_dbus_connection *connection;
  68     pa_hashmap *sessions;
  69 };
  70
  71 static void add_session(struct userdata *u, const char *id) {
  72     DBusError error;
  73     DBusMessage *m = NULL, *reply = NULL;
  74     int32_t uid;
  75     struct session *session;
  76     char *t;
  77
  78     dbus_error_init (&error);
  79
  80     if (pa_hashmap_get(u->sessions, id)) {
  81         pa_log_warn("Duplicate session %s, ignoring.", id);
  82         return;
  83     }
  84
  85     if (!(m = dbus_message_new_method_call("org.freedesktop.ConsoleKit", id, "org.freedesktop.ConsoleKit.Session", "GetUnixUser"))) {
  86         pa_log("Failed to allocate GetUnixUser() method call.");
  87         goto fail;
  88     }
  89
  90     if (!(reply = dbus_connection_send_with_reply_and_block(pa_dbus_connection_get(u->connection), m, -1, &error))) {
  91         pa_log("GetUnixUser() call failed: %s: %s", error.name, error.message);
  92         goto fail;
  93     }
  94
  95     /* FIXME: Why is this in int32? and not an uint32? */
  96     if (!dbus_message_get_args(reply, &error, DBUS_TYPE_INT32, &uid, DBUS_TYPE_INVALID)) {
  97         pa_log("Failed to parse GetUnixUser() result: %s: %s", error.name, error.message);
  98         goto fail;
  99     }
 100
 101     /* We only care about our own sessions */
 102     if ((uid_t) uid != getuid())
 103         goto fail;
 104
 105     session = pa_xnew(struct session, 1);
 106     session->id = pa_xstrdup(id);
 107
 108     t = pa_sprintf_malloc("ConsoleKit Session %s", id);
 109     session->client = pa_client_new(u->core, __FILE__, t);
 110     pa_xfree(t);
 111
 112     pa_proplist_sets(session->client->proplist, "console-kit.session", id);
 113
 114     pa_hashmap_put(u->sessions, session->id, session);
 115
 116     pa_log_debug("Added new session %s", id);
 117
 118 fail:
 119
 120     if (m)
 121         dbus_message_unref(m);
 122
 123     if (reply)
 124         dbus_message_unref(reply);
 125
 126     dbus_error_free(&error);
 127 }
 128
 129 static void free_session(struct session *session) {
 130     pa_assert(session);
 131
 132     pa_log_debug("Removing session %s", session->id);
 133
 134     pa_client_free(session->client);
 135     pa_xfree(session->id);
 136     pa_xfree(session);
 137 }
 138
 139 static void remove_session(struct userdata *u, const char *id) {
 140     struct session *session;
 141
 142     if (!(session = pa_hashmap_remove(u->sessions, id)))
 143         return;
 144
 145     free_session(session);
 146 }
 147
 148 static DBusHandlerResult filter_cb(DBusConnection *bus, DBusMessage *message, void *userdata) {
 149     struct userdata *u = userdata;
 150     DBusError error;
 151     const char *path;
 152
 153     pa_assert(bus);
 154     pa_assert(message);
 155     pa_assert(u);
 156
 157     dbus_error_init(&error);
 158
 159     pa_log_debug("dbus: interface=%s, path=%s, member=%s\n",
 160                  dbus_message_get_interface(message),
 161                  dbus_message_get_path(message),
 162                  dbus_message_get_member(message));
 163
 164     if (dbus_message_is_signal(message, "org.freedesktop.ConsoleKit.Seat", "SessionAdded")) {
 165
 166         if (!dbus_message_get_args(message, &error, DBUS_TYPE_STRING, &path, DBUS_TYPE_INVALID) || dbus_error_is_set(&error)) {
 167             pa_log_error("Failed to parse SessionAdded message: %s: %s", error.name, error.message);
 168             goto finish;
 169         }
 170
 171         add_session(u, path);
 172
 173     } else if (dbus_message_is_signal(message, "org.freedesktop.ConsoleKit.Seat", "SessionRemoved")) {
 174
 175         if (!dbus_message_get_args(message, &error, DBUS_TYPE_STRING, &path, DBUS_TYPE_INVALID) || dbus_error_is_set(&error)) {
 176             pa_log_error("Failed to parse SessionRemoved message: %s: %s", error.name, error.message);
 177             goto finish;
 178         }
 179
 180         remove_session(u, path);
 181     }
 182
 183 finish:
 184     dbus_error_free(&error);
 185
 186     return DBUS_HANDLER_RESULT_HANDLED;
 187 }
 188
 189 static int get_session_list(struct userdata *u) {
 190     DBusError error;
 191     DBusMessage *m = NULL, *reply = NULL;
 192     uint32_t uid;
 193     DBusMessageIter iter, sub;
 194     int ret = -1;
 195
 196     pa_assert(u);
 197
 198     dbus_error_init(&error);
 199
 200     if (!(m = dbus_message_new_method_call("org.freedesktop.ConsoleKit", "/org/freedesktop/ConsoleKit/Manager", "org.freedesktop.ConsoleKit.Manager", "GetSessionsForUnixUser"))) {
 201         pa_log("Failed to allocate GetSessionsForUnixUser() method call.");
 202         goto fail;
 203     }
 204
 205     uid = (uint32_t) getuid();
 206     if (!(dbus_message_append_args(m, DBUS_TYPE_UINT32, &uid, DBUS_TYPE_INVALID))) {
 207         pa_log("Failed to append arguments to GetSessionsForUnixUser() method call.");
 208         goto fail;
 209     }
 210
 211     if (!(reply = dbus_connection_send_with_reply_and_block(pa_dbus_connection_get(u->connection), m, -1, &error))) {
 212         pa_log("GetSessionsForUnixUser() call failed: %s: %s", error.name, error.message);
 213         goto fail;
 214     }
 215
 216     dbus_message_iter_init(reply, &iter);
 217
 218     if (dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_ARRAY ||
 219         dbus_message_iter_get_element_type(&iter) != DBUS_TYPE_OBJECT_PATH) {
 220         pa_log("Failed to parse GetSessionsForUnixUser() result.");
 221         goto fail;
 222     }
 223
 224     dbus_message_iter_recurse(&iter, &sub);
 225
 226     for (;;) {
 227         int at;
 228         const char *id;
 229
 230         if ((at = dbus_message_iter_get_arg_type(&sub)) == DBUS_TYPE_INVALID)
 231             break;
 232
 233         assert(at == DBUS_TYPE_OBJECT_PATH);
 234         dbus_message_iter_get_basic(&sub, &id);
 235
 236         add_session(u, id);
 237
 238         dbus_message_iter_next(&sub);
 239     }
 240
 241     ret = 0;
 242
 243 fail:
 244
 245     if (m)
 246         dbus_message_unref(m);
 247
 248     if (reply)
 249         dbus_message_unref(reply);
 250
 251     dbus_error_free(&error);
 252
 253     return ret;
 254 }
 255
 256 int pa__init(pa_module*m) {
 257     DBusError error;
 258     pa_dbus_connection *connection;
 259     struct userdata *u = NULL;
 260     pa_modargs *ma;
 261
 262     pa_assert(m);
 263
 264     dbus_error_init(&error);
 265
 266     if (!(ma = pa_modargs_new(m->argument, valid_modargs))) {
 267         pa_log("Failed to parse module arguments");
 268         goto fail;
 269     }
 270
 271     if (!(connection = pa_dbus_bus_get(m->core, DBUS_BUS_SYSTEM, &error)) || dbus_error_is_set(&error)) {
 272
 273         if (connection)
 274             pa_dbus_connection_unref(connection);
 275
 276         pa_log_error("Unable to contact D-Bus system bus: %s: %s", error.name, error.message);
 277         goto fail;
 278     }
 279
 280     m->userdata = u = pa_xnew(struct userdata, 1);
 281     u->core = m->core;
 282     u->connection = connection;
 283     u->sessions = pa_hashmap_new(pa_idxset_string_hash_func, pa_idxset_string_compare_func);
 284
 285     if (!dbus_connection_add_filter(pa_dbus_connection_get(connection), filter_cb, u, NULL)) {
 286         pa_log_error("Failed to add filter function");
 287         goto fail;
 288     }
 289
 290     dbus_bus_add_match(pa_dbus_connection_get(connection), "type='signal',sender='org.freedesktop.ConsoleKit', interface='org.freedesktop.ConsoleKit.Seat'", &error);
 291     if (dbus_error_is_set(&error)) {
 292         pa_log_error("Unable to subscribe to ConsoleKit signals: %s: %s", error.name, error.message);
 293         goto fail;
 294     }
 295
 296     if (get_session_list(u) < 0)
 297         goto fail;
 298
 299     pa_modargs_free(ma);
 300
 301     return 0;
 302
 303 fail:
 304     if (ma)
 305         pa_modargs_free(ma);
 306
 307     dbus_error_free(&error);
 308     pa__done(m);
 309
 310     return -1;
 311 }
 312
 313
 314 void pa__done(pa_module *m) {
 315     struct userdata *u;
 316     struct session *session;
 317
 318     pa_assert(m);
 319
 320     if (!(u = m->userdata))
 321         return;
 322
 323     if (u->sessions) {
 324         while ((session = pa_hashmap_steal_first(u->sessions)))
 325             free_session(session);
 326
 327         pa_hashmap_free(u->sessions, NULL, NULL);
 328     }
 329
 330     if (u->connection)
 331         pa_dbus_connection_unref(u->connection);
 332
 333     pa_xfree(u);
 334 }