f1d7c0b196
A fair bit of setup / teardown code was being duplicated in many test cases in the socketpair testsuite. Take advantage of the `setup()`, `before()`, and `after()` functions in the new ZTest API. Signed-off-by: Christopher Friedt <cfriedt@meta.com>
127 lines
2.8 KiB
C
127 lines
2.8 KiB
C
/*
|
|
* Copyright (c) 2020 Friedt Professional Engineering Services, Inc
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
#include "_main.h"
|
|
|
|
struct ctx {
|
|
/* true if test is write_block(), false if test is read_block() */
|
|
bool write;
|
|
/* the secondary-side socket of the socketpair */
|
|
int fd;
|
|
/* the count of the main thread */
|
|
atomic_t m;
|
|
};
|
|
static ZTEST_BMEM struct ctx ctx;
|
|
static ZTEST_BMEM struct k_work work;
|
|
|
|
static void work_handler(struct k_work *w)
|
|
{
|
|
int res;
|
|
char c = '\0';
|
|
|
|
(void)w;
|
|
|
|
LOG_DBG("doing work");
|
|
|
|
while (true) {
|
|
if (ctx.write) {
|
|
LOG_DBG("ctx.m: %lu", atomic_get(&ctx.m));
|
|
if (atomic_get(&ctx.m)
|
|
< CONFIG_NET_SOCKETPAIR_BUFFER_SIZE) {
|
|
continue;
|
|
}
|
|
LOG_DBG("ready to read!");
|
|
} else {
|
|
LOG_DBG("sleeping for 100ms..");
|
|
k_sleep(K_MSEC(100));
|
|
LOG_DBG("ready to write!");
|
|
}
|
|
break;
|
|
}
|
|
|
|
LOG_DBG("%sing 1 byte %s fd %d", ctx.write ? "read" : "writ",
|
|
ctx.write ? "from" : "to", ctx.fd);
|
|
if (ctx.write) {
|
|
res = recv(ctx.fd, &c, 1, 0);
|
|
} else {
|
|
res = send(ctx.fd, "x", 1, 0);
|
|
}
|
|
if (-1 == res || 1 != res) {
|
|
LOG_DBG("%s() failed: %d", ctx.write ? "recv" : "send", errno);
|
|
} else {
|
|
LOG_DBG("%s 1 byte", ctx.write ? "read" : "wrote");
|
|
}
|
|
}
|
|
|
|
ZTEST_F(net_socketpair, test_write_block)
|
|
{
|
|
int res;
|
|
|
|
for (size_t i = 0; i < 2; ++i) {
|
|
|
|
LOG_DBG("data direction %d -> %d", fixture->sv[i], fixture->sv[(!i) & 1]);
|
|
|
|
LOG_DBG("setting up context");
|
|
memset(&ctx, 0, sizeof(ctx));
|
|
ctx.write = true;
|
|
ctx.fd = fixture->sv[(!i) & 1];
|
|
|
|
LOG_DBG("queueing work");
|
|
k_work_init(&work, work_handler);
|
|
k_work_submit(&work);
|
|
|
|
/* fill up the buffer */
|
|
for (ctx.m = 0; atomic_get(&ctx.m)
|
|
< CONFIG_NET_SOCKETPAIR_BUFFER_SIZE;) {
|
|
|
|
res = send(fixture->sv[i], "x", 1, 0);
|
|
zassert_not_equal(res, -1, "send() failed: %d", errno);
|
|
zassert_equal(res, 1, "wrote %d bytes instead of 1",
|
|
res);
|
|
|
|
atomic_inc(&ctx.m);
|
|
LOG_DBG("have written %lu bytes", atomic_get(&ctx.m));
|
|
}
|
|
|
|
/* try to write one more byte */
|
|
LOG_DBG("writing to fd %d", fixture->sv[i]);
|
|
res = send(fixture->sv[i], "x", 1, 0);
|
|
zassert_not_equal(res, -1, "send() failed: %d", errno);
|
|
zassert_equal(res, 1, "wrote %d bytes instead of 1", res);
|
|
|
|
LOG_DBG("success!");
|
|
}
|
|
}
|
|
|
|
ZTEST_F(net_socketpair, test_read_block)
|
|
{
|
|
int res;
|
|
char x;
|
|
|
|
for (size_t i = 0; i < 2; ++i) {
|
|
|
|
LOG_DBG("data direction %d <- %d", fixture->sv[i], fixture->sv[(!i) & 1]);
|
|
|
|
LOG_DBG("setting up context");
|
|
memset(&ctx, 0, sizeof(ctx));
|
|
ctx.write = false;
|
|
ctx.fd = fixture->sv[(!i) & 1];
|
|
|
|
LOG_DBG("queueing work");
|
|
k_work_init(&work, work_handler);
|
|
k_work_submit(&work);
|
|
|
|
/* try to read one byte */
|
|
LOG_DBG("reading from fd %d", fixture->sv[i]);
|
|
x = '\0';
|
|
res = recv(fixture->sv[i], &x, 1, 0);
|
|
zassert_not_equal(res, -1, "recv() failed: %d", errno);
|
|
zassert_equal(res, 1, "read %d bytes instead of 1", res);
|
|
|
|
LOG_DBG("success!");
|
|
}
|
|
}
|