benchmarks: added "data" option

1 files changed, 12 insertions, 1 deletions

diff --git a/tests/benchmarks/benchmarks.c b/tests/benchmarks/benchmarks.c
index 647e5e7..0dd0d0d 100644
--- a/tests/benchmarks/benchmarks.c
+++ b/tests/benchmarks/benchmarks.c
@@ -78,6 +78,14 @@ static struct argp_option options[] = {
     .doc   = "Add a host to connect for benchmark (format user@hostname)",
     .group = 0
   },
+  {
+    .name  = "data",
+    .key   = 'd',
+    .arg   = "MBYTES",
+    .flags = 0,
+    .doc   = "MBytes of data to send/receive per test",
+    .group = 0
+  },
   {NULL, 0, NULL, 0, NULL, 0}
 };
 
@@ -100,6 +108,9 @@ static error_t parse_opt (int key, char *arg, struct argp_state *state) {
     case 'v':
       arguments->verbose++;
       break;
+    case 'd':
+      arguments->data = atoi(arg);
+      break;
     case 'h':
       if(arguments->nhosts >= MAX_HOSTS_CONNECT){
         fprintf(stderr, "Too much hosts\n");
@@ -222,7 +233,7 @@ int main(int argc, char **argv){
     return EXIT_FAILURE;
   }
   if (arguments.ntests==0){
-    for(i=1; i < BENCHMARK_NUMBER ; ++i){
+    for(i=0; i < BENCHMARK_NUMBER ; ++i){
       arguments.benchmarks[i]=1;
     }
     arguments.ntests=BENCHMARK_NUMBER;