[PATCH] BTT patch: (2/3) per-IO stream output

[blktrace.git] / btt / args.c

diff --git a/btt/args.c b/btt/args.c
index c8a79b297d6a31d926f00ece0ce704f8c47a3d3d..0e8849c5a1f20b39e89931bc8077663de4b35ad6 100644 (file)
--- a/btt/args.c
+++ b/btt/args.c
@@ -27,7 +27,7 @@
 #include <fcntl.h>
 #include "globals.h"
 
-#define S_OPTS "d:D:e:hi:I:l:M:o:q:s:S:Vv"
+#define S_OPTS "d:D:e:hi:I:l:M:o:p:q:s:S:Vv"
 static struct option l_opts[] = {
        {
                .name = "range-delta",
@@ -83,6 +83,12 @@ static struct option l_opts[] = {
                .flag = NULL,
                .val = 'o'
        },
+       {
+               .name = "per-io-dump",
+               .has_arg = required_argument,
+               .flag = NULL,
+               .val = 'p'
+       },
        {
                .name = "q2c-latencies",
                .has_arg = required_argument,
@@ -128,6 +134,7 @@ static char usage_str[] = \
        "[ -l <output name> | --d2c-latencies=<output name> ]\n" \
        "[ -M <dev map>     | --dev-maps=<dev map>\n" \
        "[ -o <output name> | --output-file=<output name> ]\n" \
+       "[ -p <output name> | --per-io-dump=<output name> ]\n" \
        "[ -q <output name> | --q2c-latencies=<output name> ]\n" \
        "[ -s <output name> | --seeks=<output name> ]\n" \
        "[ -S <interval>    | --iostat-interval=<interval> ]\n" \
@@ -174,6 +181,9 @@ void handle_args(int argc, char *argv[])
                case 'o':
                        output_name = strdup(optarg);
                        break;
+               case 'p':
+                       per_io_name = strdup(optarg);
+                       break;
                case 'q':
                        q2c_name = strdup(optarg);
                        break;
@@ -242,4 +252,12 @@ void handle_args(int argc, char *argv[])
                        exit(1);
                }
        }
+
+       if (per_io_name != NULL) {
+               per_io_ofp = fopen(per_io_name, "w");
+               if (per_io_ofp == NULL) {
+                       perror(per_io_name);
+                       exit(1);
+               }
+       }
 }