Have trace-cmd record not need a command

Currently trace-cmd record requires a command to exec.
This patch will just record what is happening if no command is given
and will stop when the user hits Ctrl^C.

Signed-off-by: Steven Rostedt <rostedt@goodmis.org>

1 files changed, 17 insertions, 8 deletions

diff --git a/trace-cmd.c b/trace-cmd.c
index 26fd883..aca6f90 100644
--- a/trace-cmd.c
+++ b/trace-cmd.c
@@ -1106,7 +1106,7 @@ void usage(char **argv)
         printf("\n"
                "%s version %s\n\n"
                "usage:\n"
-               " %s record [-e event][-p plugin] [-d] [-o file] [-O option ] command ...\n"
+               " %s record [-e event][-p plugin] [-d] [-o file] [-O option ] [command ...]\n"
                "          -e run command with event enabled\n"
                "          -p run command with plugin enabled\n"
                "          -d disable function tracer when running\n"
@@ -1152,6 +1152,7 @@ int main (int argc, char **argv)
         int events = 0;
         int options = 0;
         int record = 0;
+        int run_command = 0;
         int fset;
 
         int c;
@@ -1257,12 +1258,12 @@ int main (int argc, char **argv)
                 usage(argv);
         }
 
-        if ((argc - optind) < 2) {
-                if (record)
-                        usage(argv);
-        } else if (!record)
-                die("Command start does not take any commands\n"
-                    "Did you mean 'record'?");
+        if ((argc - optind) >= 2) {
+                if (!record)
+                        die("Command start does not take any commands\n"
+                            "Did you mean 'record'?");
+                run_command = 1;
+        }
 
         if (!events && !plugin)
                 die("no event or plugin was specified... aborting");
@@ -1308,7 +1309,15 @@ int main (int argc, char **argv)
         if (!record)
                 exit(0);
 
-        run_cmd((argc - optind) - 1, &argv[optind + 1]);
+        if (run_command)
+                run_cmd((argc - optind) - 1, &argv[optind + 1]);
+        else {
+                /* sleep till we are woken with Ctrl^C */
+                printf("Hit Ctrl^C to stop recording\n");
+                while (!finished)
+                        sleep(10);
+                
+        }
 
         disable_tracing();