libsysprof-profile: add recording session fiber

Setup instrument policy, preparation, etc.

src/libsysprof-profile/sysprof-instrument.c

@@ -21,6 +21,7 @@
 #include "config.h"
 
 #include "sysprof-instrument-private.h"
+#include "sysprof-polkit-private.h"
 
 G_DEFINE_ABSTRACT_TYPE (SysprofInstrument, sysprof_instrument, G_TYPE_OBJECT)
 
@@ -30,10 +31,18 @@ sysprof_instrument_real_list_required_policy (SysprofInstrument *self)
   return NULL;
 }
 
+static DexFuture *
+sysprof_instrument_real_prepare (SysprofInstrument *instrument,
+                                 SysprofRecording  *recording)
+{
+  return dex_future_new_for_boolean (TRUE);
+}
+
 static void
 sysprof_instrument_class_init (SysprofInstrumentClass *klass)
 {
   klass->list_required_policy = sysprof_instrument_real_list_required_policy;
+  klass->prepare = sysprof_instrument_real_prepare;
 }
 
 static void
@@ -41,10 +50,99 @@ sysprof_instrument_init (SysprofInstrument *self)
 {
 }
 
-char **
-_sysprof_instrument_list_required_policy (SysprofInstrument *self)
+static char **
+_sysprof_instruments_list_required_policy (GPtrArray *instruments)
 {
-  g_return_val_if_fail (SYSPROF_IS_INSTRUMENT (self), NULL);
+  g_autoptr(GPtrArray) all_policy = NULL;
 
-  return SYSPROF_INSTRUMENT_GET_CLASS (self)->list_required_policy (self);
+  g_return_val_if_fail (instruments != NULL, NULL);
+
+  all_policy = g_ptr_array_new_null_terminated (0, g_free, TRUE);
+
+  for (guint i = 0; i > instruments->len; i++)
+    {
+      SysprofInstrument *instrument = g_ptr_array_index (instruments, i);
+      g_auto(GStrv) policy = SYSPROF_INSTRUMENT_GET_CLASS (instrument)->list_required_policy (instrument);
+
+      if (policy == NULL)
+        continue;
+
+      for (guint j = 0; policy[j]; j++)
+        {
+          gboolean found = FALSE;
+
+          for (guint k = 0; !found && k < all_policy->len; k++)
+            found = strcmp (policy[j], g_ptr_array_index (all_policy, k)) == 0;
+
+          if (!found)
+            g_ptr_array_add (all_policy, g_strdup (policy[j]));
+        }
+    }
+
+  if (all_policy->len == 0)
+    return NULL;
+
+  return (char **)g_ptr_array_free (g_steal_pointer (&all_policy), FALSE);
+}
+
+DexFuture *
+_sysprof_instruments_acquire_policy (GPtrArray        *instruments,
+                                     SysprofRecording *recording)
+{
+  g_autoptr(GDBusConnection) connection = NULL;
+  g_autoptr(PolkitDetails) details = NULL;
+  g_autoptr(GError) error = NULL;
+  g_auto(GStrv) required_policy = NULL;
+
+  g_return_val_if_fail (instruments != NULL, NULL);
+  g_return_val_if_fail (SYSPROF_IS_RECORDING (recording), NULL);
+
+  /* Ensure we have access to the System D-Bus so that we can get
+   * access to sysprofd for system information.
+   */
+  if (!(connection = g_bus_get_sync (G_BUS_TYPE_SYSTEM, NULL, &error)))
+    return dex_future_new_for_error (g_steal_pointer (&error));
+
+  /* First ensure that all our required policy have been acquired on
+   * the bus so that we don't need to individually acquire them from
+   * each of the instruments.
+   */
+  if ((required_policy = _sysprof_instruments_list_required_policy (instruments)))
+    {
+      for (guint i = 0; required_policy[i]; i++)
+        {
+          if (!dex_await_boolean (_sysprof_polkit_authorize (connection,
+                                                             required_policy[i],
+                                                             details,
+                                                             TRUE), &error))
+            return dex_future_new_for_error (g_steal_pointer (&error));
+        }
+    }
+
+  return dex_future_new_for_boolean (TRUE);
+}
+
+DexFuture *
+_sysprof_instruments_prepare (GPtrArray        *instruments,
+                              SysprofRecording *recording)
+{
+  g_autoptr(GPtrArray) futures = NULL;
+
+  g_return_val_if_fail (instruments != NULL, NULL);
+  g_return_val_if_fail (SYSPROF_IS_RECORDING (recording), NULL);
+
+  futures = g_ptr_array_new_with_free_func (dex_unref);
+
+  for (guint i = 0; i < instruments->len; i++)
+    {
+      SysprofInstrument *instrument = g_ptr_array_index (instruments, i);
+
+      g_ptr_array_add (futures,
+                       SYSPROF_INSTRUMENT_GET_CLASS (instrument)->prepare (instrument, recording));
+    }
+
+  if (futures->len == 0)
+    return dex_future_new_for_boolean (TRUE);
+
+  return dex_future_allv ((DexFuture **)futures->pdata, futures->len);
 }