server: make struct group_run_stats network transfer friendly

[fio.git] / init.c

diff --git a/init.c b/init.c
index 5d5709110c22d1519becdbbd5d9082dc0f853b06..89c241f2500b0c689ceb0ac952d2d53bdf2a6543 100644 (file)
--- a/init.c
+++ b/init.c
@@ -46,6 +46,7 @@ char *exec_profile = NULL;
 int warnings_fatal = 0;
 int terse_version = 2;
 int is_backend = 0;
+int nr_clients = 0;
 
 int write_bw_log = 0;
 int read_only = 0;
@@ -160,21 +161,21 @@ static struct option l_opts[FIO_NR_OPTIONS] = {
                .has_arg        = no_argument,
                .val            = 'S',
        },
+       {
+               .name           = (char *) "net-port",
+               .has_arg        = required_argument,
+               .val            = 'P',
+       },
+       {
+               .name           = (char *) "client",
+               .has_arg        = required_argument,
+               .val            = 'C',
+       },
        {
                .name           = NULL,
        },
 };
 
-FILE *get_f_out()
-{
-       return f_out;
-}
-
-FILE *get_f_err()
-{
-       return f_err;
-}
-
 /*
  * Return a free job structure.
  */
@@ -1079,6 +1080,9 @@ static void usage(const char *name)
                " (def 1024)\n");
        printf("\t--warnings-fatal Fio parser warnings are fatal\n");
        printf("\t--max-jobs\tMaximum number of threads/processes to support\n");
+       printf("\t--server\tStart a backend fio server\n");
+       printf("\t--client=hostname Talk to remove backend fio server at hostname\n");
+       printf("\t--net-port=port\tUse specified port for client/server connection\n");
        printf("\nFio was written by Jens Axboe <jens.axboe@oracle.com>");
        printf("\n                   Jens Axboe <jaxboe@fusionio.com>\n");
 }
@@ -1098,6 +1102,7 @@ struct debug_level debug_levels[] = {
        { .name = "mutex",      .shift = FD_MUTEX },
        { .name = "profile",    .shift = FD_PROFILE },
        { .name = "time",       .shift = FD_TIME },
+       { .name = "net",        .shift = FD_NET },
        { .name = NULL, },
 };
 
@@ -1304,8 +1309,26 @@ static int parse_cmd_line(int argc, char *argv[])
                        }
                        break;
                case 'S':
+                       if (nr_clients) {
+                               log_err("fio: can't be both client and server\n");
+                               do_exit++;
+                               exit_val = 1;
+                               break;
+                       }
                        is_backend = 1;
                        break;
+               case 'P':
+                       fio_net_port = atoi(optarg);
+                       break;
+               case 'C':
+                       if (is_backend) {
+                               log_err("fio: can't be both client and server\n");
+                               do_exit++;
+                               exit_val = 1;
+                               break;
+                       }
+                       fio_client_add(optarg);
+                       break;
                default:
                        do_exit++;
                        exit_val = 1;
@@ -1316,8 +1339,14 @@ static int parse_cmd_line(int argc, char *argv[])
        if (do_exit)
                exit(exit_val);
 
+       if (nr_clients && fio_clients_connect()) {
+               do_exit++;
+               exit_val = 1;
+               return 1;
+       }
+
        if (is_backend)
-               fio_server();
+               return fio_server();
 
        if (td) {
                if (!ret)
@@ -1354,9 +1383,15 @@ int parse_options(int argc, char *argv[])
        for (i = 0; i < job_files; i++) {
                if (fill_def_thread())
                        return 1;
-               if (parse_jobs_ini(ini_file[i], 0, i))
-                       return 1;
-               free(ini_file[i]);
+               if (nr_clients) {
+                       if (fio_clients_send_ini(ini_file[i]))
+                               return 1;
+                       free(ini_file[i]);
+               } else if (!is_backend) {
+                       if (parse_jobs_ini(ini_file[i], 0, i))
+                               return 1;
+                       free(ini_file[i]);
+               }
        }
 
        free(ini_file);
@@ -1367,6 +1402,8 @@ int parse_options(int argc, char *argv[])
                        return 0;
                if (exec_profile)
                        return 0;
+               if (is_backend || nr_clients)
+                       return 0;
 
                log_err("No jobs(s) defined\n\n");
                usage(argv[0]);