|
Packit |
d3489f |
/* SPDX-License-Identifier: MIT */
|
|
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 <assert.h>
|
|
Packit |
d3489f |
#include <pthread.h>
|
|
Packit |
d3489f |
#include <sys/socket.h>
|
|
Packit |
d3489f |
#include <netinet/tcp.h>
|
|
Packit |
d3489f |
#include <netinet/in.h>
|
|
Packit |
d3489f |
#include <poll.h>
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
#include "liburing.h"
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
|
|
Packit |
d3489f |
pthread_cond_t cond = PTHREAD_COND_INITIALIZER;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static int recv_thread_ready = 0;
|
|
Packit |
d3489f |
static int recv_thread_done = 0;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static void signal_var(int *var)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
pthread_mutex_lock(&mutex);
|
|
Packit |
d3489f |
*var = 1;
|
|
Packit |
d3489f |
pthread_cond_signal(&cond;;
|
|
Packit |
d3489f |
pthread_mutex_unlock(&mutex);
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static void wait_for_var(int *var)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
pthread_mutex_lock(&mutex);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
while (!*var)
|
|
Packit |
d3489f |
pthread_cond_wait(&cond, &mutex);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
pthread_mutex_unlock(&mutex);
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
struct data {
|
|
Packit |
d3489f |
unsigned expected[2];
|
|
Packit |
d3489f |
unsigned is_mask[2];
|
|
Packit |
d3489f |
unsigned long timeout;
|
|
Packit |
d3489f |
int port;
|
|
Packit |
d3489f |
int stop;
|
|
Packit |
d3489f |
};
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static void *send_thread(void *arg)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
struct data *data = arg;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
wait_for_var(&recv_thread_ready);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
int s0 = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
|
|
Packit |
d3489f |
assert(s0 != -1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
struct sockaddr_in addr;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
addr.sin_family = AF_INET;
|
|
Packit |
d3489f |
addr.sin_port = data->port;
|
|
Packit |
d3489f |
addr.sin_addr.s_addr = 0x0100007fU;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (connect(s0, (struct sockaddr*)&addr, sizeof(addr)) != -1)
|
|
Packit |
d3489f |
wait_for_var(&recv_thread_done);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
close(s0);
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
void *recv_thread(void *arg)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
struct data *data = arg;
|
|
Packit |
d3489f |
struct io_uring_sqe *sqe;
|
|
Packit |
d3489f |
struct io_uring ring;
|
|
Packit |
d3489f |
int i;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
assert(io_uring_queue_init(8, &ring, 0) == 0);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
int s0 = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
|
|
Packit |
d3489f |
assert(s0 != -1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
int32_t val = 1;
|
|
Packit |
d3489f |
assert(setsockopt(s0, SOL_SOCKET, SO_REUSEPORT, &val, sizeof(val)) != -1);
|
|
Packit |
d3489f |
assert(setsockopt(s0, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val)) != -1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
struct sockaddr_in addr;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
addr.sin_family = AF_INET;
|
|
Packit |
d3489f |
addr.sin_addr.s_addr = 0x0100007fU;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
i = 0;
|
|
Packit |
d3489f |
do {
|
|
Packit |
d3489f |
data->port = 1025 + (rand() % 64510);
|
|
Packit |
d3489f |
addr.sin_port = data->port;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (bind(s0, (struct sockaddr*)&addr, sizeof(addr)) != -1)
|
|
Packit |
d3489f |
break;
|
|
Packit |
d3489f |
} while (++i < 100);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (i >= 100) {
|
|
Packit |
d3489f |
fprintf(stderr, "Can't find good port, skipped\n");
|
|
Packit |
d3489f |
data->stop = 1;
|
|
Packit |
d3489f |
signal_var(&recv_thread_ready);
|
|
Packit |
d3489f |
goto out;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
assert(listen(s0, 128) != -1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
signal_var(&recv_thread_ready);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
sqe = io_uring_get_sqe(&ring);
|
|
Packit |
d3489f |
assert(sqe != NULL);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
io_uring_prep_poll_add(sqe, s0, POLLIN | POLLHUP | POLLERR);
|
|
Packit |
d3489f |
sqe->flags |= IOSQE_IO_LINK;
|
|
Packit |
d3489f |
sqe->user_data = 1;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
sqe = io_uring_get_sqe(&ring);
|
|
Packit |
d3489f |
assert(sqe != NULL);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
struct __kernel_timespec ts;
|
|
Packit |
d3489f |
ts.tv_sec = data->timeout / 1000000000;
|
|
Packit |
d3489f |
ts.tv_nsec = data->timeout % 1000000000;
|
|
Packit |
d3489f |
io_uring_prep_link_timeout(sqe, &ts, 0);
|
|
Packit |
d3489f |
sqe->user_data = 2;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
assert(io_uring_submit(&ring) == 2);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
for (i = 0; i < 2; i++) {
|
|
Packit |
d3489f |
struct io_uring_cqe *cqe;
|
|
Packit |
d3489f |
int idx;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (io_uring_wait_cqe(&ring, &cqe)) {
|
|
Packit |
d3489f |
fprintf(stderr, "wait cqe failed\n");
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
idx = cqe->user_data - 1;
|
|
Packit |
d3489f |
if (data->is_mask[idx] && !(data->expected[idx] & cqe->res)) {
|
|
Packit |
d3489f |
fprintf(stderr, "cqe %llu got %x, wanted mask %x\n",
|
|
Packit |
d3489f |
cqe->user_data, cqe->res,
|
|
Packit |
d3489f |
data->expected[idx]);
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
} else if (!data->is_mask[idx] && cqe->res != data->expected[idx]) {
|
|
Packit |
d3489f |
fprintf(stderr, "cqe %llu got %d, wanted %d\n",
|
|
Packit |
d3489f |
cqe->user_data, cqe->res,
|
|
Packit |
d3489f |
data->expected[idx]);
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
io_uring_cqe_seen(&ring, cqe);
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
out:
|
|
Packit |
d3489f |
signal_var(&recv_thread_done);
|
|
Packit |
d3489f |
close(s0);
|
|
Packit |
d3489f |
io_uring_queue_exit(&ring);
|
|
Packit |
d3489f |
return NULL;
|
|
Packit |
d3489f |
err:
|
|
Packit |
d3489f |
signal_var(&recv_thread_done);
|
|
Packit |
d3489f |
close(s0);
|
|
Packit |
d3489f |
io_uring_queue_exit(&ring);
|
|
Packit |
d3489f |
return (void *) 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static int test_poll_timeout(int do_connect, unsigned long timeout)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
pthread_t t1, t2;
|
|
Packit |
d3489f |
struct data d;
|
|
Packit |
d3489f |
void *tret;
|
|
Packit |
d3489f |
int ret = 0;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
recv_thread_ready = 0;
|
|
Packit |
d3489f |
recv_thread_done = 0;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
memset(&d, 0, sizeof(d));
|
|
Packit |
d3489f |
d.timeout = timeout;
|
|
Packit |
d3489f |
if (!do_connect) {
|
|
Packit |
d3489f |
d.expected[0] = -ECANCELED;
|
|
Packit |
d3489f |
d.expected[1] = -ETIME;
|
|
Packit |
d3489f |
} else {
|
|
Packit |
d3489f |
d.expected[0] = POLLIN;
|
|
Packit |
d3489f |
d.is_mask[0] = 1;
|
|
Packit |
d3489f |
d.expected[1] = -ECANCELED;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
pthread_create(&t1, NULL, recv_thread, &d);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (do_connect)
|
|
Packit |
d3489f |
pthread_create(&t2, NULL, send_thread, &d);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
pthread_join(t1, &tret);
|
|
Packit |
d3489f |
if (tret)
|
|
Packit |
d3489f |
ret++;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (do_connect) {
|
|
Packit |
d3489f |
pthread_join(t2, &tret);
|
|
Packit |
d3489f |
if (tret)
|
|
Packit |
d3489f |
ret++;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
return ret;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
int main(int argc, char *argv[])
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
if (argc > 1)
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
srand(getpid());
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (test_poll_timeout(0, 200000000)) {
|
|
Packit |
d3489f |
fprintf(stderr, "poll timeout 0 failed\n");
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (test_poll_timeout(1, 1000000000)) {
|
|
Packit |
d3489f |
fprintf(stderr, "poll timeout 1 failed\n");
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
}
|