|
Packit |
d3489f |
/* SPDX-License-Identifier: MIT */
|
|
Packit |
d3489f |
/*
|
|
Packit |
d3489f |
* Description: test disable/enable notifications through eventfd
|
|
Packit |
d3489f |
*
|
|
Packit |
d3489f |
*/
|
|
Packit |
d3489f |
#include <errno.h>
|
|
Packit |
d3489f |
#include <stdio.h>
|
|
Packit |
d3489f |
#include <unistd.h>
|
|
Packit |
d3489f |
#include <stdlib.h>
|
|
Packit |
d3489f |
#include <string.h>
|
|
Packit |
d3489f |
#include <fcntl.h>
|
|
Packit |
d3489f |
#include <sys/poll.h>
|
|
Packit |
d3489f |
#include <sys/eventfd.h>
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
#include "liburing.h"
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
int main(int argc, char *argv[])
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
struct io_uring_params p = {};
|
|
Packit |
d3489f |
struct io_uring_sqe *sqe;
|
|
Packit |
d3489f |
struct io_uring_cqe *cqe;
|
|
Packit |
d3489f |
struct io_uring ring;
|
|
Packit |
d3489f |
uint64_t ptr;
|
|
Packit |
d3489f |
struct iovec vec = {
|
|
Packit |
d3489f |
.iov_base = &ptr,
|
|
Packit |
d3489f |
.iov_len = sizeof(ptr)
|
|
Packit |
d3489f |
};
|
|
Packit |
d3489f |
int ret, evfd, i;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (argc > 1)
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = io_uring_queue_init_params(64, &ring, &p);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "ring setup failed: %d\n", ret);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
evfd = eventfd(0, EFD_CLOEXEC);
|
|
Packit |
d3489f |
if (evfd < 0) {
|
|
Packit |
d3489f |
perror("eventfd");
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = io_uring_register_eventfd(&ring, evfd);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "failed to register evfd: %d\n", ret);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (!io_uring_cq_eventfd_enabled(&ring)) {
|
|
Packit |
d3489f |
fprintf(stderr, "eventfd disabled\n");
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = io_uring_cq_eventfd_toggle(&ring, false);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stdout, "Skipping, CQ flags not available!\n");
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
sqe = io_uring_get_sqe(&ring);
|
|
Packit |
d3489f |
io_uring_prep_readv(sqe, evfd, &vec, 1, 0);
|
|
Packit |
d3489f |
sqe->user_data = 1;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = io_uring_submit(&ring);
|
|
Packit |
d3489f |
if (ret != 1) {
|
|
Packit |
d3489f |
fprintf(stderr, "submit: %d\n", ret);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
for (i = 0; i < 63; i++) {
|
|
Packit |
d3489f |
sqe = io_uring_get_sqe(&ring);
|
|
Packit |
d3489f |
io_uring_prep_nop(sqe);
|
|
Packit |
d3489f |
sqe->user_data = 2;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = io_uring_submit(&ring);
|
|
Packit |
d3489f |
if (ret != 63) {
|
|
Packit |
d3489f |
fprintf(stderr, "submit: %d\n", ret);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
for (i = 0; i < 63; i++) {
|
|
Packit |
d3489f |
ret = io_uring_wait_cqe(&ring, &cqe);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "wait: %d\n", ret);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
switch (cqe->user_data) {
|
|
Packit |
d3489f |
case 1: /* eventfd */
|
|
Packit |
d3489f |
fprintf(stderr, "eventfd unexpected: %d\n", (int)ptr);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
case 2:
|
|
Packit |
d3489f |
if (cqe->res) {
|
|
Packit |
d3489f |
fprintf(stderr, "nop: %d\n", cqe->res);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
break;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
io_uring_cqe_seen(&ring, cqe);
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = io_uring_cq_eventfd_toggle(&ring, true);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "io_uring_cq_eventfd_toggle: %d\n", ret);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
sqe = io_uring_get_sqe(&ring);
|
|
Packit |
d3489f |
io_uring_prep_nop(sqe);
|
|
Packit |
d3489f |
sqe->user_data = 2;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = io_uring_submit(&ring);
|
|
Packit |
d3489f |
if (ret != 1) {
|
|
Packit |
d3489f |
fprintf(stderr, "submit: %d\n", ret);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
for (i = 0; i < 2; i++) {
|
|
Packit |
d3489f |
ret = io_uring_wait_cqe(&ring, &cqe);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "wait: %d\n", ret);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
switch (cqe->user_data) {
|
|
Packit |
d3489f |
case 1: /* eventfd */
|
|
Packit |
d3489f |
if (cqe->res != sizeof(ptr)) {
|
|
Packit |
d3489f |
fprintf(stderr, "read res: %d\n", cqe->res);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (ptr != 1) {
|
|
Packit |
d3489f |
fprintf(stderr, "eventfd: %d\n", (int)ptr);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
break;
|
|
Packit |
d3489f |
case 2:
|
|
Packit |
d3489f |
if (cqe->res) {
|
|
Packit |
d3489f |
fprintf(stderr, "nop: %d\n", cqe->res);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
break;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
io_uring_cqe_seen(&ring, cqe);
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
}
|