Skip to content

Commit 763665d

Browse files
jukkardkalowsk
jukkar
authored and
dkalowsk
committed
tests: net: dns_dispatcher: Add tests for dispatcher
Make sure that the socket service is properly unregistered when dispatcher is unregistered. Signed-off-by: Jukka Rissanen <[email protected]> (cherry picked from commit d98fe73)
1 parent d58e1ea commit 763665d

File tree

4 files changed

+250
-0
lines changed

4 files changed

+250
-0
lines changed

‎tests/net/lib/dns_dispatcher/CMakeLists.txt‎

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
# SPDX-License-Identifier: Apache-2.0
2+
3+
cmake_minimum_required(VERSION 3.20.0)
4+
find_package(Zephyr REQUIRED HINTS $ENV{ZEPHYR_BASE})
5+
project(dns_dispatcher)
6+
7+
target_include_directories(app PRIVATE ${ZEPHYR_BASE}/subsys/net/ip)
8+
FILE(GLOB app_sources src/*.c)
9+
target_sources(app PRIVATE ${app_sources})

‎tests/net/lib/dns_dispatcher/prj.conf‎

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
CONFIG_NETWORKING=y
2+
CONFIG_NET_TEST=y
3+
CONFIG_ENTROPY_GENERATOR=y
4+
CONFIG_TEST_RANDOM_GENERATOR=y
5+
CONFIG_NET_L2_DUMMY=y
6+
CONFIG_DNS_RESOLVER=y
7+
CONFIG_DNS_RESOLVER_MAX_SERVERS=2
8+
CONFIG_DNS_NUM_CONCUR_QUERIES=1
9+
CONFIG_NET_LOG=y
10+
CONFIG_NET_MGMT=y
11+
CONFIG_NET_MGMT_EVENT=y
12+
CONFIG_NET_MGMT_EVENT_QUEUE_SIZE=5
13+
CONFIG_NET_IPV4=y
14+
CONFIG_NET_IPV6=y
15+
CONFIG_PRINTK=y
16+
CONFIG_ZTEST=y
17+
CONFIG_MAIN_STACK_SIZE=2048
18+
CONFIG_ZVFS_OPEN_MAX=10
19+
CONFIG_ZVFS_POLL_MAX=10
20+
CONFIG_MDNS_RESPONDER=n
21+
CONFIG_MDNS_RESOLVER=n
22+
CONFIG_DNS_SERVER_IP_ADDRESSES=y
23+
CONFIG_DNS_RESOLVER_MAX_SERVERS=2
24+
CONFIG_DNS_SERVER1="192.0.2.1"
25+
CONFIG_DNS_SERVER2="2001:db8::1"

‎tests/net/lib/dns_dispatcher/src/main.c‎

Lines changed: 203 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,203 @@
1+
/*
2+
* Copyright (c) 2024 Nordic Semiconductor ASA
3+
*
4+
* SPDX-License-Identifier: Apache-2.0
5+
*/
6+
7+
#include <zephyr/logging/log.h>
8+
LOG_MODULE_REGISTER(net_test, CONFIG_DNS_RESOLVER_LOG_LEVEL);
9+
10+
#include <zephyr/types.h>
11+
#include <stdbool.h>
12+
#include <stddef.h>
13+
#include <string.h>
14+
#include <errno.h>
15+
#include <zephyr/sys/printk.h>
16+
#include <zephyr/random/random.h>
17+
18+
#include <zephyr/ztest.h>
19+
20+
#include <zephyr/net/ethernet.h>
21+
#include <zephyr/net/dummy.h>
22+
#include <zephyr/net_buf.h>
23+
#include <zephyr/net/net_ip.h>
24+
#include <zephyr/net/net_if.h>
25+
#include <zephyr/net/dns_resolve.h>
26+
#include <zephyr/net/net_event.h>
27+
#include <zephyr/net/net_mgmt.h>
28+
#include <zephyr/net/socket_service.h>
29+
30+
#define NET_LOG_ENABLED 1
31+
#include "net_private.h"
32+
33+
#if defined(CONFIG_DNS_RESOLVER_LOG_LEVEL_DBG)
34+
#define DBG(fmt, ...) printk(fmt, ##__VA_ARGS__)
35+
#else
36+
#define DBG(fmt, ...)
37+
#endif
38+
39+
#define NAME4 "4.zephyr.test"
40+
#define NAME6 "6.zephyr.test"
41+
#define NAME_IPV4 "192.0.2.1"
42+
#define NAME_IPV6 "2001:db8::1"
43+
44+
#define DNS_NAME_IPV4 "192.0.2.4"
45+
#define DNS2_NAME_IPV4 "192.0.2.5"
46+
#define DNS_NAME_IPV6 "2001:db8::4"
47+
48+
#define DNS_TIMEOUT 500 /* ms */
49+
50+
#if defined(CONFIG_NET_IPV6)
51+
/* Interface 1 addresses */
52+
static struct in6_addr my_addr1 = { { { 0x20, 0x01, 0x0d, 0xb8, 1, 0, 0, 0,
53+
0, 0, 0, 0, 0, 0, 0, 0x1 } } };
54+
#endif
55+
56+
#if defined(CONFIG_NET_IPV4)
57+
/* Interface 1 addresses */
58+
static struct in_addr my_addr2 = { { { 192, 0, 2, 1 } } };
59+
#endif
60+
61+
static struct net_if *iface1;
62+
63+
/* this must be higher that the DNS_TIMEOUT */
64+
#define WAIT_TIME K_MSEC((DNS_TIMEOUT + 300) * 3)
65+
66+
struct net_if_test {
67+
uint8_t idx;
68+
uint8_t mac_addr[sizeof(struct net_eth_addr)];
69+
};
70+
71+
static uint8_t *net_iface_get_mac(const struct device *dev)
72+
{
73+
struct net_if_test *data = dev->data;
74+
75+
if (data->mac_addr[2] == 0x00) {
76+
/* 00-00-5E-00-53-xx Documentation RFC 7042 */
77+
data->mac_addr[0] = 0x00;
78+
data->mac_addr[1] = 0x00;
79+
data->mac_addr[2] = 0x5E;
80+
data->mac_addr[3] = 0x00;
81+
data->mac_addr[4] = 0x53;
82+
data->mac_addr[5] = sys_rand8_get();
83+
}
84+
85+
return data->mac_addr;
86+
}
87+
88+
static void net_iface_init(struct net_if *iface)
89+
{
90+
uint8_t *mac = net_iface_get_mac(net_if_get_device(iface));
91+
92+
net_if_set_link_addr(iface, mac, sizeof(struct net_eth_addr),
93+
NET_LINK_ETHERNET);
94+
}
95+
96+
static int sender_iface(const struct device *dev, struct net_pkt *pkt)
97+
{
98+
if (!pkt->frags) {
99+
DBG("No data to send!\n");
100+
return -ENODATA;
101+
}
102+
103+
return 0;
104+
}
105+
106+
struct net_if_test net_iface1_data;
107+
108+
static struct dummy_api net_iface_api = {
109+
.iface_api.init = net_iface_init,
110+
.send = sender_iface,
111+
};
112+
113+
#define _ETH_L2_LAYER DUMMY_L2
114+
#define _ETH_L2_CTX_TYPE NET_L2_GET_CTX_TYPE(DUMMY_L2)
115+
116+
NET_DEVICE_INIT_INSTANCE(net_iface1_test,
117+
"iface1",
118+
iface1,
119+
NULL,
120+
NULL,
121+
&net_iface1_data,
122+
NULL,
123+
CONFIG_KERNEL_INIT_PRIORITY_DEFAULT,
124+
&net_iface_api,
125+
_ETH_L2_LAYER,
126+
_ETH_L2_CTX_TYPE,
127+
127);
128+
129+
static void *test_init(void)
130+
{
131+
struct net_if_addr *ifaddr;
132+
133+
iface1 = net_if_get_by_index(0);
134+
zassert_is_null(iface1, "iface1");
135+
136+
iface1 = net_if_get_by_index(1);
137+
138+
((struct net_if_test *) net_if_get_device(iface1)->data)->idx =
139+
net_if_get_by_iface(iface1);
140+
141+
#if defined(CONFIG_NET_IPV6)
142+
ifaddr = net_if_ipv6_addr_add(iface1, &my_addr1,
143+
NET_ADDR_MANUAL, 0);
144+
if (!ifaddr) {
145+
DBG("Cannot add IPv6 address %s\n",
146+
net_sprint_ipv6_addr(&my_addr1));
147+
zassert_not_null(ifaddr, "addr1");
148+
149+
return NULL;
150+
}
151+
152+
/* For testing purposes we need to set the adddresses preferred */
153+
ifaddr->addr_state = NET_ADDR_PREFERRED;
154+
#endif
155+
156+
#if defined(CONFIG_NET_IPV4)
157+
ifaddr = net_if_ipv4_addr_add(iface1, &my_addr2,
158+
NET_ADDR_MANUAL, 0);
159+
if (!ifaddr) {
160+
DBG("Cannot add IPv4 address %s\n",
161+
net_sprint_ipv4_addr(&my_addr2));
162+
zassert_not_null(ifaddr, "addr2");
163+
164+
return NULL;
165+
}
166+
167+
ifaddr->addr_state = NET_ADDR_PREFERRED;
168+
#endif
169+
170+
net_if_up(iface1);
171+
172+
return NULL;
173+
}
174+
175+
ZTEST(dns_dispatcher, test_dns_dispatcher)
176+
{
177+
struct dns_resolve_context *ctx;
178+
int sock1, sock2 = -1;
179+
180+
ctx = dns_resolve_get_default();
181+
182+
dns_resolve_init_default(ctx);
183+
184+
sock1 = ctx->servers[0].sock;
185+
186+
for (int i = 0; i < ctx->servers[0].dispatcher.fds_len; i++) {
187+
if (ctx->servers[0].dispatcher.fds[i].fd == sock1) {
188+
sock2 = i;
189+
break;
190+
}
191+
}
192+
193+
zassert_not_equal(sock2, -1, "Cannot find socket");
194+
195+
k_sleep(K_MSEC(10));
196+
197+
dns_resolve_close(ctx);
198+
199+
zassert_equal(ctx->servers[0].dispatcher.fds[sock2].fd, -1, "Socket not closed");
200+
zassert_equal(ctx->servers[0].dispatcher.sock, -1, "Dispatcher still registered");
201+
}
202+
203+
ZTEST_SUITE(dns_dispatcher, NULL, test_init, NULL, NULL, NULL);

‎tests/net/lib/dns_dispatcher/testcase.yaml‎

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
common:
2+
tags:
3+
- dns
4+
- net
5+
depends_on: netif
6+
min_ram: 21
7+
timeout: 600
8+
platform_exclude:
9+
- native_posix
10+
- native_posix/native/64
11+
tests:
12+
net.dns.dispatch:
13+
min_ram: 21

0 commit comments

Comments
 (0)