[PATCH 03/11] selftests: ublk: add --foreground command line

[Ming Lei]

Add --foreground command for helping to debug.

Signed-off-by: Ming Lei <ming....@redhat.com>
---
 tools/testing/selftests/ublk/kublk.c | 17 +++++++++++++----
 tools/testing/selftests/ublk/kublk.h |  1 +
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/tools/testing/selftests/ublk/kublk.c 
b/tools/testing/selftests/ublk/kublk.c
index 2072d880fdc4..24557a3e5508 100644
--- a/tools/testing/selftests/ublk/kublk.c
+++ b/tools/testing/selftests/ublk/kublk.c
@@ -679,7 +679,10 @@ static int ublk_start_daemon(const struct dev_ctx *ctx, 
struct ublk_dev *dev)
        }
 
        ublk_ctrl_get_info(dev);
-       ublk_send_dev_event(ctx, dev->dev_info.dev_id);
+       if (ctx->fg)
+               ublk_ctrl_dump(dev);
+       else
+               ublk_send_dev_event(ctx, dev->dev_info.dev_id);
 
        /* wait until we are terminated */
        for (i = 0; i < dinfo->nr_hw_queues; i++)
@@ -867,6 +870,9 @@ static int cmd_dev_add(struct dev_ctx *ctx)
 {
        int res;
 
+       if (ctx->fg)
+               goto run;
+
        ctx->_evtfd = eventfd(0, 0);
        if (ctx->_evtfd < 0) {
                ublk_err("%s: failed to create eventfd %s\n", __func__, 
strerror(errno));
@@ -876,8 +882,9 @@ static int cmd_dev_add(struct dev_ctx *ctx)
        setsid();
        res = fork();
        if (res == 0) {
-               __cmd_dev_add(ctx);
-               exit(EXIT_SUCCESS);
+run:
+               res = __cmd_dev_add(ctx);
+               return res;
        } else if (res > 0) {
                uint64_t id;
 
@@ -1044,6 +1051,7 @@ int main(int argc, char *argv[])
                { "debug_mask",         1,      NULL,  0  },
                { "quiet",              0,      NULL,  0  },
                { "zero_copy",          1,      NULL, 'z' },
+               { "foreground",         0,      NULL,  0  },
                { 0, 0, 0, 0 }
        };
        int option_idx, opt;
@@ -1087,7 +1095,8 @@ int main(int argc, char *argv[])
                                ublk_dbg_mask = strtol(optarg, NULL, 16);
                        if (!strcmp(longopts[option_idx].name, "quiet"))
                                ublk_dbg_mask = 0;
-                       break;
+                       if (!strcmp(longopts[option_idx].name, "foreground"))
+                               ctx.fg = 1;
                }
        }
 
diff --git a/tools/testing/selftests/ublk/kublk.h 
b/tools/testing/selftests/ublk/kublk.h
index 8f48eb8568ab..26d9aa9c5ca2 100644
--- a/tools/testing/selftests/ublk/kublk.h
+++ b/tools/testing/selftests/ublk/kublk.h
@@ -67,6 +67,7 @@ struct dev_ctx {
        char *files[MAX_BACK_FILES];
        unsigned int    logging:1;
        unsigned int    all:1;
+       unsigned int    fg:1;
 
        int _evtfd;
 };
-- 
2.47.0