1 #include "test_ip4.h"
2 
3 #include "lwip/ip4.h"
4 #include "lwip/inet_chksum.h"
5 #include "lwip/stats.h"
6 #include "lwip/prot/ip.h"
7 #include "lwip/prot/ip4.h"
8 
9 #if !LWIP_IPV4 || !IP_REASSEMBLY || !MIB2_STATS || !IPFRAG_STATS
10 #error "This tests needs LWIP_IPV4, IP_REASSEMBLY; MIB2- and IPFRAG-statistics enabled"
11 #endif
12 
13 /* Helper functions */
14 static void
create_ip4_input_fragment(u16_t ip_id,u16_t start,u16_t len,int last)15 create_ip4_input_fragment(u16_t ip_id, u16_t start, u16_t len, int last)
16 {
17   struct pbuf *p;
18   struct netif *input_netif = netif_list; /* just use any netif */
19   fail_unless((start & 7) == 0);
20   fail_unless(((len & 7) == 0) || last);
21   fail_unless(input_netif != NULL);
22 
23   p = pbuf_alloc(PBUF_RAW, len + sizeof(struct ip_hdr), PBUF_RAM);
24   fail_unless(p != NULL);
25   if (p != NULL) {
26     err_t err;
27     struct ip_hdr *iphdr = (struct ip_hdr *)p->payload;
28     IPH_VHL_SET(iphdr, 4, sizeof(struct ip_hdr) / 4);
29     IPH_TOS_SET(iphdr, 0);
30     IPH_LEN_SET(iphdr, lwip_htons(p->tot_len));
31     IPH_ID_SET(iphdr, lwip_htons(ip_id));
32     if (last) {
33       IPH_OFFSET_SET(iphdr, lwip_htons(start / 8));
34     } else {
35       IPH_OFFSET_SET(iphdr, lwip_htons((start / 8) | IP_MF));
36     }
37     IPH_TTL_SET(iphdr, 5);
38     IPH_PROTO_SET(iphdr, IP_PROTO_UDP);
39     IPH_CHKSUM_SET(iphdr, 0);
40     ip4_addr_copy(iphdr->src, *netif_ip4_addr(input_netif));
41     iphdr->src.addr = lwip_htonl(lwip_htonl(iphdr->src.addr) + 1);
42     ip4_addr_copy(iphdr->dest, *netif_ip4_addr(input_netif));
43     IPH_CHKSUM_SET(iphdr, inet_chksum(iphdr, sizeof(struct ip_hdr)));
44 
45     err = ip4_input(p, input_netif);
46     if (err != ERR_OK) {
47       pbuf_free(p);
48     }
49     fail_unless(err == ERR_OK);
50   }
51 }
52 
53 /* Setups/teardown functions */
54 
55 static void
ip4_setup(void)56 ip4_setup(void)
57 {
58 }
59 
60 static void
ip4_teardown(void)61 ip4_teardown(void)
62 {
63   if (netif_list->loop_first != NULL) {
64     pbuf_free(netif_list->loop_first);
65     netif_list->loop_first = NULL;
66   }
67   netif_list->loop_last = NULL;
68 }
69 
70 
71 /* Test functions */
72 
START_TEST(test_ip4_reass)73 START_TEST(test_ip4_reass)
74 {
75   const u16_t ip_id = 128;
76   LWIP_UNUSED_ARG(_i);
77 
78   memset(&lwip_stats.mib2, 0, sizeof(lwip_stats.mib2));
79 
80   create_ip4_input_fragment(ip_id, 8*200, 200, 1);
81   fail_unless(lwip_stats.ip_frag.recv == 1);
82   fail_unless(lwip_stats.ip_frag.err == 0);
83   fail_unless(lwip_stats.ip_frag.memerr == 0);
84   fail_unless(lwip_stats.ip_frag.drop == 0);
85   fail_unless(lwip_stats.mib2.ipreasmoks == 0);
86 
87   create_ip4_input_fragment(ip_id, 0*200, 200, 0);
88   fail_unless(lwip_stats.ip_frag.recv == 2);
89   fail_unless(lwip_stats.ip_frag.err == 0);
90   fail_unless(lwip_stats.ip_frag.memerr == 0);
91   fail_unless(lwip_stats.ip_frag.drop == 0);
92   fail_unless(lwip_stats.mib2.ipreasmoks == 0);
93 
94   create_ip4_input_fragment(ip_id, 1*200, 200, 0);
95   fail_unless(lwip_stats.ip_frag.recv == 3);
96   fail_unless(lwip_stats.ip_frag.err == 0);
97   fail_unless(lwip_stats.ip_frag.memerr == 0);
98   fail_unless(lwip_stats.ip_frag.drop == 0);
99   fail_unless(lwip_stats.mib2.ipreasmoks == 0);
100 
101   create_ip4_input_fragment(ip_id, 2*200, 200, 0);
102   fail_unless(lwip_stats.ip_frag.recv == 4);
103   fail_unless(lwip_stats.ip_frag.err == 0);
104   fail_unless(lwip_stats.ip_frag.memerr == 0);
105   fail_unless(lwip_stats.ip_frag.drop == 0);
106   fail_unless(lwip_stats.mib2.ipreasmoks == 0);
107 
108   create_ip4_input_fragment(ip_id, 3*200, 200, 0);
109   fail_unless(lwip_stats.ip_frag.recv == 5);
110   fail_unless(lwip_stats.ip_frag.err == 0);
111   fail_unless(lwip_stats.ip_frag.memerr == 0);
112   fail_unless(lwip_stats.ip_frag.drop == 0);
113   fail_unless(lwip_stats.mib2.ipreasmoks == 0);
114 
115   create_ip4_input_fragment(ip_id, 4*200, 200, 0);
116   fail_unless(lwip_stats.ip_frag.recv == 6);
117   fail_unless(lwip_stats.ip_frag.err == 0);
118   fail_unless(lwip_stats.ip_frag.memerr == 0);
119   fail_unless(lwip_stats.ip_frag.drop == 0);
120   fail_unless(lwip_stats.mib2.ipreasmoks == 0);
121 
122   create_ip4_input_fragment(ip_id, 7*200, 200, 0);
123   fail_unless(lwip_stats.ip_frag.recv == 7);
124   fail_unless(lwip_stats.ip_frag.err == 0);
125   fail_unless(lwip_stats.ip_frag.memerr == 0);
126   fail_unless(lwip_stats.ip_frag.drop == 0);
127   fail_unless(lwip_stats.mib2.ipreasmoks == 0);
128 
129   create_ip4_input_fragment(ip_id, 6*200, 200, 0);
130   fail_unless(lwip_stats.ip_frag.recv == 8);
131   fail_unless(lwip_stats.ip_frag.err == 0);
132   fail_unless(lwip_stats.ip_frag.memerr == 0);
133   fail_unless(lwip_stats.ip_frag.drop == 0);
134   fail_unless(lwip_stats.mib2.ipreasmoks == 0);
135 
136   create_ip4_input_fragment(ip_id, 5*200, 200, 0);
137   fail_unless(lwip_stats.ip_frag.recv == 9);
138   fail_unless(lwip_stats.ip_frag.err == 0);
139   fail_unless(lwip_stats.ip_frag.memerr == 0);
140   fail_unless(lwip_stats.ip_frag.drop == 0);
141   fail_unless(lwip_stats.mib2.ipreasmoks == 1);
142 }
143 END_TEST
144 
145 
146 /** Create the suite including all tests for this module */
147 Suite *
ip4_suite(void)148 ip4_suite(void)
149 {
150   testfunc tests[] = {
151     TESTFUNC(test_ip4_reass),
152   };
153   return create_suite("IPv4", tests, sizeof(tests)/sizeof(testfunc), ip4_setup, ip4_teardown);
154 }
155