rstream: Add option to test fork support

Signed-off-by: Sean Hefty <sean.hefty@intel.com>

diff --git a/examples/rstream.c b/examples/rstream.c
index 48e2771e8da7ba8e2a4451073460fb22a576145d..85956572941792059b2928110bc5122e5a46ecd6 100644 (file)
--- a/examples/rstream.c
+++ b/examples/rstream.c
@@ -84,6 +84,8 @@ static int verify;
 static int flags = MSG_DONTWAIT;
 static int poll_timeout = 0;
 static int custom;
+static int use_fork;
+static pid_t fork_pid;
 static enum rs_optimization optimization;
 static int size_option;
 static int iterations = 1;
@@ -463,7 +465,10 @@ static int server_connect(void)
                perror("raccept");
        }
 
-       set_options(rs);
+       if (use_fork)
+               fork_pid = fork();
+       if (!fork_pid)
+               set_options(rs);
        return ret;
 }
 
@@ -546,20 +551,24 @@ static int run(void)
                if (ret)
                        goto free;
 
-               for (i = 0; i < TEST_CNT; i++) {
+               for (i = 0; i < TEST_CNT && !fork_pid; i++) {
                        if (test_size[i].option > size_option)
                                continue;
                        init_latency_test(test_size[i].size);
                        run_test();
                }
-               rs_shutdown(rs, SHUT_RDWR);
+               if (!fork_pid)
+                       rs_shutdown(rs, SHUT_RDWR);
                rs_close(rs);
 
+               if (!dst_addr && use_fork && !fork_pid)
+                       goto free;
+
                optimization = opt_bandwidth;
                ret = dst_addr ? client_connect() : server_connect();
                if (ret)
                        goto free;
-               for (i = 0; i < TEST_CNT; i++) {
+               for (i = 0; i < TEST_CNT && !fork_pid; i++) {
                        if (test_size[i].option > size_option)
                                continue;
                        init_bandwidth_test(test_size[i].size);
@@ -570,10 +579,12 @@ static int run(void)
                if (ret)
                        goto free;
 
-               ret = run_test();
+               if (!fork_pid)
+                       ret = run_test();
        }
 
-       rs_shutdown(rs, SHUT_RDWR);
+       if (!fork_pid)
+               rs_shutdown(rs, SHUT_RDWR);
        rs_close(rs);
 free:
        free(buf);
@@ -593,6 +604,10 @@ static int set_test_opt(char *optarg)
                case 'b':
                        flags &= ~MSG_DONTWAIT;
                        break;
+               case 'f':
+                       use_fork = 1;
+                       use_rs = 0;
+                       break;
                case 'n':
                        flags |= MSG_DONTWAIT;
                        break;
@@ -613,6 +628,9 @@ static int set_test_opt(char *optarg)
                        flags |= MSG_DONTWAIT;
                } else if (!strncasecmp("verify", optarg, 6)) {
                        verify = 1;
+               } else if (!strncasecmp("fork", optarg, 4)) {
+                       use_fork = 1;
+                       use_rs = 0;
                } else {
                        return -1;
                }
@@ -671,6 +689,7 @@ int main(int argc, char **argv)
                        printf("\t    s|sockets - use standard tcp/ip sockets\n");
                        printf("\t    a|async - asynchronous operation (use poll)\n");
                        printf("\t    b|blocking - use blocking calls\n");
+                       printf("\t    f|fork - fork server processing\n");
                        printf("\t    n|nonblocking - use nonblocking calls\n");
                        printf("\t    v|verify - verify data\n");
                        exit(1);

diff --git a/man/rstream.1 b/man/rstream.1
index ea87b9518d0f7e03112ed1faa68b0425ddc32849..29aa66bdd7ac29d9713b0e3fcace9d37d4a642da 100644 (file)
--- a/man/rstream.1
+++ b/man/rstream.1
@@ -48,6 +48,8 @@ a | async - uses asynchronous operation (e.g. select / poll)
 .P
 b | blocking - uses blocking calls
 .P
+f | fork - fork server processing (forces -T s option)
+.P
 n | nonblocking - uses non-blocking calls
 .P
 v | verify - verifies data transfers