tests: net: igmp: Add IPv4 IGMP tests
IGMP (Internet Group Management Protocol) support tests. Signed-off-by: Jukka Rissanen <jukka.rissanen@linux.intel.com>
This commit is contained in:
Jukka Rissanen
Jukka Rissanen
parent
a1c4952dfd
commit
a4c119a23d
4 changed files with 376 additions and 0 deletions
9
tests/net/igmp/CMakeLists.txt
Normal file
9
tests/net/igmp/CMakeLists.txt
Normal file
|
|
@ -0,0 +1,9 @@
|
|||
# SPDX-License-Identifier: Apache-2.0
|
||||
|
||||
cmake_minimum_required(VERSION 3.13.1)
|
||||
find_package(Zephyr REQUIRED HINTS $ENV{ZEPHYR_BASE})
|
||||
project(mld)
|
||||
|
||||
target_include_directories(app PRIVATE ${ZEPHYR_BASE}/subsys/net/ip)
|
||||
FILE(GLOB app_sources src/*.c)
|
||||
target_sources(app PRIVATE ${app_sources})
|
||||
21
tests/net/igmp/prj.conf
Normal file
21
tests/net/igmp/prj.conf
Normal file
|
|
@ -0,0 +1,21 @@
|
|||
CONFIG_ZTEST=y
|
||||
CONFIG_MAIN_STACK_SIZE=2048
|
||||
CONFIG_NET_TEST=y
|
||||
CONFIG_NETWORKING=y
|
||||
CONFIG_NET_IPV6=n
|
||||
CONFIG_NET_UDP=y
|
||||
CONFIG_NET_TCP=n
|
||||
CONFIG_NET_IPV4=y
|
||||
CONFIG_NET_IPV4_IGMP=y
|
||||
CONFIG_NET_L2_DUMMY=y
|
||||
CONFIG_NET_LOG=y
|
||||
CONFIG_ENTROPY_GENERATOR=y
|
||||
CONFIG_TEST_RANDOM_GENERATOR=y
|
||||
CONFIG_NET_PKT_TX_COUNT=20
|
||||
CONFIG_NET_PKT_RX_COUNT=20
|
||||
CONFIG_NET_BUF_RX_COUNT=20
|
||||
CONFIG_NET_BUF_TX_COUNT=20
|
||||
CONFIG_NET_MGMT=y
|
||||
CONFIG_NET_MGMT_EVENT=y
|
||||
CONFIG_NET_IF_UNICAST_IPV4_ADDR_COUNT=2
|
||||
CONFIG_NET_IF_MCAST_IPV4_ADDR_COUNT=2
|
||||
336
tests/net/igmp/src/main.c
Normal file
336
tests/net/igmp/src/main.c
Normal file
|
|
@ -0,0 +1,336 @@
|
|||
/* main.c - Application main entry point */
|
||||
|
||||
/*
|
||||
* Copyright (c) 2021 Intel Corporation
|
||||
*
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*/
|
||||
|
||||
#include <logging/log.h>
|
||||
LOG_MODULE_REGISTER(net_test, CONFIG_NET_IPV4_LOG_LEVEL);
|
||||
|
||||
#include <zephyr/types.h>
|
||||
#include <stdbool.h>
|
||||
#include <stddef.h>
|
||||
#include <string.h>
|
||||
#include <errno.h>
|
||||
#include <linker/sections.h>
|
||||
|
||||
#include <ztest.h>
|
||||
|
||||
#include <net/net_if.h>
|
||||
#include <net/net_pkt.h>
|
||||
#include <net/net_ip.h>
|
||||
#include <net/net_core.h>
|
||||
#include <net/ethernet.h>
|
||||
#include <net/dummy.h>
|
||||
#include <net/net_mgmt.h>
|
||||
#include <net/net_event.h>
|
||||
#include <net/igmp.h>
|
||||
|
||||
#include <random/rand32.h>
|
||||
|
||||
#include "ipv4.h"
|
||||
|
||||
#define THREAD_SLEEP 50 /* ms */
|
||||
|
||||
#define NET_LOG_ENABLED 1
|
||||
#include "net_private.h"
|
||||
|
||||
#if defined(CONFIG_NET_IPV4_LOG_LEVEL_DBG)
|
||||
#define DBG(fmt, ...) printk(fmt, ##__VA_ARGS__)
|
||||
#else
|
||||
#define DBG(fmt, ...)
|
||||
#endif
|
||||
|
||||
static struct in_addr my_addr = { { { 192, 0, 2, 1 } } };
|
||||
static struct in_addr mcast_addr = { { { 224, 0, 2, 63 } } };
|
||||
|
||||
static struct net_if *iface;
|
||||
static bool is_group_joined;
|
||||
static bool is_group_left;
|
||||
static bool is_join_msg_ok;
|
||||
static bool is_leave_msg_ok;
|
||||
static bool is_query_received;
|
||||
static bool is_report_sent;
|
||||
static bool ignore_already;
|
||||
K_SEM_DEFINE(wait_data, 0, UINT_MAX);
|
||||
|
||||
#define WAIT_TIME 500
|
||||
#define WAIT_TIME_LONG MSEC_PER_SEC
|
||||
#define MY_PORT 1969
|
||||
#define PEER_PORT 13856
|
||||
|
||||
struct net_test_igmp {
|
||||
uint8_t mac_addr[sizeof(struct net_eth_addr)];
|
||||
struct net_linkaddr ll_addr;
|
||||
};
|
||||
|
||||
int net_test_dev_init(const struct device *dev)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
static uint8_t *net_test_get_mac(const struct device *dev)
|
||||
{
|
||||
struct net_test_igmp *context = dev->data;
|
||||
|
||||
if (context->mac_addr[2] == 0x00) {
|
||||
/* 00-00-5E-00-53-xx Documentation RFC 7042 */
|
||||
context->mac_addr[0] = 0x00;
|
||||
context->mac_addr[1] = 0x00;
|
||||
context->mac_addr[2] = 0x5E;
|
||||
context->mac_addr[3] = 0x00;
|
||||
context->mac_addr[4] = 0x53;
|
||||
context->mac_addr[5] = sys_rand32_get();
|
||||
}
|
||||
|
||||
return context->mac_addr;
|
||||
}
|
||||
|
||||
static void net_test_iface_init(struct net_if *iface)
|
||||
{
|
||||
uint8_t *mac = net_test_get_mac(net_if_get_device(iface));
|
||||
|
||||
net_if_set_link_addr(iface, mac, sizeof(struct net_eth_addr),
|
||||
NET_LINK_ETHERNET);
|
||||
}
|
||||
|
||||
static struct net_ipv4_igmp_v2_query *get_igmp_hdr(struct net_pkt *pkt)
|
||||
{
|
||||
net_pkt_cursor_init(pkt);
|
||||
|
||||
net_pkt_skip(pkt, net_pkt_ip_hdr_len(pkt) +
|
||||
net_pkt_ipv4_opts_len(pkt));
|
||||
|
||||
return (struct net_ipv4_igmp_v2_query *)net_pkt_cursor_get_pos(pkt);
|
||||
}
|
||||
|
||||
static int tester_send(const struct device *dev, struct net_pkt *pkt)
|
||||
{
|
||||
struct net_ipv4_igmp_v2_query *igmp;
|
||||
|
||||
if (!pkt->buffer) {
|
||||
TC_ERROR("No data to send!\n");
|
||||
return -ENODATA;
|
||||
}
|
||||
|
||||
igmp = get_igmp_hdr(pkt);
|
||||
|
||||
if (igmp->type == NET_IPV4_IGMP_QUERY) {
|
||||
NET_DBG("Received query....");
|
||||
is_query_received = true;
|
||||
k_sem_give(&wait_data);
|
||||
} else if (igmp->type == NET_IPV4_IGMP_REPORT_V2) {
|
||||
NET_DBG("Received v2 report....");
|
||||
is_join_msg_ok = true;
|
||||
is_report_sent = true;
|
||||
k_sem_give(&wait_data);
|
||||
} else if (igmp->type == NET_IPV4_IGMP_LEAVE) {
|
||||
NET_DBG("Received leave....");
|
||||
is_leave_msg_ok = true;
|
||||
k_sem_give(&wait_data);
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
struct net_test_igmp net_test_data;
|
||||
|
||||
static struct dummy_api net_test_if_api = {
|
||||
.iface_api.init = net_test_iface_init,
|
||||
.send = tester_send,
|
||||
};
|
||||
|
||||
#define _ETH_L2_LAYER DUMMY_L2
|
||||
#define _ETH_L2_CTX_TYPE NET_L2_GET_CTX_TYPE(DUMMY_L2)
|
||||
|
||||
NET_DEVICE_INIT(net_test_igmp, "net_test_igmp",
|
||||
net_test_dev_init, device_pm_control_nop, &net_test_data, NULL,
|
||||
CONFIG_KERNEL_INIT_PRIORITY_DEFAULT,
|
||||
&net_test_if_api, _ETH_L2_LAYER, _ETH_L2_CTX_TYPE,
|
||||
127);
|
||||
|
||||
static void group_joined(struct net_mgmt_event_callback *cb,
|
||||
uint32_t nm_event, struct net_if *iface)
|
||||
{
|
||||
if (nm_event != NET_EVENT_IPV4_MCAST_JOIN) {
|
||||
/* Spurious callback. */
|
||||
return;
|
||||
}
|
||||
|
||||
is_group_joined = true;
|
||||
|
||||
k_sem_give(&wait_data);
|
||||
}
|
||||
|
||||
static void group_left(struct net_mgmt_event_callback *cb,
|
||||
uint32_t nm_event, struct net_if *iface)
|
||||
{
|
||||
if (nm_event != NET_EVENT_IPV4_MCAST_LEAVE) {
|
||||
/* Spurious callback. */
|
||||
return;
|
||||
}
|
||||
|
||||
is_group_left = true;
|
||||
|
||||
k_sem_give(&wait_data);
|
||||
}
|
||||
|
||||
static struct mgmt_events {
|
||||
uint32_t event;
|
||||
net_mgmt_event_handler_t handler;
|
||||
struct net_mgmt_event_callback cb;
|
||||
} mgmt_events[] = {
|
||||
{ .event = NET_EVENT_IPV4_MCAST_JOIN, .handler = group_joined },
|
||||
{ .event = NET_EVENT_IPV4_MCAST_LEAVE, .handler = group_left },
|
||||
{ 0 }
|
||||
};
|
||||
|
||||
static void setup_mgmt_events(void)
|
||||
{
|
||||
int i;
|
||||
|
||||
for (i = 0; mgmt_events[i].event; i++) {
|
||||
net_mgmt_init_event_callback(&mgmt_events[i].cb,
|
||||
mgmt_events[i].handler,
|
||||
mgmt_events[i].event);
|
||||
|
||||
net_mgmt_add_event_callback(&mgmt_events[i].cb);
|
||||
}
|
||||
}
|
||||
|
||||
static void test_igmp_setup(void)
|
||||
{
|
||||
struct net_if_addr *ifaddr;
|
||||
|
||||
setup_mgmt_events();
|
||||
|
||||
iface = net_if_get_first_by_type(&NET_L2_GET_NAME(DUMMY));
|
||||
|
||||
zassert_not_null(iface, "Interface is NULL");
|
||||
|
||||
ifaddr = net_if_ipv4_addr_add(iface, &my_addr, NET_ADDR_MANUAL, 0);
|
||||
|
||||
zassert_not_null(ifaddr, "Cannot add IPv4 address");
|
||||
}
|
||||
|
||||
static void test_join_group(void)
|
||||
{
|
||||
int ret;
|
||||
|
||||
ret = net_ipv4_igmp_join(iface, &mcast_addr);
|
||||
|
||||
if (ignore_already) {
|
||||
zassert_true(ret == 0 || ret == -EALREADY,
|
||||
"Cannot join IPv4 multicast group");
|
||||
} else {
|
||||
zassert_equal(ret, 0, "Cannot join IPv4 multicast group");
|
||||
}
|
||||
|
||||
/* Let the network stack to proceed */
|
||||
k_msleep(THREAD_SLEEP);
|
||||
}
|
||||
|
||||
static void test_leave_group(void)
|
||||
{
|
||||
int ret;
|
||||
|
||||
ret = net_ipv4_igmp_leave(iface, &mcast_addr);
|
||||
|
||||
zassert_equal(ret, 0, "Cannot leave IPv4 multicast group");
|
||||
|
||||
if (IS_ENABLED(CONFIG_NET_TC_THREAD_PREEMPTIVE)) {
|
||||
/* Let the network stack to proceed */
|
||||
k_msleep(THREAD_SLEEP);
|
||||
} else {
|
||||
k_yield();
|
||||
}
|
||||
}
|
||||
|
||||
static void test_catch_join_group(void)
|
||||
{
|
||||
is_group_joined = false;
|
||||
|
||||
ignore_already = false;
|
||||
|
||||
test_join_group();
|
||||
|
||||
if (k_sem_take(&wait_data, K_MSEC(WAIT_TIME))) {
|
||||
zassert_true(0, "Timeout while waiting join event");
|
||||
}
|
||||
|
||||
if (!is_group_joined) {
|
||||
zassert_true(0, "Did not catch join event");
|
||||
}
|
||||
|
||||
is_group_joined = false;
|
||||
}
|
||||
|
||||
static void test_catch_leave_group(void)
|
||||
{
|
||||
is_group_joined = false;
|
||||
|
||||
test_leave_group();
|
||||
|
||||
if (k_sem_take(&wait_data, K_MSEC(WAIT_TIME))) {
|
||||
zassert_true(0, "Timeout while waiting leave event");
|
||||
}
|
||||
|
||||
if (!is_group_left) {
|
||||
zassert_true(0, "Did not catch leave event");
|
||||
}
|
||||
|
||||
is_group_left = false;
|
||||
}
|
||||
|
||||
static void test_verify_join_group(void)
|
||||
{
|
||||
is_join_msg_ok = false;
|
||||
|
||||
ignore_already = false;
|
||||
|
||||
test_join_group();
|
||||
|
||||
if (k_sem_take(&wait_data, K_MSEC(WAIT_TIME))) {
|
||||
zassert_true(0, "Timeout while waiting join event");
|
||||
}
|
||||
|
||||
if (!is_join_msg_ok) {
|
||||
zassert_true(0, "Join msg invalid");
|
||||
}
|
||||
|
||||
is_join_msg_ok = false;
|
||||
}
|
||||
|
||||
static void test_verify_leave_group(void)
|
||||
{
|
||||
is_leave_msg_ok = false;
|
||||
|
||||
test_leave_group();
|
||||
|
||||
if (k_sem_take(&wait_data, K_MSEC(WAIT_TIME))) {
|
||||
zassert_true(0, "Timeout while waiting leave event");
|
||||
}
|
||||
|
||||
if (!is_leave_msg_ok) {
|
||||
zassert_true(0, "Leave msg invalid");
|
||||
}
|
||||
|
||||
is_leave_msg_ok = false;
|
||||
}
|
||||
|
||||
void test_main(void)
|
||||
{
|
||||
ztest_test_suite(net_igmp_test,
|
||||
ztest_unit_test(test_igmp_setup),
|
||||
ztest_unit_test(test_join_group),
|
||||
ztest_unit_test(test_leave_group),
|
||||
ztest_unit_test(test_catch_join_group),
|
||||
ztest_unit_test(test_catch_leave_group),
|
||||
ztest_unit_test(test_verify_join_group),
|
||||
ztest_unit_test(test_verify_leave_group)
|
||||
);
|
||||
|
||||
ztest_run_test_suite(net_igmp_test);
|
||||
}
|
||||
10
tests/net/igmp/testcase.yaml
Normal file
10
tests/net/igmp/testcase.yaml
Normal file
|
|
@ -0,0 +1,10 @@
|
|||
common:
|
||||
tags: net igmp
|
||||
depends_on: netif
|
||||
tests:
|
||||
net.igmp:
|
||||
extra_configs:
|
||||
- CONFIG_NET_TC_THREAD_COOPERATIVE=y
|
||||
net.igmp.preempt:
|
||||
extra_configs:
|
||||
- CONFIG_NET_TC_THREAD_PREEMPTIVE=y
|
||||
Loading…
Add table
Add a link
Reference in a new issue