Skip to content

Commit fe0ac4f

Browse files
authored
Merge pull request #11836 from kivaisan/add_sender_address_and_port_check
Add sender address and port verification to UDPSOCKET_ECHOTEST
2 parents dd70938 + 582b16f commit fe0ac4f

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

TESTS/netsocket/udp/udpsocket_echotest.cpp

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@ static void _sigio_handler()
6363
void UDPSOCKET_ECHOTEST()
6464
{
6565
SocketAddress udp_addr;
66+
SocketAddress recv_addr;
6667
NetworkInterface::get_default_instance()->gethostbyname(ECHO_SERVER_ADDR, &udp_addr);
6768
udp_addr.set_port(ECHO_SERVER_PORT);
6869

@@ -94,7 +95,7 @@ void UDPSOCKET_ECHOTEST()
9495

9596
do {
9697
received_duplicate_packet = false;
97-
recvd = sock.recvfrom(NULL, rx_buffer, pkt_s);
98+
recvd = sock.recvfrom(&recv_addr, rx_buffer, pkt_s);
9899
//Check if received duplicated packet
99100
for (unsigned int d_idx = 0; d_idx < PKTS; ++d_idx) {
100101
if (pkt_received[d_idx] && d_idx != s_idx && recvd == pkt_sizes[d_idx]) {
@@ -111,6 +112,10 @@ void UDPSOCKET_ECHOTEST()
111112
printf("[Round#%02d - Receiver] error, returned %d\n", s_idx, recvd);
112113
}
113114
}
115+
// Verify received address is correct
116+
TEST_ASSERT(udp_addr == recv_addr);
117+
TEST_ASSERT_EQUAL(udp_addr.get_port(), recv_addr.get_port());
118+
114119
if (memcmp(tx_buffer, rx_buffer, pkt_s) == 0) {
115120
packets_recv++;
116121
pkt_received[s_idx] = true;

0 commit comments

Comments
 (0)