198e150aaf
* grub-core/Makefile.core.def (net): Add net/dns.c, net/tcp.c, net/icmp.c and net/icmp6.c. (http): New module. (priority_queue): Likewise. * grub-core/io/bufio.c: Rewritten. * grub-core/lib/legacy_parse.c (legacy_command): New argument type TYPE_WITH_CONFIGFILE_OPTION. (legacy_commands): Add bootp and dhcp. (is_option): Handle TYPE_WITH_CONFIGFILE_OPTION. (grub_legacy_parse): Likewise. * grub-core/lib/priority_queue.c: New file. * grub-core/net/arp.c: Add missing license header. (arp_find_entry): Removed. (arp_find_entry): Likewise. (grub_net_arp_resolve): Rename to ... (grub_net_arp_send_request): ...this. (grub_net_arp_receive): New card argument. * grub-core/net/bootp.c (parse_dhcp_vendor): Clean up. Set router and DNS server. (grub_net_configure_by_dhcp_ack): Handle routing information. (grub_cmd_bootp): Set checksum. (grub_bootp_init): Remove net_dhcp. * grub-core/net/dns.c: New file. * grub-core/net/drivers/efi/efinet.c (send_card_buffer): Wait for completion. (get_card_packet): Handle allocation. (grub_efinet_findcards): Set mtu. * grub-core/net/drivers/emu/emunet.c: Add missing license header. (get_card_packet): Handle allocation. (emucard): Set mtu. * grub-core/net/drivers/i386/pc/pxe.c (grub_pxe_recv): Handle allocation (GRUB_MOD_INIT): Set mtu. * grub-core/net/drivers/ieee1275/ofnet.c (grub_ofnetcard_data): Remove mtu. (get_card_packet): Handle allocation. (grub_ofnet_findcards): Set mtu. * grub-core/net/ethernet.c (send_ethernet_packet): Add compile time assert. (grub_net_recv_ethernet_packet): Handle IPv6. * grub-core/net/http.c: New file. * grub-core/net/icmp.c: Likewise. * grub-core/net/icmp6.c: Likewise. * grub-core/net/ip.c (ip6addr): New type. (ip6hdr): Likewise. (reassemble): Likewise. (cmp): New function. (reassembles): New variable. (grub_net_ip_chksum): Handle 0xffff sum and unaligned buffers. (id): New variable. (send_fragmented): New function. (grub_net_send_ip_packet): Rename to ... (grub_net_send_ip4_packet): ... this. Send fragmented if needed. Handle non-UDP. (grub_net_recv_ip_packets): Rename to ... (handle_dgram): ... this. Check checksum. Handle non-UDP. (free_rsm): New function. (free_old_fragments): Likewise. (grub_net_recv_ip4_packets): New function. (grub_net_send_ip6_packet): Likewise. (grub_net_send_ip_packet): Likewise. (grub_net_recv_ip6_packets): Likewise. (grub_net_recv_ip_packets): Likewise. * grub-core/net/net.c (grub_net_link_layer_entry): New struct. (LINK_LAYER_CACHE_SIZE): New const. (link_layer_find_entry): New function. (grub_net_link_layer_add_address): Likewise. (grub_net_link_layer_resolve_check): Likewise. (grub_net_link_layer_resolve): Likewise. (grub_net_ipv6_get_slaac): Likewise. (grub_net_ipv6_get_link_local): Likewise. (grub_cmd_ipv6_autoconf): Likewise. (parse_ip): Handle one number representation. (parse_ip6): New functoion. (match_net): Handle IPv6. (grub_net_resolve_address): Handle IPv6 and DNS. (grub_net_resolve_net_address): Handle IPv6. (route_cmp): New function. (grub_net_route_address): Find best route. (grub_net_addr_to_str): Handle IPv6. (grub_net_addr_cmp): New function. (grub_net_add_addr): Register local route. (print_net_address): Handle net address. (grub_net_poll_cards): Retransmit TCP. (grub_net_poll_cards_idle_real): Likewise. (have_ahead): New function. (grub_net_seek_real): Use underlying seek. (GRUB_MOD_INIT): Register net_ipv6_autoconf and init dns. * grub-core/net/tcp.c: New file. * grub-core/net/tftp.c (tftp_data): Add priority_queue. (cmp): New function. (ack): Likewise. (tftp_receive): Handle unordered input. (destroy_pq): New function. (tftp_close): Close pq. * grub-core/net/udp.c: Put missing license header. (grub_net_udp_socket): New function. (udp_socket_register): Likewise. (grub_net_udp_close): Likewise. (grub_net_recv_udp_packet): Check checksum. * include/grub/efi/api.h (grub_efi_simple_network): Add status. * include/grub/misc.h (grub_memchr): New function. * include/grub/net.h (GRUB_NET_*_SIZE): New enum. (grub_net_card_driver): Return buf in recv. (grub_net_slaac_mac_list): New struct. (grub_network_level_protocol_id): Add ipv6. (grub_net_network_level_addr): Likewise. (grub_net_network_level_net_addr): Likewise. (grub_net_app_protocol): Add seek. (grub_net_socket): Removed. (grub_net_sockets): Likewise. (grub_net_socket_register): Likewise. (grub_net_socket_unregister): Likewise. (FOR_NET_SOCKETS): Likewise. (grub_net_add_addr): Add const. (GRUB_NET_BOOTP_*): New enum. (grub_net_addr_cmp): New proto. (GRUB_NET_MAX_STR_ADDR_LEN): Take IPV6 into account. (GRUB_NET_MAX_STR_HWADDR_LEN): New define. (grub_net_hwaddr_to_str): NEw proto. (FOR_NET_NETWORK_LEVEL_INTERFACES): New macro. (FOR_NET_NETWORK_LEVEL_INTERFACES_SAFE): Handle NULL. (grub_dns_init): New proto. (grub_dns_fini): Likewise. (grub_net_tcp_retransmit): Likewise. (grub_net_link_layer_add_address): Likewise. (grub_net_link_layer_resolve_check): Likewise. (grub_net_link_layer_resolve): Likewise. (grub_net_dns_lookup): Likewise. (grub_net_add_dns_server): Likewise. (grub_net_remove_dns_server): Likewise. (GRUB_NET_TRIES): New const. (GRUB_NET_INTERVAL): Likewise. * include/grub/net/arp.h: Mostly rewritten. * include/grub/net/ethernet.h (grub_net_ethertype_t): New enum. * include/grub/net/ip.h: Mostly rewritten. * include/grub/net/netbuff.h: Indent. * include/grub/net/tcp.h: New file. * include/grub/net/udp.h: Mostly rewritten. * include/grub/priority_queue.h: New file. * include/grub/types.h (PRIdGRUB_SSIZE): New define. (grub_swap_bytes64_compile_time): Likewise. (grub_cpu_to_be16_compile_time): Likewise. (grub_cpu_to_be32_compile_time): Likewise. (grub_cpu_to_be64_compile_time): Likewise. (grub_be_to_cpu64_compile_time): Likewise.
700 lines
19 KiB
C
700 lines
19 KiB
C
/*
|
|
* GRUB -- GRand Unified Bootloader
|
|
* Copyright (C) 2010,2011 Free Software Foundation, Inc.
|
|
*
|
|
* GRUB is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* GRUB is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with GRUB. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <grub/net/ip.h>
|
|
#include <grub/misc.h>
|
|
#include <grub/net/arp.h>
|
|
#include <grub/net/udp.h>
|
|
#include <grub/net/ethernet.h>
|
|
#include <grub/net.h>
|
|
#include <grub/net/netbuff.h>
|
|
#include <grub/mm.h>
|
|
#include <grub/priority_queue.h>
|
|
#include <grub/time.h>
|
|
|
|
struct iphdr {
|
|
grub_uint8_t verhdrlen;
|
|
grub_uint8_t service;
|
|
grub_uint16_t len;
|
|
grub_uint16_t ident;
|
|
grub_uint16_t frags;
|
|
grub_uint8_t ttl;
|
|
grub_uint8_t protocol;
|
|
grub_uint16_t chksum;
|
|
grub_uint32_t src;
|
|
grub_uint32_t dest;
|
|
} __attribute__ ((packed)) ;
|
|
|
|
enum
|
|
{
|
|
DONT_FRAGMENT = 0x4000,
|
|
MORE_FRAGMENTS = 0x2000,
|
|
OFFSET_MASK = 0x1fff
|
|
};
|
|
|
|
typedef grub_uint64_t ip6addr[2];
|
|
|
|
struct ip6hdr {
|
|
grub_uint32_t version_class_flow;
|
|
grub_uint16_t len;
|
|
grub_uint8_t protocol;
|
|
grub_uint8_t ttl;
|
|
ip6addr src;
|
|
ip6addr dest;
|
|
} __attribute__ ((packed)) ;
|
|
|
|
static int
|
|
cmp (const void *a__, const void *b__)
|
|
{
|
|
struct grub_net_buff *a_ = *(struct grub_net_buff **) a__;
|
|
struct grub_net_buff *b_ = *(struct grub_net_buff **) b__;
|
|
struct iphdr *a = (struct iphdr *) a_->data;
|
|
struct iphdr *b = (struct iphdr *) b_->data;
|
|
/* We want the first elements to be on top. */
|
|
if ((grub_be_to_cpu16 (a->frags) & OFFSET_MASK)
|
|
< (grub_be_to_cpu16 (b->frags) & OFFSET_MASK))
|
|
return +1;
|
|
if ((grub_be_to_cpu16 (a->frags) & OFFSET_MASK)
|
|
> (grub_be_to_cpu16 (b->frags) & OFFSET_MASK))
|
|
return -1;
|
|
return 0;
|
|
}
|
|
|
|
struct reassemble
|
|
{
|
|
struct reassemble *next;
|
|
grub_uint32_t source;
|
|
grub_uint32_t dest;
|
|
grub_uint16_t id;
|
|
grub_uint8_t proto;
|
|
grub_uint64_t last_time;
|
|
grub_priority_queue_t pq;
|
|
grub_uint8_t *asm_buffer;
|
|
grub_size_t total_len;
|
|
grub_size_t cur_ptr;
|
|
grub_uint8_t ttl;
|
|
};
|
|
|
|
static struct reassemble *reassembles;
|
|
|
|
grub_uint16_t
|
|
grub_net_ip_chksum (void *ipv, grub_size_t len)
|
|
{
|
|
grub_uint16_t *ip = (grub_uint16_t *) ipv;
|
|
grub_uint32_t sum = 0;
|
|
|
|
for (; len >= 2; len -= 2)
|
|
{
|
|
sum += grub_be_to_cpu16 (grub_get_unaligned16 (ip++));
|
|
if (sum > 0xFFFF)
|
|
sum -= 0xFFFF;
|
|
}
|
|
if (len)
|
|
{
|
|
sum += *((grub_uint8_t *) ip) << 8;
|
|
if (sum > 0xFFFF)
|
|
sum -= 0xFFFF;
|
|
}
|
|
|
|
if (sum >= 0xFFFF)
|
|
sum -= 0xFFFF;
|
|
|
|
return grub_cpu_to_be16 ((~sum) & 0x0000FFFF);
|
|
}
|
|
|
|
static int id = 0x2400;
|
|
|
|
static grub_err_t
|
|
send_fragmented (struct grub_net_network_level_interface * inf,
|
|
const grub_net_network_level_address_t * target,
|
|
struct grub_net_buff * nb,
|
|
grub_net_ip_protocol_t proto,
|
|
grub_net_link_level_address_t ll_target_addr)
|
|
{
|
|
grub_size_t off = 0;
|
|
grub_size_t fraglen;
|
|
grub_err_t err;
|
|
|
|
fraglen = (inf->card->mtu - sizeof (struct iphdr)) & ~7;
|
|
id++;
|
|
|
|
while (nb->tail - nb->data)
|
|
{
|
|
grub_size_t len = fraglen;
|
|
struct grub_net_buff *nb2;
|
|
struct iphdr *iph;
|
|
|
|
if ((grub_ssize_t) len > nb->tail - nb->data)
|
|
len = nb->tail - nb->data;
|
|
nb2 = grub_netbuff_alloc (fraglen + sizeof (struct iphdr)
|
|
+ GRUB_NET_MAX_LINK_HEADER_SIZE);
|
|
if (!nb2)
|
|
return grub_errno;
|
|
err = grub_netbuff_reserve (nb2, GRUB_NET_MAX_LINK_HEADER_SIZE);
|
|
if (err)
|
|
return err;
|
|
err = grub_netbuff_put (nb2, sizeof (struct iphdr));
|
|
if (err)
|
|
return err;
|
|
|
|
iph = (struct iphdr *) nb2->data;
|
|
iph->verhdrlen = ((4 << 4) | 5);
|
|
iph->service = 0;
|
|
iph->len = grub_cpu_to_be16 (len + sizeof (struct iphdr));
|
|
iph->ident = grub_cpu_to_be16 (id);
|
|
iph->frags = grub_cpu_to_be16 (off | (((grub_ssize_t) len
|
|
== nb->tail - nb->data)
|
|
? 0 : MORE_FRAGMENTS));
|
|
iph->ttl = 0xff;
|
|
iph->protocol = proto;
|
|
iph->src = inf->address.ipv4;
|
|
iph->dest = target->ipv4;
|
|
off += len / 8;
|
|
|
|
iph->chksum = 0;
|
|
iph->chksum = grub_net_ip_chksum ((void *) nb2->data, sizeof (*iph));
|
|
err = grub_netbuff_put (nb2, len);
|
|
if (err)
|
|
return err;
|
|
grub_memcpy (iph + 1, nb->data, len);
|
|
err = grub_netbuff_pull (nb, len);
|
|
if (err)
|
|
return err;
|
|
err = send_ethernet_packet (inf, nb2, ll_target_addr,
|
|
GRUB_NET_ETHERTYPE_IP);
|
|
if (err)
|
|
return err;
|
|
}
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
|
|
static grub_err_t
|
|
grub_net_send_ip4_packet (struct grub_net_network_level_interface *inf,
|
|
const grub_net_network_level_address_t *target,
|
|
const grub_net_link_level_address_t *ll_target_addr,
|
|
struct grub_net_buff *nb,
|
|
grub_net_ip_protocol_t proto)
|
|
{
|
|
struct iphdr *iph;
|
|
|
|
COMPILE_TIME_ASSERT (GRUB_NET_OUR_IPV4_HEADER_SIZE == sizeof (*iph));
|
|
|
|
if (nb->tail - nb->data + sizeof (struct iphdr) > inf->card->mtu)
|
|
return send_fragmented (inf, target, nb, proto, *ll_target_addr);
|
|
|
|
grub_netbuff_push (nb, sizeof (*iph));
|
|
iph = (struct iphdr *) nb->data;
|
|
|
|
iph->verhdrlen = ((4 << 4) | 5);
|
|
iph->service = 0;
|
|
iph->len = grub_cpu_to_be16 (nb->tail - nb->data);
|
|
iph->ident = grub_cpu_to_be16 (++id);
|
|
iph->frags = 0;
|
|
iph->ttl = 0xff;
|
|
iph->protocol = proto;
|
|
iph->src = inf->address.ipv4;
|
|
iph->dest = target->ipv4;
|
|
|
|
iph->chksum = 0;
|
|
iph->chksum = grub_net_ip_chksum ((void *) nb->data, sizeof (*iph));
|
|
|
|
return send_ethernet_packet (inf, nb, *ll_target_addr,
|
|
GRUB_NET_ETHERTYPE_IP);
|
|
}
|
|
|
|
static grub_err_t
|
|
handle_dgram (struct grub_net_buff *nb,
|
|
struct grub_net_card *card,
|
|
const grub_net_link_level_address_t *source_hwaddress,
|
|
const grub_net_link_level_address_t *hwaddress,
|
|
grub_net_ip_protocol_t proto,
|
|
const grub_net_network_level_address_t *source,
|
|
const grub_net_network_level_address_t *dest,
|
|
grub_uint8_t ttl)
|
|
{
|
|
struct grub_net_network_level_interface *inf = NULL;
|
|
grub_err_t err;
|
|
int multicast = 0;
|
|
|
|
/* DHCP needs special treatment since we don't know IP yet. */
|
|
{
|
|
struct udphdr *udph;
|
|
udph = (struct udphdr *) nb->data;
|
|
if (proto == GRUB_NET_IP_UDP && grub_be_to_cpu16 (udph->dst) == 68)
|
|
{
|
|
FOR_NET_NETWORK_LEVEL_INTERFACES (inf)
|
|
if (inf->card == card
|
|
&& inf->address.type == GRUB_NET_NETWORK_LEVEL_PROTOCOL_DHCP_RECV
|
|
&& grub_net_hwaddr_cmp (&inf->hwaddress, hwaddress) == 0)
|
|
{
|
|
if (udph->chksum)
|
|
{
|
|
grub_uint16_t chk, expected;
|
|
chk = udph->chksum;
|
|
udph->chksum = 0;
|
|
expected = grub_net_ip_transport_checksum (nb,
|
|
GRUB_NET_IP_UDP,
|
|
source,
|
|
dest);
|
|
if (expected != chk)
|
|
{
|
|
grub_dprintf ("net", "Invalid UDP checksum. "
|
|
"Expected %x, got %x\n",
|
|
grub_be_to_cpu16 (expected),
|
|
grub_be_to_cpu16 (chk));
|
|
grub_netbuff_free (nb);
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
udph->chksum = chk;
|
|
}
|
|
|
|
err = grub_netbuff_pull (nb, sizeof (*udph));
|
|
if (err)
|
|
return err;
|
|
grub_net_process_dhcp (nb, inf->card);
|
|
grub_netbuff_free (nb);
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
grub_netbuff_free (nb);
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
}
|
|
|
|
FOR_NET_NETWORK_LEVEL_INTERFACES (inf)
|
|
{
|
|
if (inf->card == card
|
|
&& grub_net_addr_cmp (&inf->address, dest) == 0
|
|
&& grub_net_hwaddr_cmp (&inf->hwaddress, hwaddress) == 0)
|
|
break;
|
|
/* Solicited node multicast. */
|
|
if (inf->card == card
|
|
&& inf->address.type == GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6
|
|
&& dest->type == GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6
|
|
&& dest->ipv6[0] == grub_be_to_cpu64_compile_time (0xff02ULL << 48)
|
|
&& dest->ipv6[1] == (grub_be_to_cpu64_compile_time (0x01ff000000ULL)
|
|
| (inf->address.ipv6[1]
|
|
& grub_be_to_cpu64_compile_time (0xffffff)))
|
|
&& hwaddress->type == GRUB_NET_LINK_LEVEL_PROTOCOL_ETHERNET
|
|
&& hwaddress->mac[0] == 0x33 && hwaddress->mac[1] == 0x33
|
|
&& hwaddress->mac[2] == 0xff
|
|
&& hwaddress->mac[3] == ((grub_be_to_cpu64 (inf->address.ipv6[1])
|
|
>> 16) & 0xff)
|
|
&& hwaddress->mac[4] == ((grub_be_to_cpu64 (inf->address.ipv6[1])
|
|
>> 8) & 0xff)
|
|
&& hwaddress->mac[5] == ((grub_be_to_cpu64 (inf->address.ipv6[1])
|
|
>> 0) & 0xff))
|
|
{
|
|
multicast = 1;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!inf && !(dest->type == GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6
|
|
&& dest->ipv6[0] == grub_be_to_cpu64_compile_time (0xff02ULL
|
|
<< 48)
|
|
&& dest->ipv6[1] == grub_be_to_cpu64_compile_time (1)))
|
|
{
|
|
grub_netbuff_free (nb);
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
if (multicast)
|
|
inf = NULL;
|
|
|
|
switch (proto)
|
|
{
|
|
case GRUB_NET_IP_UDP:
|
|
return grub_net_recv_udp_packet (nb, inf, source);
|
|
case GRUB_NET_IP_TCP:
|
|
return grub_net_recv_tcp_packet (nb, inf, source);
|
|
case GRUB_NET_IP_ICMP:
|
|
return grub_net_recv_icmp_packet (nb, inf, source_hwaddress, source);
|
|
case GRUB_NET_IP_ICMPV6:
|
|
return grub_net_recv_icmp6_packet (nb, card, inf, source_hwaddress,
|
|
source, dest, ttl);
|
|
default:
|
|
grub_netbuff_free (nb);
|
|
break;
|
|
}
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
|
|
static void
|
|
free_rsm (struct reassemble *rsm)
|
|
{
|
|
struct grub_net_buff **nb;
|
|
while ((nb = grub_priority_queue_top (rsm->pq)))
|
|
{
|
|
grub_netbuff_free (*nb);
|
|
grub_priority_queue_pop (rsm->pq);
|
|
}
|
|
grub_free (rsm->asm_buffer);
|
|
grub_priority_queue_destroy (rsm->pq);
|
|
}
|
|
|
|
static void
|
|
free_old_fragments (void)
|
|
{
|
|
struct reassemble *rsm, **prev;
|
|
grub_uint64_t limit_time = grub_get_time_ms () - 90000;
|
|
|
|
for (prev = &reassembles, rsm = *prev; rsm; prev = &rsm->next, rsm = *prev)
|
|
if (rsm->last_time < limit_time)
|
|
{
|
|
*prev = rsm->next;
|
|
free_rsm (rsm);
|
|
}
|
|
}
|
|
|
|
static grub_err_t
|
|
grub_net_recv_ip4_packets (struct grub_net_buff *nb,
|
|
struct grub_net_card *card,
|
|
const grub_net_link_level_address_t *hwaddress,
|
|
const grub_net_link_level_address_t *src_hwaddress)
|
|
{
|
|
struct iphdr *iph = (struct iphdr *) nb->data;
|
|
grub_err_t err;
|
|
struct reassemble *rsm, **prev;
|
|
|
|
if ((iph->verhdrlen >> 4) != 4)
|
|
{
|
|
grub_dprintf ("net", "Bad IP version: %d\n", (iph->verhdrlen >> 4));
|
|
grub_netbuff_free (nb);
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
|
|
if ((iph->verhdrlen & 0xf) < 5)
|
|
{
|
|
grub_dprintf ("net", "IP header too short: %d\n",
|
|
(iph->verhdrlen & 0xf));
|
|
grub_netbuff_free (nb);
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
|
|
if (nb->tail - nb->data < (grub_ssize_t) ((iph->verhdrlen & 0xf)
|
|
* sizeof (grub_uint32_t)))
|
|
{
|
|
grub_dprintf ("net", "IP packet too short: %" PRIdGRUB_SSIZE "\n",
|
|
(nb->tail - nb->data));
|
|
grub_netbuff_free (nb);
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
|
|
/* Check size. */
|
|
{
|
|
grub_size_t expected_size = grub_be_to_cpu16 (iph->len);
|
|
grub_size_t actual_size = (nb->tail - nb->data);
|
|
if (actual_size > expected_size)
|
|
{
|
|
err = grub_netbuff_unput (nb, actual_size - expected_size);
|
|
if (err)
|
|
{
|
|
grub_netbuff_free (nb);
|
|
return err;
|
|
}
|
|
}
|
|
if (actual_size < expected_size)
|
|
{
|
|
grub_dprintf ("net", "Cut IP packet actual: %" PRIuGRUB_SIZE
|
|
", expected %" PRIuGRUB_SIZE "\n", actual_size,
|
|
expected_size);
|
|
grub_netbuff_free (nb);
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
}
|
|
|
|
/* Unfragmented packet. Good. */
|
|
if (((grub_be_to_cpu16 (iph->frags) & MORE_FRAGMENTS) == 0)
|
|
&& (grub_be_to_cpu16 (iph->frags) & OFFSET_MASK) == 0)
|
|
{
|
|
grub_net_network_level_address_t source;
|
|
grub_net_network_level_address_t dest;
|
|
|
|
err = grub_netbuff_pull (nb, ((iph->verhdrlen & 0xf)
|
|
* sizeof (grub_uint32_t)));
|
|
if (err)
|
|
{
|
|
grub_netbuff_free (nb);
|
|
return err;
|
|
}
|
|
|
|
source.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4;
|
|
source.ipv4 = iph->src;
|
|
|
|
dest.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4;
|
|
dest.ipv4 = iph->dest;
|
|
|
|
return handle_dgram (nb, card, src_hwaddress, hwaddress, iph->protocol,
|
|
&source, &dest, iph->ttl);
|
|
}
|
|
|
|
for (prev = &reassembles, rsm = *prev; rsm; prev = &rsm->next, rsm = *prev)
|
|
if (rsm->source == iph->src && rsm->dest == iph->dest
|
|
&& rsm->id == iph->ident && rsm->proto == iph->protocol)
|
|
break;
|
|
if (!rsm)
|
|
{
|
|
rsm = grub_malloc (sizeof (*rsm));
|
|
if (!rsm)
|
|
return grub_errno;
|
|
rsm->source = iph->src;
|
|
rsm->dest = iph->dest;
|
|
rsm->id = iph->ident;
|
|
rsm->proto = iph->protocol;
|
|
rsm->next = reassembles;
|
|
reassembles = rsm;
|
|
prev = &reassembles;
|
|
rsm->pq = grub_priority_queue_new (sizeof (struct grub_net_buff **), cmp);
|
|
if (!rsm->pq)
|
|
{
|
|
grub_free (rsm);
|
|
return grub_errno;
|
|
}
|
|
rsm->asm_buffer = 0;
|
|
rsm->total_len = 0;
|
|
rsm->cur_ptr = 0;
|
|
rsm->ttl = 0xff;
|
|
}
|
|
if (rsm->ttl > iph->ttl)
|
|
rsm->ttl = iph->ttl;
|
|
rsm->last_time = grub_get_time_ms ();
|
|
free_old_fragments ();
|
|
|
|
err = grub_priority_queue_push (rsm->pq, &nb);
|
|
if (err)
|
|
return err;
|
|
|
|
if (!(grub_be_to_cpu16 (iph->frags) & MORE_FRAGMENTS))
|
|
{
|
|
rsm->total_len = (8 * (grub_be_to_cpu16 (iph->frags) & OFFSET_MASK)
|
|
+ (nb->tail - nb->data));
|
|
rsm->total_len -= ((iph->verhdrlen & 0xf) * sizeof (grub_uint32_t));
|
|
rsm->asm_buffer = grub_zalloc (rsm->total_len);
|
|
if (!rsm->asm_buffer)
|
|
{
|
|
*prev = rsm->next;
|
|
free_rsm (rsm);
|
|
return grub_errno;
|
|
}
|
|
}
|
|
if (!rsm->asm_buffer)
|
|
return GRUB_ERR_NONE;
|
|
|
|
while (1)
|
|
{
|
|
struct grub_net_buff **nb_top_p, *nb_top;
|
|
grub_size_t copy;
|
|
grub_uint8_t *res;
|
|
grub_size_t res_len;
|
|
struct grub_net_buff *ret;
|
|
grub_net_ip_protocol_t proto;
|
|
grub_uint32_t src;
|
|
grub_uint32_t dst;
|
|
grub_net_network_level_address_t source;
|
|
grub_net_network_level_address_t dest;
|
|
grub_uint8_t ttl;
|
|
|
|
nb_top_p = grub_priority_queue_top (rsm->pq);
|
|
if (!nb_top_p)
|
|
return GRUB_ERR_NONE;
|
|
nb_top = *nb_top_p;
|
|
grub_priority_queue_pop (rsm->pq);
|
|
iph = (struct iphdr *) nb_top->data;
|
|
err = grub_netbuff_pull (nb_top, ((iph->verhdrlen & 0xf)
|
|
* sizeof (grub_uint32_t)));
|
|
if (err)
|
|
{
|
|
grub_netbuff_free (nb_top);
|
|
return err;
|
|
}
|
|
if (rsm->cur_ptr < (grub_size_t) 8 * (grub_be_to_cpu16 (iph->frags)
|
|
& OFFSET_MASK))
|
|
return GRUB_ERR_NONE;
|
|
|
|
rsm->cur_ptr = (8 * (grub_be_to_cpu16 (iph->frags) & OFFSET_MASK)
|
|
+ (nb_top->tail - nb_top->head));
|
|
if ((grub_size_t) 8 * (grub_be_to_cpu16 (iph->frags) & OFFSET_MASK)
|
|
>= rsm->total_len)
|
|
{
|
|
grub_netbuff_free (nb_top);
|
|
continue;
|
|
}
|
|
copy = nb_top->tail - nb_top->data;
|
|
if (rsm->total_len - 8 * (grub_be_to_cpu16 (iph->frags) & OFFSET_MASK)
|
|
< copy)
|
|
copy = rsm->total_len - 8 * (grub_be_to_cpu16 (iph->frags)
|
|
& OFFSET_MASK);
|
|
grub_memcpy (&rsm->asm_buffer[8 * (grub_be_to_cpu16 (iph->frags)
|
|
& OFFSET_MASK)],
|
|
nb_top->data, copy);
|
|
|
|
if ((grub_be_to_cpu16 (iph->frags) & MORE_FRAGMENTS))
|
|
continue;
|
|
|
|
res = rsm->asm_buffer;
|
|
proto = rsm->proto;
|
|
src = rsm->source;
|
|
dst = rsm->dest;
|
|
ttl = rsm->ttl;
|
|
|
|
rsm->asm_buffer = 0;
|
|
res_len = rsm->total_len;
|
|
*prev = rsm->next;
|
|
free_rsm (rsm);
|
|
ret = grub_malloc (sizeof (*ret));
|
|
if (!ret)
|
|
{
|
|
grub_free (res);
|
|
return grub_errno;
|
|
}
|
|
ret->data = ret->head = res;
|
|
ret->tail = ret->end = res + res_len;
|
|
|
|
source.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4;
|
|
source.ipv4 = src;
|
|
|
|
dest.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4;
|
|
dest.ipv4 = dst;
|
|
|
|
return handle_dgram (ret, card, src_hwaddress,
|
|
hwaddress, proto, &source, &dest,
|
|
ttl);
|
|
}
|
|
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
|
|
static grub_err_t
|
|
grub_net_send_ip6_packet (struct grub_net_network_level_interface *inf,
|
|
const grub_net_network_level_address_t *target,
|
|
const grub_net_link_level_address_t *ll_target_addr,
|
|
struct grub_net_buff *nb,
|
|
grub_net_ip_protocol_t proto)
|
|
{
|
|
struct ip6hdr *iph;
|
|
|
|
COMPILE_TIME_ASSERT (GRUB_NET_OUR_IPV6_HEADER_SIZE == sizeof (*iph));
|
|
|
|
if (nb->tail - nb->data + sizeof (struct iphdr) > inf->card->mtu)
|
|
return grub_error (GRUB_ERR_NET_PACKET_TOO_BIG, "packet too big");
|
|
|
|
grub_netbuff_push (nb, sizeof (*iph));
|
|
iph = (struct ip6hdr *) nb->data;
|
|
|
|
iph->version_class_flow = grub_cpu_to_be32 ((6 << 28));
|
|
iph->len = grub_cpu_to_be16 (nb->tail - nb->data - sizeof (*iph));
|
|
iph->protocol = proto;
|
|
iph->ttl = 0xff;
|
|
grub_memcpy (&iph->src, inf->address.ipv6, sizeof (iph->src));
|
|
grub_memcpy (&iph->dest, target->ipv6, sizeof (iph->dest));
|
|
|
|
return send_ethernet_packet (inf, nb, *ll_target_addr,
|
|
GRUB_NET_ETHERTYPE_IP6);
|
|
}
|
|
|
|
grub_err_t
|
|
grub_net_send_ip_packet (struct grub_net_network_level_interface *inf,
|
|
const grub_net_network_level_address_t *target,
|
|
const grub_net_link_level_address_t *ll_target_addr,
|
|
struct grub_net_buff *nb,
|
|
grub_net_ip_protocol_t proto)
|
|
{
|
|
switch (target->type)
|
|
{
|
|
case GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4:
|
|
return grub_net_send_ip4_packet (inf, target, ll_target_addr, nb, proto);
|
|
case GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6:
|
|
return grub_net_send_ip6_packet (inf, target, ll_target_addr, nb, proto);
|
|
default:
|
|
return grub_error (GRUB_ERR_BAD_ARGUMENT, "not an IP");
|
|
}
|
|
}
|
|
|
|
static grub_err_t
|
|
grub_net_recv_ip6_packets (struct grub_net_buff *nb,
|
|
struct grub_net_card *card,
|
|
const grub_net_link_level_address_t *hwaddress,
|
|
const grub_net_link_level_address_t *src_hwaddress)
|
|
{
|
|
struct ip6hdr *iph = (struct ip6hdr *) nb->data;
|
|
grub_err_t err;
|
|
grub_net_network_level_address_t source;
|
|
grub_net_network_level_address_t dest;
|
|
|
|
if (nb->tail - nb->data < (grub_ssize_t) sizeof (*iph))
|
|
{
|
|
grub_dprintf ("net", "IP packet too short: %" PRIdGRUB_SSIZE "\n",
|
|
nb->tail - nb->data);
|
|
grub_netbuff_free (nb);
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
|
|
err = grub_netbuff_pull (nb, sizeof (*iph));
|
|
if (err)
|
|
{
|
|
grub_netbuff_free (nb);
|
|
return err;
|
|
}
|
|
|
|
/* Check size. */
|
|
{
|
|
grub_size_t expected_size = grub_be_to_cpu16 (iph->len);
|
|
grub_size_t actual_size = (nb->tail - nb->data);
|
|
if (actual_size > expected_size)
|
|
{
|
|
err = grub_netbuff_unput (nb, actual_size - expected_size);
|
|
if (err)
|
|
{
|
|
grub_netbuff_free (nb);
|
|
return err;
|
|
}
|
|
}
|
|
if (actual_size < expected_size)
|
|
{
|
|
grub_dprintf ("net", "Cut IP packet actual: %" PRIuGRUB_SIZE
|
|
", expected %" PRIuGRUB_SIZE "\n", actual_size,
|
|
expected_size);
|
|
grub_netbuff_free (nb);
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
}
|
|
|
|
source.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6;
|
|
dest.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6;
|
|
grub_memcpy (source.ipv6, &iph->src, sizeof (source.ipv6));
|
|
grub_memcpy (dest.ipv6, &iph->dest, sizeof (dest.ipv6));
|
|
|
|
return handle_dgram (nb, card, src_hwaddress, hwaddress, iph->protocol,
|
|
&source, &dest, iph->ttl);
|
|
}
|
|
|
|
grub_err_t
|
|
grub_net_recv_ip_packets (struct grub_net_buff *nb,
|
|
struct grub_net_card *card,
|
|
const grub_net_link_level_address_t *hwaddress,
|
|
const grub_net_link_level_address_t *src_hwaddress)
|
|
{
|
|
struct iphdr *iph = (struct iphdr *) nb->data;
|
|
|
|
if ((iph->verhdrlen >> 4) == 4)
|
|
return grub_net_recv_ip4_packets (nb, card, hwaddress, src_hwaddress);
|
|
if ((iph->verhdrlen >> 4) == 6)
|
|
return grub_net_recv_ip6_packets (nb, card, hwaddress, src_hwaddress);
|
|
grub_dprintf ("net", "Bad IP version: %d\n", (iph->verhdrlen >> 4));
|
|
grub_netbuff_free (nb);
|
|
return GRUB_ERR_NONE;
|
|
}
|