| #include "test_tcp.h" |
| |
| #include "lwip/tcp_impl.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}; |
| ip_addr_t 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); |
| } |