|
Packit |
d3489f |
/* SPDX-License-Identifier: MIT */
|
|
Packit |
d3489f |
/*
|
|
Packit |
d3489f |
* Simple test case showing using send and recv through io_uring
|
|
Packit |
d3489f |
*/
|
|
Packit |
d3489f |
#include <errno.h>
|
|
Packit |
d3489f |
#include <stdio.h>
|
|
Packit |
d3489f |
#include <stdlib.h>
|
|
Packit |
d3489f |
#include <string.h>
|
|
Packit |
d3489f |
#include <unistd.h>
|
|
Packit |
d3489f |
#include <arpa/inet.h>
|
|
Packit |
d3489f |
#include <sys/types.h>
|
|
Packit |
d3489f |
#include <sys/socket.h>
|
|
Packit |
d3489f |
#include <pthread.h>
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
#include "liburing.h"
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static char str[] = "This is a test of send and recv over io_uring!";
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
#define MAX_MSG 128
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
#define PORT 10200
|
|
Packit |
d3489f |
#define HOST "127.0.0.1"
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
#if 0
|
|
Packit |
d3489f |
# define io_uring_prep_send io_uring_prep_write
|
|
Packit |
d3489f |
# define io_uring_prep_recv io_uring_prep_read
|
|
Packit |
d3489f |
#endif
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static int recv_prep(struct io_uring *ring, struct iovec *iov, int *sock)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
struct sockaddr_in saddr;
|
|
Packit |
d3489f |
struct io_uring_sqe *sqe;
|
|
Packit |
d3489f |
int sockfd, ret, val;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
memset(&saddr, 0, sizeof(saddr));
|
|
Packit |
d3489f |
saddr.sin_family = AF_INET;
|
|
Packit |
d3489f |
saddr.sin_addr.s_addr = htonl(INADDR_ANY);
|
|
Packit |
d3489f |
saddr.sin_port = htons(PORT);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
sockfd = socket(AF_INET, SOCK_DGRAM, 0);
|
|
Packit |
d3489f |
if (sockfd < 0) {
|
|
Packit |
d3489f |
perror("socket");
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
val = 1;
|
|
Packit |
d3489f |
setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val));
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = bind(sockfd, (struct sockaddr *)&saddr, sizeof(saddr));
|
|
Packit |
d3489f |
if (ret < 0) {
|
|
Packit |
d3489f |
perror("bind");
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = io_uring_register_files(ring, &sockfd, 1);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "file reg failed\n");
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
sqe = io_uring_get_sqe(ring);
|
|
Packit |
d3489f |
io_uring_prep_recv(sqe, 0, iov->iov_base, iov->iov_len, 0);
|
|
Packit |
d3489f |
sqe->flags |= IOSQE_FIXED_FILE;
|
|
Packit |
d3489f |
sqe->user_data = 2;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = io_uring_submit(ring);
|
|
Packit |
d3489f |
if (ret <= 0) {
|
|
Packit |
d3489f |
fprintf(stderr, "submit failed: %d\n", ret);
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
*sock = sockfd;
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
err:
|
|
Packit |
d3489f |
close(sockfd);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static int do_recv(struct io_uring *ring, struct iovec *iov)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
struct io_uring_cqe *cqe;
|
|
Packit |
d3489f |
int ret;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = io_uring_wait_cqe(ring, &cqe);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stdout, "wait_cqe: %d\n", ret);
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
if (cqe->res == -EINVAL) {
|
|
Packit |
d3489f |
fprintf(stdout, "recv not supported, skipping\n");
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
if (cqe->res < 0) {
|
|
Packit |
d3489f |
fprintf(stderr, "failed cqe: %d\n", cqe->res);
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (cqe->res -1 != strlen(str)) {
|
|
Packit |
d3489f |
fprintf(stderr, "got wrong length: %d/%d\n", cqe->res,
|
|
Packit |
d3489f |
(int) strlen(str) + 1);
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (strcmp(str, iov->iov_base)) {
|
|
Packit |
d3489f |
fprintf(stderr, "string mismatch\n");
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
err:
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
struct recv_data {
|
|
Packit |
d3489f |
pthread_mutex_t mutex;
|
|
Packit |
d3489f |
int use_sqthread;
|
|
Packit |
d3489f |
};
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static void *recv_fn(void *data)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
struct recv_data *rd = data;
|
|
Packit |
d3489f |
char buf[MAX_MSG + 1];
|
|
Packit |
d3489f |
struct iovec iov = {
|
|
Packit |
d3489f |
.iov_base = buf,
|
|
Packit |
d3489f |
.iov_len = sizeof(buf) - 1,
|
|
Packit |
d3489f |
};
|
|
Packit |
d3489f |
struct io_uring_params p = { };
|
|
Packit |
d3489f |
struct io_uring ring;
|
|
Packit |
d3489f |
int ret, sock;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (rd->use_sqthread)
|
|
Packit |
d3489f |
p.flags = IORING_SETUP_SQPOLL;
|
|
Packit |
d3489f |
ret = io_uring_queue_init_params(1, &ring, &p);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "queue init failed: %d\n", ret);
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = recv_prep(&ring, &iov, &sock);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "recv_prep failed: %d\n", ret);
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
pthread_mutex_unlock(&rd->mutex);
|
|
Packit |
d3489f |
ret = do_recv(&ring, &iov;;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
close(sock);
|
|
Packit |
d3489f |
io_uring_queue_exit(&ring);
|
|
Packit |
d3489f |
err:
|
|
Packit |
d3489f |
return (void *)(intptr_t)ret;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static int do_send(void)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
struct sockaddr_in saddr;
|
|
Packit |
d3489f |
struct iovec iov = {
|
|
Packit |
d3489f |
.iov_base = str,
|
|
Packit |
d3489f |
.iov_len = sizeof(str),
|
|
Packit |
d3489f |
};
|
|
Packit |
d3489f |
struct io_uring ring;
|
|
Packit |
d3489f |
struct io_uring_cqe *cqe;
|
|
Packit |
d3489f |
struct io_uring_sqe *sqe;
|
|
Packit |
d3489f |
int sockfd, ret;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = io_uring_queue_init(1, &ring, 0);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "queue init failed: %d\n", ret);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
memset(&saddr, 0, sizeof(saddr));
|
|
Packit |
d3489f |
saddr.sin_family = AF_INET;
|
|
Packit |
d3489f |
saddr.sin_port = htons(PORT);
|
|
Packit |
d3489f |
inet_pton(AF_INET, HOST, &saddr.sin_addr);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
sockfd = socket(AF_INET, SOCK_DGRAM, 0);
|
|
Packit |
d3489f |
if (sockfd < 0) {
|
|
Packit |
d3489f |
perror("socket");
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = connect(sockfd, &saddr, sizeof(saddr));
|
|
Packit |
d3489f |
if (ret < 0) {
|
|
Packit |
d3489f |
perror("connect");
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
sqe = io_uring_get_sqe(&ring);
|
|
Packit |
d3489f |
io_uring_prep_send(sqe, sockfd, iov.iov_base, iov.iov_len, 0);
|
|
Packit |
d3489f |
sqe->user_data = 1;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = io_uring_submit(&ring);
|
|
Packit |
d3489f |
if (ret <= 0) {
|
|
Packit |
d3489f |
fprintf(stderr, "submit failed: %d\n", ret);
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = io_uring_wait_cqe(&ring, &cqe);
|
|
Packit |
d3489f |
if (cqe->res == -EINVAL) {
|
|
Packit |
d3489f |
fprintf(stdout, "send not supported, skipping\n");
|
|
Packit |
d3489f |
close(sockfd);
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
if (cqe->res != iov.iov_len) {
|
|
Packit |
d3489f |
fprintf(stderr, "failed cqe: %d\n", cqe->res);
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
close(sockfd);
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
err:
|
|
Packit |
d3489f |
close(sockfd);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static int test(int use_sqthread)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
pthread_mutexattr_t attr;
|
|
Packit |
d3489f |
pthread_t recv_thread;
|
|
Packit |
d3489f |
struct recv_data rd;
|
|
Packit |
d3489f |
int ret;
|
|
Packit |
d3489f |
void *retval;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
pthread_mutexattr_init(&attr);
|
|
Packit |
d3489f |
pthread_mutexattr_setpshared(&attr, 1);
|
|
Packit |
d3489f |
pthread_mutex_init(&rd.mutex, &attr);
|
|
Packit |
d3489f |
pthread_mutex_lock(&rd.mutex);
|
|
Packit |
d3489f |
rd.use_sqthread = use_sqthread;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = pthread_create(&recv_thread, NULL, recv_fn, &rd);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "Thread create failed: %d\n", ret);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
pthread_mutex_lock(&rd.mutex);
|
|
Packit |
d3489f |
do_send();
|
|
Packit |
d3489f |
pthread_join(recv_thread, &retval);
|
|
Packit |
d3489f |
return (int)(intptr_t)retval;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
int main(int argc, char *argv[])
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
int ret;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (argc > 1)
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = test(0);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "test sqthread=0 failed\n");
|
|
Packit |
d3489f |
return ret;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (geteuid()) {
|
|
Packit |
d3489f |
fprintf(stdout, "%s: skipping SQPOLL variant\n", argv[0]);
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = test(1);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "test sqthread=1 failed\n");
|
|
Packit |
d3489f |
return ret;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
}
|