|
Packit |
d3489f |
/* SPDX-License-Identifier: MIT */
|
|
Packit |
d3489f |
/*
|
|
Packit |
d3489f |
* Check that a readv on a socket queued before a writev doesn't hang
|
|
Packit |
d3489f |
* the processing.
|
|
Packit |
d3489f |
*
|
|
Packit |
d3489f |
* From Hrvoje Zeba <zeba.hrvoje@gmail.com>
|
|
Packit |
d3489f |
*/
|
|
Packit |
d3489f |
#include <stdio.h>
|
|
Packit |
d3489f |
#include <stdlib.h>
|
|
Packit |
d3489f |
#include <stdint.h>
|
|
Packit |
d3489f |
#include <assert.h>
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
#include <errno.h>
|
|
Packit |
d3489f |
#include <fcntl.h>
|
|
Packit |
d3489f |
#include <unistd.h>
|
|
Packit |
d3489f |
#include <sys/socket.h>
|
|
Packit |
d3489f |
#include <sys/un.h>
|
|
Packit |
d3489f |
#include <netinet/tcp.h>
|
|
Packit |
d3489f |
#include <netinet/in.h>
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
#include "liburing.h"
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
int main(int argc, char *argv[])
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
int p_fd[2];
|
|
Packit |
d3489f |
int32_t recv_s0;
|
|
Packit |
d3489f |
int32_t val = 1;
|
|
Packit |
d3489f |
struct sockaddr_in addr;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (argc > 1)
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
recv_s0 = socket(AF_INET, SOCK_STREAM | SOCK_CLOEXEC, IPPROTO_TCP);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
assert(setsockopt(recv_s0, SOL_SOCKET, SO_REUSEPORT, &val, sizeof(val)) != -1);
|
|
Packit |
d3489f |
assert(setsockopt(recv_s0, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val)) != -1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
addr.sin_family = AF_INET;
|
|
Packit |
d3489f |
addr.sin_port = 0x1235;
|
|
Packit |
d3489f |
addr.sin_addr.s_addr = 0x0100007fU;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
assert(bind(recv_s0, (struct sockaddr*)&addr, sizeof(addr)) != -1);
|
|
Packit |
d3489f |
assert(listen(recv_s0, 128) != -1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
p_fd[1] = socket(AF_INET, SOCK_STREAM | SOCK_CLOEXEC, IPPROTO_TCP);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
val = 1;
|
|
Packit |
d3489f |
assert(setsockopt(p_fd[1], IPPROTO_TCP, TCP_NODELAY, &val, sizeof(val)) != -1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
int32_t flags = fcntl(p_fd[1], F_GETFL, 0);
|
|
Packit |
d3489f |
assert(flags != -1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
flags |= O_NONBLOCK;
|
|
Packit |
d3489f |
assert(fcntl(p_fd[1], F_SETFL, flags) != -1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
assert(connect(p_fd[1], (struct sockaddr*)&addr, sizeof(addr)) == -1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
flags = fcntl(p_fd[1], F_GETFL, 0);
|
|
Packit |
d3489f |
assert(flags != -1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
flags &= ~O_NONBLOCK;
|
|
Packit |
d3489f |
assert(fcntl(p_fd[1], F_SETFL, flags) != -1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
p_fd[0] = accept(recv_s0, NULL, NULL);
|
|
Packit |
d3489f |
assert(p_fd[0] != -1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
while (1) {
|
|
Packit |
d3489f |
int32_t code;
|
|
Packit |
d3489f |
socklen_t code_len = sizeof(code);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
assert(getsockopt(p_fd[1], SOL_SOCKET, SO_ERROR, &code, &code_len) != -1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (!code)
|
|
Packit |
d3489f |
break;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
struct io_uring m_io_uring;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
assert(io_uring_queue_init(32, &m_io_uring, 0) >= 0);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
char recv_buff[128];
|
|
Packit |
d3489f |
char send_buff[128];
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
struct iovec iov[1];
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
iov[0].iov_base = recv_buff;
|
|
Packit |
d3489f |
iov[0].iov_len = sizeof(recv_buff);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
struct io_uring_sqe* sqe = io_uring_get_sqe(&m_io_uring);
|
|
Packit |
d3489f |
assert(sqe != NULL);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
io_uring_prep_readv(sqe, p_fd[0], iov, 1, 0);
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
struct iovec iov[1];
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
iov[0].iov_base = send_buff;
|
|
Packit |
d3489f |
iov[0].iov_len = sizeof(send_buff);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
struct io_uring_sqe* sqe = io_uring_get_sqe(&m_io_uring);
|
|
Packit |
d3489f |
assert(sqe != NULL);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
io_uring_prep_writev(sqe, p_fd[1], iov, 1, 0);
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
assert(io_uring_submit_and_wait(&m_io_uring, 2) != -1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
struct io_uring_cqe* cqe;
|
|
Packit |
d3489f |
uint32_t head;
|
|
Packit |
d3489f |
uint32_t count = 0;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
while (count != 2) {
|
|
Packit |
d3489f |
io_uring_for_each_cqe(&m_io_uring, head, cqe) {
|
|
Packit |
d3489f |
assert(cqe->res == 128);
|
|
Packit |
d3489f |
count++;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
assert(count <= 2);
|
|
Packit |
d3489f |
io_uring_cq_advance(&m_io_uring, count);
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
io_uring_queue_exit(&m_io_uring);
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
}
|