mirror of
https://github.com/varun-r-mallya/sysprof.git
synced 2025-12-31 20:36:25 +00:00
libsysprof: add authorize_async() to authorize bus
We want a single authorization request from the user when starting a profile, and to do it as early as possible. This will allow for the local profiler to set this up before starting things.
This commit is contained in:
@ -23,6 +23,7 @@
|
||||
#include "config.h"
|
||||
|
||||
#include <gio/gunixfdlist.h>
|
||||
#include <polkit/polkit.h>
|
||||
|
||||
#include "sysprof-helpers.h"
|
||||
|
||||
@ -524,3 +525,102 @@ sysprof_helpers_perf_event_open_finish (SysprofHelpers *self,
|
||||
return FALSE;
|
||||
}
|
||||
#endif /* __linux__ */
|
||||
|
||||
static void
|
||||
sysprof_helpers_check_authorization_cb (GObject *object,
|
||||
GAsyncResult *result,
|
||||
gpointer user_data)
|
||||
{
|
||||
PolkitAuthority *authority = (PolkitAuthority *)object;
|
||||
g_autoptr(PolkitAuthorizationResult) res = NULL;
|
||||
g_autoptr(GError) error = NULL;
|
||||
g_autoptr(GTask) task = user_data;
|
||||
|
||||
g_assert (POLKIT_IS_AUTHORITY (authority));
|
||||
g_assert (G_IS_ASYNC_RESULT (result));
|
||||
g_assert (G_IS_TASK (task));
|
||||
|
||||
if (!(res = polkit_authority_check_authorization_finish (authority, result, &error)))
|
||||
g_task_return_error (task, g_steal_pointer (&error));
|
||||
else if (!polkit_authorization_result_get_is_authorized (res))
|
||||
g_task_return_new_error (task,
|
||||
G_IO_ERROR,
|
||||
G_IO_ERROR_PROXY_AUTH_FAILED,
|
||||
"Failed to authorize user credentials");
|
||||
else
|
||||
g_task_return_boolean (task, TRUE);
|
||||
}
|
||||
|
||||
static void
|
||||
sysprof_helpers_get_authority_cb (GObject *object,
|
||||
GAsyncResult *result,
|
||||
gpointer user_data)
|
||||
{
|
||||
g_autoptr(PolkitAuthority) authority = user_data;
|
||||
g_autoptr(GError) error = NULL;
|
||||
g_autoptr(GTask) task = user_data;
|
||||
PolkitSubject *subject;
|
||||
GCancellable *cancellable;
|
||||
|
||||
g_assert (G_IS_ASYNC_RESULT (result));
|
||||
g_assert (G_IS_TASK (task));
|
||||
|
||||
cancellable = g_task_get_cancellable (task);
|
||||
subject = g_task_get_task_data (task);
|
||||
|
||||
g_assert (!cancellable || G_IS_CANCELLABLE (cancellable));
|
||||
g_assert (POLKIT_IS_SUBJECT (subject));
|
||||
|
||||
if (!(authority = polkit_authority_get_finish (result, &error)))
|
||||
g_task_return_error (task, g_steal_pointer (&error));
|
||||
else
|
||||
polkit_authority_check_authorization (authority,
|
||||
subject,
|
||||
"org.gnome.sysprof3.profile",
|
||||
NULL,
|
||||
POLKIT_CHECK_AUTHORIZATION_FLAGS_ALLOW_USER_INTERACTION,
|
||||
cancellable,
|
||||
sysprof_helpers_check_authorization_cb,
|
||||
g_steal_pointer (&task));
|
||||
}
|
||||
|
||||
void
|
||||
sysprof_helpers_authorize_async (SysprofHelpers *self,
|
||||
GCancellable *cancellable,
|
||||
GAsyncReadyCallback callback,
|
||||
gpointer user_data)
|
||||
{
|
||||
g_autoptr(GTask) task = NULL;
|
||||
g_autoptr(PolkitSubject) subject = NULL;
|
||||
GDBusConnection *bus;
|
||||
const gchar *unique_name;
|
||||
|
||||
g_return_if_fail (SYSPROF_IS_HELPERS (self));
|
||||
g_return_if_fail (!cancellable || G_IS_CANCELLABLE (cancellable));
|
||||
|
||||
task = g_task_new (self, cancellable, callback, user_data);
|
||||
g_task_set_source_tag (task, sysprof_helpers_authorize_async);
|
||||
|
||||
if (fail_if_no_proxy (self, task))
|
||||
return;
|
||||
|
||||
bus = g_dbus_proxy_get_connection (G_DBUS_PROXY (self->proxy));
|
||||
unique_name = g_dbus_connection_get_unique_name (bus);
|
||||
subject = polkit_system_bus_name_new (unique_name);
|
||||
g_task_set_task_data (task, g_steal_pointer (&subject), g_object_unref);
|
||||
|
||||
polkit_authority_get_async (cancellable,
|
||||
sysprof_helpers_get_authority_cb,
|
||||
g_steal_pointer (&task));
|
||||
}
|
||||
|
||||
gboolean
|
||||
sysprof_helpers_authorize_finish (SysprofHelpers *self,
|
||||
GAsyncResult *result,
|
||||
GError **error)
|
||||
{
|
||||
g_return_val_if_fail (SYSPROF_IS_HELPERS (self), FALSE);
|
||||
g_return_val_if_fail (G_IS_TASK (result), FALSE);
|
||||
|
||||
return g_task_propagate_boolean (G_TASK (result), error);
|
||||
}
|
||||
|
||||
@ -33,6 +33,13 @@ G_BEGIN_DECLS
|
||||
G_DECLARE_FINAL_TYPE (SysprofHelpers, sysprof_helpers, SYSPROF, HELPERS, GObject)
|
||||
|
||||
SysprofHelpers *sysprof_helpers_get_default (void);
|
||||
void sysprof_helpers_authorize_async (SysprofHelpers *self,
|
||||
GCancellable *cancellable,
|
||||
GAsyncReadyCallback callback,
|
||||
gpointer user_data);
|
||||
gboolean sysprof_helpers_authorize_finish (SysprofHelpers *self,
|
||||
GAsyncResult *result,
|
||||
GError **error);
|
||||
gboolean sysprof_helpers_list_processes (SysprofHelpers *self,
|
||||
GCancellable *cancellable,
|
||||
gint32 **processes,
|
||||
@ -62,6 +69,15 @@ gboolean sysprof_helpers_get_proc_file_finish (SysprofHelpers
|
||||
gchar **contents,
|
||||
GError **error);
|
||||
#ifdef __linux__
|
||||
gboolean sysprof_helpers_perf_event_open (SysprofHelpers *self,
|
||||
struct perf_event_attr *attr,
|
||||
gint32 pid,
|
||||
gint32 cpu,
|
||||
gint32 group_fd,
|
||||
guint64 flags,
|
||||
GCancellable *cancellable,
|
||||
gint *out_fd,
|
||||
GError **error);
|
||||
void sysprof_helpers_perf_event_open_async (SysprofHelpers *self,
|
||||
struct perf_event_attr *attr,
|
||||
gint32 pid,
|
||||
|
||||
Reference in New Issue
Block a user