mirror of
https://github.com/torvalds/linux.git
synced 2026-05-21 21:37:25 +02:00
selftests: iou-zcrx: Get the page size at runtime
Use the API `sysconf()` to query page size at runtime, instead of using hard code number 4096. And use `posix_memalign` to allocate the page size aligned momory. Signed-off-by: Haiyue Wang <haiyuewa@163.com> Reviewed-by: Simon Horman <horms@kernel.org> Reviewed-by: David Wei <dw@davidwei.uk> Link: https://patch.msgid.link/20250419141044.10304-1-haiyuewa@163.com Signed-off-by: Jakub Kicinski <kuba@kernel.org>
This commit is contained in:
parent
5565acd1e6
commit
df8cf32413
|
|
@ -37,8 +37,8 @@
|
|||
|
||||
#include <liburing.h>
|
||||
|
||||
#define PAGE_SIZE (4096)
|
||||
#define AREA_SIZE (8192 * PAGE_SIZE)
|
||||
static long page_size;
|
||||
#define AREA_SIZE (8192 * page_size)
|
||||
#define SEND_SIZE (512 * 4096)
|
||||
#define min(a, b) \
|
||||
({ \
|
||||
|
|
@ -66,7 +66,7 @@ static int cfg_oneshot_recvs;
|
|||
static int cfg_send_size = SEND_SIZE;
|
||||
static struct sockaddr_in6 cfg_addr;
|
||||
|
||||
static char payload[SEND_SIZE] __attribute__((aligned(PAGE_SIZE)));
|
||||
static char *payload;
|
||||
static void *area_ptr;
|
||||
static void *ring_ptr;
|
||||
static size_t ring_size;
|
||||
|
|
@ -114,8 +114,8 @@ static inline size_t get_refill_ring_size(unsigned int rq_entries)
|
|||
|
||||
ring_size = rq_entries * sizeof(struct io_uring_zcrx_rqe);
|
||||
/* add space for the header (head/tail/etc.) */
|
||||
ring_size += PAGE_SIZE;
|
||||
return ALIGN_UP(ring_size, 4096);
|
||||
ring_size += page_size;
|
||||
return ALIGN_UP(ring_size, page_size);
|
||||
}
|
||||
|
||||
static void setup_zcrx(struct io_uring *ring)
|
||||
|
|
@ -219,7 +219,7 @@ static void process_accept(struct io_uring *ring, struct io_uring_cqe *cqe)
|
|||
|
||||
connfd = cqe->res;
|
||||
if (cfg_oneshot)
|
||||
add_recvzc_oneshot(ring, connfd, PAGE_SIZE);
|
||||
add_recvzc_oneshot(ring, connfd, page_size);
|
||||
else
|
||||
add_recvzc(ring, connfd);
|
||||
}
|
||||
|
|
@ -245,7 +245,7 @@ static void process_recvzc(struct io_uring *ring, struct io_uring_cqe *cqe)
|
|||
|
||||
if (cfg_oneshot) {
|
||||
if (cqe->res == 0 && cqe->flags == 0 && cfg_oneshot_recvs) {
|
||||
add_recvzc_oneshot(ring, connfd, PAGE_SIZE);
|
||||
add_recvzc_oneshot(ring, connfd, page_size);
|
||||
cfg_oneshot_recvs--;
|
||||
}
|
||||
} else if (!(cqe->flags & IORING_CQE_F_MORE)) {
|
||||
|
|
@ -370,7 +370,7 @@ static void usage(const char *filepath)
|
|||
|
||||
static void parse_opts(int argc, char **argv)
|
||||
{
|
||||
const int max_payload_len = sizeof(payload) -
|
||||
const int max_payload_len = SEND_SIZE -
|
||||
sizeof(struct ipv6hdr) -
|
||||
sizeof(struct tcphdr) -
|
||||
40 /* max tcp options */;
|
||||
|
|
@ -443,6 +443,13 @@ int main(int argc, char **argv)
|
|||
const char *cfg_test = argv[argc - 1];
|
||||
int i;
|
||||
|
||||
page_size = sysconf(_SC_PAGESIZE);
|
||||
if (page_size < 0)
|
||||
return 1;
|
||||
|
||||
if (posix_memalign((void **)&payload, page_size, SEND_SIZE))
|
||||
return 1;
|
||||
|
||||
parse_opts(argc, argv);
|
||||
|
||||
for (i = 0; i < SEND_SIZE; i++)
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user