From 731841a52397ad6a02d1cc4cb04e71083c6da35e Mon Sep 17 00:00:00 2001
From: Christian Hergert <chergert@redhat.com>
Date: Thu, 25 May 2023 17:25:13 -0700
Subject: [PATCH] libsysprof-profile: wait for recording and seutp ctrl+c

---
 src/libsysprof-profile/tests/test-profiler.c | 50 ++++++++++++++++++++
 1 file changed, 50 insertions(+)

diff --git a/src/libsysprof-profile/tests/test-profiler.c b/src/libsysprof-profile/tests/test-profiler.c
index 2a6fb469..6624762c 100644
--- a/src/libsysprof-profile/tests/test-profiler.c
+++ b/src/libsysprof-profile/tests/test-profiler.c
@@ -18,15 +18,34 @@
  * SPDX-License-Identifier: GPL-3.0-or-later
  */
 
+#include <glib-unix.h>
+
 #include <sysprof-profile.h>
 
 static GMainLoop *main_loop;
 static char *capture_file;
+static SysprofRecording *active_recording;
 static const GOptionEntry entries[] = {
   { "capture", 'c', 0, G_OPTION_ARG_FILENAME, &capture_file, "The file to capture into", "CAPTURE" },
   { 0 }
 };
 
+static void
+wait_cb (GObject      *object,
+         GAsyncResult *result,
+         gpointer      user_data)
+{
+  SysprofRecording *recording = (SysprofRecording *)object;
+  g_autoptr(GError) error = NULL;
+  gboolean r;
+
+  r = sysprof_recording_wait_finish (recording, result, &error);
+  g_assert_no_error (error);
+  g_assert_true (r);
+
+  g_main_loop_quit (main_loop);
+}
+
 static void
 record_cb (GObject      *object,
            GAsyncResult *result,
@@ -38,6 +57,34 @@ record_cb (GObject      *object,
   g_assert_no_error (error);
   g_assert_nonnull (recording);
   g_assert_true (SYSPROF_IS_RECORDING (recording));
+
+  sysprof_recording_wait_async (recording, NULL, wait_cb, NULL);
+
+  active_recording = recording;
+}
+
+static gboolean
+sigint_handler (gpointer user_data)
+{
+  static int count;
+
+  if (count >= 2)
+    {
+      g_main_loop_quit (main_loop);
+      return G_SOURCE_REMOVE;
+    }
+
+  g_printerr ("\n");
+
+  if (count == 0)
+    {
+      g_printerr ("%s\n", "Stopping profiler. Press twice more ^C to force exit.");
+      sysprof_recording_stop (active_recording);
+    }
+
+  count++;
+
+  return G_SOURCE_CONTINUE;
 }
 
 int
@@ -68,6 +115,9 @@ main (int       argc,
 
   sysprof_profiler_record_async (profiler, writer, NULL, record_cb, NULL);
 
+  g_unix_signal_add (SIGINT, sigint_handler, main_loop);
+  g_unix_signal_add (SIGTERM, sigint_handler, main_loop);
+
   g_main_loop_run (main_loop);
 
   g_clear_pointer (&writer, sysprof_capture_writer_unref);