mirror of
https://github.com/varun-r-mallya/sysprof.git
synced 2025-12-31 20:36:25 +00:00
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).
This commit is contained in:
Christian Hergert
@ -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.',
|
||||
|
||||
10
src/org.gnome.Sysprof3.Profiler.xml
Normal file
10
src/org.gnome.Sysprof3.Profiler.xml
Normal file
@ -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>
|
||||
@ -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,
|
||||
)
|
||||
|
||||
197
src/tools/sysprof-profiler-ctl.c
Normal file
197
src/tools/sysprof-profiler-ctl.c
Normal file
@ -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;
|
||||
}
|
||||
Reference in New Issue
Block a user