[sysprof] libsysprof: implement various serialization hooks

[Christian Hergert <chergert src gnome org>]

commit 404317d49df595fd6bce48c33435f84214d9a34d
Author: Christian Hergert <chergert redhat com>
Date:   Thu May 23 18:09:42 2019 -0700

    libsysprof: implement various serialization hooks

 src/libsysprof/sysprof-local-profiler.c | 41 ++++++++++++++++++++++++++++++---
 src/libsysprof/sysprof-source.c         | 26 +++++++++++++++++++++
 src/libsysprof/sysprof-source.h         | 33 ++++++++++++++++++++++++++
 3 files changed, 97 insertions(+), 3 deletions(-)
---
diff --git a/src/libsysprof/sysprof-local-profiler.c b/src/libsysprof/sysprof-local-profiler.c
index 03ab028..836e514 100644
--- a/src/libsysprof/sysprof-local-profiler.c
+++ b/src/libsysprof/sysprof-local-profiler.c
@@ -33,6 +33,13 @@
 #include "sysprof-local-profiler.h"
 #include "sysprof-platform.h"
 
+#include "sysprof-hostinfo-source.h"
+#ifdef __linux__
+# include "sysprof-perf-source.h"
+#endif
+#include "sysprof-proc-source.h"
+#include "sysprof-proxy-source.h"
+
 #define CSTRV(s) ((const gchar * const *)s)
 
 typedef struct
@@ -386,6 +393,13 @@ sysprof_local_profiler_class_init (SysprofLocalProfilerClass *klass)
   g_object_class_override_property (object_class, PROP_SPAWN_ENV, "spawn-env");
   g_object_class_override_property (object_class, PROP_SPAWN_INHERIT_ENVIRON, "spawn-inherit-environ");
   g_object_class_override_property (object_class, PROP_WHOLE_SYSTEM, "whole-system");
+
+  g_type_ensure (SYSPROF_TYPE_HOSTINFO_SOURCE);
+  g_type_ensure (SYSPROF_TYPE_PROC_SOURCE);
+#ifdef __linux__
+  g_type_ensure (SYSPROF_TYPE_PERF_SOURCE);
+#endif
+  g_type_ensure (SYSPROF_TYPE_PROXY_SOURCE);
 }
 
 static void
@@ -627,13 +641,15 @@ sysprof_local_profiler_authorize_cb (GObject      *object,
         }
     }
 
