Whoo, hoo, it builds

1 files changed, 214 insertions, 0 deletions

diff --git a/src/gtk-dialog/ck-pk-helper.c b/src/gtk-dialog/ck-pk-helper.c
new file mode 100644
index 0000000..9278905
--- /dev/null
+++ b/src/gtk-dialog/ck-pk-helper.c
@@ -0,0 +1,214 @@
+
+#include <unistd.h>
+#include <glib.h>
+#include <dbus/dbus-glib.h>
+#include <polkit-gnome/polkit-gnome.h>
+
+#include "logout-dialog.h"
+#include "ck-pk-helper.h"
+
+static gboolean
+ck_multiple_users (void)
+{
+	DBusGConnection * sbus = dbus_g_bus_get(DBUS_BUS_SYSTEM, NULL);
+	g_return_val_if_fail(sbus != NULL, TRUE); /* worst case */
+	DBusGProxy * proxy = dbus_g_proxy_new_for_name(sbus, "org.freedesktop.ConsoleKit",
+	                                                     "/org/freedesktop/ConsoleKit/Manager",
+	                                                     "org.freedesktop.ConsoleKit.Manager");
+
+	if (proxy == NULL) {
+		return TRUE;
+	}
+
+	gboolean result;
+	GPtrArray * seats = NULL;
+
+	result = dbus_g_proxy_call(proxy, "GetSeats", NULL, G_TYPE_INVALID,
+	                           dbus_g_type_get_collection("GPtrArray", DBUS_TYPE_G_OBJECT_PATH), &seats, G_TYPE_INVALID);
+
+	if (!result) {
+		g_warning("Unable to get the seats for ConsoleKit");
+		g_object_unref(proxy);
+		return TRUE;
+	}
+	
+	gchar * this_session_id = NULL;
+	
+	result = dbus_g_proxy_call(proxy, "GetCurrentSession", NULL, G_TYPE_INVALID,
+							DBUS_TYPE_G_OBJECT_PATH, &this_session_id, G_TYPE_INVALID);
+							
+	g_object_unref(proxy);
+							
+	if (!result) {
+		g_warning("Unable to get current session from ConsoleKit");
+		return TRUE;
+	}
+	
+	proxy = dbus_g_proxy_new_for_name(sbus, "org.freedesktop.ConsoleKit",
+									this_session_id, "org.freedesktop.ConsoleKit.Session");
+
+	if (proxy == NULL) {
+		return TRUE;
+	}
+	
+	guint this_session_uid;
+	
+	result = dbus_g_proxy_call(proxy, "GetUnixUser", NULL, G_TYPE_INVALID,
+							G_TYPE_UINT, &this_session_uid, G_TYPE_INVALID);										
+
+	if (!result) {
+		g_warning("Unable to get UID from ConsoleKit");
+		return TRUE;
+	}
+	
+	guint seat;
+	gboolean multiple_users = FALSE;
+	for (seat = 0; seat < seats->len; seat++) {
+		gchar * seat_id = g_ptr_array_index(seats, seat);
+		DBusGProxy * seat_proxy = dbus_g_proxy_new_for_name(sbus, "org.freedesktop.ConsoleKit",
+		                                                    seat_id, "org.freedesktop.ConsoleKit.Seat");
+		g_free(seat_id);
+
+		if (seat_proxy == NULL) {
+			continue;
+		}
+
+		GPtrArray * sessions = NULL;
+
+		gboolean result = dbus_g_proxy_call(seat_proxy,
+		                                    "GetSessions", NULL, G_TYPE_INVALID,
+	                                        dbus_g_type_get_collection("GPtrArray", DBUS_TYPE_G_OBJECT_PATH), &sessions, G_TYPE_INVALID);
+
+		g_object_unref(seat_proxy);
+		if (!result) {
+			continue;
+		}
+
+		guint session;
+		for (session = 0; session < sessions->len; session++) {
+			gchar * session_id = g_ptr_array_index(sessions, session);
+			if (g_strcmp0(this_session_id, session_id) == 0) {
+				continue;
+			}
+			DBusGProxy * session_proxy = dbus_g_proxy_new_for_name(sbus, "org.freedesktop.ConsoleKit",
+		                                                    	session_id, "org.freedesktop.ConsoleKit.Session");
+		    g_free(session_id);
+		    
+		    if (session_proxy == NULL) {
+		    	continue;
+		    }
+		    
+		    guint session_uid;
+		    result =  dbus_g_proxy_call(session_proxy, "GetUnixUser", NULL, G_TYPE_INVALID,
+		    							G_TYPE_UINT, &session_uid, G_TYPE_INVALID);	
+		    g_object_unref(session_proxy);
+		    							
+		    if (!result) {
+		    	continue;
+		    }
+		    
+		    if (session_uid != this_session_uid) {
+		    	multiple_users = TRUE;
+		    	break;
+		    }
+		}
+		
+		g_ptr_array_free(sessions, TRUE);
+		
+		if (multiple_users) {
+			break;
+		}
+	}
+
+	g_ptr_array_free(seats, TRUE);
+	g_object_unref(proxy);
+	g_free(this_session_id);
+
+	return multiple_users;
+}
+
+gboolean
+pk_require_auth (LogoutDialogAction action) {
+	if (action == LOGOUT_DIALOG_LOGOUT) {
+		return FALSE;
+	}
+
+	gchar * pk_action;
+	if (ck_multiple_users()) {
+		if (action == LOGOUT_DIALOG_RESTART) {
+			pk_action = "org.freedesktop.consolekit.system.restart-multiple-users";
+		} else {
+			pk_action = "org.freedesktop.consolekit.system.stop-multiple-users";
+		}
+	} else {
+		if (action == LOGOUT_DIALOG_RESTART) {
+			pk_action = "org.freedesktop.consolekit.system.restart";
+		} else {
+			pk_action = "org.freedesktop.consolekit.system.stop";
+		}
+	}
+
+	PolKitResult polres;
+	if (pk_can_do_action(pk_action, &polres)) {
+		if (polres == POLKIT_RESULT_YES) {
+			return FALSE;
+		}
+		return TRUE;
+	}
+	return FALSE;
+}
+
+gboolean
+pk_can_do_action (const gchar    *action_id, PolKitResult * pol_result)
+{
+        PolKitGnomeContext *gnome_context;
+        PolKitAction *action;
+        PolKitCaller *caller;
+        DBusError dbus_error;
+        PolKitError *error;
+        PolKitResult result;
+
+        gnome_context = polkit_gnome_context_get (NULL);
+
+        if (gnome_context == NULL) {
+                return FALSE;
+        }
+
+        if (gnome_context->pk_tracker == NULL) {
+                return FALSE;
+        }
+
+        dbus_error_init (&dbus_error);
+        caller = polkit_tracker_get_caller_from_pid (gnome_context->pk_tracker,
+                                                     getpid (),
+                                                     &dbus_error);
+        dbus_error_free (&dbus_error);
+
+        if (caller == NULL) {
+                return FALSE;
+        }
+
+        action = polkit_action_new ();
+        if (!polkit_action_set_action_id (action, action_id)) {
+                polkit_action_unref (action);
+                polkit_caller_unref (caller);
+                return FALSE;
+        }
+
+        result = POLKIT_RESULT_UNKNOWN;
+        error = NULL;
+        result = polkit_context_is_caller_authorized (gnome_context->pk_context,
+                                                      action, caller, FALSE,
+                                                      &error);
+        if (polkit_error_is_set (error)) {
+                polkit_error_free (error);
+        }
+        polkit_action_unref (action);
+                polkit_caller_unref (caller);
+
+		if (pol_result != NULL) {
+			*pol_result = result;
+		}
+
+        return result != POLKIT_RESULT_NO && result != POLKIT_RESULT_UNKNOWN;
+}