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