?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Indentation fix, cleanup.
[AROS.git] / arch / all-pc / boot / grub2-aros / grub-core / net / ip.c
blob8c56baaf7894e7551c817bf925c0ecfba9b54eaf
1 /*
2 * GRUB -- GRand Unified Bootloader
3 * Copyright (C) 2010,2011 Free Software Foundation, Inc.
4 *
5 * GRUB is free software: you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation, either version 3 of the License, or
8 * (at your option) any later version.
9 *
10 * GRUB is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License
16 * along with GRUB. If not, see <http://www.gnu.org/licenses/>.
17 */
18
19 #include <grub/net/ip.h>
20 #include <grub/misc.h>
21 #include <grub/net/arp.h>
22 #include <grub/net/udp.h>
23 #include <grub/net/ethernet.h>
24 #include <grub/net.h>
25 #include <grub/net/netbuff.h>
26 #include <grub/mm.h>
27 #include <grub/priority_queue.h>
28 #include <grub/time.h>
29
30 struct iphdr {
31 grub_uint8_t verhdrlen;
32 grub_uint8_t service;
33 grub_uint16_t len;
34 grub_uint16_t ident;
35 grub_uint16_t frags;
36 grub_uint8_t ttl;
37 grub_uint8_t protocol;
38 grub_uint16_t chksum;
39 grub_uint32_t src;
40 grub_uint32_t dest;
41 } GRUB_PACKED ;
42
43 enum
44 {
45 DONT_FRAGMENT = 0x4000,
46 MORE_FRAGMENTS = 0x2000,
47 OFFSET_MASK = 0x1fff
48 };
49
50 typedef grub_uint64_t ip6addr[2];
51
52 struct ip6hdr {
53 grub_uint32_t version_class_flow;
54 grub_uint16_t len;
55 grub_uint8_t protocol;
56 grub_uint8_t ttl;
57 ip6addr src;
58 ip6addr dest;
59 } GRUB_PACKED ;
60
61 static int
62 cmp (const void *a__, const void *b__)
63 {
64 struct grub_net_buff *a_ = *(struct grub_net_buff **) a__;
65 struct grub_net_buff *b_ = *(struct grub_net_buff **) b__;
66 struct iphdr *a = (struct iphdr *) a_->data;
67 struct iphdr *b = (struct iphdr *) b_->data;
68 /* We want the first elements to be on top. */
69 if ((grub_be_to_cpu16 (a->frags) & OFFSET_MASK)
70 < (grub_be_to_cpu16 (b->frags) & OFFSET_MASK))
71 return +1;
72 if ((grub_be_to_cpu16 (a->frags) & OFFSET_MASK)
73 > (grub_be_to_cpu16 (b->frags) & OFFSET_MASK))
74 return -1;
75 return 0;
76 }
77
78 struct reassemble
79 {
80 struct reassemble *next;
81 grub_uint32_t source;
82 grub_uint32_t dest;
83 grub_uint16_t id;
84 grub_uint8_t proto;
85 grub_uint64_t last_time;
86 grub_priority_queue_t pq;
87 struct grub_net_buff *asm_netbuff;
88 grub_size_t total_len;
89 grub_size_t cur_ptr;
90 grub_uint8_t ttl;
91 };
92
93 static struct reassemble *reassembles;
94
95 grub_uint16_t
96 grub_net_ip_chksum (void *ipv, grub_size_t len)
97 {
98 grub_uint16_t *ip = (grub_uint16_t *) ipv;
99 grub_uint32_t sum = 0;
100
101 for (; len >= 2; len -= 2)
102 {
103 sum += grub_be_to_cpu16 (grub_get_unaligned16 (ip++));
104 if (sum > 0xFFFF)
105 sum -= 0xFFFF;
106 }
107 if (len)
108 {
109 sum += *((grub_uint8_t *) ip) << 8;
110 if (sum > 0xFFFF)
111 sum -= 0xFFFF;
112 }
113
114 if (sum >= 0xFFFF)
115 sum -= 0xFFFF;
116
117 return grub_cpu_to_be16 ((~sum) & 0x0000FFFF);
118 }
119
120 static int id = 0x2400;
121
122 static grub_err_t
123 send_fragmented (struct grub_net_network_level_interface * inf,
124 const grub_net_network_level_address_t * target,
125 struct grub_net_buff * nb,
126 grub_net_ip_protocol_t proto,
127 grub_net_link_level_address_t ll_target_addr)
128 {
129 grub_size_t off = 0;
130 grub_size_t fraglen;
131 grub_err_t err;
132
133 fraglen = (inf->card->mtu - sizeof (struct iphdr)) & ~7;
134 id++;
135
136 while (nb->tail - nb->data)
137 {
138 grub_size_t len = fraglen;
139 struct grub_net_buff *nb2;
140 struct iphdr *iph;
141
142 if ((grub_ssize_t) len > nb->tail - nb->data)
143 len = nb->tail - nb->data;
144 nb2 = grub_netbuff_alloc (fraglen + sizeof (struct iphdr)
145 + GRUB_NET_MAX_LINK_HEADER_SIZE);
146 if (!nb2)
147 return grub_errno;
148 err = grub_netbuff_reserve (nb2, GRUB_NET_MAX_LINK_HEADER_SIZE);
149 if (err)
150 return err;
151 err = grub_netbuff_put (nb2, sizeof (struct iphdr));
152 if (err)
153 return err;
154
155 iph = (struct iphdr *) nb2->data;
156 iph->verhdrlen = ((4 << 4) | 5);
157 iph->service = 0;
158 iph->len = grub_cpu_to_be16 (len + sizeof (struct iphdr));
159 iph->ident = grub_cpu_to_be16 (id);
160 iph->frags = grub_cpu_to_be16 (off | (((grub_ssize_t) len
161 == nb->tail - nb->data)
162 ? 0 : MORE_FRAGMENTS));
163 iph->ttl = 0xff;
164 iph->protocol = proto;
165 iph->src = inf->address.ipv4;
166 iph->dest = target->ipv4;
167 off += len / 8;
168
169 iph->chksum = 0;
170 iph->chksum = grub_net_ip_chksum ((void *) nb2->data, sizeof (*iph));
171 err = grub_netbuff_put (nb2, len);
172 if (err)
173 return err;
174 grub_memcpy (iph + 1, nb->data, len);
175 err = grub_netbuff_pull (nb, len);
176 if (err)
177 return err;
178 err = send_ethernet_packet (inf, nb2, ll_target_addr,
179 GRUB_NET_ETHERTYPE_IP);
180 if (err)
181 return err;
182 }
183 return GRUB_ERR_NONE;
184 }
185
186 static grub_err_t
187 grub_net_send_ip4_packet (struct grub_net_network_level_interface *inf,
188 const grub_net_network_level_address_t *target,
189 const grub_net_link_level_address_t *ll_target_addr,
190 struct grub_net_buff *nb,
191 grub_net_ip_protocol_t proto)
192 {
193 struct iphdr *iph;
194 grub_err_t err;
195
196 COMPILE_TIME_ASSERT (GRUB_NET_OUR_IPV4_HEADER_SIZE == sizeof (*iph));
197
198 if (nb->tail - nb->data + sizeof (struct iphdr) > inf->card->mtu)
199 return send_fragmented (inf, target, nb, proto, *ll_target_addr);
200
201 err = grub_netbuff_push (nb, sizeof (*iph));
202 if (err)
203 return err;
204
205 iph = (struct iphdr *) nb->data;
206 iph->verhdrlen = ((4 << 4) | 5);
207 iph->service = 0;
208 iph->len = grub_cpu_to_be16 (nb->tail - nb->data);
209 iph->ident = grub_cpu_to_be16 (++id);
210 iph->frags = 0;
211 iph->ttl = 0xff;
212 iph->protocol = proto;
213 iph->src = inf->address.ipv4;
214 iph->dest = target->ipv4;
215
216 iph->chksum = 0;
217 iph->chksum = grub_net_ip_chksum ((void *) nb->data, sizeof (*iph));
218
219 return send_ethernet_packet (inf, nb, *ll_target_addr,
220 GRUB_NET_ETHERTYPE_IP);
221 }
222
223 static grub_err_t
224 handle_dgram (struct grub_net_buff *nb,
225 struct grub_net_card *card,
226 const grub_net_link_level_address_t *source_hwaddress,
227 const grub_net_link_level_address_t *hwaddress,
228 grub_net_ip_protocol_t proto,
229 const grub_net_network_level_address_t *source,
230 const grub_net_network_level_address_t *dest,
231 grub_uint8_t ttl)
232 {
233 struct grub_net_network_level_interface *inf = NULL;
234 grub_err_t err;
235 int multicast = 0;
236
237 /* DHCP needs special treatment since we don't know IP yet. */
238 {
239 struct udphdr *udph;
240 udph = (struct udphdr *) nb->data;
241 if (proto == GRUB_NET_IP_UDP && grub_be_to_cpu16 (udph->dst) == 68)
242 {
243 const struct grub_net_bootp_packet *bootp;
244 if (udph->chksum)
245 {
246 grub_uint16_t chk, expected;
247 chk = udph->chksum;
248 udph->chksum = 0;
249 expected = grub_net_ip_transport_checksum (nb,
250 GRUB_NET_IP_UDP,
251 source,
252 dest);
253 if (expected != chk)
254 {
255 grub_dprintf ("net", "Invalid UDP checksum. "
256 "Expected %x, got %x\n",
257 grub_be_to_cpu16 (expected),
258 grub_be_to_cpu16 (chk));
259 grub_netbuff_free (nb);
260 return GRUB_ERR_NONE;
261 }
262 udph->chksum = chk;
263 }
264
265 err = grub_netbuff_pull (nb, sizeof (*udph));
266 if (err)
267 {
268 grub_netbuff_free (nb);
269 return err;
270 }
271
272 bootp = (const struct grub_net_bootp_packet *) nb->data;
273
274 FOR_NET_NETWORK_LEVEL_INTERFACES (inf)
275 if (inf->card == card
276 && inf->address.type == GRUB_NET_NETWORK_LEVEL_PROTOCOL_DHCP_RECV
277 && inf->hwaddress.type == GRUB_NET_LINK_LEVEL_PROTOCOL_ETHERNET
278 && grub_memcmp (inf->hwaddress.mac, &bootp->mac_addr,
279 sizeof (inf->hwaddress.mac)) == 0)
280 {
281 grub_net_process_dhcp (nb, inf->card);
282 grub_netbuff_free (nb);
283 return GRUB_ERR_NONE;
284 }
285 grub_netbuff_free (nb);
286 return GRUB_ERR_NONE;
287 }
288 }
289
290 FOR_NET_NETWORK_LEVEL_INTERFACES (inf)
291 {
292 if (inf->card == card
293 && grub_net_addr_cmp (&inf->address, dest) == 0
294 && grub_net_hwaddr_cmp (&inf->hwaddress, hwaddress) == 0)
295 break;
296 /* Solicited node multicast. */
297 if (inf->card == card
298 && inf->address.type == GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6
299 && dest->type == GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6
300 && dest->ipv6[0] == grub_be_to_cpu64_compile_time (0xff02ULL << 48)
301 && dest->ipv6[1] == (grub_be_to_cpu64_compile_time (0x01ff000000ULL)
302 | (inf->address.ipv6[1]
303 & grub_be_to_cpu64_compile_time (0xffffff)))
304 && hwaddress->type == GRUB_NET_LINK_LEVEL_PROTOCOL_ETHERNET
305 && hwaddress->mac[0] == 0x33 && hwaddress->mac[1] == 0x33
306 && hwaddress->mac[2] == 0xff
307 && hwaddress->mac[3] == ((grub_be_to_cpu64 (inf->address.ipv6[1])
308 >> 16) & 0xff)
309 && hwaddress->mac[4] == ((grub_be_to_cpu64 (inf->address.ipv6[1])
310 >> 8) & 0xff)
311 && hwaddress->mac[5] == ((grub_be_to_cpu64 (inf->address.ipv6[1])
312 >> 0) & 0xff))
313 {
314 multicast = 1;
315 break;
316 }
317 }
318
319 if (!inf && !(dest->type == GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6
320 && dest->ipv6[0] == grub_be_to_cpu64_compile_time (0xff02ULL
321 << 48)
322 && dest->ipv6[1] == grub_be_to_cpu64_compile_time (1)))
323 {
324 grub_netbuff_free (nb);
325 return GRUB_ERR_NONE;
326 }
327 if (multicast)
328 inf = NULL;
329
330 switch (proto)
331 {
332 case GRUB_NET_IP_UDP:
333 return grub_net_recv_udp_packet (nb, inf, source);
334 case GRUB_NET_IP_TCP:
335 return grub_net_recv_tcp_packet (nb, inf, source);
336 case GRUB_NET_IP_ICMP:
337 return grub_net_recv_icmp_packet (nb, inf, source_hwaddress, source);
338 case GRUB_NET_IP_ICMPV6:
339 return grub_net_recv_icmp6_packet (nb, card, inf, source_hwaddress,
340 source, dest, ttl);
341 default:
342 grub_netbuff_free (nb);
343 break;
344 }
345 return GRUB_ERR_NONE;
346 }
347
348 static void
349 free_rsm (struct reassemble *rsm)
350 {
351 struct grub_net_buff **nb;
352 while ((nb = grub_priority_queue_top (rsm->pq)))
353 {
354 grub_netbuff_free (*nb);
355 grub_priority_queue_pop (rsm->pq);
356 }
357 grub_netbuff_free (rsm->asm_netbuff);
358 grub_priority_queue_destroy (rsm->pq);
359 grub_free (rsm);
360 }
361
362 static void
363 free_old_fragments (void)
364 {
365 struct reassemble *rsm, **prev;
366 grub_uint64_t limit_time = grub_get_time_ms () - 90000;
367
368 for (prev = &reassembles, rsm = *prev; rsm; rsm = *prev)
369 if (rsm->last_time < limit_time)
370 {
371 *prev = rsm->next;
372 free_rsm (rsm);
373 }
374 else
375 {
376 prev = &rsm->next;
377 }
378 }
379
380 static grub_err_t
381 grub_net_recv_ip4_packets (struct grub_net_buff *nb,
382 struct grub_net_card *card,
383 const grub_net_link_level_address_t *hwaddress,
384 const grub_net_link_level_address_t *src_hwaddress)
385 {
386 struct iphdr *iph = (struct iphdr *) nb->data;
387 grub_err_t err;
388 struct reassemble *rsm, **prev;
389
390 if ((iph->verhdrlen >> 4) != 4)
391 {
392 grub_dprintf ("net", "Bad IP version: %d\n", (iph->verhdrlen >> 4));
393 grub_netbuff_free (nb);
394 return GRUB_ERR_NONE;
395 }
396
397 if ((iph->verhdrlen & 0xf) < 5)
398 {
399 grub_dprintf ("net", "IP header too short: %d\n",
400 (iph->verhdrlen & 0xf));
401 grub_netbuff_free (nb);
402 return GRUB_ERR_NONE;
403 }
404
405 if (nb->tail - nb->data < (grub_ssize_t) ((iph->verhdrlen & 0xf)
406 * sizeof (grub_uint32_t)))
407 {
408 grub_dprintf ("net", "IP packet too short: %" PRIdGRUB_SSIZE "\n",
409 (grub_ssize_t) (nb->tail - nb->data));
410 grub_netbuff_free (nb);
411 return GRUB_ERR_NONE;
412 }
413
414 /* Check size. */
415 {
416 grub_size_t expected_size = grub_be_to_cpu16 (iph->len);
417 grub_size_t actual_size = (nb->tail - nb->data);
418 if (actual_size > expected_size)
419 {
420 err = grub_netbuff_unput (nb, actual_size - expected_size);
421 if (err)
422 {
423 grub_netbuff_free (nb);
424 return err;
425 }
426 }
427 if (actual_size < expected_size)
428 {
429 grub_dprintf ("net", "Cut IP packet actual: %" PRIuGRUB_SIZE
430 ", expected %" PRIuGRUB_SIZE "\n", actual_size,
431 expected_size);
432 grub_netbuff_free (nb);
433 return GRUB_ERR_NONE;
434 }
435 }
436
437 /* Unfragmented packet. Good. */
438 if (((grub_be_to_cpu16 (iph->frags) & MORE_FRAGMENTS) == 0)
439 && (grub_be_to_cpu16 (iph->frags) & OFFSET_MASK) == 0)
440 {
441 grub_net_network_level_address_t source;
442 grub_net_network_level_address_t dest;
443
444 err = grub_netbuff_pull (nb, ((iph->verhdrlen & 0xf)
445 * sizeof (grub_uint32_t)));
446 if (err)
447 {
448 grub_netbuff_free (nb);
449 return err;
450 }
451
452 source.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4;
453 source.ipv4 = iph->src;
454
455 dest.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4;
456 dest.ipv4 = iph->dest;
457
458 return handle_dgram (nb, card, src_hwaddress, hwaddress, iph->protocol,
459 &source, &dest, iph->ttl);
460 }
461
462 for (prev = &reassembles, rsm = *prev; rsm; prev = &rsm->next, rsm = *prev)
463 if (rsm->source == iph->src && rsm->dest == iph->dest
464 && rsm->id == iph->ident && rsm->proto == iph->protocol)
465 break;
466 if (!rsm)
467 {
468 rsm = grub_malloc (sizeof (*rsm));
469 if (!rsm)
470 return grub_errno;
471 rsm->source = iph->src;
472 rsm->dest = iph->dest;
473 rsm->id = iph->ident;
474 rsm->proto = iph->protocol;
475 rsm->next = reassembles;
476 reassembles = rsm;
477 prev = &reassembles;
478 rsm->pq = grub_priority_queue_new (sizeof (struct grub_net_buff **), cmp);
479 if (!rsm->pq)
480 {
481 grub_free (rsm);
482 return grub_errno;
483 }
484 rsm->asm_netbuff = 0;
485 rsm->total_len = 0;
486 rsm->cur_ptr = 0;
487 rsm->ttl = 0xff;
488 }
489 if (rsm->ttl > iph->ttl)
490 rsm->ttl = iph->ttl;
491 rsm->last_time = grub_get_time_ms ();
492 free_old_fragments ();
493
494 err = grub_priority_queue_push (rsm->pq, &nb);
495 if (err)
496 return err;
497
498 if (!(grub_be_to_cpu16 (iph->frags) & MORE_FRAGMENTS))
499 {
500 rsm->total_len = (8 * (grub_be_to_cpu16 (iph->frags) & OFFSET_MASK)
501 + (nb->tail - nb->data));
502 rsm->total_len -= ((iph->verhdrlen & 0xf) * sizeof (grub_uint32_t));
503 rsm->asm_netbuff = grub_netbuff_alloc (rsm->total_len);
504 if (!rsm->asm_netbuff)
505 {
506 *prev = rsm->next;
507 free_rsm (rsm);
508 return grub_errno;
509 }
510 }
511 if (!rsm->asm_netbuff)
512 return GRUB_ERR_NONE;
513
514 while (1)
515 {
516 struct grub_net_buff **nb_top_p, *nb_top;
517 grub_size_t copy;
518 grub_size_t res_len;
519 struct grub_net_buff *ret;
520 grub_net_ip_protocol_t proto;
521 grub_uint32_t src;
522 grub_uint32_t dst;
523 grub_net_network_level_address_t source;
524 grub_net_network_level_address_t dest;
525 grub_uint8_t ttl;
526
527 nb_top_p = grub_priority_queue_top (rsm->pq);
528 if (!nb_top_p)
529 return GRUB_ERR_NONE;
530 nb_top = *nb_top_p;
531 grub_priority_queue_pop (rsm->pq);
532 iph = (struct iphdr *) nb_top->data;
533 err = grub_netbuff_pull (nb_top, ((iph->verhdrlen & 0xf)
534 * sizeof (grub_uint32_t)));
535 if (err)
536 {
537 grub_netbuff_free (nb_top);
538 return err;
539 }
540 if (rsm->cur_ptr < (grub_size_t) 8 * (grub_be_to_cpu16 (iph->frags)
541 & OFFSET_MASK))
542 {
543 grub_netbuff_free (nb_top);
544 return GRUB_ERR_NONE;
545 }
546
547 rsm->cur_ptr = (8 * (grub_be_to_cpu16 (iph->frags) & OFFSET_MASK)
548 + (nb_top->tail - nb_top->head));
549 if ((grub_size_t) 8 * (grub_be_to_cpu16 (iph->frags) & OFFSET_MASK)
550 >= rsm->total_len)
551 {
552 grub_netbuff_free (nb_top);
553 continue;
554 }
555 copy = nb_top->tail - nb_top->data;
556 if (rsm->total_len - 8 * (grub_be_to_cpu16 (iph->frags) & OFFSET_MASK)
557 < copy)
558 copy = rsm->total_len - 8 * (grub_be_to_cpu16 (iph->frags)
559 & OFFSET_MASK);
560 grub_memcpy (&rsm->asm_netbuff->data[8 * (grub_be_to_cpu16 (iph->frags)
561 & OFFSET_MASK)],
562 nb_top->data, copy);
563
564 if ((grub_be_to_cpu16 (iph->frags) & MORE_FRAGMENTS))
565 {
566 grub_netbuff_free (nb_top);
567 continue;
568 }
569 grub_netbuff_free (nb_top);
570
571 ret = rsm->asm_netbuff;
572 proto = rsm->proto;
573 src = rsm->source;
574 dst = rsm->dest;
575 ttl = rsm->ttl;
576
577 rsm->asm_netbuff = 0;
578 res_len = rsm->total_len;
579 *prev = rsm->next;
580 free_rsm (rsm);
581
582 if (grub_netbuff_put (ret, res_len))
583 {
584 grub_netbuff_free (ret);
585 return GRUB_ERR_NONE;
586 }
587
588 source.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4;
589 source.ipv4 = src;
590
591 dest.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4;
592 dest.ipv4 = dst;
593
594 return handle_dgram (ret, card, src_hwaddress,
595 hwaddress, proto, &source, &dest,
596 ttl);
597 }
598 }
599
600 static grub_err_t
601 grub_net_send_ip6_packet (struct grub_net_network_level_interface *inf,
602 const grub_net_network_level_address_t *target,
603 const grub_net_link_level_address_t *ll_target_addr,
604 struct grub_net_buff *nb,
605 grub_net_ip_protocol_t proto)
606 {
607 struct ip6hdr *iph;
608 grub_err_t err;
609
610 COMPILE_TIME_ASSERT (GRUB_NET_OUR_IPV6_HEADER_SIZE == sizeof (*iph));
611
612 if (nb->tail - nb->data + sizeof (struct iphdr) > inf->card->mtu)
613 return grub_error (GRUB_ERR_NET_PACKET_TOO_BIG, "packet too big");
614
615 err = grub_netbuff_push (nb, sizeof (*iph));
616 if (err)
617 return err;
618
619 iph = (struct ip6hdr *) nb->data;
620 iph->version_class_flow = grub_cpu_to_be32_compile_time ((6 << 28));
621 iph->len = grub_cpu_to_be16 (nb->tail - nb->data - sizeof (*iph));
622 iph->protocol = proto;
623 iph->ttl = 0xff;
624 grub_memcpy (&iph->src, inf->address.ipv6, sizeof (iph->src));
625 grub_memcpy (&iph->dest, target->ipv6, sizeof (iph->dest));
626
627 return send_ethernet_packet (inf, nb, *ll_target_addr,
628 GRUB_NET_ETHERTYPE_IP6);
629 }
630
631 grub_err_t
632 grub_net_send_ip_packet (struct grub_net_network_level_interface *inf,
633 const grub_net_network_level_address_t *target,
634 const grub_net_link_level_address_t *ll_target_addr,
635 struct grub_net_buff *nb,
636 grub_net_ip_protocol_t proto)
637 {
638 switch (target->type)
639 {
640 case GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4:
641 return grub_net_send_ip4_packet (inf, target, ll_target_addr, nb, proto);
642 case GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6:
643 return grub_net_send_ip6_packet (inf, target, ll_target_addr, nb, proto);
644 default:
645 return grub_error (GRUB_ERR_BUG, "not an IP");
646 }
647 }
648
649 static grub_err_t
650 grub_net_recv_ip6_packets (struct grub_net_buff *nb,
651 struct grub_net_card *card,
652 const grub_net_link_level_address_t *hwaddress,
653 const grub_net_link_level_address_t *src_hwaddress)
654 {
655 struct ip6hdr *iph = (struct ip6hdr *) nb->data;
656 grub_err_t err;
657 grub_net_network_level_address_t source;
658 grub_net_network_level_address_t dest;
659
660 if (nb->tail - nb->data < (grub_ssize_t) sizeof (*iph))
661 {
662 grub_dprintf ("net", "IP packet too short: %" PRIdGRUB_SSIZE "\n",
663 (grub_ssize_t) (nb->tail - nb->data));
664 grub_netbuff_free (nb);
665 return GRUB_ERR_NONE;
666 }
667
668 err = grub_netbuff_pull (nb, sizeof (*iph));
669 if (err)
670 {
671 grub_netbuff_free (nb);
672 return err;
673 }
674
675 /* Check size. */
676 {
677 grub_size_t expected_size = grub_be_to_cpu16 (iph->len);
678 grub_size_t actual_size = (nb->tail - nb->data);
679 if (actual_size > expected_size)
680 {
681 err = grub_netbuff_unput (nb, actual_size - expected_size);
682 if (err)
683 {
684 grub_netbuff_free (nb);
685 return err;
686 }
687 }
688 if (actual_size < expected_size)
689 {
690 grub_dprintf ("net", "Cut IP packet actual: %" PRIuGRUB_SIZE
691 ", expected %" PRIuGRUB_SIZE "\n", actual_size,
692 expected_size);
693 grub_netbuff_free (nb);
694 return GRUB_ERR_NONE;
695 }
696 }
697
698 source.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6;
699 dest.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6;
700 grub_memcpy (source.ipv6, &iph->src, sizeof (source.ipv6));
701 grub_memcpy (dest.ipv6, &iph->dest, sizeof (dest.ipv6));
702
703 return handle_dgram (nb, card, src_hwaddress, hwaddress, iph->protocol,
704 &source, &dest, iph->ttl);
705 }
706
707 grub_err_t
708 grub_net_recv_ip_packets (struct grub_net_buff *nb,
709 struct grub_net_card *card,
710 const grub_net_link_level_address_t *hwaddress,
711 const grub_net_link_level_address_t *src_hwaddress)
712 {
713 struct iphdr *iph = (struct iphdr *) nb->data;
714
715 if ((iph->verhdrlen >> 4) == 4)
716 return grub_net_recv_ip4_packets (nb, card, hwaddress, src_hwaddress);
717 if ((iph->verhdrlen >> 4) == 6)
718 return grub_net_recv_ip6_packets (nb, card, hwaddress, src_hwaddress);
719 grub_dprintf ("net", "Bad IP version: %d\n", (iph->verhdrlen >> 4));
720 grub_netbuff_free (nb);
721 return GRUB_ERR_NONE;
722 }
Mirror of AROS' svn.