riot-ci/RIOT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

tests/nanocoap: allow server to ignore requests

Browse Source
This commit is contained in:
Ken Bannister 2018-12-25 00:41:05 -05:00
parent 27ba81eed0
commit 2e2847372c
1 changed files with 90 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

114
tests/nanocoap_cli/nanocli_server.c
View File

@ -27,38 +27,104 @@
#define ENABLE_DEBUG (0)
#include "debug.h"
static void _start_server(uint16_t port)
/*
* Customized implementation of nanocoap_server() to ignore a count of
* requests. Allows testing confirmable messaging.
*/
static int _nanocoap_server(sock_udp_ep_t *local, uint8_t *buf, size_t bufsize,
int ignore_count)
{
sock_udp_t sock;
sock_udp_ep_t remote;
if (!local->port) {
local->port = COAP_PORT;
}
ssize_t res = sock_udp_create(&sock, local, NULL, 0);
if (res == -1) {
return -1;
}
int recv_count = 0;
while (1) {
res = sock_udp_recv(&sock, buf, bufsize, -1, &remote);
if (++recv_count <= ignore_count) {
DEBUG("ignoring request\n");
continue;
}
if (res == -1) {
DEBUG("nanocoap: error receiving coap request, \n");
return -1;
}
else {
coap_pkt_t pkt;
if (coap_parse(&pkt, (uint8_t *)buf, res) < 0) {
DEBUG("nanocoap: error parsing packet\n");
continue;
}
if ((res = coap_handle_req(&pkt, buf, bufsize)) > 0) {
res = sock_udp_send(&sock, buf, res, &remote);
}
}
}
return 0;
}
static void _start_server(uint16_t port, int ignore_count)
{
uint8_t buf[128];
sock_udp_ep_t local = { .port=port, .family=AF_INET6 };
nanocoap_server(&local, buf, sizeof(buf));
_nanocoap_server(&local, buf, sizeof(buf), ignore_count);
}
int nanotest_server_cmd(int argc, char **argv)
{
uint16_t port = COAP_PORT;
if (argc >= 2) {
if (strncmp("start", argv[1], 5) != 0) {
goto end;
}
if (argc == 3) {
port = atoi(argv[2]);
if (port == 0) {
puts("nanocli: unable to parse port");
return 1;
}
}
puts("starting server\n");
_start_server(port);
/* server executes run loop; never reaches this point*/
return 0;
if (argc < 2) {
goto error;
}
end:
printf("usage: %s start [port]\n", argv[0]);
printf(" Port defaults to %u\n", COAP_PORT);
if (strncmp("start", argv[1], 5) != 0) {
goto error;
}
int arg_pos = 2;
int ignore_count = 0;
if ((argc >= (arg_pos+1)) && (strcmp(argv[arg_pos], "-i") == 0)) {
/* need count of requests to ignore*/
if (argc == 3) {
goto error;
}
arg_pos++;
ignore_count = atoi(argv[arg_pos]);
if (ignore_count <= 0) {
puts("nanocli: unable to parse ignore_count");
goto error;
}
arg_pos++;
}
uint16_t port = COAP_PORT;
if (argc == (arg_pos+1)) {
port = atoi(argv[arg_pos]);
if (port == 0) {
puts("nanocli: unable to parse port");
goto error;
}
}
printf("starting server on port %u\n", port);
_start_server(port, ignore_count);
/* server executes run loop; never reaches this point*/
return 0;
error:
printf("usage: %s start [-i ignore_count] [port]\n", argv[0]);
printf("Options\n");
printf(" -i ignore a number of requests\n");
printf(" port defaults to %u\n", COAP_PORT);
return 1;
}