summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJens Axboe <axboe@kernel.dk>2019-10-17 13:58:41 -0600
committerJens Axboe <axboe@kernel.dk>2019-10-17 13:58:41 -0600
commit08bd815170ab4352d71019f4d3e532cd3f6f0489 (patch)
treefef2783c61236f10406b0a0deaa3dce346a3ea63
parent989630a7d7643fac30534db602c031b7147fbf30 (diff)
downloadliburing-08bd815170ab4352d71019f4d3e532cd3f6f0489.tar.gz
liburing-08bd815170ab4352d71019f4d3e532cd3f6f0489.tar.bz2
Un-DOSify test/232c93d07b74-test.c
Signed-off-by: Jens Axboe <axboe@kernel.dk>
-rw-r--r--test/232c93d07b74-test.c564
1 files changed, 282 insertions, 282 deletions
diff --git a/test/232c93d07b74-test.c b/test/232c93d07b74-test.c
index 51fdb0a..b321668 100644
--- a/test/232c93d07b74-test.c
+++ b/test/232c93d07b74-test.c
@@ -1,282 +1,282 @@
-/*
- * Test case for socket read/write through IORING_OP_READV and
- * IORING_OP_WRITEV, using both TCP and sockets and blocking and
- * non-blocking IO.
- *
- * Heavily based on a test case from Hrvoje Zeba <zeba.hrvoje@gmail.com>
- */
-#include <stdio.h>
-#include <stdlib.h>
-#include <stdint.h>
-#include <assert.h>
-
-#include <pthread.h>
-#include <errno.h>
-#include <fcntl.h>
-#include <unistd.h>
-#include <sys/socket.h>
-#include <sys/un.h>
-#include <netinet/tcp.h>
-#include <netinet/in.h>
-
-#include "liburing.h"
-
-#define RECV_BUFF_SIZE 2
-#define SEND_BUFF_SIZE 3
-
-#define PORT 0x1235
-
-struct params {
- int tcp;
- int non_blocking;
-};
-
-pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
-pthread_cond_t cond = PTHREAD_COND_INITIALIZER;
-int rcv_ready = 0;
-
-static void set_rcv_ready(void)
-{
- pthread_mutex_lock(&mutex);
-
- rcv_ready = 1;
- pthread_cond_signal(&cond);
-
- pthread_mutex_unlock(&mutex);
-}
-
-static void wait_for_rcv_ready(void)
-{
- pthread_mutex_lock(&mutex);
-
- while (!rcv_ready)
- pthread_cond_wait(&cond, &mutex);
-
- pthread_mutex_unlock(&mutex);
-}
-
-static void *rcv(void *arg)
-{
- struct params *p = arg;
- int s0;
-
- if (p->tcp) {
- int val = 1;
-
- s0 = socket(AF_INET, SOCK_STREAM | SOCK_CLOEXEC, IPPROTO_TCP);
- assert(setsockopt(s0, SOL_SOCKET, SO_REUSEPORT, &val, sizeof(val)) != -1);
- assert(setsockopt(s0, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val)) != -1);
-
- struct sockaddr_in addr;
-
- addr.sin_family = AF_INET;
- addr.sin_port = PORT;
- addr.sin_addr.s_addr = 0x0100007fU;
- assert(bind(s0, (struct sockaddr *) &addr, sizeof(addr)) != -1);
- } else {
- s0 = socket(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0);
- assert(s0 != -1);
-
- struct sockaddr_un addr;
- memset(&addr, 0, sizeof(addr));
-
- addr.sun_family = AF_UNIX;
- memcpy(addr.sun_path, "\0sock", 6);
- assert(bind(s0, (struct sockaddr *) &addr, sizeof(addr)) != -1);
- }
-
- assert(listen(s0, 128) != -1);
-
- set_rcv_ready();
-
- int s1 = accept(s0, NULL, NULL);
- assert(s1 != -1);
-
- if (p->non_blocking) {
- int flags = fcntl(s1, F_GETFL, 0);
- assert(flags != -1);
-
- flags |= O_NONBLOCK;
- assert(fcntl(s1, F_SETFL, flags) != -1);
- }
-
- struct io_uring m_io_uring;
- void *ret = NULL;
-
- assert(io_uring_queue_init(32, &m_io_uring, 0) >= 0);
-
- int bytes_read = 0;
- int expected_byte = 0;
- int done = 0;
-
- while (!done && bytes_read != 33) {
- char buff[RECV_BUFF_SIZE];
- struct iovec iov;
-
- iov.iov_base = buff;
- iov.iov_len = sizeof(buff);
-
- struct io_uring_sqe *sqe = io_uring_get_sqe(&m_io_uring);
- assert(sqe != NULL);
-
- io_uring_prep_readv(sqe, s1, &iov, 1, 0);
-
- assert(io_uring_submit(&m_io_uring) != -1);
-
- struct io_uring_cqe *cqe;
- unsigned head;
- unsigned count = 0;
-
- while (!done && count != 1) {
- io_uring_for_each_cqe(&m_io_uring, head, cqe) {
- if (cqe->res < 0)
- assert(cqe->res == -EAGAIN);
- else {
- for (int i = 0; i < cqe->res; i++) {
- if (buff[i] != expected_byte) {
- fprintf(stderr,
- "Received %d, wanted %d\n",
- buff[i], expected_byte);
- ret++;
- done = 1;
- }
- expected_byte++;
- }
- bytes_read += cqe->res;
- }
-
- count++;
- }
-
- assert(count <= 1);
- io_uring_cq_advance(&m_io_uring, count);
- }
- }
-
- shutdown(s1, SHUT_RDWR);
- close(s1);
- close(s0);
- io_uring_queue_exit(&m_io_uring);
- return ret;
-}
-
-static void *snd(void *arg)
-{
- struct params *p = arg;
- int s0;
-
- wait_for_rcv_ready();
-
- if (p->tcp) {
- int val = 1;
-
- s0 = socket(AF_INET, SOCK_STREAM | SOCK_CLOEXEC, IPPROTO_TCP);
- assert(setsockopt(s0, IPPROTO_TCP, TCP_NODELAY, &val, sizeof(val)) != -1);
-
- struct sockaddr_in addr;
-
- addr.sin_family = AF_INET;
- addr.sin_port = PORT;
- addr.sin_addr.s_addr = 0x0100007fU;
- assert(connect(s0, (struct sockaddr*) &addr, sizeof(addr)) != -1);
- } else {
- s0 = socket(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0);
- assert(s0 != -1);
-
- struct sockaddr_un addr;
- memset(&addr, 0, sizeof(addr));
-
- addr.sun_family = AF_UNIX;
- memcpy(addr.sun_path, "\0sock", 6);
- assert(connect(s0, (struct sockaddr*) &addr, sizeof(addr)) != -1);
- }
-
- if (p->non_blocking) {
- int flags = fcntl(s0, F_GETFL, 0);
- assert(flags != -1);
-
- flags |= O_NONBLOCK;
- assert(fcntl(s0, F_SETFL, flags) != -1);
- }
-
- struct io_uring m_io_uring;
-
- assert(io_uring_queue_init(32, &m_io_uring, 0) >= 0);
-
- int bytes_written = 0;
- int done = 0;
-
- while (!done && bytes_written != 33) {
- char buff[SEND_BUFF_SIZE];
-
- for (int i = 0; i < SEND_BUFF_SIZE; i++)
- buff[i] = i + bytes_written;
-
- struct iovec iov;
-
- iov.iov_base = buff;
- iov.iov_len = sizeof(buff);
-
- struct io_uring_sqe *sqe = io_uring_get_sqe(&m_io_uring);
- assert(sqe != NULL);
-
- io_uring_prep_writev(sqe, s0, &iov, 1, 0);
-
- assert(io_uring_submit(&m_io_uring) != -1);
-
- struct io_uring_cqe *cqe;
- unsigned head;
- unsigned count = 0;
-
- while (!done && count != 1) {
- io_uring_for_each_cqe(&m_io_uring, head, cqe) {
- if (cqe->res < 0) {
- if (cqe->res == -EPIPE) {
- done = 1;
- break;
- }
- assert(cqe->res == -EAGAIN);
- } else {
- bytes_written += cqe->res;
- }
-
- count++;
- }
-
- assert(count <= 1);
- io_uring_cq_advance(&m_io_uring, count);
- }
- usleep(100000);
- }
-
- shutdown(s0, SHUT_RDWR);
- close(s0);
- io_uring_queue_exit(&m_io_uring);
- return NULL;
-}
-
-int main(int argc, char *argv[])
-{
- struct params p;
- pthread_t t1, t2;
- void *res1, *res2;
- int i, exit_val = 0;
-
- for (i = 0; i < 4; i++) {
- p.tcp = i & 1;
- p.non_blocking = (i & 2) >> 1;
-
- rcv_ready = 0;
-
- pthread_create(&t1, NULL, rcv, &p);
- pthread_create(&t2, NULL, snd, &p);
- pthread_join(t1, &res1);
- pthread_join(t2, &res2);
- if (res1 || res2) {
- fprintf(stderr, "Failed tcp=%d, non_blocking=%d\n", p.tcp, p.non_blocking);
- exit_val = 1;
- }
- }
-
- return exit_val;
-}
+/*
+ * Test case for socket read/write through IORING_OP_READV and
+ * IORING_OP_WRITEV, using both TCP and sockets and blocking and
+ * non-blocking IO.
+ *
+ * Heavily based on a test case from Hrvoje Zeba <zeba.hrvoje@gmail.com>
+ */
+#include <stdio.h>
+#include <stdlib.h>
+#include <stdint.h>
+#include <assert.h>
+
+#include <pthread.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <sys/socket.h>
+#include <sys/un.h>
+#include <netinet/tcp.h>
+#include <netinet/in.h>
+
+#include "liburing.h"
+
+#define RECV_BUFF_SIZE 2
+#define SEND_BUFF_SIZE 3
+
+#define PORT 0x1235
+
+struct params {
+ int tcp;
+ int non_blocking;
+};
+
+pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
+pthread_cond_t cond = PTHREAD_COND_INITIALIZER;
+int rcv_ready = 0;
+
+static void set_rcv_ready(void)
+{
+ pthread_mutex_lock(&mutex);
+
+ rcv_ready = 1;
+ pthread_cond_signal(&cond);
+
+ pthread_mutex_unlock(&mutex);
+}
+
+static void wait_for_rcv_ready(void)
+{
+ pthread_mutex_lock(&mutex);
+
+ while (!rcv_ready)
+ pthread_cond_wait(&cond, &mutex);
+
+ pthread_mutex_unlock(&mutex);
+}
+
+static void *rcv(void *arg)
+{
+ struct params *p = arg;
+ int s0;
+
+ if (p->tcp) {
+ int val = 1;
+
+ s0 = socket(AF_INET, SOCK_STREAM | SOCK_CLOEXEC, IPPROTO_TCP);
+ assert(setsockopt(s0, SOL_SOCKET, SO_REUSEPORT, &val, sizeof(val)) != -1);
+ assert(setsockopt(s0, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val)) != -1);
+
+ struct sockaddr_in addr;
+
+ addr.sin_family = AF_INET;
+ addr.sin_port = PORT;
+ addr.sin_addr.s_addr = 0x0100007fU;
+ assert(bind(s0, (struct sockaddr *) &addr, sizeof(addr)) != -1);
+ } else {
+ s0 = socket(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0);
+ assert(s0 != -1);
+
+ struct sockaddr_un addr;
+ memset(&addr, 0, sizeof(addr));
+
+ addr.sun_family = AF_UNIX;
+ memcpy(addr.sun_path, "\0sock", 6);
+ assert(bind(s0, (struct sockaddr *) &addr, sizeof(addr)) != -1);
+ }
+
+ assert(listen(s0, 128) != -1);
+
+ set_rcv_ready();
+
+ int s1 = accept(s0, NULL, NULL);
+ assert(s1 != -1);
+
+ if (p->non_blocking) {
+ int flags = fcntl(s1, F_GETFL, 0);
+ assert(flags != -1);
+
+ flags |= O_NONBLOCK;
+ assert(fcntl(s1, F_SETFL, flags) != -1);
+ }
+
+ struct io_uring m_io_uring;
+ void *ret = NULL;
+
+ assert(io_uring_queue_init(32, &m_io_uring, 0) >= 0);
+
+ int bytes_read = 0;
+ int expected_byte = 0;
+ int done = 0;
+
+ while (!done && bytes_read != 33) {
+ char buff[RECV_BUFF_SIZE];
+ struct iovec iov;
+
+ iov.iov_base = buff;
+ iov.iov_len = sizeof(buff);
+
+ struct io_uring_sqe *sqe = io_uring_get_sqe(&m_io_uring);
+ assert(sqe != NULL);
+
+ io_uring_prep_readv(sqe, s1, &iov, 1, 0);
+
+ assert(io_uring_submit(&m_io_uring) != -1);
+
+ struct io_uring_cqe *cqe;
+ unsigned head;
+ unsigned count = 0;
+
+ while (!done && count != 1) {
+ io_uring_for_each_cqe(&m_io_uring, head, cqe) {
+ if (cqe->res < 0)
+ assert(cqe->res == -EAGAIN);
+ else {
+ for (int i = 0; i < cqe->res; i++) {
+ if (buff[i] != expected_byte) {
+ fprintf(stderr,
+ "Received %d, wanted %d\n",
+ buff[i], expected_byte);
+ ret++;
+ done = 1;
+ }
+ expected_byte++;
+ }
+ bytes_read += cqe->res;
+ }
+
+ count++;
+ }
+
+ assert(count <= 1);
+ io_uring_cq_advance(&m_io_uring, count);
+ }
+ }
+
+ shutdown(s1, SHUT_RDWR);
+ close(s1);
+ close(s0);
+ io_uring_queue_exit(&m_io_uring);
+ return ret;
+}
+
+static void *snd(void *arg)
+{
+ struct params *p = arg;
+ int s0;
+
+ wait_for_rcv_ready();
+
+ if (p->tcp) {
+ int val = 1;
+
+ s0 = socket(AF_INET, SOCK_STREAM | SOCK_CLOEXEC, IPPROTO_TCP);
+ assert(setsockopt(s0, IPPROTO_TCP, TCP_NODELAY, &val, sizeof(val)) != -1);
+
+ struct sockaddr_in addr;
+
+ addr.sin_family = AF_INET;
+ addr.sin_port = PORT;
+ addr.sin_addr.s_addr = 0x0100007fU;
+ assert(connect(s0, (struct sockaddr*) &addr, sizeof(addr)) != -1);
+ } else {
+ s0 = socket(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0);
+ assert(s0 != -1);
+
+ struct sockaddr_un addr;
+ memset(&addr, 0, sizeof(addr));
+
+ addr.sun_family = AF_UNIX;
+ memcpy(addr.sun_path, "\0sock", 6);
+ assert(connect(s0, (struct sockaddr*) &addr, sizeof(addr)) != -1);
+ }
+
+ if (p->non_blocking) {
+ int flags = fcntl(s0, F_GETFL, 0);
+ assert(flags != -1);
+
+ flags |= O_NONBLOCK;
+ assert(fcntl(s0, F_SETFL, flags) != -1);
+ }
+
+ struct io_uring m_io_uring;
+
+ assert(io_uring_queue_init(32, &m_io_uring, 0) >= 0);
+
+ int bytes_written = 0;
+ int done = 0;
+
+ while (!done && bytes_written != 33) {
+ char buff[SEND_BUFF_SIZE];
+
+ for (int i = 0; i < SEND_BUFF_SIZE; i++)
+ buff[i] = i + bytes_written;
+
+ struct iovec iov;
+
+ iov.iov_base = buff;
+ iov.iov_len = sizeof(buff);
+
+ struct io_uring_sqe *sqe = io_uring_get_sqe(&m_io_uring);
+ assert(sqe != NULL);
+
+ io_uring_prep_writev(sqe, s0, &iov, 1, 0);
+
+ assert(io_uring_submit(&m_io_uring) != -1);
+
+ struct io_uring_cqe *cqe;
+ unsigned head;
+ unsigned count = 0;
+
+ while (!done && count != 1) {
+ io_uring_for_each_cqe(&m_io_uring, head, cqe) {
+ if (cqe->res < 0) {
+ if (cqe->res == -EPIPE) {
+ done = 1;
+ break;
+ }
+ assert(cqe->res == -EAGAIN);
+ } else {
+ bytes_written += cqe->res;
+ }
+
+ count++;
+ }
+
+ assert(count <= 1);
+ io_uring_cq_advance(&m_io_uring, count);
+ }
+ usleep(100000);
+ }
+
+ shutdown(s0, SHUT_RDWR);
+ close(s0);
+ io_uring_queue_exit(&m_io_uring);
+ return NULL;
+}
+
+int main(int argc, char *argv[])
+{
+ struct params p;
+ pthread_t t1, t2;
+ void *res1, *res2;
+ int i, exit_val = 0;
+
+ for (i = 0; i < 4; i++) {
+ p.tcp = i & 1;
+ p.non_blocking = (i & 2) >> 1;
+
+ rcv_ready = 0;
+
+ pthread_create(&t1, NULL, rcv, &p);
+ pthread_create(&t2, NULL, snd, &p);
+ pthread_join(t1, &res1);
+ pthread_join(t2, &res2);
+ if (res1 || res2) {
+ fprintf(stderr, "Failed tcp=%d, non_blocking=%d\n", p.tcp, p.non_blocking);
+ exit_val = 1;
+ }
+ }
+
+ return exit_val;
+}