openharmony/third_party_liburing
1
0
Fork 0
mirror of https://github.com/openharmony/third_party_liburing.git synced 2026-08-28 05:10:09 -04:00
Code Issues Packages Projects Releases Wiki Activity

test/zc: test short zc send

Browse Source 
Zerocopy send should hide short writes for stream sockets, test it.

Signed-off-by: Pavel Begunkov <asml.silence@gmail.com>
Link: https://lore.kernel.org/r/6ee389d6b720356ec2a8c677a34a852c761c1627.1659622771.git.asml.silence@gmail.com
Signed-off-by: Jens Axboe <axboe@kernel.dk>
This commit is contained in:
Pavel Begunkov
2022-08-04 15:20:25 +01:00
committed by Jens Axboe
parent 5bcd0fd97d
commit 8e061130fc
1 changed files with 28 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/send-zerocopy.c
+28 -6
View File
@@ -54,7 +54,7 @@
static int seqs[NR_SLOTS];
static char *tx_buffer, *rx_buffer;
static struct iovec buffers_iov[2];
static struct iovec buffers_iov[3];
static inline bool tag_userdata(__u64 user_data)
{
@@ -662,7 +662,7 @@ static int do_test_inet_send(struct io_uring *ring, int sock_client, int sock_se
for (i = 0; i < nr_reqs; i++) {
bool cur_fixed_buf = fixed_buf;
size_t cur_size = chunk_size;
int msg_flags = 0;
int msg_flags = MSG_WAITALL;
if (mix_register)
cur_fixed_buf = rand() & 1;
@@ -791,15 +791,26 @@ static int test_inet_send(struct io_uring *ring)
return 1;
}
for (i = 0; i < 64; i++) {
for (i = 0; i < 128; i++) {
bool fixed_buf = i & 1;
struct sockaddr_storage *addr_arg = (i & 2) ? &addr : NULL;
size_t size = (i & 4) ? 137 : 4096;
bool cork = i & 8;
bool mix_register = i & 16;
bool aligned = i & 32;
bool large_buf = i & 64;
int buf_idx = aligned ? 0 : 1;
if (!tcp || !large_buf)
continue;
if (large_buf) {
buf_idx = 2;
size = buffers_iov[buf_idx].iov_len;
if (!aligned || !tcp)
continue;
}
if (!buffers_iov[buf_idx].iov_base)
continue;
if (tcp && cork)
continue;
if (mix_register && (!cork || fixed_buf))
@@ -829,22 +840,33 @@ int main(int argc, char *argv[])
{
struct io_uring ring;
int i, ret, sp[2];
size_t len = 8096;
size_t len;
if (argc > 1)
return T_EXIT_SKIP;
len = 1U << 25; /* 32MB, should be enough to trigger a short send */
tx_buffer = aligned_alloc(4096, len);
rx_buffer = aligned_alloc(4096, len);
if (tx_buffer && rx_buffer) {
buffers_iov[2].iov_base = tx_buffer;
buffers_iov[2].iov_len = len;
} else {
printf("skip large buffer tests, can't alloc\n");
len = 8192;
tx_buffer = aligned_alloc(4096, len);
rx_buffer = aligned_alloc(4096, len);
}
if (!tx_buffer || !rx_buffer) {
fprintf(stderr, "can't allocate buffers\n");
return T_EXIT_FAIL;
}
buffers_iov[0].iov_base = tx_buffer;
buffers_iov[0].iov_len = len;
buffers_iov[0].iov_len = 8192;
buffers_iov[1].iov_base = tx_buffer + BUFFER_OFFSET;
buffers_iov[1].iov_len = len - BUFFER_OFFSET - 13;
buffers_iov[1].iov_len = 8192 - BUFFER_OFFSET - 13;
ret = io_uring_queue_init(32, &ring, 0);
if (ret) {