You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
tdebase/tdm/backend/consolekit.c

558 lines
13 KiB

/*
Copyright (C) 2006-2007 William Jon McCann <mccann@jhu.edu>
Copyright (C) 2007 Kevin Kofler <Kevin@tigcc.ticalc.org>
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU Library General Public License
along with this library; see the file COPYING.LIB. If not, write to
the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
Boston, MA 02110-1301, USA.
*/
#include "dm.h"
#include "dm_auth.h"
#include "dm_error.h"
#include <stdlib.h>
#include <string.h>
#include <pwd.h>
#define DBUS_API_SUBJECT_TO_CHANGE
#include <dbus/dbus.h>
#include "consolekit.h"
#define CK_NAME "org.freedesktop.ConsoleKit"
#define CK_PATH "/org/freedesktop/ConsoleKit"
#define CK_INTERFACE "org.freedesktop.ConsoleKit"
#define CK_MANAGER_PATH "/org/freedesktop/ConsoleKit/Manager"
#define CK_MANAGER_INTERFACE "org.freedesktop.ConsoleKit.Manager"
#define CK_SESSION_INTERFACE "org.freedesktop.ConsoleKit.Session"
static DBusConnection *private_connection = NULL;
static void
add_param_int (DBusMessageIter *iter_struct,
const char *key,
int value)
{
DBusMessageIter iter_struct_entry;
DBusMessageIter iter_var;
dbus_message_iter_open_container (iter_struct,
DBUS_TYPE_STRUCT,
NULL,
&iter_struct_entry);
dbus_message_iter_append_basic (&iter_struct_entry,
DBUS_TYPE_STRING,
&key);
dbus_message_iter_open_container (&iter_struct_entry,
DBUS_TYPE_VARIANT,
DBUS_TYPE_INT32_AS_STRING,
&iter_var);
dbus_message_iter_append_basic (&iter_var,
DBUS_TYPE_INT32,
&value);
dbus_message_iter_close_container (&iter_struct_entry,
&iter_var);
dbus_message_iter_close_container (iter_struct, &iter_struct_entry);
}
static void
add_param_boolean (DBusMessageIter *iter_struct,
const char *key,
int value)
{
DBusMessageIter iter_struct_entry;
DBusMessageIter iter_var;
dbus_message_iter_open_container (iter_struct,
DBUS_TYPE_STRUCT,
NULL,
&iter_struct_entry);
dbus_message_iter_append_basic (&iter_struct_entry,
DBUS_TYPE_STRING,
&key);
dbus_message_iter_open_container (&iter_struct_entry,
DBUS_TYPE_VARIANT,
DBUS_TYPE_BOOLEAN_AS_STRING,
&iter_var);
dbus_message_iter_append_basic (&iter_var,
DBUS_TYPE_BOOLEAN,
&value);
dbus_message_iter_close_container (&iter_struct_entry,
&iter_var);
dbus_message_iter_close_container (iter_struct, &iter_struct_entry);
}
static void
add_param_string (DBusMessageIter *iter_struct,
const char *key,
const char *value)
{
DBusMessageIter iter_struct_entry;
DBusMessageIter iter_var;
dbus_message_iter_open_container (iter_struct,
DBUS_TYPE_STRUCT,
NULL,
&iter_struct_entry);
dbus_message_iter_append_basic (&iter_struct_entry,
DBUS_TYPE_STRING,
&key);
dbus_message_iter_open_container (&iter_struct_entry,
DBUS_TYPE_VARIANT,
DBUS_TYPE_STRING_AS_STRING,
&iter_var);
dbus_message_iter_append_basic (&iter_var,
DBUS_TYPE_STRING,
&value);
dbus_message_iter_close_container (&iter_struct_entry,
&iter_var);
dbus_message_iter_close_container (iter_struct, &iter_struct_entry);
}
static int
session_get_x11_display (DBusConnection *connection,
const char *ssid,
char **str)
{
DBusError error;
DBusMessage *message;
DBusMessage *reply;
DBusMessageIter iter;
const char *value;
if (str != NULL) {
*str = NULL;
}
message = dbus_message_new_method_call (CK_NAME,
ssid,
CK_SESSION_INTERFACE,
"GetX11Display");
if (message == NULL) {
Debug ("ConsoleKit: Couldn't allocate the D-Bus message");
return FALSE;
}
dbus_error_init (&error);
reply = dbus_connection_send_with_reply_and_block (connection,
message,
-1, &error);
if (dbus_error_is_set (&error)) {
Debug ("ConsoleKit: %s raised:\n %s\n\n", error.name, error.message);
reply = NULL;
}
dbus_connection_flush (connection);
dbus_message_unref (message);
if (reply == NULL) {
return FALSE;
}
dbus_message_iter_init (reply, &iter);
dbus_message_iter_get_basic (&iter, &value);
if (str != NULL) {
*str = strdup (value);
}
dbus_message_unref (reply);
return TRUE;
}
static int
session_unlock (DBusConnection *connection,
const char *ssid)
{
DBusError error;
DBusMessage *message;
DBusMessage *reply;
Debug ("ConsoleKit: Unlocking session %s", ssid);
message = dbus_message_new_method_call (CK_NAME,
ssid,
CK_SESSION_INTERFACE,
"Unlock");
if (message == NULL) {
Debug ("ConsoleKit: Couldn't allocate the D-Bus message");
return FALSE;
}
dbus_error_init (&error);
reply = dbus_connection_send_with_reply_and_block (connection,
message,
-1, &error);
dbus_message_unref (message);
dbus_message_unref (reply);
dbus_connection_flush (connection);
if (dbus_error_is_set (&error)) {
Debug ("ConsoleKit: %s raised:\n %s\n\n", error.name, error.message);
return FALSE;
}
return TRUE;
}
/* from libhal */
static char **
get_path_array_from_iter (DBusMessageIter *iter,
int *num_elements)
{
int count;
char **buffer;
count = 0;
buffer = (char **)malloc (sizeof (char *) * 8);
if (buffer == NULL)
goto oom;
buffer[0] = NULL;
while (dbus_message_iter_get_arg_type (iter) == DBUS_TYPE_OBJECT_PATH) {
const char *value;
char *str;
if ((count % 8) == 0 && count != 0) {
buffer = realloc (buffer, sizeof (char *) * (count + 8));
if (buffer == NULL)
goto oom;
}
dbus_message_iter_get_basic (iter, &value);
str = strdup (value);
if (str == NULL)
goto oom;
buffer[count] = str;
dbus_message_iter_next (iter);
count++;
}
if ((count % 8) == 0) {
buffer = realloc (buffer, sizeof (char *) * (count + 1));
if (buffer == NULL)
goto oom;
}
buffer[count] = NULL;
if (num_elements != NULL)
*num_elements = count;
return buffer;
oom:
LogWarn ("%s %d : error allocating memory\n", __FILE__, __LINE__);
return NULL;
}
static char **
get_sessions_for_user (DBusConnection *connection,
const char *user,
const char *x11_display)
{
DBusError error;
DBusMessage *message;
DBusMessage *reply;
DBusMessageIter iter;
DBusMessageIter iter_reply;
DBusMessageIter iter_array;
struct passwd *pwent;
char **sessions;
sessions = NULL;
message = NULL;
reply = NULL;
pwent = getpwnam (user);
dbus_error_init (&error);
message = dbus_message_new_method_call (CK_NAME,
CK_MANAGER_PATH,
CK_MANAGER_INTERFACE,
"GetSessionsForUser");
if (message == NULL) {
Debug ("ConsoleKit: Couldn't allocate the D-Bus message");
goto out;
}
dbus_message_iter_init_append (message, &iter);
dbus_message_iter_append_basic (&iter,
DBUS_TYPE_UINT32,
&pwent->pw_uid);
dbus_error_init (&error);
reply = dbus_connection_send_with_reply_and_block (connection,
message,
-1, &error);
dbus_connection_flush (connection);
if (dbus_error_is_set (&error)) {
Debug ("ConsoleKit: %s raised:\n %s\n\n", error.name, error.message);
goto out;
}
if (reply == NULL) {
Debug ("ConsoleKit: No reply for GetSessionsForUser");
goto out;
}
dbus_message_iter_init (reply, &iter_reply);
if (dbus_message_iter_get_arg_type (&iter_reply) != DBUS_TYPE_ARRAY) {
Debug ("ConsoleKit: Wrong reply for GetSessionsForUser - expecting an array.");
goto out;
}
dbus_message_iter_recurse (&iter_reply, &iter_array);
sessions = get_path_array_from_iter (&iter_array, NULL);
out:
if (message != NULL) {
dbus_message_unref (message);
}
if (reply != NULL) {
dbus_message_unref (reply);
}
return sessions;
}
void
unlock_ck_session (const char *user,
const char *x11_display)
{
DBusError error;
DBusConnection *connection;
char **sessions;
int i;
Debug ("ConsoleKit: Unlocking session for %s on %s", user, x11_display);
dbus_error_init (&error);
connection = dbus_bus_get (DBUS_BUS_SYSTEM, &error);
if (connection == NULL) {
Debug ("ConsoleKit: Failed to connect to the D-Bus daemon: %s", error.message);
dbus_error_free (&error);
return;
}
sessions = get_sessions_for_user (connection, user, x11_display);
if (sessions == NULL || sessions[0] == NULL) {
Debug ("ConsoleKit: no sessions found");
return;
}
for (i = 0; sessions[i] != NULL; i++) {
char *ssid;
char *xdisplay;
ssid = sessions[i];
session_get_x11_display (connection, ssid, &xdisplay);
Debug ("ConsoleKit: session %s has DISPLAY %s", ssid, xdisplay);
if (xdisplay != NULL
&& x11_display != NULL
&& strcmp (xdisplay, x11_display) == 0) {
int res;
res = session_unlock (connection, ssid);
if (! res) {
LogError ("ConsoleKit: Unable to unlock %s", ssid);
}
}
free (xdisplay);
}
freeStrArr (sessions);
}
char *
open_ck_session (struct passwd *pwent,
struct display *d)
{
DBusConnection *connection;
DBusError error;
DBusMessage *message;
DBusMessage *reply;
DBusMessageIter iter;
DBusMessageIter iter_struct;
char *cookie;
cookie = NULL;
if (pwent == NULL) {
Debug ("ConsoleKit: NULL user passed as parameter");
return NULL;
}
Debug ("ConsoleKit: Opening session for %s", pwent->pw_name);
dbus_error_init (&error);
connection = dbus_bus_get_private (DBUS_BUS_SYSTEM, &error);
private_connection = connection;
if (connection == NULL) {
Debug ("ConsoleKit: Failed to connect to the D-Bus daemon: %s", error.message);
dbus_error_free (&error);
return NULL;
}
dbus_connection_set_exit_on_disconnect (connection, FALSE);
/* FIXME: What to do about these?
dbus_connection_set_watch_functions( connection,
dbusAddWatch,
dbusRemoveWatch,
dbusToggleWatch,
data, 0 );
dbus_connection_set_timeout_functions( connection,
dbusAddTimeout,
dbusRemoveTimeout,
dbusToggleTimeout,
data, 0 );
dbus_connection_set_wakeup_main_function( connection,
dbusWakeupMain,
data, 0 ); */
dbus_error_init (&error);
message = dbus_message_new_method_call (CK_NAME,
CK_MANAGER_PATH,
CK_MANAGER_INTERFACE,
"OpenSessionWithParameters");
if (message == NULL) {
Debug ("ConsoleKit: Couldn't allocate the D-Bus message");
return NULL;
}
dbus_message_iter_init_append (message, &iter);
dbus_message_iter_open_container (&iter,
DBUS_TYPE_ARRAY,
DBUS_STRUCT_BEGIN_CHAR_AS_STRING
DBUS_TYPE_STRING_AS_STRING
DBUS_TYPE_VARIANT_AS_STRING
DBUS_STRUCT_END_CHAR_AS_STRING,
&iter_struct);
add_param_int (&iter_struct, "user", pwent->pw_uid);
add_param_string (&iter_struct, "x11-display", d->name);
add_param_boolean (&iter_struct, "is-local", ((d->displayType & d_location) == dLocal));
#ifdef XDMCP
if ((d->displayType & d_location) != dLocal) {
add_param_string (&iter_struct, "remote-host-name", d->remoteHost);
}
#endif
#ifdef HAVE_VTS
if (d->serverVT > 0) {
char device[20];
/* FIXME: how does xorg construct this */
sprintf(device, "/dev/tty%d", d->serverVT);
add_param_string (&iter_struct, "x11-display-device", device);
}
#endif
dbus_message_iter_close_container (&iter, &iter_struct);
reply = dbus_connection_send_with_reply_and_block (connection,
message,
-1, &error);
if (dbus_error_is_set (&error)) {
Debug ("ConsoleKit: %s raised:\n %s\n\n", error.name, error.message);
reply = NULL;
}
dbus_connection_flush (connection);
dbus_message_unref (message);
dbus_error_free (&error);
if (reply != NULL) {
const char *value;
dbus_message_iter_init (reply, &iter);
dbus_message_iter_get_basic (&iter, &value);
cookie = strdup (value);
dbus_message_unref (reply);
}
return cookie;
}
void
close_ck_session (const char *cookie)
{
DBusError error;
DBusMessage *message;
DBusMessage *reply;
DBusMessageIter iter;
if (cookie == NULL) {
return;
}
if (private_connection == NULL) {
return;
}
dbus_error_init (&error);
message = dbus_message_new_method_call (CK_NAME,
CK_MANAGER_PATH,
CK_MANAGER_INTERFACE,
"CloseSession");
if (message == NULL) {
Debug ("ConsoleKit: Couldn't allocate the D-Bus message");
return;
}
dbus_message_iter_init_append (message, &iter);
dbus_message_iter_append_basic (&iter,
DBUS_TYPE_STRING,
&cookie);
reply = dbus_connection_send_with_reply_and_block (private_connection,
message,
-1, &error);
if (dbus_error_is_set (&error)) {
Debug ("ConsoleKit: %s raised:\n %s\n\n", error.name, error.message);
reply = NULL;
}
dbus_connection_flush (private_connection);
dbus_message_unref (message);
dbus_error_free (&error);
dbus_connection_close (private_connection);
private_connection = NULL;
}