diff options
author | thegeorg <thegeorg@yandex-team.com> | 2024-06-09 11:55:21 +0300 |
---|---|---|
committer | thegeorg <thegeorg@yandex-team.com> | 2024-06-09 12:07:55 +0300 |
commit | afd4899380eea1c70e2a68714b5da1c9919ccdbd (patch) | |
tree | cd5120708784139bc6a0f8881da1ed8389a065b3 /contrib/libs/liburing/test/truncate.c | |
parent | a83bd2dd3c21e38c6c0807ec5e679497ab567f24 (diff) | |
download | ydb-afd4899380eea1c70e2a68714b5da1c9919ccdbd.tar.gz |
Update contrib/libs/liburing to 2.6
3b51a9fb14de805208d11f1c077c78bb5d487e0f
Diffstat (limited to 'contrib/libs/liburing/test/truncate.c')
-rw-r--r-- | contrib/libs/liburing/test/truncate.c | 187 |
1 files changed, 187 insertions, 0 deletions
diff --git a/contrib/libs/liburing/test/truncate.c b/contrib/libs/liburing/test/truncate.c new file mode 100644 index 0000000000..395d09e70d --- /dev/null +++ b/contrib/libs/liburing/test/truncate.c @@ -0,0 +1,187 @@ +#include "../config-host.h" +/* SPDX-License-Identifier: MIT */ +/* + * Description: run various truncate tests + * + */ +#include <errno.h> +#include <stdio.h> +#include <unistd.h> +#include <stdlib.h> +#include <string.h> +#include <fcntl.h> +#include <sys/stat.h> +#include <sys/ioctl.h> + +#include "liburing.h" +#include "helpers.h" + +#define TWO_GIG_SIZE ((loff_t)2 * 1024 * 1024 * 1024) +#define ONE_GIG_SIZE ((loff_t)1024 * 1024 * 1024) +#define HALF_GIG_SIZE ((loff_t)512 * 1024 * 1024) + +static int test_truncate(struct io_uring *ring, int fd) +{ + struct io_uring_cqe *cqe; + struct io_uring_sqe *sqe; + int ret = -1; + + sqe = io_uring_get_sqe(ring); + if (!sqe) { + fprintf(stderr, "get sqe failed\n"); + return T_EXIT_FAIL; + } + + memset(sqe, 0, sizeof(*sqe)); + + io_uring_prep_rw(IORING_OP_FTRUNCATE, sqe, fd, "fail", 0, 4); + + ret = io_uring_submit(ring); + if (ret <= 0) { + fprintf(stderr, "sqe submit failed: %d\n", ret); + return T_EXIT_FAIL; + } + + ret = io_uring_wait_cqe(ring, &cqe); + if (ret < 0) { + fprintf(stderr, "wait completion %d\n", ret); + return T_EXIT_FAIL; + } + ret = cqe->res; + io_uring_cqe_seen(ring, cqe); + if (ret == -EINVAL) + return T_EXIT_PASS; + + fprintf(stderr, "unexpected truncate res %d\n", ret); + return T_EXIT_FAIL; +} + +static int test_ftruncate(struct io_uring *ring, int fd, loff_t len) +{ + struct io_uring_cqe *cqe; + struct io_uring_sqe *sqe; + int ret; + + sqe = io_uring_get_sqe(ring); + if (!sqe) { + fprintf(stderr, "get sqe failed\n"); + goto err; + } + + memset(sqe, 0, sizeof(*sqe)); + + io_uring_prep_ftruncate(sqe, fd, len); + + ret = io_uring_submit(ring); + if (ret <= 0) { + fprintf(stderr, "sqe submit failed: %d\n", ret); + goto err; + } + + ret = io_uring_wait_cqe(ring, &cqe); + if (ret < 0) { + fprintf(stderr, "wait completion %d\n", ret); + goto err; + } + ret = cqe->res; + io_uring_cqe_seen(ring, cqe); + return ret; +err: + return 1; +} + +static int get_file_size(int fd, loff_t *size) +{ + struct stat st; + + if (fstat(fd, &st) < 0) { + perror("fstat"); + return -1; + } + if (S_ISREG(st.st_mode)) { + *size = st.st_size; + return 0; + } else if (S_ISBLK(st.st_mode)) { + unsigned long long bytes; + + if (ioctl(fd, BLKGETSIZE64, &bytes) != 0) { + perror("ioctl"); + return -1; + } + + *size = bytes; + return 0; + } + + return -1; +} + +int main(int argc, char *argv[]) +{ + struct io_uring ring; + char path[32] = ".truncate.XXXXXX"; + int ret; + int fd; + int i; + loff_t size; + loff_t test_sizes[3]; + + if (argc > 1) + return T_EXIT_SKIP; + + ret = io_uring_queue_init(1, &ring, 0); + if (ret) { + fprintf(stderr, "ring setup failed: %d\n", ret); + return T_EXIT_FAIL; + } + + fd = mkostemp(path, O_WRONLY | O_CREAT | O_TRUNC); + if (fd < 0) { + perror("mkostemp"); + return T_EXIT_FAIL; + } + + test_sizes[0] = TWO_GIG_SIZE; + test_sizes[1] = ONE_GIG_SIZE; + test_sizes[2] = HALF_GIG_SIZE; + + for (i = 0; i < 3; i++) { + ret = test_ftruncate(&ring, fd, test_sizes[i]); + if (ret < 0) { + if (ret == -EBADF || ret == -EINVAL) { + if (i == 0) { + fprintf(stdout, "Ftruncate not supported, skipping\n"); + ret = T_EXIT_SKIP; + goto out; + } + goto err; + } + fprintf(stderr, "ftruncate: %s\n", strerror(-ret)); + goto err; + } else if (ret) { + fprintf(stderr, "unexpected cqe->res %d\n", ret); + goto err; + } + if (get_file_size(fd, &size)) + goto err; + if (size != test_sizes[i]) { + fprintf(stderr, "fail %d size=%llu, %llu\n", i, + (unsigned long long) size, + (unsigned long long) test_sizes[i]); + goto err; + } + } + + ret = test_truncate(&ring, fd); + if (ret != T_EXIT_PASS) + goto err; + +out: + unlink(path); + close(fd); + return T_EXIT_PASS; +err: + unlink(path); + close(fd); + return T_EXIT_FAIL; +} |