|
Packit |
d3489f |
/* SPDX-License-Identifier: MIT */
|
|
Packit |
d3489f |
/*
|
|
Packit |
d3489f |
* Description: test many files being polled for
|
|
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 <signal.h>
|
|
Packit |
d3489f |
#include <sys/poll.h>
|
|
Packit |
d3489f |
#include <sys/resource.h>
|
|
Packit |
d3489f |
#include <fcntl.h>
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
#include "liburing.h"
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
#define NFILES 5000
|
|
Packit |
d3489f |
#define BATCH 500
|
|
Packit |
d3489f |
#define NLOOPS 1000
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
#define RING_SIZE 512
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
struct p {
|
|
Packit |
d3489f |
int fd[2];
|
|
Packit |
d3489f |
int triggered;
|
|
Packit |
d3489f |
};
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static struct p p[NFILES];
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static int arm_poll(struct io_uring *ring, int off)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
struct io_uring_sqe *sqe;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
sqe = io_uring_get_sqe(ring);
|
|
Packit |
d3489f |
if (!sqe) {
|
|
Packit |
d3489f |
fprintf(stderr, "failed getting sqe\n");
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
io_uring_prep_poll_add(sqe, p[off].fd[0], POLLIN);
|
|
Packit |
d3489f |
sqe->user_data = off;
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static int reap_polls(struct io_uring *ring)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
struct io_uring_cqe *cqe;
|
|
Packit |
d3489f |
int i, ret, off;
|
|
Packit |
d3489f |
char c;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
for (i = 0; i < BATCH; i++) {
|
|
Packit |
d3489f |
ret = io_uring_wait_cqe(ring, &cqe);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "wait cqe %d\n", ret);
|
|
Packit |
d3489f |
return ret;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
off = cqe->user_data;
|
|
Packit |
d3489f |
p[off].triggered = 0;
|
|
Packit |
d3489f |
ret = read(p[off].fd[0], &c, 1);
|
|
Packit |
d3489f |
if (ret != 1) {
|
|
Packit |
d3489f |
fprintf(stderr, "read got %d/%d\n", ret, errno);
|
|
Packit |
d3489f |
break;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
if (arm_poll(ring, off))
|
|
Packit |
d3489f |
break;
|
|
Packit |
d3489f |
io_uring_cqe_seen(ring, cqe);
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (i != BATCH) {
|
|
Packit |
d3489f |
fprintf(stderr, "gave up at %d\n", i);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = io_uring_submit(ring);
|
|
Packit |
d3489f |
if (ret != BATCH) {
|
|
Packit |
d3489f |
fprintf(stderr, "submitted %d, %d\n", ret, BATCH);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static int trigger_polls(void)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
char c = 89;
|
|
Packit |
d3489f |
int i, ret;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
for (i = 0; i < BATCH; i++) {
|
|
Packit |
d3489f |
int off;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
do {
|
|
Packit |
d3489f |
off = rand() % NFILES;
|
|
Packit |
d3489f |
if (!p[off].triggered)
|
|
Packit |
d3489f |
break;
|
|
Packit |
d3489f |
} while (1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
p[off].triggered = 1;
|
|
Packit |
d3489f |
ret = write(p[off].fd[1], &c, 1);
|
|
Packit |
d3489f |
if (ret != 1) {
|
|
Packit |
d3489f |
fprintf(stderr, "write got %d/%d\n", ret, errno);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static int arm_polls(struct io_uring *ring)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
int ret, to_arm = NFILES, i, off;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
off = 0;
|
|
Packit |
d3489f |
while (to_arm) {
|
|
Packit |
d3489f |
int this_arm;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
this_arm = to_arm;
|
|
Packit |
d3489f |
if (this_arm > RING_SIZE)
|
|
Packit |
d3489f |
this_arm = RING_SIZE;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
for (i = 0; i < this_arm; i++) {
|
|
Packit |
d3489f |
if (arm_poll(ring, off)) {
|
|
Packit |
d3489f |
fprintf(stderr, "arm failed at %d\n", off);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
off++;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = io_uring_submit(ring);
|
|
Packit |
d3489f |
if (ret != this_arm) {
|
|
Packit |
d3489f |
fprintf(stderr, "submitted %d, %d\n", ret, this_arm);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
to_arm -= this_arm;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
int main(int argc, char *argv[])
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
struct io_uring ring;
|
|
Packit |
d3489f |
struct rlimit rlim;
|
|
Packit |
d3489f |
int i, ret;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (argc > 1)
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (getrlimit(RLIMIT_NOFILE, &rlim) < 0) {
|
|
Packit |
d3489f |
perror("getrlimit");
|
|
Packit |
d3489f |
goto err_noring;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (rlim.rlim_cur < (2 * NFILES + 5)) {
|
|
Packit |
d3489f |
rlim.rlim_cur = (2 * NFILES + 5);
|
|
Packit |
d3489f |
rlim.rlim_max = rlim.rlim_cur;
|
|
Packit |
d3489f |
if (setrlimit(RLIMIT_NOFILE, &rlim) < 0) {
|
|
Packit |
d3489f |
if (errno == EPERM)
|
|
Packit |
d3489f |
goto err_nofail;
|
|
Packit |
d3489f |
perror("setrlimit");
|
|
Packit |
d3489f |
goto err_noring;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
for (i = 0; i < NFILES; i++) {
|
|
Packit |
d3489f |
if (pipe(p[i].fd) < 0) {
|
|
Packit |
d3489f |
perror("pipe");
|
|
Packit |
d3489f |
goto err_noring;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (io_uring_queue_init(RING_SIZE, &ring, 0)) {
|
|
Packit |
d3489f |
fprintf(stderr, "failed ring init\n");
|
|
Packit |
d3489f |
goto err_noring;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (arm_polls(&ring))
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
for (i = 0; i < NLOOPS; i++) {
|
|
Packit |
d3489f |
trigger_polls();
|
|
Packit |
d3489f |
ret = reap_polls(&ring);
|
|
Packit |
d3489f |
if (ret)
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
io_uring_queue_exit(&ring);
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
err:
|
|
Packit |
d3489f |
io_uring_queue_exit(&ring);
|
|
Packit |
d3489f |
err_noring:
|
|
Packit |
d3489f |
fprintf(stderr, "poll-many failed\n");
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
err_nofail:
|
|
Packit |
d3489f |
fprintf(stderr, "poll-many: not enough files available (and not root), "
|
|
Packit |
d3489f |
"skipped\n");
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
}
|