#include <stdio.h>
#include <assert.h>
#include <liburing.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/udp.h>

void recvmsg_syscall(int s)
{
    struct sockaddr_in addr;

    char buff[1024];
    struct iovec iov = {
        .iov_base = buff,
        .iov_len = sizeof(buff)
    };

    struct msghdr msg = {
        .msg_name = &addr,
        .msg_namelen = sizeof(addr),
        .msg_iov = &iov,
        .msg_iovlen = 1,
    };

    int res = recvmsg(s, &msg, 0/*MSG_DONTWAIT*/);

    assert(res == -1);
    assert(errno == EAGAIN);
}

void recvmsg_iouring(int s)
{
    struct sockaddr_in addr;

    char buff[1024];
    struct iovec iov = {
        .iov_base = buff,
        .iov_len = sizeof(buff)
    };

    struct msghdr msg = {
        .msg_name = &addr,
        .msg_namelen = sizeof(addr),
        .msg_iov = &iov,
        .msg_iovlen = 1,
    };

    struct io_uring ring;

    int res = io_uring_queue_init(1024, &ring, 0);
    assert(res == 0);

    struct io_uring_cqe *cqe;
    struct io_uring_sqe* sqe;

    sqe = io_uring_get_sqe(&ring);
    io_uring_prep_recvmsg(sqe, s, &msg, 0/*MSG_DONTWAIT*/);

    res = io_uring_submit(&ring);
    assert(res == 1);

    res = io_uring_wait_cqe(&ring, &cqe);
    assert(res == 0);

    assert(cqe->res == -EAGAIN);

    io_uring_cqe_seen(&ring, cqe);

    io_uring_queue_exit(&ring);
}

int main(int argc, const char* argv[])
{
    int s = socket(AF_INET, SOCK_DGRAM | SOCK_NONBLOCK, 0);
    assert(s != -1);

    struct sockaddr_in addr = {
        .sin_family = AF_INET,
        .sin_port = 0x0010,
        .sin_addr.s_addr = 0x0100007fUL
    };

    int res = bind(s, (struct sockaddr*)(&addr), sizeof(addr));
    assert(res == 0);

    recvmsg_syscall(s);
    recvmsg_iouring(s);

    return 0;
}