mirror of
https://github.com/lwip-tcpip/lwip.git
synced 2024-10-01 12:24:22 +00:00
Initial check-in of some unit tests
This commit is contained in:
goldsimon
parent
478ccee5fc
commit
902ad897b8
37
test/unit/lwip_check.h
Normal file
37
test/unit/lwip_check.h
Normal file
@ -0,0 +1,37 @@
|
|||||||
|
#ifndef __LWIP_CHECK_H__
|
||||||
|
#define __LWIP_CHECK_H__
|
||||||
|
|
||||||
|
/* Common header file for lwIP unit tests using the check framework */
|
||||||
|
|
||||||
|
#include <config.h>
|
||||||
|
#include <check.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
|
||||||
|
#define FAIL_RET() do { fail(); return; } while(0)
|
||||||
|
#define EXPECT(x) fail_unless(x)
|
||||||
|
#define EXPECT_RET(x) do { fail_unless(x); if(!(x)) { return; }} while(0)
|
||||||
|
#define EXPECT_RETX(x, y) do { fail_unless(x); if(!(x)) { return y; }} while(0)
|
||||||
|
#define EXPECT_RETNULL(x) EXPECT_RETX(x, NULL)
|
||||||
|
|
||||||
|
/** typedef for a function returning a test suite */
|
||||||
|
typedef Suite* (suite_getter_fn)(void);
|
||||||
|
|
||||||
|
/** Create a test suite */
|
||||||
|
static Suite* create_suite(const char* name, TFun *tests, size_t num_tests, SFun setup, SFun teardown)
|
||||||
|
{
|
||||||
|
size_t i;
|
||||||
|
Suite *s = suite_create(name);
|
||||||
|
|
||||||
|
for(i = 0; i < num_tests; i++) {
|
||||||
|
// Core test case
|
||||||
|
TCase *tc_core = tcase_create("Core");
|
||||||
|
if ((setup != NULL) || (teardown != NULL)) {
|
||||||
|
tcase_add_checked_fixture(tc_core, setup, teardown);
|
||||||
|
}
|
||||||
|
tcase_add_test(tc_core, tests[i]);
|
||||||
|
suite_add_tcase(s, tc_core);
|
||||||
|
}
|
||||||
|
return s;
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif /* __LWIP_CHECK_H__ */
|
||||||
42
test/unit/lwip_unittests.c
Normal file
42
test/unit/lwip_unittests.c
Normal file
@ -0,0 +1,42 @@
|
|||||||
|
#include "lwip_check.h"
|
||||||
|
|
||||||
|
#include "udp/test_udp.h"
|
||||||
|
#include "tcp/test_tcp.h"
|
||||||
|
#include "tcp/test_tcp_oos.h"
|
||||||
|
|
||||||
|
#include "lwip/init.h"
|
||||||
|
|
||||||
|
|
||||||
|
int main()
|
||||||
|
{
|
||||||
|
int number_failed;
|
||||||
|
SRunner *sr;
|
||||||
|
size_t i;
|
||||||
|
suite_getter_fn* suites[] = {
|
||||||
|
udp_suite,
|
||||||
|
tcp_suite,
|
||||||
|
tcp_oos_suite,
|
||||||
|
};
|
||||||
|
size_t num = sizeof(suites)/sizeof(void*);
|
||||||
|
LWIP_ASSERT("No suites defined", num > 0);
|
||||||
|
|
||||||
|
lwip_init();
|
||||||
|
|
||||||
|
sr = srunner_create((suites[0])());
|
||||||
|
for(i = 1; i < num; i++) {
|
||||||
|
srunner_add_suite(sr, ((suite_getter_fn*)suites[i])());
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef LWIP_UNITTESTS_NOFORK
|
||||||
|
srunner_set_fork_status(sr, CK_NOFORK);
|
||||||
|
#endif
|
||||||
|
#ifdef LWIP_UNITTESTS_FORK
|
||||||
|
srunner_set_fork_status(sr, CK_FORK);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
srunner_run_all(sr, CK_NORMAL);
|
||||||
|
number_failed = srunner_ntests_failed(sr);
|
||||||
|
srunner_free(sr);
|
||||||
|
return (number_failed == 0) ? EXIT_SUCCESS : EXIT_FAILURE;
|
||||||
|
}
|
||||||
|
|
||||||
191
test/unit/tcp/tcp_helper.c
Normal file
191
test/unit/tcp/tcp_helper.c
Normal file
@ -0,0 +1,191 @@
|
|||||||
|
#include "tcp_helper.h"
|
||||||
|
|
||||||
|
#include "lwip/tcp.h"
|
||||||
|
#include "lwip/stats.h"
|
||||||
|
#include "lwip/pbuf.h"
|
||||||
|
#include "lwip/inet_chksum.h"
|
||||||
|
|
||||||
|
#if !LWIP_STATS || !TCP_STATS || !MEMP_STATS
|
||||||
|
#error "This tests needs TCP- and MEMP-statistics enabled"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
/** Remove all pcbs on the given list. */
|
||||||
|
static void
|
||||||
|
tcp_remove(struct tcp_pcb* pcb_list)
|
||||||
|
{
|
||||||
|
struct tcp_pcb *pcb = pcb_list;
|
||||||
|
struct tcp_pcb *pcb2;
|
||||||
|
|
||||||
|
while(pcb != NULL) {
|
||||||
|
pcb2 = pcb;
|
||||||
|
pcb = pcb->next;
|
||||||
|
tcp_abort(pcb2);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Remove all pcbs on listen-, active- and time-wait-list (bound- isn't exported). */
|
||||||
|
void
|
||||||
|
tcp_remove_all(void)
|
||||||
|
{
|
||||||
|
//tcp_remove(tcp_bound_pcbs);
|
||||||
|
tcp_remove(tcp_listen_pcbs.pcbs);
|
||||||
|
tcp_remove(tcp_active_pcbs);
|
||||||
|
tcp_remove(tcp_tw_pcbs);
|
||||||
|
fail_unless(lwip_stats.memp[MEMP_TCP_PCB].used == 0);
|
||||||
|
fail_unless(lwip_stats.memp[MEMP_TCP_PCB_LISTEN].used == 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Create a TCP segment usable for passing to tcp_input
|
||||||
|
* - IP-addresses, ports, seqno and ackno are taken from pcb
|
||||||
|
* - seqno and ackno can be altered with an offset
|
||||||
|
*/
|
||||||
|
struct pbuf*
|
||||||
|
tcp_create_rx_segment(struct tcp_pcb* pcb, void* data, size_t data_len, u32_t seqno_offset,
|
||||||
|
u32_t ackno_offset, u8_t headerflags)
|
||||||
|
{
|
||||||
|
return tcp_create_segment(&pcb->remote_ip, &pcb->local_ip, pcb->remote_port, pcb->local_port,
|
||||||
|
data, data_len, pcb->rcv_nxt + seqno_offset, pcb->snd_nxt + ackno_offset, headerflags);
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Create a TCP segment usable for passing to tcp_input */
|
||||||
|
struct pbuf*
|
||||||
|
tcp_create_segment(struct ip_addr* src_ip, struct ip_addr* dst_ip,
|
||||||
|
u16_t src_port, u16_t dst_port, void* data, size_t data_len,
|
||||||
|
u32_t seqno, u32_t ackno, u8_t headerflags)
|
||||||
|
{
|
||||||
|
struct pbuf* p;
|
||||||
|
struct ip_hdr* iphdr;
|
||||||
|
struct tcp_hdr* tcphdr;
|
||||||
|
|
||||||
|
p = pbuf_alloc(PBUF_RAW, sizeof(struct ip_hdr) + sizeof(struct tcp_hdr) + data_len, PBUF_RAM);
|
||||||
|
EXPECT_RETNULL(p != NULL);
|
||||||
|
EXPECT_RETNULL(p->next == NULL);
|
||||||
|
|
||||||
|
memset(p->payload, 0, p->len);
|
||||||
|
|
||||||
|
iphdr = p->payload;
|
||||||
|
/* fill IP header */
|
||||||
|
iphdr->dest.addr = dst_ip->addr;
|
||||||
|
iphdr->src.addr = src_ip->addr;
|
||||||
|
IPH_VHLTOS_SET(iphdr, 4, IP_HLEN / 4, 0);
|
||||||
|
IPH_LEN_SET(iphdr, htons(p->tot_len));
|
||||||
|
IPH_CHKSUM_SET(iphdr, inet_chksum(iphdr, IP_HLEN));
|
||||||
|
|
||||||
|
pbuf_header(p, -(s16_t)sizeof(struct ip_hdr));
|
||||||
|
|
||||||
|
tcphdr = p->payload;
|
||||||
|
tcphdr->src = htons(src_port);
|
||||||
|
tcphdr->dest = htons(dst_port);
|
||||||
|
tcphdr->seqno = htonl(seqno);
|
||||||
|
tcphdr->ackno = htonl(ackno);
|
||||||
|
TCPH_HDRLEN_SET(tcphdr, sizeof(struct tcp_hdr)/4);
|
||||||
|
TCPH_FLAGS_SET(tcphdr, headerflags);
|
||||||
|
tcphdr->wnd = htonl(TCP_WND);
|
||||||
|
|
||||||
|
/* copy data */
|
||||||
|
memcpy((char*)tcphdr + sizeof(struct tcp_hdr), data, data_len);
|
||||||
|
|
||||||
|
/* calculate checksum */
|
||||||
|
tcphdr->chksum = inet_chksum_pseudo(p, &(iphdr->src), &(iphdr->dest),
|
||||||
|
IP_PROTO_TCP, p->tot_len);
|
||||||
|
|
||||||
|
pbuf_header(p, sizeof(struct ip_hdr));
|
||||||
|
|
||||||
|
return p;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Safely bring a tcp_pcb into the requested state */
|
||||||
|
void
|
||||||
|
tcp_set_state(struct tcp_pcb* pcb, enum tcp_state state, struct ip_addr* local_ip,
|
||||||
|
struct ip_addr* remote_ip, u16_t local_port, u16_t remote_port)
|
||||||
|
{
|
||||||
|
/* @todo: are these all states? */
|
||||||
|
/* @todo: remove from previous list */
|
||||||
|
pcb->state = state;
|
||||||
|
if (state == ESTABLISHED) {
|
||||||
|
TCP_REG(&tcp_active_pcbs, pcb);
|
||||||
|
pcb->local_ip.addr = local_ip->addr;
|
||||||
|
pcb->local_port = local_port;
|
||||||
|
pcb->remote_ip.addr = remote_ip->addr;
|
||||||
|
pcb->remote_port = remote_port;
|
||||||
|
} else if(state == LISTEN) {
|
||||||
|
TCP_REG(&tcp_listen_pcbs.pcbs, pcb);
|
||||||
|
pcb->local_ip.addr = local_ip->addr;
|
||||||
|
pcb->local_port = local_port;
|
||||||
|
} else if(state == TIME_WAIT) {
|
||||||
|
TCP_REG(&tcp_tw_pcbs, pcb);
|
||||||
|
pcb->local_ip.addr = local_ip->addr;
|
||||||
|
pcb->local_port = local_port;
|
||||||
|
pcb->remote_ip.addr = remote_ip->addr;
|
||||||
|
pcb->remote_port = remote_port;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
test_tcp_counters_err(void* arg, err_t err)
|
||||||
|
{
|
||||||
|
struct test_tcp_counters* counters = arg;
|
||||||
|
EXPECT_RET(arg != NULL);
|
||||||
|
counters->err_calls++;
|
||||||
|
counters->last_err = err;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
test_tcp_counters_check_rxdata(struct test_tcp_counters* counters, struct pbuf* p)
|
||||||
|
{
|
||||||
|
struct pbuf* q;
|
||||||
|
u32_t i, received;
|
||||||
|
if(counters->expected_data == NULL) {
|
||||||
|
/* no data to compare */
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
EXPECT_RET(counters->recved_bytes + p->tot_len <= counters->expected_data_len);
|
||||||
|
received = counters->recved_bytes;
|
||||||
|
for(q = p; q != NULL; q = q->next) {
|
||||||
|
char *data = q->payload;
|
||||||
|
for(i = 0; i < q->len; i++) {
|
||||||
|
EXPECT_RET(data[i] == counters->expected_data[received]);
|
||||||
|
received++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
EXPECT(received == counters->recved_bytes + p->tot_len);
|
||||||
|
}
|
||||||
|
|
||||||
|
err_t
|
||||||
|
test_tcp_counters_recv(void* arg, struct tcp_pcb* pcb, struct pbuf* p, err_t err)
|
||||||
|
{
|
||||||
|
struct test_tcp_counters* counters = arg;
|
||||||
|
EXPECT_RETX(arg != NULL, ERR_OK);
|
||||||
|
EXPECT_RETX(pcb != NULL, ERR_OK);
|
||||||
|
EXPECT_RETX(err == ERR_OK, ERR_OK);
|
||||||
|
|
||||||
|
if (p != NULL) {
|
||||||
|
if (counters->close_calls == 0) {
|
||||||
|
counters->recv_calls++;
|
||||||
|
test_tcp_counters_check_rxdata(counters, p);
|
||||||
|
counters->recved_bytes += p->tot_len;
|
||||||
|
} else {
|
||||||
|
counters->recv_calls_after_close++;
|
||||||
|
counters->recved_bytes_after_close += p->tot_len;
|
||||||
|
}
|
||||||
|
pbuf_free(p);
|
||||||
|
} else {
|
||||||
|
counters->close_calls++;
|
||||||
|
}
|
||||||
|
EXPECT(counters->recv_calls_after_close == 0 && counters->recved_bytes_after_close == 0);
|
||||||
|
return ERR_OK;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Allocate a pcb and set up the test_tcp_counters_* callbacks */
|
||||||
|
struct tcp_pcb*
|
||||||
|
test_tcp_new_counters_pcb(struct test_tcp_counters* counters)
|
||||||
|
{
|
||||||
|
struct tcp_pcb* pcb = tcp_new();
|
||||||
|
if (pcb != NULL) {
|
||||||
|
/* set up args and callbacks */
|
||||||
|
tcp_arg(pcb, counters);
|
||||||
|
tcp_recv(pcb, test_tcp_counters_recv);
|
||||||
|
tcp_err(pcb, test_tcp_counters_err);
|
||||||
|
}
|
||||||
|
return pcb;
|
||||||
|
}
|
||||||
36
test/unit/tcp/tcp_helper.h
Normal file
36
test/unit/tcp/tcp_helper.h
Normal file
@ -0,0 +1,36 @@
|
|||||||
|
#ifndef __TCP_HELPER_H__
|
||||||
|
#define __TCP_HELPER_H__
|
||||||
|
|
||||||
|
#include "../lwip_check.h"
|
||||||
|
#include "lwip/arch.h"
|
||||||
|
#include "lwip/tcp.h"
|
||||||
|
|
||||||
|
/* counters used for test_tcp_counters_* callback functions */
|
||||||
|
struct test_tcp_counters {
|
||||||
|
u32_t recv_calls;
|
||||||
|
u32_t recved_bytes;
|
||||||
|
u32_t recv_calls_after_close;
|
||||||
|
u32_t recved_bytes_after_close;
|
||||||
|
u32_t close_calls;
|
||||||
|
u32_t err_calls;
|
||||||
|
err_t last_err;
|
||||||
|
char* expected_data;
|
||||||
|
u32_t expected_data_len;
|
||||||
|
};
|
||||||
|
|
||||||
|
/* Helper functions */
|
||||||
|
void tcp_remove_all(void);
|
||||||
|
|
||||||
|
struct pbuf* tcp_create_segment(struct ip_addr* src_ip, struct ip_addr* dst_ip,
|
||||||
|
u16_t src_port, u16_t dst_port, void* data, size_t data_len,
|
||||||
|
u32_t seqno, u32_t ackno, u8_t headerflags);
|
||||||
|
struct pbuf* tcp_create_rx_segment(struct tcp_pcb* pcb, void* data, size_t data_len,
|
||||||
|
u32_t seqno_offset, u32_t ackno_offset, u8_t headerflags);
|
||||||
|
void tcp_set_state(struct tcp_pcb* pcb, enum tcp_state state, struct ip_addr* local_ip,
|
||||||
|
struct ip_addr* remote_ip, u16_t local_port, u16_t remote_port);
|
||||||
|
void test_tcp_counters_err(void* arg, err_t err);
|
||||||
|
err_t test_tcp_counters_recv(void* arg, struct tcp_pcb* pcb, struct pbuf* p, err_t err);
|
||||||
|
|
||||||
|
struct tcp_pcb* test_tcp_new_counters_pcb(struct test_tcp_counters* counters);
|
||||||
|
|
||||||
|
#endif
|
||||||
104
test/unit/tcp/test_tcp.c
Normal file
104
test/unit/tcp/test_tcp.c
Normal file
@ -0,0 +1,104 @@
|
|||||||
|
#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);
|
||||||
|
}
|
||||||
8
test/unit/tcp/test_tcp.h
Normal file
8
test/unit/tcp/test_tcp.h
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
#ifndef __TEST_TCP_H__
|
||||||
|
#define __TEST_TCP_H__
|
||||||
|
|
||||||
|
#include "../lwip_check.h"
|
||||||
|
|
||||||
|
Suite *tcp_suite(void);
|
||||||
|
|
||||||
|
#endif
|
||||||
148
test/unit/tcp/test_tcp_oos.c
Normal file
148
test/unit/tcp/test_tcp_oos.c
Normal file
@ -0,0 +1,148 @@
|
|||||||
|
#include "test_tcp_oos.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
|
||||||
|
#if !TCP_QUEUE_OOSEQ
|
||||||
|
#error "This tests needs TCP_QUEUE_OOSEQ enabled"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
||||||
|
/* Setups/teardown functions */
|
||||||
|
|
||||||
|
static void
|
||||||
|
tcp_oos_setup(void)
|
||||||
|
{
|
||||||
|
tcp_remove_all();
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
tcp_oos_teardown(void)
|
||||||
|
{
|
||||||
|
tcp_remove_all();
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/* Test functions */
|
||||||
|
|
||||||
|
/** create multiple segments and pass them to tcp_input in a wrong
|
||||||
|
* order to see if ooseq-caching works correctly */
|
||||||
|
START_TEST(test_tcp_recv_ooseq)
|
||||||
|
{
|
||||||
|
struct test_tcp_counters counters;
|
||||||
|
struct tcp_pcb* pcb;
|
||||||
|
struct pbuf *p1, *p2, *p3, *p4, *pinseq;
|
||||||
|
char data[] = {
|
||||||
|
1, 2, 3, 4,
|
||||||
|
5, 6, 7, 8,
|
||||||
|
9, 10, 11, 12,
|
||||||
|
13, 14, 15, 16};
|
||||||
|
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 segments */
|
||||||
|
/* pinseq is sent as last segment! */
|
||||||
|
pinseq = tcp_create_rx_segment(pcb, &data[0], 4, 0, 0, TCP_ACK);
|
||||||
|
/* p1: 8 bytes before FIN */
|
||||||
|
/* seqno: 8..15 */
|
||||||
|
p1 = tcp_create_rx_segment(pcb, &data[8], 8, 8, 0, TCP_ACK|TCP_FIN);
|
||||||
|
/* p2: 4 bytes before p1, including the first 4 bytes of p1 (partly duplicate) */
|
||||||
|
/* seqno: 4..11 */
|
||||||
|
p2 = tcp_create_rx_segment(pcb, &data[4], 8, 4, 0, TCP_ACK);
|
||||||
|
/* p3: same as p2 but 2 bytes longer */
|
||||||
|
/* seqno: 4..13 */
|
||||||
|
p3 = tcp_create_rx_segment(pcb, &data[4], 10, 4, 0, TCP_ACK);
|
||||||
|
/* p4: 14 bytes before FIN, includes data from p1 and p2, plus partly from pinseq */
|
||||||
|
/* seqno: 2..15 */
|
||||||
|
p4 = tcp_create_rx_segment(pcb, &data[2], 14, 2, 0, TCP_ACK|TCP_FIN);
|
||||||
|
EXPECT(pinseq != NULL);
|
||||||
|
EXPECT(p1 != NULL);
|
||||||
|
EXPECT(p2 != NULL);
|
||||||
|
EXPECT(p3 != NULL);
|
||||||
|
if ((pinseq != NULL) && (p1 != NULL) && (p2 != NULL) && (p3 != NULL)) {
|
||||||
|
/* pass the segment to tcp_input */
|
||||||
|
tcp_input(p1, &netif);
|
||||||
|
/* check if counters are as expected */
|
||||||
|
EXPECT(counters.close_calls == 0);
|
||||||
|
EXPECT(counters.recv_calls == 0);
|
||||||
|
EXPECT(counters.recved_bytes == 0);
|
||||||
|
EXPECT(counters.err_calls == 0);
|
||||||
|
/* @todo check ooseq queue? */
|
||||||
|
|
||||||
|
/* pass the segment to tcp_input */
|
||||||
|
tcp_input(p2, &netif);
|
||||||
|
/* check if counters are as expected */
|
||||||
|
EXPECT(counters.close_calls == 0);
|
||||||
|
EXPECT(counters.recv_calls == 0);
|
||||||
|
EXPECT(counters.recved_bytes == 0);
|
||||||
|
EXPECT(counters.err_calls == 0);
|
||||||
|
/* @todo check ooseq queue? */
|
||||||
|
|
||||||
|
/* pass the segment to tcp_input */
|
||||||
|
tcp_input(p3, &netif);
|
||||||
|
/* check if counters are as expected */
|
||||||
|
EXPECT(counters.close_calls == 0);
|
||||||
|
EXPECT(counters.recv_calls == 0);
|
||||||
|
EXPECT(counters.recved_bytes == 0);
|
||||||
|
EXPECT(counters.err_calls == 0);
|
||||||
|
/* @todo check ooseq queue? */
|
||||||
|
|
||||||
|
/* pass the segment to tcp_input */
|
||||||
|
tcp_input(p4, &netif);
|
||||||
|
/* check if counters are as expected */
|
||||||
|
EXPECT(counters.close_calls == 0);
|
||||||
|
EXPECT(counters.recv_calls == 0);
|
||||||
|
EXPECT(counters.recved_bytes == 0);
|
||||||
|
EXPECT(counters.err_calls == 0);
|
||||||
|
/* @todo check ooseq queue? */
|
||||||
|
|
||||||
|
/* pass the segment to tcp_input */
|
||||||
|
tcp_input(pinseq, &netif);
|
||||||
|
/* check if counters are as expected */
|
||||||
|
EXPECT(counters.close_calls == 1);
|
||||||
|
EXPECT(counters.recv_calls == 1);
|
||||||
|
EXPECT(counters.recved_bytes == data_len);
|
||||||
|
EXPECT(counters.err_calls == 0);
|
||||||
|
EXPECT(pcb->ooseq == NULL);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 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_oos_suite(void)
|
||||||
|
{
|
||||||
|
TFun tests[] = {
|
||||||
|
test_tcp_recv_ooseq,
|
||||||
|
};
|
||||||
|
return create_suite("TCP_OOS", tests, sizeof(tests)/sizeof(TFun), tcp_oos_setup, tcp_oos_teardown);
|
||||||
|
}
|
||||||
8
test/unit/tcp/test_tcp_oos.h
Normal file
8
test/unit/tcp/test_tcp_oos.h
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
#ifndef __TEST_TCP_OOS_H__
|
||||||
|
#define __TEST_TCP_OOS_H__
|
||||||
|
|
||||||
|
#include "../lwip_check.h"
|
||||||
|
|
||||||
|
Suite *tcp_oos_suite(void);
|
||||||
|
|
||||||
|
#endif
|
||||||
80
test/unit/udp/test_udp.c
Normal file
80
test/unit/udp/test_udp.c
Normal file
@ -0,0 +1,80 @@
|
|||||||
|
#include "test_udp.h"
|
||||||
|
|
||||||
|
#include "lwip/udp.h"
|
||||||
|
#include "lwip/stats.h"
|
||||||
|
|
||||||
|
#if !LWIP_STATS || !UDP_STATS || !MEMP_STATS
|
||||||
|
#error "This tests needs UDP- and MEMP-statistics enabled"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
/* Helper functions */
|
||||||
|
static void
|
||||||
|
udp_remove_all(void)
|
||||||
|
{
|
||||||
|
struct udp_pcb *pcb = udp_pcbs;
|
||||||
|
struct udp_pcb *pcb2;
|
||||||
|
|
||||||
|
while(pcb != NULL) {
|
||||||
|
pcb2 = pcb;
|
||||||
|
pcb = pcb->next;
|
||||||
|
udp_remove(pcb2);
|
||||||
|
}
|
||||||
|
fail_unless(lwip_stats.memp[MEMP_UDP_PCB].used == 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Setups/teardown functions */
|
||||||
|
|
||||||
|
static void
|
||||||
|
udp_setup(void)
|
||||||
|
{
|
||||||
|
udp_remove_all();
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
udp_teardown(void)
|
||||||
|
{
|
||||||
|
udp_remove_all();
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/* Test functions */
|
||||||
|
|
||||||
|
START_TEST(test_udp_new_remove)
|
||||||
|
{
|
||||||
|
struct udp_pcb* pcb;
|
||||||
|
LWIP_UNUSED_ARG(_i);
|
||||||
|
|
||||||
|
fail_unless(lwip_stats.memp[MEMP_UDP_PCB].used == 0);
|
||||||
|
|
||||||
|
pcb = udp_new();
|
||||||
|
fail_unless(pcb != NULL);
|
||||||
|
if (pcb != NULL) {
|
||||||
|
fail_unless(lwip_stats.memp[MEMP_UDP_PCB].used == 1);
|
||||||
|
udp_remove(pcb);
|
||||||
|
fail_unless(lwip_stats.memp[MEMP_UDP_PCB].used == 0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
END_TEST
|
||||||
|
|
||||||
|
START_TEST(test_udp_remove)
|
||||||
|
{
|
||||||
|
struct udp_pcb* pcb;
|
||||||
|
LWIP_UNUSED_ARG(_i);
|
||||||
|
|
||||||
|
pcb = NULL;
|
||||||
|
//pcb = udp_new();
|
||||||
|
//fail_unless(pcb != NULL);
|
||||||
|
}
|
||||||
|
END_TEST
|
||||||
|
|
||||||
|
|
||||||
|
/** Create the suite including all tests for this module */
|
||||||
|
Suite *
|
||||||
|
udp_suite(void)
|
||||||
|
{
|
||||||
|
TFun tests[] = {
|
||||||
|
test_udp_new_remove,
|
||||||
|
test_udp_remove
|
||||||
|
};
|
||||||
|
return create_suite("UDP", tests, sizeof(tests)/sizeof(TFun), udp_setup, udp_teardown);
|
||||||
|
}
|
||||||
8
test/unit/udp/test_udp.h
Normal file
8
test/unit/udp/test_udp.h
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
#ifndef __TEST_UDP_H__
|
||||||
|
#define __TEST_UDP_H__
|
||||||
|
|
||||||
|
#include "../lwip_check.h"
|
||||||
|
|
||||||
|
Suite* udp_suite(void);
|
||||||
|
|
||||||
|
#endif
|
||||||
Loading…
Reference in New Issue
Block a user