ustctl cli rewrite

[ust.git] / ustctl / channel_cmds.c

diff --git a/ustctl/channel_cmds.c b/ustctl/channel_cmds.c
new file mode 100644 (file)
index 0000000..51f7d50
--- /dev/null
+++ b/ustctl/channel_cmds.c
@@ -0,0 +1,153 @@
+/* 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/ustcmd.h>
+#include "scanning_functions.h"
+#include "usterr.h"
+#include "cli.h"
+
+static int set_subbuf_size(int argc, char *argv[])
+{
+       int result = 0;
+       pid_t pid;
+       char *channel = NULL;
+       unsigned int size;
+
+       pid = parse_pid(argv[1]);
+
+       if (scan_ch_and_num(argv[3], &channel, &size)) {
+               fprintf(stderr, "Failed to scan channel and size from"
+                               " %s\n", argv[3]);
+               if (channel)
+                       free(channel);
+               return -1;
+       }
+       if (ustcmd_set_subbuf_size(argv[2], channel, size, pid)) {
+               ERR("error while trying to set the size of subbuffers "
+                   "for PID %u\n",
+                   pid);
+               result = -1;
+       }
+
+       free(channel);
+
+       return result;
+}
+
+static int set_subbuf_num(int argc, char *argv[])
+{
+       int result = 0;
+       pid_t pid;
+       char *channel = NULL;
+       unsigned int num;
+
+       pid = parse_pid(argv[1]);
+
+       if (scan_ch_and_num(argv[3], &channel, &num)) {
+               fprintf(stderr, "Failed to scan channel and number from"
+                               " %s\n", argv[3]);
+               if (channel)
+                       free(channel);
+               return -1;
+       }
+       if (ustcmd_set_subbuf_num(argv[2], channel, num, pid)) {
+               ERR("error while trying to set the number of subbuffers for PID %u\n",
+                   pid);
+               result = -1;
+       }
+
+       free(channel);
+
+       return result;
+}
+
+static int get_subbuf_size(int argc, char *argv[])
+{
+       pid_t pid;
+       unsigned int size;
+
+       pid = parse_pid(argv[1]);
+
+       if ((size = ustcmd_get_subbuf_size(argv[2], argv[3], pid)) < 0) {
+               ERR("error while trying to get the subbuffer size from PID %u\n",
+                   pid);
+               return -1;
+       }
+
+       printf("The subbufer size is %d bytes\n", size);
+
+       return 0;
+}
+
+static int get_subbuf_num(int argc, char *argv[])
+{
+       pid_t pid;
+       unsigned int num;
+
+       pid = parse_pid(argv[1]);
+
+       if ((num = ustcmd_get_subbuf_num(argv[2], argv[3], pid)) < 0) {
+               ERR("error while trying to get the subbuffer size from PID %u\n",
+                   pid);
+               return -1;
+       }
+
+       printf("There are %u subbufers in each buffer\n", num);
+
+       return 0;
+}
+
+struct cli_cmd __cli_cmds channel_cmds[] = {
+       {
+               .name = "set-subbuf-size",
+               .description = "Set the subbuffer size for a channel",
+               .help_text = "set-subbuf-size <pid> <trace> <channel>/<size> \n"
+               "Set the subbuffer size for a channel\n",
+               .function = set_subbuf_size,
+               .desired_args = 3,
+               .desired_args_op = CLI_EQ,
+       },
+       {
+               .name = "set-subbuf-num",
+               .description = "Set the number of subbuffers for a channel",
+               .help_text = "set-subbuf-num <pid> <trace> <channel>/<num> \n"
+               "Set the number of subbuffers for a channel\n",
+               .function = set_subbuf_num,
+               .desired_args = 3,
+               .desired_args_op = CLI_EQ,
+       },
+       {
+               .name = "get-subbuf-size",
+               .description = "Get the subbuffer size for a channel",
+               .help_text = "get-subbuf-size <pid> <trace> <channel>\n"
+               "Get the subbuffer size for a channel\n",
+               .function = get_subbuf_size,
+               .desired_args = 3,
+               .desired_args_op = CLI_EQ,
+       },
+       {
+               .name = "get-subbuf-num",
+               .description = "Get the number of subbuffers for a channel",
+               .help_text = "get-subbuf-num <pid> <trace> <channel>\n"
+               "Get the number of subbuffers for a channel\n",
+               .function = get_subbuf_num,
+               .desired_args = 3,
+               .desired_args_op = CLI_EQ,
+       },
+};