You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
1.6 KiB
74 lines
1.6 KiB
#include "uring.h"
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
#include "log.h"
|
|
#include "loglib.h"
|
|
#include "socket.h"
|
|
#include "poller.h"
|
|
|
|
|
|
static ssize_t __socket_sendmsg(socket_t *s, struct msghdr *m, const endpoint_t *e,
|
|
struct sockaddr_storage *ss, struct uring_req *r)
|
|
{
|
|
ssize_t ret = socket_sendmsg(s, m, e);
|
|
r->handler(r, 0, 0);
|
|
return ret;
|
|
}
|
|
|
|
|
|
__thread __typeof(__socket_sendmsg) (*uring_sendmsg) = __socket_sendmsg;
|
|
|
|
|
|
#ifdef HAVE_LIBURING
|
|
|
|
#include <liburing.h>
|
|
|
|
|
|
static __thread struct io_uring rtpe_uring;
|
|
|
|
|
|
static ssize_t __uring_sendmsg(socket_t *s, struct msghdr *m, const endpoint_t *e,
|
|
struct sockaddr_storage *ss, struct uring_req *r)
|
|
{
|
|
struct io_uring_sqe *sqe = io_uring_get_sqe(&rtpe_uring);
|
|
assert(sqe != NULL);
|
|
s->family->endpoint2sockaddr(ss, e);
|
|
m->msg_name = ss;
|
|
m->msg_namelen = s->family->sockaddr_size;
|
|
io_uring_sqe_set_data(sqe, r);
|
|
io_uring_prep_sendmsg(sqe, s->fd, m, 0);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static unsigned int __uring_thread_loop(void) {
|
|
io_uring_submit_and_get_events(&rtpe_uring);
|
|
|
|
struct io_uring_cqe *cqe;
|
|
unsigned int head, num = 0;
|
|
io_uring_for_each_cqe(&rtpe_uring, head, cqe) {
|
|
struct uring_req *req = io_uring_cqe_get_data(cqe);
|
|
req->handler(req, cqe->res, cqe->flags);
|
|
num++;
|
|
}
|
|
|
|
io_uring_cq_advance(&rtpe_uring, num);
|
|
|
|
return num;
|
|
}
|
|
|
|
void uring_thread_init(void) {
|
|
struct io_uring_params params = {0};
|
|
int ret = io_uring_queue_init_params(4096, &rtpe_uring, ¶ms);
|
|
if (ret)
|
|
die("io_uring init failed (%s)", strerror(errno));
|
|
|
|
uring_sendmsg = __uring_sendmsg;
|
|
uring_thread_loop = __uring_thread_loop;
|
|
}
|
|
|
|
void uring_thread_cleanup(void) {
|
|
io_uring_queue_exit(&rtpe_uring);
|
|
}
|
|
|
|
#endif
|