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:
Christian Hergert
2019-05-10 09:23:17 -07:00
parent 1d62916295
commit 0e20e98801
2 changed files with 116 additions and 0 deletions

View File

@ -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);
}

View File

@ -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,