Add tests for removing a pending link timeout, including a chain with a linked nop that should still run after the head read completes. Signed-off-by: Yang Xiuwei --- test/link-timeout.c | 230 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 230 insertions(+) diff --git a/test/link-timeout.c b/test/link-timeout.c index 07856ff3..e432e630 100644 --- a/test/link-timeout.c +++ b/test/link-timeout.c @@ -768,6 +768,224 @@ err: return 1; } +/* + * Test removal of a pending link timeout + */ +static int test_link_timeout_remove(struct io_uring *ring) +{ + struct __kernel_timespec ts; + struct io_uring_cqe *cqe; + struct io_uring_sqe *sqe; + int fds[2], ret, i; + struct iovec iov; + char buffer[128]; + + if (pipe(fds)) { + perror("pipe"); + return 1; + } + + sqe = io_uring_get_sqe(ring); + if (!sqe) { + printf("get sqe failed\n"); + goto err; + } + + iov.iov_base = buffer; + iov.iov_len = sizeof(buffer); + io_uring_prep_readv(sqe, fds[0], &iov, 1, 0); + sqe->flags |= IOSQE_IO_LINK; + sqe->user_data = 1; + + sqe = io_uring_get_sqe(ring); + if (!sqe) { + printf("get sqe failed\n"); + goto err; + } + + ts.tv_sec = 3600; + ts.tv_nsec = 0; + io_uring_prep_link_timeout(sqe, &ts, 0); + sqe->user_data = 2; + + ret = io_uring_submit(ring); + if (ret != 2) { + printf("sqe submit failed: %d\n", ret); + goto err; + } + + sqe = io_uring_get_sqe(ring); + if (!sqe) { + printf("get sqe failed\n"); + goto err; + } + io_uring_prep_timeout_remove(sqe, 2, 0); + sqe->user_data = 3; + + ret = io_uring_submit(ring); + if (ret != 1) { + printf("sqe submit failed: %d\n", ret); + goto err; + } + + for (i = 0; i < 2; i++) { + ret = io_uring_wait_cqe(ring, &cqe); + if (ret < 0) { + printf("wait completion %d\n", ret); + goto err; + } + switch (cqe->user_data) { + case 2: + if (cqe->res != -ECANCELED) { + fprintf(stderr, "Link timeout got %d, wanted -ECANCELED\n", + cqe->res); + goto err; + } + break; + case 3: + if (cqe->res) { + fprintf(stderr, "Req %" PRIu64 " got %d\n", + (uint64_t) cqe->user_data, cqe->res); + goto err; + } + break; + } + io_uring_cqe_seen(ring, cqe); + } + + return 0; +err: + return 1; +} + +/* + * Test removal of a pending link timeout with a linked nop behind it + */ +static int test_link_timeout_remove_chain(struct io_uring *ring) +{ + struct __kernel_timespec ts; + struct io_uring_cqe *cqe; + struct io_uring_sqe *sqe; + int fds[2], ret, i; + struct iovec iov; + char buffer[128]; + char byte = 'x'; + + if (pipe(fds)) { + perror("pipe"); + return 1; + } + + sqe = io_uring_get_sqe(ring); + if (!sqe) { + printf("get sqe failed\n"); + goto err; + } + + iov.iov_base = buffer; + iov.iov_len = sizeof(buffer); + io_uring_prep_readv(sqe, fds[0], &iov, 1, 0); + sqe->flags |= IOSQE_IO_LINK; + sqe->user_data = 1; + + sqe = io_uring_get_sqe(ring); + if (!sqe) { + printf("get sqe failed\n"); + goto err; + } + + ts.tv_sec = 3600; + ts.tv_nsec = 0; + io_uring_prep_link_timeout(sqe, &ts, 0); + sqe->flags |= IOSQE_IO_LINK; + sqe->user_data = 2; + + sqe = io_uring_get_sqe(ring); + if (!sqe) { + printf("get sqe failed\n"); + goto err; + } + io_uring_prep_nop(sqe); + sqe->user_data = 3; + + ret = io_uring_submit(ring); + if (ret != 3) { + printf("sqe submit failed: %d\n", ret); + goto err; + } + + sqe = io_uring_get_sqe(ring); + if (!sqe) { + printf("get sqe failed\n"); + goto err; + } + io_uring_prep_timeout_remove(sqe, 2, 0); + sqe->user_data = 4; + + ret = io_uring_submit(ring); + if (ret != 1) { + printf("sqe submit failed: %d\n", ret); + goto err; + } + + for (i = 0; i < 2; i++) { + ret = io_uring_wait_cqe(ring, &cqe); + if (ret < 0) { + printf("wait completion %d\n", ret); + goto err; + } + switch (cqe->user_data) { + case 2: + if (cqe->res != -ECANCELED) { + fprintf(stderr, "Link timeout got %d, wanted -ECANCELED\n", + cqe->res); + goto err; + } + break; + case 4: + if (cqe->res) { + fprintf(stderr, "Req %" PRIu64 " got %d\n", + (uint64_t) cqe->user_data, cqe->res); + goto err; + } + break; + } + io_uring_cqe_seen(ring, cqe); + } + + if (write(fds[1], &byte, 1) != 1) { + perror("write"); + goto err; + } + + ret = io_uring_wait_cqe(ring, &cqe); + if (ret < 0) { + printf("wait completion %d\n", ret); + goto err; + } + if (cqe->user_data != 1 || cqe->res != 1) { + fprintf(stderr, "Read got %d\n", cqe->res); + goto err; + } + io_uring_cqe_seen(ring, cqe); + + ret = io_uring_wait_cqe(ring, &cqe); + if (ret < 0) { + printf("wait completion %d\n", ret); + goto err; + } + if (cqe->user_data != 3 || cqe->res) { + fprintf(stderr, "Req %" PRIu64 " got %d\n", + (uint64_t) cqe->user_data, cqe->res); + goto err; + } + io_uring_cqe_seen(ring, cqe); + + return 0; +err: + return 1; +} + static int test_timeout_link_chain1(struct io_uring *ring) { struct __kernel_timespec ts; @@ -1365,5 +1583,17 @@ int main(int argc, char *argv[]) return ret; } + ret = test_link_timeout_remove(&ring); + if (ret) { + printf("test_link_timeout_remove failed\n"); + return ret; + } + + ret = test_link_timeout_remove_chain(&ring); + if (ret) { + printf("test_link_timeout_remove_chain failed\n"); + return ret; + } + return T_EXIT_PASS; } -- 2.25.1