105 lines
2.4 KiB
C
105 lines
2.4 KiB
C
|
#include "test_tcp.h"
|
||
|
|
|
||
|
|
#include "lwip/tcp.h"
|
||
|
|
#include "lwip/stats.h"
|
||
|
|
#include "tcp_helper.h"
|
||
|
|
|
||
|
|
#if !LWIP_STATS || !TCP_STATS || !MEMP_STATS
|
||
|
|
#error "This tests needs TCP- and MEMP-statistics enabled"
|
||
|
|
#endif
|
||
|
|
|
||
|
|
/* Setups/teardown functions */
|
||
|
|
|
||
|
|
static void
|
||
|
|
tcp_setup(void)
|
||
|
|
{
|
||
|
|
tcp_remove_all();
|
||
|
|
}
|
||
|
|
|
||
|
|
static void
|
||
|
|
tcp_teardown(void)
|
||
|
|
{
|
||
|
|
tcp_remove_all();
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/* Test functions */
|
||
|
|
|
||
|
|
/** Call tcp_new() and tcp_abort() and test memp stats */
|
||
|
|
START_TEST(test_tcp_new_abort)
|
||
|
|
{
|
||
|
|
struct tcp_pcb* pcb;
|
||
|
|
LWIP_UNUSED_ARG(_i);
|
||
|
|
|
||
|
|
fail_unless(lwip_stats.memp[MEMP_TCP_PCB].used == 0);
|
||
|
|
|
||
|
|
pcb = tcp_new();
|
||
|
|
fail_unless(pcb != NULL);
|
||
|
|
if (pcb != NULL) {
|
||
|
|
fail_unless(lwip_stats.memp[MEMP_TCP_PCB].used == 1);
|
||
|
|
tcp_abort(pcb);
|
||
|
|
fail_unless(lwip_stats.memp[MEMP_TCP_PCB].used == 0);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
END_TEST
|
||
|
|
|
||
|
|
/** Create an ESTABLISHED pcb and check if receive callback is called */
|
||
|
|
START_TEST(test_tcp_recv_inseq)
|
||
|
|
{
|
||
|
|
struct test_tcp_counters counters;
|
||
|
|
struct tcp_pcb* pcb;
|
||
|
|
struct pbuf* p;
|
||
|
|
char data[] = {1, 2, 3, 4};
|
||
|
|
struct ip_addr remote_ip, local_ip;
|
||
|
|
u16_t data_len;
|
||
|
|
u16_t remote_port = 0x100, local_port = 0x101;
|
||
|
|
struct netif netif;
|
||
|
|
LWIP_UNUSED_ARG(_i);
|
||
|
|
|
||
|
|
/* initialize local vars */
|
||
|
|
memset(&netif, 0, sizeof(netif));
|
||
|
|
IP4_ADDR(&local_ip, 192, 168, 1, 1);
|
||
|
|
IP4_ADDR(&remote_ip, 192, 168, 1, 2);
|
||
|
|
data_len = sizeof(data);
|
||
|
|
/* initialize counter struct */
|
||
|
|
memset(&counters, 0, sizeof(counters));
|
||
|
|
counters.expected_data_len = data_len;
|
||
|
|
counters.expected_data = data;
|
||
|
|
|
||
|
|
/* create and initialize the pcb */
|
||
|
|
pcb = test_tcp_new_counters_pcb(&counters);
|
||
|
|
EXPECT_RET(pcb != NULL);
|
||
|
|
tcp_set_state(pcb, ESTABLISHED, &local_ip, &remote_ip, local_port, remote_port);
|
||
|
|
|
||
|
|
/* create a segment */
|
||
|
|
p = tcp_create_rx_segment(pcb, counters.expected_data, data_len, 0, 0, 0);
|
||
|
|
EXPECT(p != NULL);
|
||
|
|
if (p != NULL) {
|
||
|
|
/* pass the segment to tcp_input */
|
||
|
|
tcp_input(p, &netif);
|
||
|
|
/* check if counters are as expected */
|
||
|
|
EXPECT(counters.close_calls == 0);
|
||
|
|
EXPECT(counters.recv_calls == 1);
|
||
|
|
EXPECT(counters.recved_bytes == data_len);
|
||
|
|
EXPECT(counters.err_calls == 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
/* make sure the pcb is freed */
|
||
|
|
EXPECT(lwip_stats.memp[MEMP_TCP_PCB].used == 1);
|
||
|
|
tcp_abort(pcb);
|
||
|
|
EXPECT(lwip_stats.memp[MEMP_TCP_PCB].used == 0);
|
||
|
|
}
|
||
|
|
END_TEST
|
||
|
|
|
||
|
|
|
||
|
|
/** Create the suite including all tests for this module */
|
||
|
|
Suite *
|
||
|
|
tcp_suite(void)
|
||
|
|
{
|
||
|
|
TFun tests[] = {
|
||
|
|
test_tcp_new_abort,
|
||
|
|
test_tcp_recv_inseq,
|
||
|
|
};
|
||
|
|
return create_suite("TCP", tests, sizeof(tests)/sizeof(TFun), tcp_setup, tcp_teardown);
|
||
|
|
}
|