[sysprof] tools: add tool to toggle a profiler

[Christian Hergert <chergert src gnome org>]

commit adcf667b5a86e2e043d073d09c78978b7ce7227b
Author: Christian Hergert <chergert redhat com>
Date:   Thu May 16 15:29:55 2019 -0700

    tools: add tool to toggle a profiler
    
    This is in-tree only currently, although we may consider installing it
    at some point (or adding it to sysprof-cli).

 src/meson.build                     |   6 ++
 src/org.gnome.Sysprof3.Profiler.xml |  10 ++
 src/tools/meson.build               |   6 ++
 src/tools/sysprof-profiler-ctl.c    | 197 ++++++++++++++++++++++++++++++++++++
 4 files changed, 219 insertions(+)
---
diff --git a/src/meson.build b/src/meson.build
index f0db4b7..c3440a4 100644
--- a/src/meson.build
+++ b/src/meson.build
@@ -7,6 +7,12 @@ sysprof_version_conf.set('MINOR_VERSION', sysprof_version[1])
 sysprof_version_conf.set('MICRO_VERSION', sysprof_version[2])
 sysprof_version_conf.set('VERSION', meson.project_version())
 
+ipc_profiler_src = gnome.gdbus_codegen('ipc-profiler',
+           sources: 'org.gnome.Sysprof3.Profiler.xml',
+  interface_prefix: 'org.gnome.Sysprof3.',
+         namespace: 'Ipc',
+)
+
 ipc_service_src = gnome.gdbus_codegen('ipc-service',
            sources: 'org.gnome.Sysprof3.Service.xml',
   interface_prefix: 'org.gnome.Sysprof3.',
diff --git a/src/org.gnome.Sysprof3.Profiler.xml b/src/org.gnome.Sysprof3.Profiler.xml
new file mode 100644
index 0000000..4ca4ca0
--- /dev/null
+++ b/src/org.gnome.Sysprof3.Profiler.xml
@@ -0,0 +1,10 @@
+<!DOCTYPE node PUBLIC "-//freedesktop//DTD D-BUS Object Introspection 1.0//EN"
+"http://www.freedesktop.org/standards/dbus/1.0/introspect.dtd";>
+<node>
+  <interface name="org.gnome.Sysprof3.Profiler">
+    <method name='Start'>
+      <arg type='h' name='fd' direction='in'/>
+    </method>
+    <method name='Stop'/>
+  </interface>
+</node>
diff --git a/src/tools/meson.build b/src/tools/meson.build
index 1c7a0e4..3c81235 100644
--- a/src/tools/meson.build
+++ b/src/tools/meson.build
@@ -21,3 +21,9 @@ sysprof_dump = executable('sysprof-dump', 'sysprof-dump.c',
    install_dir: get_option('bindir'),
        install: true,
 )
+
+sysprof_profiler_ctl = executable('sysprof-profiler-ctl',
+  [ 'sysprof-profiler-ctl.c', ipc_profiler_src ],
+  dependencies: [ tools_deps, gio_unix_dep ],
+       install: false,
+)
diff --git a/src/tools/sysprof-profiler-ctl.c b/src/tools/sysprof-profiler-ctl.c
new file mode 100644
index 0000000..5adc592
--- /dev/null
+++ b/src/tools/sysprof-profiler-ctl.c
@@ -0,0 +1,197 @@
+/* sysprof-profiler-ctl.c
+ *
+ * Copyright 2019 Christian Hergert <chergert redhat com>
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ * SPDX-License-Identifier: GPL-3.0-or-later
+ */
+
+#define G_LOG_DOMAIN "sysprof-profiler-ctl"
+
+#include "config.h"
+
+#include <fcntl.h>
+#include <glib-unix.h>
+#include <glib/gi18n.h>
+#include <glib/gstdio.h>
+#include <gio/gunixfdlist.h>
+#include <stdlib.h>
+#include <gio/gio.h>
+#include <unistd.h>
+
+#include "ipc-profiler.h"
+
+static gboolean opt_force;
+static gboolean opt_system;
+static gboolean opt_session;
+static gchar *opt_address;
+static gchar *opt_dest;
+static gchar *opt_object_path;
+static gint opt_timeout = -1;
+
+static const GOptionEntry main_entries[] = {
+  { "system", 'y', 0, G_OPTION_ARG_NONE, &opt_system, N_("Connect to the system bus") },
+  { "session", 'e', 0, G_OPTION_ARG_NONE, &opt_session, N_("Connect to the session bus") },
+  { "address", 'a', 0, G_OPTION_ARG_STRING, &opt_address, N_("Connect to the given D-Bus address") },
+  { "dest", 'd', 0, G_OPTION_ARG_STRING, &opt_dest, N_("Destination D-Bus name to invoke method on") },
+  { "object-path", 'o', 0, G_OPTION_ARG_STRING, &opt_object_path, N_("Object path to invoke method on"), 
N_("/org/gnome/Sysprof3/Profiler") },
+  { "timeout", 't', 0, G_OPTION_ARG_INT, &opt_timeout, N_("Timeout in seconds") },
+  { "force", 'f', 0, G_OPTION_ARG_NONE, &opt_force, N_("Overwrite FILENAME if it exists") },
+  { 0 }
+};
+
+static gboolean
+handle_sigint (gpointer data)
+{
+  GMainLoop *main_loop = data;
+  g_printerr ("\nSIGINT received, stopping profiler.\n");
+  g_main_loop_quit (main_loop);
+  return G_SOURCE_REMOVE;
+}
+
+gint
+main (gint   argc,
+      gchar *argv[])
+{
+  g_autoptr(GOptionContext) context = NULL;
+  g_autoptr(GDBusConnection) bus = NULL;
+  g_autoptr(GUnixFDList) fd_list = NULL;
+  g_autoptr(GVariant) reply = NULL;
+  g_autoptr(GError) error = NULL;
+  g_autoptr(GMainLoop) main_loop = NULL;
+  g_autofree gchar *filename = NULL;
+  guint flags = O_RDWR;
+  gint fd;
+  gint handle;
+
+  context = g_option_context_new (_("--dest=BUS_NAME [FILENAME] - connect to an embedded sysprof profiler"));
+  g_option_context_add_main_entries (context, main_entries, GETTEXT_PACKAGE);
+
+  if (!g_option_context_parse (context, &argc, &argv, &error))
+    {
+      g_printerr ("%s\n", error->message);
+      return EXIT_FAILURE;
+    }
+
+  if (opt_dest == NULL)
+    {
+      g_printerr ("--dest= must contain a peer bus name on the D-Bus.\n");
+      return EXIT_FAILURE;
+    }
+
+  if ((opt_system && opt_session) ||
+      (opt_address && (opt_system || opt_session)))
+    {
+      g_printerr ("Only one of --system --session or --address may be specified.");
+      return EXIT_FAILURE;
+    }
+
+  if (!opt_address && !opt_session && !opt_system)
+    opt_session = TRUE;
+
+  main_loop = g_main_loop_new (NULL, FALSE);
+  g_unix_signal_add (SIGINT, handle_sigint, main_loop);
+
+  if (opt_session)
+    bus = g_bus_get_sync (G_BUS_TYPE_SESSION, NULL, &error);
+  else if (opt_system)
+    bus = g_bus_get_sync (G_BUS_TYPE_SYSTEM, NULL, &error);
+  else
+    bus = g_dbus_connection_new_for_address_sync (opt_address,
+                                                  G_DBUS_CONNECTION_FLAGS_MESSAGE_BUS_CONNECTION,
+                                                  NULL,
+                                                  NULL,
+                                                  &error);
+
+  if (bus == NULL)
+    {
+      g_printerr ("Failed to connect to D-Bus: %s\n", error->message);
+      return EXIT_FAILURE;
+    }
+
+  filename = (argc == 1) ? g_strdup_printf ("sysprof.%d.syscap", getpid ())
+                         : g_strdup (argv[1]);
+
+  if (!opt_force)
+    flags |= O_CREAT;
+
+  if (-1 == (fd = g_open (filename, flags, 0644)))
+    {
+      g_printerr ("Failed to open file \"%s\". Try --force.\n", filename);
+      return EXIT_FAILURE;
+    }
+
+  fd_list = g_unix_fd_list_new ();
+  handle = g_unix_fd_list_append (fd_list, fd, &error);
+
+  close (fd);
+  fd = -1;
+
+  if (handle == -1)
+    {
+      g_printerr ("Failed to build FD list for peer.\n");
+      g_unlink (filename);
+      return EXIT_FAILURE;
+    }
+
+  reply = g_dbus_connection_call_with_unix_fd_list_sync (bus,
+                                                         opt_dest,
+                                                         opt_object_path,
+                                                         "org.gnome.Sysprof3.Profiler",
+                                                         "Start",
+                                                         g_variant_new ("(h)", handle),
+                                                         NULL,
+                                                         G_DBUS_CALL_FLAGS_ALLOW_INTERACTIVE_AUTHORIZATION,
+                                                         opt_timeout,
+                                                         fd_list,
+                                                         NULL,
+                                                         NULL,
+                                                         &error);
+
+  if (error != NULL)
+    {
+      g_printerr ("Failed to start profiler: %s\n", error->message);
+      g_unlink (filename);
+      return EXIT_FAILURE;
+    }
+
+  g_printerr ("Profiler capturing to file \"%s\"\n", filename);
+
+  g_clear_pointer (&reply, g_variant_unref);
+
+  g_main_loop_run (main_loop);
+
+  reply = g_dbus_connection_call_sync (bus,
+                                       opt_dest,
+                                       opt_object_path,
+                                       "org.gnome.Sysprof3.Profiler",
+                                       "Stop",
+                                       g_variant_new ("()"),
+                                       NULL,
+                                       G_DBUS_CALL_FLAGS_NONE,
+                                       opt_timeout,
+                                       NULL,
+                                       &error);
+
+  if (error != NULL)
+    {
+      g_printerr ("Failed to stop profiler: %s\n", error->message);
+      return EXIT_FAILURE;
+    }
+
+  g_printerr ("Profiler stopped.\n");
+
+  return EXIT_SUCCESS;
+}