michaelh/zephyr
1
0
Fork 0
Code Releases Activity

net: shell: Add support for 'net tcp recv' command

Browse source 
Add command to register a receive callback to the net_context used for
the TCP connection that is opened with 'net tcp connect'. The receive
callback will simply print the number of bytes received and inform if
the connection is closed. This makes it possible to test both the tx
and rx paths with the net shell.

Signed-off-by: Tobias Svehagen <tobias.svehagen@gmail.com>
This commit is contained in:
Tobias Svehagen 2020-04-11 09:42:27 +02:00 • committed by Anas Nashif
commit 8b3a606009
1 changed files with 60 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

60
subsys/net/ip/net_shell.c
View file

@ -3744,6 +3744,34 @@ static void tcp_sent_cb(struct net_context *context,
{ {
PR_SHELL(tcp_shell, "Message sent\n"); PR_SHELL(tcp_shell, "Message sent\n");
} }
static void tcp_recv_cb(struct net_context *context, struct net_pkt *pkt,
union net_ip_header *ip_hdr,
union net_proto_header *proto_hdr,
int status, void *user_data)
{
int ret;
if (pkt == NULL) {
if (!tcp_ctx || !net_context_is_used(tcp_ctx)) {
return;
}
ret = net_context_put(tcp_ctx);
if (ret < 0) {
PR_SHELL(tcp_shell,
"Cannot close the connection (%d)\n", ret);
return;
}
PR_SHELL(tcp_shell, "Connection closed by remote peer.\n");
tcp_ctx = NULL;
return;
}
PR_SHELL(tcp_shell, "%d bytes received\n", net_pkt_get_len(pkt));
}
#endif #endif
static int cmd_net_tcp_connect(const struct shell *shell, size_t argc, static int cmd_net_tcp_connect(const struct shell *shell, size_t argc,
@ -3827,6 +3855,35 @@ static int cmd_net_tcp_send(const struct shell *shell, size_t argc,
return 0; return 0;
} }
static int cmd_net_tcp_recv(const struct shell *shell, size_t argc,
char *argv[])
{
#if defined(CONFIG_NET_TCP1) && defined(CONFIG_NET_NATIVE_TCP)
int ret;
struct net_shell_user_data user_data;
/* tcp recv */
if (!tcp_ctx || !net_context_is_used(tcp_ctx)) {
PR_WARNING("Not connected\n");
return -ENOEXEC;
}
user_data.shell = shell;
ret = net_context_recv(tcp_ctx, tcp_recv_cb, K_NO_WAIT, &user_data);
if (ret < 0) {
PR_WARNING("Cannot recv data (%d)\n", ret);
return -ENOEXEC;
}
#else
PR_INFO("Set %s to enable %s support.\n",
"CONFIG_NET_TCP and CONFIG_NET_NATIVE", "TCP");
#endif /* CONFIG_NET_NATIVE_TCP */
return 0;
}
static int cmd_net_tcp_close(const struct shell *shell, size_t argc, static int cmd_net_tcp_close(const struct shell *shell, size_t argc,
char *argv[]) char *argv[])
{ {
@ -4469,6 +4526,9 @@ SHELL_STATIC_SUBCMD_SET_CREATE(net_cmd_tcp,
SHELL_CMD(send, NULL, SHELL_CMD(send, NULL,
"'net tcp send <data>' sends data to peer using TCP.", "'net tcp send <data>' sends data to peer using TCP.",
cmd_net_tcp_send), cmd_net_tcp_send),
SHELL_CMD(recv, NULL,
"'net tcp recv' receives data using TCP.",
cmd_net_tcp_recv),
SHELL_CMD(close, NULL, SHELL_CMD(close, NULL,
"'net tcp close' closes TCP connection.", cmd_net_tcp_close), "'net tcp close' closes TCP connection.", cmd_net_tcp_close),
SHELL_SUBCMD_SET_END SHELL_SUBCMD_SET_END