selftests: ublk: replace assert() with ublk_assert()

Replace assert() with ublk_assert() since it is often triggered in daemon,
and we may get nothing shown in terminal.

Add ublk_assert(), so we can log something to syslog when assert() is
triggered.

Signed-off-by: Ming Lei <ming.lei@redhat.com>
Signed-off-by: Jens Axboe <axboe@kernel.dk>

diff --git a/tools/testing/selftests/ublk/common.c b/tools/testing/selftests/ublk/common.c
index d9873d4..530f987 100644 (file)
--- a/tools/testing/selftests/ublk/common.c
+++ b/tools/testing/selftests/ublk/common.c
@@ -16,7 +16,7 @@ int backing_file_tgt_init(struct ublk_dev *dev, unsigned int nr_direct)
 {
        int fd, i;
 
-       assert(dev->nr_fds == 1);
+       ublk_assert(dev->nr_fds == 1);
 
        for (i = 0; i < dev->tgt.nr_backing_files; i++) {
                char *file = dev->tgt.backing_file[i];

diff --git a/tools/testing/selftests/ublk/file_backed.c b/tools/testing/selftests/ublk/file_backed.c
index c3ce5ff..889047b 100644 (file)
--- a/tools/testing/selftests/ublk/file_backed.c
+++ b/tools/testing/selftests/ublk/file_backed.c
@@ -10,7 +10,7 @@ static enum io_uring_op ublk_to_uring_op(const struct ublksrv_io_desc *iod, int
                return zc ? IORING_OP_READ_FIXED : IORING_OP_READ;
        else if (ublk_op == UBLK_IO_OP_WRITE)
                return zc ? IORING_OP_WRITE_FIXED : IORING_OP_WRITE;
-       assert(0);
+       ublk_assert(0);
 }
 
 static int loop_queue_flush_io(struct ublk_thread *t, struct ublk_queue *q,

diff --git a/tools/testing/selftests/ublk/kublk.c b/tools/testing/selftests/ublk/kublk.c
index 3472ce7..e98999b 100644 (file)
--- a/tools/testing/selftests/ublk/kublk.c
+++ b/tools/testing/selftests/ublk/kublk.c
@@ -825,7 +825,7 @@ static void ublk_handle_uring_cmd(struct ublk_thread *t,
        }
 
        if (cqe->res == UBLK_IO_RES_OK) {
-               assert(tag < q->q_depth);
+               ublk_assert(tag < q->q_depth);
 
                if (ublk_queue_use_user_copy(q))
                        ublk_user_copy(io, UBLK_IO_OP_WRITE);

diff --git a/tools/testing/selftests/ublk/kublk.h b/tools/testing/selftests/ublk/kublk.h
index 69fd579..48634d2 100644 (file)
--- a/tools/testing/selftests/ublk/kublk.h
+++ b/tools/testing/selftests/ublk/kublk.h
@@ -260,7 +260,7 @@ static inline __u64 build_user_data(unsigned tag, unsigned op,
 {
        /* we only have 7 bits to encode q_id */
        _Static_assert(UBLK_MAX_QUEUES_SHIFT <= 7);
-       assert(!(tag >> 16) && !(op >> 8) && !(tgt_data >> 16) && !(q_id >> 7));
+       ublk_assert(!(tag >> 16) && !(op >> 8) && !(tgt_data >> 16) && !(q_id >> 7));
 
        return tag | ((__u64)op << 16) | ((__u64)tgt_data << 24) |
                (__u64)q_id << 56 | (__u64)is_target_io << 63;

diff --git a/tools/testing/selftests/ublk/stripe.c b/tools/testing/selftests/ublk/stripe.c
index 2be1c36..b967447 100644 (file)
--- a/tools/testing/selftests/ublk/stripe.c
+++ b/tools/testing/selftests/ublk/stripe.c
@@ -96,12 +96,12 @@ static void calculate_stripe_array(const struct stripe_conf *conf,
                        this->seq = seq;
                        s->nr += 1;
                } else {
-                       assert(seq == this->seq);
-                       assert(this->start + this->nr_sects == stripe_off);
+                       ublk_assert(seq == this->seq);
+                       ublk_assert(this->start + this->nr_sects == stripe_off);
                        this->nr_sects += nr_sects;
                }
 
-               assert(this->nr_vec < this->cap);
+               ublk_assert(this->nr_vec < this->cap);
                this->vec[this->nr_vec].iov_base = (void *)(base + done);
                this->vec[this->nr_vec++].iov_len = nr_sects << 9;
 
@@ -120,7 +120,7 @@ static inline enum io_uring_op stripe_to_uring_op(
                return zc ? IORING_OP_READV_FIXED : IORING_OP_READV;
        else if (ublk_op == UBLK_IO_OP_WRITE)
                return zc ? IORING_OP_WRITEV_FIXED : IORING_OP_WRITEV;
-       assert(0);
+       ublk_assert(0);
 }
 
 static int stripe_queue_tgt_rw_io(struct ublk_thread *t, struct ublk_queue *q,
@@ -322,7 +322,7 @@ static int ublk_stripe_tgt_init(const struct dev_ctx *ctx, struct ublk_dev *dev)
        if (!dev->tgt.nr_backing_files || dev->tgt.nr_backing_files > NR_STRIPE)
                return -EINVAL;
 
-       assert(dev->nr_fds == dev->tgt.nr_backing_files + 1);
+       ublk_assert(dev->nr_fds == dev->tgt.nr_backing_files + 1);
 
        for (i = 0; i < dev->tgt.nr_backing_files; i++)
                dev->tgt.backing_file_size[i] &= ~((1 << chunk_shift) - 1);

diff --git a/tools/testing/selftests/ublk/utils.h b/tools/testing/selftests/ublk/utils.h
index a852e0b..17eefed 100644 (file)
--- a/tools/testing/selftests/ublk/utils.h
+++ b/tools/testing/selftests/ublk/utils.h
@@ -43,6 +43,7 @@ static inline void ublk_err(const char *fmt, ...)
 
        va_start(ap, fmt);
        vfprintf(stderr, fmt, ap);
+       va_end(ap);
 }
 
 static inline void ublk_log(const char *fmt, ...)
@@ -52,6 +53,7 @@ static inline void ublk_log(const char *fmt, ...)
 
                va_start(ap, fmt);
                vfprintf(stdout, fmt, ap);
+               va_end(ap);
        }
 }
 
@@ -62,7 +64,15 @@ static inline void ublk_dbg(int level, const char *fmt, ...)
 
                va_start(ap, fmt);
                vfprintf(stdout, fmt, ap);
+               va_end(ap);
        }
 }
 
+#define ublk_assert(x)  do { \
+       if (!(x)) {     \
+               ublk_err("%s %d: assert!\n", __func__, __LINE__); \
+               assert(x);      \
+       }       \
+} while (0)
+
 #endif