BadVPN – Blame information for rev 1
?pathlinks?
Rev | Author | Line No. | Line |
---|---|---|---|
1 | office | 1 | #include "test_etharp.h" |
2 | |||
3 | #include "lwip/udp.h" |
||
4 | #include "lwip/etharp.h" |
||
5 | #include "netif/ethernet.h" |
||
6 | #include "lwip/stats.h" |
||
7 | #include "lwip/prot/iana.h" |
||
8 | |||
9 | #if !LWIP_STATS || !UDP_STATS || !MEMP_STATS || !ETHARP_STATS |
||
10 | #error "This tests needs UDP-, MEMP- and ETHARP-statistics enabled" |
||
11 | #endif |
||
12 | #if !ETHARP_SUPPORT_STATIC_ENTRIES |
||
13 | #error "This test needs ETHARP_SUPPORT_STATIC_ENTRIES enabled" |
||
14 | #endif |
||
15 | |||
16 | static struct netif test_netif; |
||
17 | static ip4_addr_t test_ipaddr, test_netmask, test_gw; |
||
18 | struct eth_addr test_ethaddr = {{1,1,1,1,1,1}}; |
||
19 | struct eth_addr test_ethaddr2 = {{1,1,1,1,1,2}}; |
||
20 | struct eth_addr test_ethaddr3 = {{1,1,1,1,1,3}}; |
||
21 | struct eth_addr test_ethaddr4 = {{1,1,1,1,1,4}}; |
||
22 | static int linkoutput_ctr; |
||
23 | |||
24 | /* Helper functions */ |
||
25 | static void |
||
26 | etharp_remove_all(void) |
||
27 | { |
||
28 | int i; |
||
29 | /* call etharp_tmr often enough to have all entries cleaned */ |
||
30 | for(i = 0; i < 0xff; i++) { |
||
31 | etharp_tmr(); |
||
32 | } |
||
33 | } |
||
34 | |||
35 | static err_t |
||
36 | default_netif_linkoutput(struct netif *netif, struct pbuf *p) |
||
37 | { |
||
38 | fail_unless(netif == &test_netif); |
||
39 | fail_unless(p != NULL); |
||
40 | linkoutput_ctr++; |
||
41 | return ERR_OK; |
||
42 | } |
||
43 | |||
44 | static err_t |
||
45 | default_netif_init(struct netif *netif) |
||
46 | { |
||
47 | fail_unless(netif != NULL); |
||
48 | netif->linkoutput = default_netif_linkoutput; |
||
49 | netif->output = etharp_output; |
||
50 | netif->mtu = 1500; |
||
51 | netif->flags = NETIF_FLAG_BROADCAST | NETIF_FLAG_ETHARP | NETIF_FLAG_LINK_UP; |
||
52 | netif->hwaddr_len = ETHARP_HWADDR_LEN; |
||
53 | return ERR_OK; |
||
54 | } |
||
55 | |||
56 | static void |
||
57 | default_netif_add(void) |
||
58 | { |
||
59 | IP4_ADDR(&test_gw, 192,168,0,1); |
||
60 | IP4_ADDR(&test_ipaddr, 192,168,0,1); |
||
61 | IP4_ADDR(&test_netmask, 255,255,0,0); |
||
62 | |||
63 | fail_unless(netif_default == NULL); |
||
64 | netif_set_default(netif_add(&test_netif, &test_ipaddr, &test_netmask, |
||
65 | &test_gw, NULL, default_netif_init, NULL)); |
||
66 | netif_set_up(&test_netif); |
||
67 | } |
||
68 | |||
69 | static void |
||
70 | default_netif_remove(void) |
||
71 | { |
||
72 | fail_unless(netif_default == &test_netif); |
||
73 | netif_remove(&test_netif); |
||
74 | } |
||
75 | |||
76 | static void |
||
77 | create_arp_response(ip4_addr_t *adr) |
||
78 | { |
||
79 | int k; |
||
80 | struct eth_hdr *ethhdr; |
||
81 | struct etharp_hdr *etharphdr; |
||
82 | struct pbuf *p = pbuf_alloc(PBUF_RAW, sizeof(struct eth_hdr) + sizeof(struct etharp_hdr), PBUF_RAM); |
||
83 | if(p == NULL) { |
||
84 | FAIL_RET(); |
||
85 | } |
||
86 | ethhdr = (struct eth_hdr*)p->payload; |
||
87 | etharphdr = (struct etharp_hdr*)(ethhdr + 1); |
||
88 | |||
89 | ethhdr->dest = test_ethaddr; |
||
90 | ethhdr->src = test_ethaddr2; |
||
91 | ethhdr->type = htons(ETHTYPE_ARP); |
||
92 | |||
93 | etharphdr->hwtype = htons(LWIP_IANA_HWTYPE_ETHERNET); |
||
94 | etharphdr->proto = htons(ETHTYPE_IP); |
||
95 | etharphdr->hwlen = ETHARP_HWADDR_LEN; |
||
96 | etharphdr->protolen = sizeof(ip4_addr_t); |
||
97 | etharphdr->opcode = htons(ARP_REPLY); |
||
98 | |||
99 | SMEMCPY(ðarphdr->sipaddr, adr, sizeof(ip4_addr_t)); |
||
100 | SMEMCPY(ðarphdr->dipaddr, &test_ipaddr, sizeof(ip4_addr_t)); |
||
101 | |||
102 | k = 6; |
||
103 | while(k > 0) { |
||
104 | k--; |
||
105 | /* Write the ARP MAC-Addresses */ |
||
106 | etharphdr->shwaddr.addr[k] = test_ethaddr2.addr[k]; |
||
107 | etharphdr->dhwaddr.addr[k] = test_ethaddr.addr[k]; |
||
108 | /* Write the Ethernet MAC-Addresses */ |
||
109 | ethhdr->dest.addr[k] = test_ethaddr.addr[k]; |
||
110 | ethhdr->src.addr[k] = test_ethaddr2.addr[k]; |
||
111 | } |
||
112 | |||
113 | ethernet_input(p, &test_netif); |
||
114 | } |
||
115 | |||
116 | /* Setups/teardown functions */ |
||
117 | |||
118 | static void |
||
119 | etharp_setup(void) |
||
120 | { |
||
121 | etharp_remove_all(); |
||
122 | default_netif_add(); |
||
123 | lwip_check_ensure_no_alloc(SKIP_POOL(MEMP_SYS_TIMEOUT)); |
||
124 | } |
||
125 | |||
126 | static void |
||
127 | etharp_teardown(void) |
||
128 | { |
||
129 | etharp_remove_all(); |
||
130 | default_netif_remove(); |
||
131 | lwip_check_ensure_no_alloc(SKIP_POOL(MEMP_SYS_TIMEOUT)); |
||
132 | } |
||
133 | |||
134 | |||
135 | /* Test functions */ |
||
136 | |||
137 | START_TEST(test_etharp_table) |
||
138 | { |
||
139 | #if ETHARP_SUPPORT_STATIC_ENTRIES |
||
140 | err_t err; |
||
141 | #endif /* ETHARP_SUPPORT_STATIC_ENTRIES */ |
||
142 | s8_t idx; |
||
143 | const ip4_addr_t *unused_ipaddr; |
||
144 | struct eth_addr *unused_ethaddr; |
||
145 | struct udp_pcb* pcb; |
||
146 | LWIP_UNUSED_ARG(_i); |
||
147 | |||
148 | if (netif_default != &test_netif) { |
||
149 | fail("This test needs a default netif"); |
||
150 | } |
||
151 | |||
152 | linkoutput_ctr = 0; |
||
153 | |||
154 | pcb = udp_new(); |
||
155 | fail_unless(pcb != NULL); |
||
156 | if (pcb != NULL) { |
||
157 | ip4_addr_t adrs[ARP_TABLE_SIZE + 2]; |
||
158 | int i; |
||
159 | for(i = 0; i < ARP_TABLE_SIZE + 2; i++) { |
||
160 | IP4_ADDR(&adrs[i], 192,168,0,i+2); |
||
161 | } |
||
162 | /* fill ARP-table with dynamic entries */ |
||
163 | for(i = 0; i < ARP_TABLE_SIZE; i++) { |
||
164 | struct pbuf *p = pbuf_alloc(PBUF_TRANSPORT, 10, PBUF_RAM); |
||
165 | fail_unless(p != NULL); |
||
166 | if (p != NULL) { |
||
167 | err_t err2; |
||
168 | ip_addr_t dst; |
||
169 | ip_addr_copy_from_ip4(dst, adrs[i]); |
||
170 | err2 = udp_sendto(pcb, p, &dst, 123); |
||
171 | fail_unless(err2 == ERR_OK); |
||
172 | /* etharp request sent? */ |
||
173 | fail_unless(linkoutput_ctr == (2*i) + 1); |
||
174 | pbuf_free(p); |
||
175 | |||
176 | /* create an ARP response */ |
||
177 | create_arp_response(&adrs[i]); |
||
178 | /* queued UDP packet sent? */ |
||
179 | fail_unless(linkoutput_ctr == (2*i) + 2); |
||
180 | |||
181 | idx = etharp_find_addr(NULL, &adrs[i], &unused_ethaddr, &unused_ipaddr); |
||
182 | fail_unless(idx == i); |
||
183 | etharp_tmr(); |
||
184 | } |
||
185 | } |
||
186 | linkoutput_ctr = 0; |
||
187 | #if ETHARP_SUPPORT_STATIC_ENTRIES |
||
188 | /* create one static entry */ |
||
189 | err = etharp_add_static_entry(&adrs[ARP_TABLE_SIZE], &test_ethaddr3); |
||
190 | fail_unless(err == ERR_OK); |
||
191 | idx = etharp_find_addr(NULL, &adrs[ARP_TABLE_SIZE], &unused_ethaddr, &unused_ipaddr); |
||
192 | fail_unless(idx == 0); |
||
193 | fail_unless(linkoutput_ctr == 0); |
||
194 | #endif /* ETHARP_SUPPORT_STATIC_ENTRIES */ |
||
195 | |||
196 | linkoutput_ctr = 0; |
||
197 | /* fill ARP-table with dynamic entries */ |
||
198 | for(i = 0; i < ARP_TABLE_SIZE; i++) { |
||
199 | struct pbuf *p = pbuf_alloc(PBUF_TRANSPORT, 10, PBUF_RAM); |
||
200 | fail_unless(p != NULL); |
||
201 | if (p != NULL) { |
||
202 | err_t err2; |
||
203 | ip_addr_t dst; |
||
204 | ip_addr_copy_from_ip4(dst, adrs[i]); |
||
205 | err2 = udp_sendto(pcb, p, &dst, 123); |
||
206 | fail_unless(err2 == ERR_OK); |
||
207 | /* etharp request sent? */ |
||
208 | fail_unless(linkoutput_ctr == (2*i) + 1); |
||
209 | pbuf_free(p); |
||
210 | |||
211 | /* create an ARP response */ |
||
212 | create_arp_response(&adrs[i]); |
||
213 | /* queued UDP packet sent? */ |
||
214 | fail_unless(linkoutput_ctr == (2*i) + 2); |
||
215 | |||
216 | idx = etharp_find_addr(NULL, &adrs[i], &unused_ethaddr, &unused_ipaddr); |
||
217 | if (i < ARP_TABLE_SIZE - 1) { |
||
218 | fail_unless(idx == i+1); |
||
219 | } else { |
||
220 | /* the last entry must not overwrite the static entry! */ |
||
221 | fail_unless(idx == 1); |
||
222 | } |
||
223 | etharp_tmr(); |
||
224 | } |
||
225 | } |
||
226 | #if ETHARP_SUPPORT_STATIC_ENTRIES |
||
227 | /* create a second static entry */ |
||
228 | err = etharp_add_static_entry(&adrs[ARP_TABLE_SIZE+1], &test_ethaddr4); |
||
229 | fail_unless(err == ERR_OK); |
||
230 | idx = etharp_find_addr(NULL, &adrs[ARP_TABLE_SIZE], &unused_ethaddr, &unused_ipaddr); |
||
231 | fail_unless(idx == 0); |
||
232 | idx = etharp_find_addr(NULL, &adrs[ARP_TABLE_SIZE+1], &unused_ethaddr, &unused_ipaddr); |
||
233 | fail_unless(idx == 2); |
||
234 | /* and remove it again */ |
||
235 | err = etharp_remove_static_entry(&adrs[ARP_TABLE_SIZE+1]); |
||
236 | fail_unless(err == ERR_OK); |
||
237 | idx = etharp_find_addr(NULL, &adrs[ARP_TABLE_SIZE], &unused_ethaddr, &unused_ipaddr); |
||
238 | fail_unless(idx == 0); |
||
239 | idx = etharp_find_addr(NULL, &adrs[ARP_TABLE_SIZE+1], &unused_ethaddr, &unused_ipaddr); |
||
240 | fail_unless(idx == -1); |
||
241 | #endif /* ETHARP_SUPPORT_STATIC_ENTRIES */ |
||
242 | |||
243 | /* check that static entries don't time out */ |
||
244 | etharp_remove_all(); |
||
245 | idx = etharp_find_addr(NULL, &adrs[ARP_TABLE_SIZE], &unused_ethaddr, &unused_ipaddr); |
||
246 | fail_unless(idx == 0); |
||
247 | |||
248 | #if ETHARP_SUPPORT_STATIC_ENTRIES |
||
249 | /* remove the first static entry */ |
||
250 | err = etharp_remove_static_entry(&adrs[ARP_TABLE_SIZE]); |
||
251 | fail_unless(err == ERR_OK); |
||
252 | idx = etharp_find_addr(NULL, &adrs[ARP_TABLE_SIZE], &unused_ethaddr, &unused_ipaddr); |
||
253 | fail_unless(idx == -1); |
||
254 | idx = etharp_find_addr(NULL, &adrs[ARP_TABLE_SIZE+1], &unused_ethaddr, &unused_ipaddr); |
||
255 | fail_unless(idx == -1); |
||
256 | #endif /* ETHARP_SUPPORT_STATIC_ENTRIES */ |
||
257 | |||
258 | udp_remove(pcb); |
||
259 | } |
||
260 | } |
||
261 | END_TEST |
||
262 | |||
263 | |||
264 | /** Create the suite including all tests for this module */ |
||
265 | Suite * |
||
266 | etharp_suite(void) |
||
267 | { |
||
268 | testfunc tests[] = { |
||
269 | TESTFUNC(test_etharp_table) |
||
270 | }; |
||
271 | return create_suite("ETHARP", tests, sizeof(tests)/sizeof(testfunc), etharp_setup, etharp_teardown); |
||
272 | } |