Test empty argument tracepoint

[lttng-ust.git] / ustctl / marker_cmds.c

/* Copyright (C) 2011  Ericsson AB, Nils Carlson <nils.carlson@ericsson.com>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
 */
#include <stdio.h>
#include <sys/types.h>
#include <unistd.h>
#include <ust/ustctl.h>
#include "scanning_functions.h"
#include "usterr.h"
#include "cli.h"

static int list_markers(int argc, char *argv[])
{
	struct marker_status *cmsf = NULL;
	int i, sock;

	sock = parse_and_connect_pid(argv[1]);

	if (ustctl_get_cmsf(sock, &cmsf)) {
		ERR("error while trying to list markers for PID %s\n", argv[1]);
		return -1;
	}
	for (i = 0; cmsf[i].channel; i++) {
		printf("{PID: %s, channel/marker: %s/%s, "
		       "state: %u, fmt: %s}\n",
		       argv[1],
		       cmsf[i].channel,
		       cmsf[i].marker,
		       cmsf[i].state,
		       cmsf[i].fs);
	}
	ustctl_free_cmsf(cmsf);
	return 0;
}

static int enable_marker(int argc, char *argv[])
{
	int i, sock, result = 0;
	char *channel, *marker;

	sock = parse_and_connect_pid(argv[1]);

	for (i = 3; i < argc; i++) {
		channel = NULL;
		marker = NULL;
		if (scan_ch_marker(argv[i],
				   &channel, &marker)) {
			result = -1;
			fprintf(stderr, "Failed to scan channel and marker from"
				" %s\n", argv[i]);
			if (channel)
				free(channel);
			if (marker)
				free(marker);
		}
		if (ustctl_set_marker_state(sock, argv[2], channel, marker, 1)) {
			PERROR("error while trying to enable marker %s with PID %s",
			       argv[i], argv[1]);
			result = -1;
		}
		free(channel);
		free(marker);
	}

	return result;
}

static int disable_marker(int argc, char *argv[])
{
	int i, sock, result = 0;
	char *channel, *marker;

	sock = parse_and_connect_pid(argv[1]);

	for (i = 3; i < argc; i++) {
		channel = NULL;
		marker = NULL;
		if (scan_ch_marker(argv[i],
				   &channel, &marker)) {
			fprintf(stderr, "Failed to scan channel and marker from"
				" %s\n", argv[i]);
			if (channel)
				free(channel);
			if (marker)
				free(marker);
			return -1;
		}
		if (ustctl_set_marker_state(sock, argv[2], channel, marker, 0)) {
			PERROR("error while trying to disable marker %s with PID %s",
			       argv[i], argv[1]);
			result = -1;
		}
		free(channel);
		free(marker);
	}

	return result;
}

struct cli_cmd __cli_cmds marker_cmds[] = {
	{
		.name = "list-markers",
		.description = "List markers for a given pid",
		.help_text = "list-markers <pid>\n"
		"List the markers in a process\n",
		.function = list_markers,
		.desired_args = 1,
		.desired_args_op = CLI_EQ,
	},
	{
		.name = "enable-marker",
		.description = "Enable markers for a given pid",
		.help_text = "enable-marker <pid> <trace> <channel>/<marker>... \n"
		"Enable the listed markers for the trace in process pid\n",
		.function = enable_marker,
		.desired_args = 3,
		.desired_args_op = CLI_GE,
	},
	{
	       .name = "disable-marker",
	       .description = "Disable markers for a given pid",
	       .help_text = "disable-marker <pid> <trace> <channel>/<marker>... \n"
	       "Disable the listed markers for the trace in process pid\n",
	       .function = disable_marker,
	       .desired_args = 3,
	       .desired_args_op = CLI_GE,
	}
};