mirror of
https://github.com/lwip-tcpip/lwip.git
synced 2024-08-24 15:16:02 +00:00
161 lines
4.9 KiB
C
161 lines
4.9 KiB
C
#include "test_ip4.h"
|
|
|
|
#include "lwip/ip4.h"
|
|
#include "lwip/inet_chksum.h"
|
|
#include "lwip/stats.h"
|
|
#include "lwip/prot/ip.h"
|
|
#include "lwip/prot/ip4.h"
|
|
|
|
#include "lwip/tcpip.h"
|
|
|
|
#if !LWIP_IPV4 || !IP_REASSEMBLY || !MIB2_STATS || !IPFRAG_STATS
|
|
#error "This tests needs LWIP_IPV4, IP_REASSEMBLY; MIB2- and IPFRAG-statistics enabled"
|
|
#endif
|
|
|
|
/* Helper functions */
|
|
static void
|
|
create_ip4_input_fragment(u16_t ip_id, u16_t start, u16_t len, int last)
|
|
{
|
|
struct pbuf *p;
|
|
struct netif *input_netif = netif_list; /* just use any netif */
|
|
fail_unless((start & 7) == 0);
|
|
fail_unless(((len & 7) == 0) || last);
|
|
fail_unless(input_netif != NULL);
|
|
|
|
p = pbuf_alloc(PBUF_RAW, len + sizeof(struct ip_hdr), PBUF_RAM);
|
|
fail_unless(p != NULL);
|
|
if (p != NULL) {
|
|
err_t err;
|
|
struct ip_hdr *iphdr = (struct ip_hdr *)p->payload;
|
|
IPH_VHL_SET(iphdr, 4, sizeof(struct ip_hdr) / 4);
|
|
IPH_TOS_SET(iphdr, 0);
|
|
IPH_LEN_SET(iphdr, lwip_htons(p->tot_len));
|
|
IPH_ID_SET(iphdr, lwip_htons(ip_id));
|
|
if (last) {
|
|
IPH_OFFSET_SET(iphdr, lwip_htons(start / 8));
|
|
} else {
|
|
IPH_OFFSET_SET(iphdr, lwip_htons((start / 8) | IP_MF));
|
|
}
|
|
IPH_TTL_SET(iphdr, 5);
|
|
IPH_PROTO_SET(iphdr, IP_PROTO_UDP);
|
|
IPH_CHKSUM_SET(iphdr, 0);
|
|
ip4_addr_copy(iphdr->src, *netif_ip4_addr(input_netif));
|
|
iphdr->src.addr = lwip_htonl(lwip_htonl(iphdr->src.addr) + 1);
|
|
ip4_addr_copy(iphdr->dest, *netif_ip4_addr(input_netif));
|
|
IPH_CHKSUM_SET(iphdr, inet_chksum(iphdr, sizeof(struct ip_hdr)));
|
|
|
|
err = ip4_input(p, input_netif);
|
|
if (err != ERR_OK) {
|
|
pbuf_free(p);
|
|
}
|
|
fail_unless(err == ERR_OK);
|
|
}
|
|
}
|
|
|
|
/* Setups/teardown functions */
|
|
|
|
static void
|
|
ip4_setup(void)
|
|
{
|
|
lwip_check_ensure_no_alloc(SKIP_POOL(MEMP_SYS_TIMEOUT));
|
|
}
|
|
|
|
static void
|
|
ip4_teardown(void)
|
|
{
|
|
if (netif_list->loop_first != NULL) {
|
|
pbuf_free(netif_list->loop_first);
|
|
netif_list->loop_first = NULL;
|
|
}
|
|
netif_list->loop_last = NULL;
|
|
/* poll until all memory is released... */
|
|
tcpip_thread_poll_one();
|
|
lwip_check_ensure_no_alloc(SKIP_POOL(MEMP_SYS_TIMEOUT));
|
|
}
|
|
|
|
|
|
/* Test functions */
|
|
|
|
START_TEST(test_ip4_reass)
|
|
{
|
|
const u16_t ip_id = 128;
|
|
LWIP_UNUSED_ARG(_i);
|
|
|
|
memset(&lwip_stats.mib2, 0, sizeof(lwip_stats.mib2));
|
|
|
|
create_ip4_input_fragment(ip_id, 8*200, 200, 1);
|
|
fail_unless(lwip_stats.ip_frag.recv == 1);
|
|
fail_unless(lwip_stats.ip_frag.err == 0);
|
|
fail_unless(lwip_stats.ip_frag.memerr == 0);
|
|
fail_unless(lwip_stats.ip_frag.drop == 0);
|
|
fail_unless(lwip_stats.mib2.ipreasmoks == 0);
|
|
|
|
create_ip4_input_fragment(ip_id, 0*200, 200, 0);
|
|
fail_unless(lwip_stats.ip_frag.recv == 2);
|
|
fail_unless(lwip_stats.ip_frag.err == 0);
|
|
fail_unless(lwip_stats.ip_frag.memerr == 0);
|
|
fail_unless(lwip_stats.ip_frag.drop == 0);
|
|
fail_unless(lwip_stats.mib2.ipreasmoks == 0);
|
|
|
|
create_ip4_input_fragment(ip_id, 1*200, 200, 0);
|
|
fail_unless(lwip_stats.ip_frag.recv == 3);
|
|
fail_unless(lwip_stats.ip_frag.err == 0);
|
|
fail_unless(lwip_stats.ip_frag.memerr == 0);
|
|
fail_unless(lwip_stats.ip_frag.drop == 0);
|
|
fail_unless(lwip_stats.mib2.ipreasmoks == 0);
|
|
|
|
create_ip4_input_fragment(ip_id, 2*200, 200, 0);
|
|
fail_unless(lwip_stats.ip_frag.recv == 4);
|
|
fail_unless(lwip_stats.ip_frag.err == 0);
|
|
fail_unless(lwip_stats.ip_frag.memerr == 0);
|
|
fail_unless(lwip_stats.ip_frag.drop == 0);
|
|
fail_unless(lwip_stats.mib2.ipreasmoks == 0);
|
|
|
|
create_ip4_input_fragment(ip_id, 3*200, 200, 0);
|
|
fail_unless(lwip_stats.ip_frag.recv == 5);
|
|
fail_unless(lwip_stats.ip_frag.err == 0);
|
|
fail_unless(lwip_stats.ip_frag.memerr == 0);
|
|
fail_unless(lwip_stats.ip_frag.drop == 0);
|
|
fail_unless(lwip_stats.mib2.ipreasmoks == 0);
|
|
|
|
create_ip4_input_fragment(ip_id, 4*200, 200, 0);
|
|
fail_unless(lwip_stats.ip_frag.recv == 6);
|
|
fail_unless(lwip_stats.ip_frag.err == 0);
|
|
fail_unless(lwip_stats.ip_frag.memerr == 0);
|
|
fail_unless(lwip_stats.ip_frag.drop == 0);
|
|
fail_unless(lwip_stats.mib2.ipreasmoks == 0);
|
|
|
|
create_ip4_input_fragment(ip_id, 7*200, 200, 0);
|
|
fail_unless(lwip_stats.ip_frag.recv == 7);
|
|
fail_unless(lwip_stats.ip_frag.err == 0);
|
|
fail_unless(lwip_stats.ip_frag.memerr == 0);
|
|
fail_unless(lwip_stats.ip_frag.drop == 0);
|
|
fail_unless(lwip_stats.mib2.ipreasmoks == 0);
|
|
|
|
create_ip4_input_fragment(ip_id, 6*200, 200, 0);
|
|
fail_unless(lwip_stats.ip_frag.recv == 8);
|
|
fail_unless(lwip_stats.ip_frag.err == 0);
|
|
fail_unless(lwip_stats.ip_frag.memerr == 0);
|
|
fail_unless(lwip_stats.ip_frag.drop == 0);
|
|
fail_unless(lwip_stats.mib2.ipreasmoks == 0);
|
|
|
|
create_ip4_input_fragment(ip_id, 5*200, 200, 0);
|
|
fail_unless(lwip_stats.ip_frag.recv == 9);
|
|
fail_unless(lwip_stats.ip_frag.err == 0);
|
|
fail_unless(lwip_stats.ip_frag.memerr == 0);
|
|
fail_unless(lwip_stats.ip_frag.drop == 0);
|
|
fail_unless(lwip_stats.mib2.ipreasmoks == 1);
|
|
}
|
|
END_TEST
|
|
|
|
|
|
/** Create the suite including all tests for this module */
|
|
Suite *
|
|
ip4_suite(void)
|
|
{
|
|
testfunc tests[] = {
|
|
TESTFUNC(test_ip4_reass),
|
|
};
|
|
return create_suite("IPv4", tests, sizeof(tests)/sizeof(testfunc), ip4_setup, ip4_teardown);
|
|
}
|