1
0
mirror of https://github.com/RIOT-OS/RIOT.git synced 2024-12-29 04:50:03 +01:00
RIOT/tests/net/gnrc_sock_async_event/main.c

167 lines
5.3 KiB
C

/*
* Copyright (C) 2019 Freie Universität Berlin
*
* This file is subject to the terms and conditions of the GNU Lesser
* General Public License v2.1. See the file LICENSE in the top level
* directory for more details.
*/
/**
* @ingroup tests
* @{
*
* @file
*
* @author Martine Lenders <m.lenders@fu-berlin.de>
*
* @}
*/
#include <stdio.h>
#include "event.h"
#include "net/ipv6/addr.h"
#include "net/ipv6/hdr.h"
#include "net/sock/ip.h"
#include "net/sock/udp.h"
#include "net/gnrc.h"
#include "net/gnrc/ipv6/hdr.h"
#include "net/gnrc/pktdump.h"
#include "net/gnrc/udp.h"
#include "net/protnum.h"
#include "od.h"
#include "test_utils/expect.h"
#include "thread.h"
#include "net/sock/async/event.h"
#define RECV_BUFFER_SIZE (128)
#define TEST_PORT (38664U)
#define TEST_LOCAL { 0xfe, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, \
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01 }
#define TEST_REMOTE { 0xfe, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, \
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02 }
#define TEST_PAYLOAD { 0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef }
static const uint8_t _test_local[] = TEST_LOCAL;
static const uint8_t _test_remote[] = TEST_REMOTE;
static const uint8_t _test_payload[] = TEST_PAYLOAD;
static gnrc_netreg_entry_t _pktdump;
static char _addr_str[IPV6_ADDR_MAX_STR_LEN];
static event_queue_t _ev_queue;
static uint8_t _buffer[128];
static sock_ip_t _ip_sock;
static sock_udp_t _udp_sock;
/* module is not compiled in, so provide this function for the test */
ipv6_hdr_t *gnrc_ipv6_get_header(gnrc_pktsnip_t *pkt)
{
gnrc_pktsnip_t *tmp = gnrc_pktsnip_search_type(pkt, GNRC_NETTYPE_IPV6);
if (tmp == NULL) {
return NULL;
}
expect(tmp->data != NULL);
expect(tmp->size >= sizeof(ipv6_hdr_t));
expect(ipv6_hdr_is(tmp->data));
return ((ipv6_hdr_t*) tmp->data);
}
static void _recv_udp(sock_udp_t *sock, sock_async_flags_t flags, void *arg)
{
expect(strcmp(arg, "test") == 0);
printf("UDP event triggered: %04X\n", flags);
if (flags & SOCK_ASYNC_MSG_RECV) {
sock_udp_ep_t remote;
ssize_t res;
if ((res = sock_udp_recv(sock, _buffer, sizeof(_buffer), 0,
&remote)) >= 0) {
printf("Received UDP packet from [%s]:%u:\n",
ipv6_addr_to_str(_addr_str, (ipv6_addr_t *)remote.addr.ipv6,
sizeof(_addr_str)),
remote.port);
od_hex_dump(_buffer, res, OD_WIDTH_DEFAULT);
}
}
if (flags & SOCK_ASYNC_MSG_SENT) {
puts("UDP message successfully sent");
}
}
static void _recv_ip(sock_ip_t *sock, sock_async_flags_t flags, void *arg)
{
expect(strcmp(arg, "test") == 0);
printf("IP event triggered: %04X\n", flags);
if (flags & SOCK_ASYNC_MSG_RECV) {
sock_ip_ep_t remote;
ssize_t res;
if ((res = sock_ip_recv(sock, _buffer, sizeof(_buffer), 0,
&remote)) >= 0) {
printf("Received IP packet from [%s]:\n",
ipv6_addr_to_str(_addr_str, (ipv6_addr_t *)remote.addr.ipv6,
sizeof(_addr_str)));
od_hex_dump(_buffer, res, OD_WIDTH_DEFAULT);
}
}
if (flags & SOCK_ASYNC_MSG_SENT) {
puts("IP message successfully sent");
}
}
int main(void)
{
gnrc_pktsnip_t *pkt;
sock_udp_ep_t local = SOCK_IPV6_EP_ANY;
sock_udp_ep_t remote = SOCK_IPV6_EP_ANY;
event_queue_init(&_ev_queue);
/* register for IPv6 to have a target */
gnrc_netreg_entry_init_pid(&_pktdump, GNRC_NETREG_DEMUX_CTX_ALL,
gnrc_pktdump_pid);
gnrc_netreg_register(GNRC_NETTYPE_IPV6, &_pktdump);
local.port = TEST_PORT;
sock_udp_create(&_udp_sock, &local, NULL, 0);
sock_ip_create(&_ip_sock, (sock_ip_ep_t *)&local, NULL, PROTNUM_UDP, 0);
sock_udp_event_init(&_udp_sock, &_ev_queue, _recv_udp, "test");
sock_ip_event_init(&_ip_sock, &_ev_queue, _recv_ip, "test");
memcpy(remote.addr.ipv6, _test_remote, sizeof(_test_remote));
remote.port = TEST_PORT - 1;
sock_udp_send(&_udp_sock, _test_payload, sizeof(_test_payload), &remote);
sock_ip_send(&_ip_sock, _test_payload, sizeof(_test_payload),
PROTNUM_RESERVED, (sock_ip_ep_t *)&remote);
/* create packet to inject for reception */
pkt = gnrc_netif_hdr_build(NULL, 0, NULL, 0);
expect(pkt != NULL);
memset(pkt->data, 0, pkt->size);
pkt = gnrc_ipv6_hdr_build(pkt, (ipv6_addr_t *)&_test_remote,
(ipv6_addr_t *)&_test_local);
expect(pkt != NULL);
/* module is not compiled in, so set header type manually */
pkt->type = GNRC_NETTYPE_IPV6;
pkt = gnrc_udp_hdr_build(pkt, TEST_PORT - 1, TEST_PORT);
expect(pkt != NULL);
pkt = gnrc_pktbuf_add(pkt, _test_payload, sizeof(_test_payload),
GNRC_NETTYPE_UNDEF);
expect(pkt != NULL);
/* we dispatch twice, so hold one time */
gnrc_pktbuf_hold(pkt, 1);
/* trigger receive on UDP sock */
gnrc_netapi_dispatch_receive(GNRC_NETTYPE_UDP, TEST_PORT, pkt);
/* trigger receive on IP sock */
gnrc_netapi_dispatch_receive(GNRC_NETTYPE_IPV6, PROTNUM_UDP, pkt);
event_loop(&_ev_queue);
return 0;
}
/** @} */