ustctl: Fix memory allocation problem with compatibility args

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

/* Copyright (C) 2009  Pierre-Marc Fournier
 * 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
 */

#define _GNU_SOURCE
#include <stdio.h>
#include <unistd.h>
#include <getopt.h>
#include <stdlib.h>
#include <fcntl.h>

#include "ust/ustctl.h"
#include "usterr.h"
#include "cli.h"
#include "scanning_functions.h"

void usage(const char *process_name)
{
	fprintf(stderr, "Usage: %s COMMAND [ARGS]...\n", process_name);
	fprintf(stderr,
		"Control tracing within a process that supports UST,\n"
		" the Userspace Tracing libary\n"
		"Options:\n"
		"  -h[<cmd>], --help[=<cmd>]        "
		"help, for a command if provided\n"
		"  -l, --list                       "
		"short list of commands\n"
		"  -e, --extended-list              "
	       "extented list of commands with help\n"
		"Commands:\n");
	list_cli_cmds(CLI_DESCRIPTIVE_LIST);
}

/*
 * Provide backward compatibility for scripts that make use of the
 * --commands in ustctl version <= 0.11
 */
enum command {
	CREATE_TRACE=1000,
	ALLOC_TRACE,
	START_TRACE,
	STOP_TRACE,
	DESTROY_TRACE,
	LIST_MARKERS,
	LIST_TRACE_EVENTS,
	ENABLE_MARKER,
	DISABLE_MARKER,
};

struct option options[] =
{
	{ "create-trace", 0, 0, CREATE_TRACE },
	{ "alloc-trace", 0, 0, ALLOC_TRACE },
	{ "start-trace", 0, 0, START_TRACE },
	{ "stop-trace", 0, 0, STOP_TRACE },
	{ "destroy-trace", 0, 0, DESTROY_TRACE },
	{ "list-markers", 0, 0, LIST_MARKERS },
	{ "list-trace-events", 0, 0, LIST_TRACE_EVENTS},
	{ "enable-marker", 0, 0, ENABLE_MARKER },
	{ "disable-marker", 0, 0, DISABLE_MARKER },
	{"help", 2, NULL, 'h'},
	{"list", 0, NULL, 'l'},
	{"extended-list", 0, NULL, 'e'},
	{NULL, 0, NULL, 0},
};

int main(int argc, char *argv[])
{
	struct cli_cmd *cli_cmd;
	char **args = argv;
	int opt;
	int i;

	if(argc <= 1) {
		fprintf(stderr, "No operation specified.\n");
		usage(argv[0]);
		exit(EXIT_FAILURE);
	}

	while ((opt = getopt_long(argc, argv, "+h::le",
				  options, NULL)) != -1) {
		switch (opt) {
		case 'h':
			if (!optarg) {
				usage(argv[0]);
			} else {
				if (cli_print_help(optarg)) {
					fprintf(stderr, "No such command %s\n",
						optarg);
				}
			}
			exit(EXIT_FAILURE);
			break;
		case 'l':
			list_cli_cmds(CLI_SIMPLE_LIST);
			exit(EXIT_FAILURE);
			break;
		case 'e':
			list_cli_cmds(CLI_EXTENDED_LIST);
			exit(EXIT_FAILURE);
		case LIST_MARKERS:
		case LIST_TRACE_EVENTS:
		case CREATE_TRACE:
		case ALLOC_TRACE:
		case START_TRACE:
		case STOP_TRACE:
		case DESTROY_TRACE:
		case ENABLE_MARKER:
		case DISABLE_MARKER:
			args = (char **)malloc(sizeof(char *) * (argc + 3));
			optind--;
			args[optind] = strdup(&argv[optind][2]);
			for (i = optind + 1; i < argc; i++) {
				args[i] = argv[i];
			}
			if (opt >= CREATE_TRACE && opt <= DESTROY_TRACE) {
				args[argc] = strdup("auto");
				argc++;
			}
			if (opt >= ENABLE_MARKER && opt <= DISABLE_MARKER) {
				args[argc] = args[argc - 2];
				args[argc - 2] = args[argc - 1];
				args[argc - 1] = strdup("auto");
				argc++;
			}
			args[argc] = NULL;
			goto do_cli;
		default:
			fprintf(stderr, "Unknown option\n");
			break;
		}
	}

do_cli:
	cli_cmd = find_cli_cmd(args[optind]);
	if (!cli_cmd) {
		fprintf(stderr, "No such command %s\n",
			args[optind]);
		exit(EXIT_FAILURE);
	}

	cli_dispatch_cmd(cli_cmd, argc - optind, &args[optind]);

	return 0;
}

static int list_trace_events(int argc, char *argv[])
{
	struct trace_event_status *tes = NULL;
	int i, sock;

	sock = parse_and_connect_pid(argv[1]);

	if (ustctl_get_tes(sock, &tes)) {
		ERR("error while trying to list "
		    "trace_events for PID %s\n",
		    argv[1]);
		return -1;
	}
	i = 0;
	for (i = 0; tes[i].name; i++) {
		printf("{PID: %s, trace_event: %s}\n",
		       argv[1],
		       tes[i].name);
	}
	ustctl_free_tes(tes);

	return 0;
}

static int set_sock_path(int argc, char *argv[])
{
	int sock;

	sock = parse_and_connect_pid(argv[1]);

	if (ustctl_set_sock_path(sock, argv[2])) {
		ERR("error while trying to set sock path for PID %s\n", argv[1]);
		return -1;
	}

	return 0;
}

static int get_sock_path(int argc, char *argv[])
{
	int sock;
	char *sock_path;

	sock = parse_and_connect_pid(argv[1]);

	if (ustctl_get_sock_path(sock, &sock_path)) {
		ERR("error while trying to get sock path for PID %s\n", argv[1]);
		return -1;
	}
	printf("The socket path is %s\n", sock_path);
	free(sock_path);

	return 0;
}

struct cli_cmd __cli_cmds general_cmds[] = {
	{
		.name = "list-trace-events",
		.description = "List trace-events for a given pid",
		.help_text = "list-trace-events <pid>\n"
		"List the trace-events in a process\n",
		.function = list_trace_events,
		.desired_args = 1,
		.desired_args_op = CLI_EQ,
	},
	{
		.name = "set-sock-path",
		.description = "Set the path to the consumer daemon socket",
		.help_text = "set-sock-path <pid> <sock-path>\n"
		"Set the path to the consumer daemon socket\n",
		.function = set_sock_path,
		.desired_args = 2,
		.desired_args_op = CLI_EQ,
	},
	{
		.name = "get-sock-path",
		.description = "Get the path to the consumer daemon socket",
		.help_text = "get-sock-path <pid>\n"
		"Get the path to the consumer daemon socket\n",
		.function = get_sock_path,
		.desired_args = 1,
		.desired_args_op = CLI_EQ,
	},
};