+  g_key_file_set_integer (keyfile, "profiler", "n-sources", priv->sources->len);
+
   for (guint i = 0; i < priv->sources->len; i++)
     {
       SysprofSource *source = g_ptr_array_index (priv->sources, i);
-      g_autofree gchar *name = g_strdup_printf ("Source.%s", G_OBJECT_TYPE_NAME (source));
+      g_autofree gchar *group = g_strdup_printf ("source-%u", i);
 
-      /* TODO: Allow sources to add information here */
-      g_key_file_set_boolean (keyfile, name, "enabled", TRUE);
+      g_key_file_set_string (keyfile, group, "gtype", G_OBJECT_TYPE_NAME (source));
+      sysprof_source_serialize (source, keyfile, group);
 
       if (priv->whole_system == FALSE)
         {
@@ -980,6 +996,7 @@ sysprof_local_profiler_new_replay (SysprofCaptureReader *reader)
   g_auto(GStrv) env = NULL;
   gboolean inherit;
   gboolean spawn;
+  gint n_sources;
 
   g_return_val_if_fail (reader != NULL, NULL);
 
@@ -997,11 +1014,29 @@ sysprof_local_profiler_new_replay (SysprofCaptureReader *reader)
   inherit = g_key_file_get_boolean (keyfile, "profiler", "spawn-inherit-environ", NULL);
   argv = g_key_file_get_string_list (keyfile, "profiler", "spawn-argv", NULL, NULL);
   env = g_key_file_get_string_list (keyfile, "profiler", "spawn-env", NULL, NULL);
+  n_sources = g_key_file_get_integer (keyfile, "profiler", "n-sources", NULL);
 
   sysprof_profiler_set_spawn (SYSPROF_PROFILER (self), spawn);
   sysprof_profiler_set_spawn_argv (SYSPROF_PROFILER (self), CSTRV (argv));
   sysprof_profiler_set_spawn_env (SYSPROF_PROFILER (self), CSTRV (env));
   sysprof_profiler_set_spawn_inherit_environ (SYSPROF_PROFILER (self), inherit);
 
+  for (guint i = 0; i < n_sources; i++)
+    {
+      g_autofree gchar *group = g_strdup_printf ("source-%u", i);
+      g_autofree gchar *type_name = NULL;
+      g_autoptr(SysprofSource) source = NULL;
+      GType gtype;
+
+      if (!g_key_file_has_group (keyfile, group) ||
+          !(type_name = g_key_file_get_string (keyfile, group, "gtype", NULL)) ||
+          !(gtype = g_type_from_name (type_name)) ||
+          !(source = g_object_new (gtype, NULL)))
+        continue;
+
+      sysprof_source_deserialize (source, keyfile, group);
+      sysprof_local_profiler_add_source (SYSPROF_PROFILER (self), source);
+    }
+
   return SYSPROF_PROFILER (g_steal_pointer (&self));
 }
diff --git a/src/libsysprof/sysprof-source.c b/src/libsysprof/sysprof-source.c
index 420cf14..bf0a23d 100644
--- a/src/libsysprof/sysprof-source.c
+++ b/src/libsysprof/sysprof-source.c
@@ -152,3 +152,29 @@ sysprof_source_modify_spawn (SysprofSource       *self,
   if (SYSPROF_SOURCE_GET_IFACE (self)->modify_spawn)
     SYSPROF_SOURCE_GET_IFACE (self)->modify_spawn (self, launcher, argv);
 }
+
+void
+sysprof_source_serialize (SysprofSource *self,
+                          GKeyFile      *keyfile,
+                          const gchar   *group)
+{
+  g_return_if_fail (SYSPROF_IS_SOURCE (self));
+  g_return_if_fail (keyfile != NULL);
+  g_return_if_fail (group != NULL);
+
+  if (SYSPROF_SOURCE_GET_IFACE (self)->serialize)
+    SYSPROF_SOURCE_GET_IFACE (self)->serialize (self, keyfile, group);
+}
+
+void
+sysprof_source_deserialize (SysprofSource *self,
+                            GKeyFile      *keyfile,
+                            const gchar   *group)
+{
+  g_return_if_fail (SYSPROF_IS_SOURCE (self));
+  g_return_if_fail (keyfile != NULL);
+  g_return_if_fail (group != NULL);
+
+  if (SYSPROF_SOURCE_GET_IFACE (self)->deserialize)
+    SYSPROF_SOURCE_GET_IFACE (self)->deserialize (self, keyfile, group);
+}
diff --git a/src/libsysprof/sysprof-source.h b/src/libsysprof/sysprof-source.h
index 31c8989..bc391b4 100644
--- a/src/libsysprof/sysprof-source.h
+++ b/src/libsysprof/sysprof-source.h
@@ -132,6 +132,31 @@ struct _SysprofSourceInterface
   void (*modify_spawn) (SysprofSource       *self,
                         GSubprocessLauncher *launcher,
                         GPtrArray           *argv);
+
+  /**
+   * SysprofSource::serialize:
+   * @self: a #SysprofSource
+   * @keyfile: a #GKeyFile
+   * @group: the keyfile group to use
+   *
+   * Requests that the source serialize itself into the keyfile
+   * so that it may be replayed at a future point in time.
+   */
+  void (*serialize) (SysprofSource *self,
+                     GKeyFile      *keyfile,
+                     const gchar   *group);
+
+  /**
+   * SysprofSource::deserialize:
+   * @self: a #SysprofSource
+   * @keyfile: a #GKeyFile
+   * @group: the keyfile group to use
+   *
+   * Deserialize from the saved state.
+   */
+  void (*deserialize) (SysprofSource *self,
+                       GKeyFile      *keyfile,
+                       const gchar   *group);
 };
 
 SYSPROF_AVAILABLE_IN_ALL
@@ -159,5 +184,13 @@ SYSPROF_AVAILABLE_IN_ALL
 void     sysprof_source_modify_spawn  (SysprofSource        *self,
                                        GSubprocessLauncher  *launcher,
                                        GPtrArray            *argv);
+SYSPROF_AVAILABLE_IN_ALL
+void     sysprof_source_serialize     (SysprofSource        *self,
+                                       GKeyFile             *keyfile,
+                                       const gchar          *group);
+SYSPROF_AVAILABLE_IN_ALL
+void     sysprof_source_deserialize   (SysprofSource        *self,
+                                       GKeyFile             *keyfile,
+                                       const gchar          *group);
 
 G_END_DECLS