/* * GRUB -- GRand Unified Bootloader * Copyright (C) 2010,2011,2012,2013 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 . */ #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include GRUB_MOD_LICENSE ("GPLv3+"); char *grub_net_default_server; struct grub_net_route { struct grub_net_route *next; struct grub_net_route **prev; grub_net_network_level_netaddress_t target; char *name; struct grub_net_network_level_protocol *prot; int is_gateway; union { struct grub_net_network_level_interface *interface; grub_net_network_level_address_t gw; }; }; struct grub_net_route *grub_net_routes = NULL; struct grub_net_network_level_interface *grub_net_network_level_interfaces = NULL; struct grub_net_card *grub_net_cards = NULL; struct grub_net_network_level_protocol *grub_net_network_level_protocols = NULL; static struct grub_fs grub_net_fs; struct grub_net_link_layer_entry { int avail; grub_net_network_level_address_t nl_address; grub_net_link_level_address_t ll_address; }; #define LINK_LAYER_CACHE_SIZE 256 static struct grub_net_link_layer_entry * link_layer_find_entry (const grub_net_network_level_address_t *proto, const struct grub_net_card *card) { unsigned i; if (!card->link_layer_table) return NULL; for (i = 0; i < LINK_LAYER_CACHE_SIZE; i++) { if (card->link_layer_table[i].avail == 1 && grub_net_addr_cmp (&card->link_layer_table[i].nl_address, proto) == 0) return &card->link_layer_table[i]; } return NULL; } void grub_net_link_layer_add_address (struct grub_net_card *card, const grub_net_network_level_address_t *nl, const grub_net_link_level_address_t *ll, int override) { struct grub_net_link_layer_entry *entry; /* Check if the sender is in the cache table. */ entry = link_layer_find_entry (nl, card); /* Update sender hardware address. */ if (entry && override) grub_memcpy (&entry->ll_address, ll, sizeof (entry->ll_address)); if (entry) return; /* Add sender to cache table. */ if (card->link_layer_table == NULL) card->link_layer_table = grub_zalloc (LINK_LAYER_CACHE_SIZE * sizeof (card->link_layer_table[0])); entry = &(card->link_layer_table[card->new_ll_entry]); entry->avail = 1; grub_memcpy (&entry->ll_address, ll, sizeof (entry->ll_address)); grub_memcpy (&entry->nl_address, nl, sizeof (entry->nl_address)); card->new_ll_entry++; if (card->new_ll_entry == LINK_LAYER_CACHE_SIZE) card->new_ll_entry = 0; } int grub_net_link_layer_resolve_check (struct grub_net_network_level_interface *inf, const grub_net_network_level_address_t *proto_addr) { struct grub_net_link_layer_entry *entry; if (proto_addr->type == GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4 && proto_addr->ipv4 == 0xffffffff) return 1; entry = link_layer_find_entry (proto_addr, inf->card); if (entry) return 1; return 0; } grub_err_t grub_net_link_layer_resolve (struct grub_net_network_level_interface *inf, const grub_net_network_level_address_t *proto_addr, grub_net_link_level_address_t *hw_addr) { struct grub_net_link_layer_entry *entry; grub_err_t err; if ((proto_addr->type == GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4 && proto_addr->ipv4 == 0xffffffff) || proto_addr->type == GRUB_NET_NETWORK_LEVEL_PROTOCOL_DHCP_RECV || (proto_addr->type == GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6 && proto_addr->ipv6[0] == grub_be_to_cpu64_compile_time (0xff02ULL << 48) && proto_addr->ipv6[1] == (grub_be_to_cpu64_compile_time (1)))) { hw_addr->type = GRUB_NET_LINK_LEVEL_PROTOCOL_ETHERNET; grub_memset (hw_addr->mac, -1, 6); return GRUB_ERR_NONE; } if (proto_addr->type == GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6 && ((grub_be_to_cpu64 (proto_addr->ipv6[0]) >> 56) == 0xff)) { hw_addr->type = GRUB_NET_LINK_LEVEL_PROTOCOL_ETHERNET; hw_addr->mac[0] = 0x33; hw_addr->mac[1] = 0x33; hw_addr->mac[2] = ((grub_be_to_cpu64 (proto_addr->ipv6[1]) >> 24) & 0xff); hw_addr->mac[3] = ((grub_be_to_cpu64 (proto_addr->ipv6[1]) >> 16) & 0xff); hw_addr->mac[4] = ((grub_be_to_cpu64 (proto_addr->ipv6[1]) >> 8) & 0xff); hw_addr->mac[5] = ((grub_be_to_cpu64 (proto_addr->ipv6[1]) >> 0) & 0xff); return GRUB_ERR_NONE; } /* Check cache table. */ entry = link_layer_find_entry (proto_addr, inf->card); if (entry) { *hw_addr = entry->ll_address; return GRUB_ERR_NONE; } switch (proto_addr->type) { case GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4: err = grub_net_arp_send_request (inf, proto_addr); break; case GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6: err = grub_net_icmp6_send_request (inf, proto_addr); break; case GRUB_NET_NETWORK_LEVEL_PROTOCOL_DHCP_RECV: return grub_error (GRUB_ERR_BUG, "shouldn't reach here"); default: return grub_error (GRUB_ERR_BUG, "unsupported address type %d", proto_addr->type); } if (err) return err; entry = link_layer_find_entry (proto_addr, inf->card); if (entry) { *hw_addr = entry->ll_address; return GRUB_ERR_NONE; } return grub_error (GRUB_ERR_TIMEOUT, N_("timeout: could not resolve hardware address")); } void grub_net_card_unregister (struct grub_net_card *card) { struct grub_net_network_level_interface *inf, *next; FOR_NET_NETWORK_LEVEL_INTERFACES_SAFE(inf, next) if (inf->card == card) grub_net_network_level_interface_unregister (inf); if (card->opened) { if (card->driver->close) card->driver->close (card); card->opened = 0; } grub_list_remove (GRUB_AS_LIST (card)); } static struct grub_net_slaac_mac_list * grub_net_ipv6_get_slaac (struct grub_net_card *card, const grub_net_link_level_address_t *hwaddr) { struct grub_net_slaac_mac_list *slaac; char *ptr; for (slaac = card->slaac_list; slaac; slaac = slaac->next) if (grub_net_hwaddr_cmp (&slaac->address, hwaddr) == 0) return slaac; slaac = grub_zalloc (sizeof (*slaac)); if (!slaac) return NULL; slaac->name = grub_malloc (grub_strlen (card->name) + GRUB_NET_MAX_STR_HWADDR_LEN + sizeof (":slaac")); ptr = grub_stpcpy (slaac->name, card->name); if (grub_net_hwaddr_cmp (&card->default_address, hwaddr) != 0) { ptr = grub_stpcpy (ptr, ":"); grub_net_hwaddr_to_str (hwaddr, ptr); ptr += grub_strlen (ptr); } ptr = grub_stpcpy (ptr, ":slaac"); grub_memcpy (&slaac->address, hwaddr, sizeof (slaac->address)); slaac->next = card->slaac_list; card->slaac_list = slaac; return slaac; } static void grub_net_network_level_interface_register (struct grub_net_network_level_interface *inter); static struct grub_net_network_level_interface * grub_net_add_addr_real (char *name, struct grub_net_card *card, const grub_net_network_level_address_t *addr, const grub_net_link_level_address_t *hwaddress, grub_net_interface_flags_t flags) { struct grub_net_network_level_interface *inter; inter = grub_zalloc (sizeof (*inter)); if (!inter) return NULL; inter->name = name; grub_memcpy (&(inter->address), addr, sizeof (inter->address)); grub_memcpy (&(inter->hwaddress), hwaddress, sizeof (inter->hwaddress)); inter->flags = flags; inter->card = card; inter->dhcp_ack = NULL; inter->dhcp_acklen = 0; grub_net_network_level_interface_register (inter); return inter; } struct grub_net_network_level_interface * grub_net_add_addr (const char *name, struct grub_net_card *card, const grub_net_network_level_address_t *addr, const grub_net_link_level_address_t *hwaddress, grub_net_interface_flags_t flags) { char *name_dup = grub_strdup (name); struct grub_net_network_level_interface *ret; if (!name_dup) return NULL; ret = grub_net_add_addr_real (name_dup, card, addr, hwaddress, flags); if (!ret) grub_free (name_dup); return ret; } struct grub_net_network_level_interface * grub_net_ipv6_get_link_local (struct grub_net_card *card, const grub_net_link_level_address_t *hwaddr) { struct grub_net_network_level_interface *inf; char *name; char *ptr; grub_net_network_level_address_t addr; name = grub_malloc (grub_strlen (card->name) + GRUB_NET_MAX_STR_HWADDR_LEN + sizeof (":link")); if (!name) return NULL; addr.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6; addr.ipv6[0] = grub_cpu_to_be64_compile_time (0xfe80ULL << 48); addr.ipv6[1] = grub_net_ipv6_get_id (hwaddr); FOR_NET_NETWORK_LEVEL_INTERFACES (inf) { if (inf->card == card && grub_net_hwaddr_cmp (&inf->hwaddress, hwaddr) == 0 && grub_net_addr_cmp (&inf->address, &addr) == 0) return inf; } ptr = grub_stpcpy (name, card->name); if (grub_net_hwaddr_cmp (&card->default_address, hwaddr) != 0) { ptr = grub_stpcpy (ptr, ":"); grub_net_hwaddr_to_str (hwaddr, ptr); ptr += grub_strlen (ptr); } ptr = grub_stpcpy (ptr, ":link"); return grub_net_add_addr_real (name, card, &addr, hwaddr, 0); } /* FIXME: allow to specify mac address. */ static grub_err_t grub_cmd_ipv6_autoconf (struct grub_command *cmd __attribute__ ((unused)), int argc, char **args) { struct grub_net_card *card; struct grub_net_network_level_interface **ifaces; grub_size_t ncards = 0; unsigned j = 0; int interval; grub_err_t err; struct grub_net_slaac_mac_list **slaacs; FOR_NET_CARDS (card) { if (argc > 0 && grub_strcmp (card->name, args[0]) != 0) continue; ncards++; } ifaces = grub_zalloc (ncards * sizeof (ifaces[0])); slaacs = grub_zalloc (ncards * sizeof (slaacs[0])); if (!ifaces || !slaacs) { grub_free (ifaces); grub_free (slaacs); return grub_errno; } FOR_NET_CARDS (card) { if (argc > 0 && grub_strcmp (card->name, args[0]) != 0) continue; ifaces[j] = grub_net_ipv6_get_link_local (card, &card->default_address); if (!ifaces[j]) { grub_free (ifaces); grub_free (slaacs); return grub_errno; } slaacs[j] = grub_net_ipv6_get_slaac (card, &card->default_address); if (!slaacs[j]) { grub_free (ifaces); grub_free (slaacs); return grub_errno; } j++; } for (interval = 200; interval < 10000; interval *= 2) { int done = 1; for (j = 0; j < ncards; j++) { if (slaacs[j]->slaac_counter) continue; err = grub_net_icmp6_send_router_solicit (ifaces[j]); if (err) err = GRUB_ERR_NONE; done = 0; } if (done) break; grub_net_poll_cards (interval, 0); } err = GRUB_ERR_NONE; for (j = 0; j < ncards; j++) { if (slaacs[j]->slaac_counter) continue; err = grub_error (GRUB_ERR_FILE_NOT_FOUND, N_("couldn't autoconfigure %s"), ifaces[j]->card->name); } grub_free (ifaces); grub_free (slaacs); return err; } static inline void grub_net_route_register (struct grub_net_route *route) { grub_list_push (GRUB_AS_LIST_P (&grub_net_routes), GRUB_AS_LIST (route)); } #define FOR_NET_ROUTES(var) for (var = grub_net_routes; var; var = var->next) static int parse_ip (const char *val, grub_uint32_t *ip, const char **rest) { grub_uint32_t newip = 0; int i; const char *ptr = val; for (i = 0; i < 4; i++) { unsigned long t; t = grub_strtoul (ptr, (char **) &ptr, 0); if (grub_errno) { grub_errno = GRUB_ERR_NONE; return 0; } if (*ptr != '.' && i == 0) { newip = t; break; } if (t & ~0xff) return 0; newip >>= 8; newip |= (t << 24); if (i != 3 && *ptr != '.') return 0; ptr++; } *ip = grub_cpu_to_le32 (newip); if (rest) *rest = (ptr - 1); return 1; } static int parse_ip6 (const char *val, grub_uint64_t *ip, const char **rest) { grub_uint16_t newip[8]; const char *ptr = val; int word, quaddot = -1; if (ptr[0] == ':' && ptr[1] != ':') return 0; if (ptr[0] == ':') ptr++; for (word = 0; word < 8; word++) { unsigned long t; if (*ptr == ':') { quaddot = word; word--; ptr++; continue; } t = grub_strtoul (ptr, (char **) &ptr, 16); if (grub_errno) { grub_errno = GRUB_ERR_NONE; break; } if (t & ~0xffff) return 0; newip[word] = grub_cpu_to_be16 (t); if (*ptr != ':') break; ptr++; } if (quaddot == -1 && word < 7) return 0; if (quaddot != -1) { grub_memmove (&newip[quaddot + 7 - word], &newip[quaddot], (word - quaddot + 1) * sizeof (newip[0])); grub_memset (&newip[quaddot], 0, (7 - word) * sizeof (newip[0])); } grub_memcpy (ip, newip, 16); if (rest) *rest = ptr; return 1; } static int match_net (const grub_net_network_level_netaddress_t *net, const grub_net_network_level_address_t *addr) { if (net->type != addr->type) return 0; switch (net->type) { case GRUB_NET_NETWORK_LEVEL_PROTOCOL_DHCP_RECV: return 0; case GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4: { grub_uint32_t mask = (0xffffffffU << (32 - net->ipv4.masksize)); if (net->ipv4.masksize == 0) mask = 0; return ((grub_be_to_cpu32 (net->ipv4.base) & mask) == (grub_be_to_cpu32 (addr->ipv4) & mask)); } case GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6: { grub_uint64_t mask[2]; if (net->ipv6.masksize <= 64) { mask[0] = 0xffffffffffffffffULL << (64 - net->ipv6.masksize); mask[1] = 0; } else { mask[0] = 0xffffffffffffffffULL; mask[1] = 0xffffffffffffffffULL << (128 - net->ipv6.masksize); } return (((grub_be_to_cpu64 (net->ipv6.base[0]) & mask[0]) == (grub_be_to_cpu64 (addr->ipv6[0]) & mask[0])) && ((grub_be_to_cpu64 (net->ipv6.base[1]) & mask[1]) == (grub_be_to_cpu64 (addr->ipv6[1]) & mask[1]))); } } return 0; } grub_err_t grub_net_resolve_address (const char *name, grub_net_network_level_address_t *addr) { const char *rest; grub_err_t err; grub_size_t naddresses; struct grub_net_network_level_address *addresses = 0; if (parse_ip (name, &addr->ipv4, &rest) && *rest == 0) { addr->type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4; return GRUB_ERR_NONE; } if (parse_ip6 (name, addr->ipv6, &rest) && *rest == 0) { addr->type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6; return GRUB_ERR_NONE; } err = grub_net_dns_lookup (name, 0, 0, &naddresses, &addresses, 1); if (err) return err; if (!naddresses) grub_error (GRUB_ERR_NET_BAD_ADDRESS, N_("unresolvable address %s"), name); /* FIXME: use other results as well. */ *addr = addresses[0]; grub_free (addresses); return GRUB_ERR_NONE; } grub_err_t grub_net_resolve_net_address (const char *name, grub_net_network_level_netaddress_t *addr) { const char *rest; if (parse_ip (name, &addr->ipv4.base, &rest)) { addr->type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4; if (*rest == '/') { addr->ipv4.masksize = grub_strtoul (rest + 1, (char **) &rest, 0); if (!grub_errno && *rest == 0) return GRUB_ERR_NONE; grub_errno = GRUB_ERR_NONE; } else if (*rest == 0) { addr->ipv4.masksize = 32; return GRUB_ERR_NONE; } } if (parse_ip6 (name, addr->ipv6.base, &rest)) { addr->type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6; if (*rest == '/') { addr->ipv6.masksize = grub_strtoul (rest + 1, (char **) &rest, 0); if (!grub_errno && *rest == 0) return GRUB_ERR_NONE; grub_errno = GRUB_ERR_NONE; } else if (*rest == 0) { addr->ipv6.masksize = 128; return GRUB_ERR_NONE; } } return grub_error (GRUB_ERR_NET_BAD_ADDRESS, N_("unrecognised network address `%s'"), name); } static int route_cmp (const struct grub_net_route *a, const struct grub_net_route *b) { if (a == NULL && b == NULL) return 0; if (b == NULL) return +1; if (a == NULL) return -1; if (a->target.type < b->target.type) return -1; if (a->target.type > b->target.type) return +1; switch (a->target.type) { case GRUB_NET_NETWORK_LEVEL_PROTOCOL_DHCP_RECV: break; case GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6: if (a->target.ipv6.masksize > b->target.ipv6.masksize) return +1; if (a->target.ipv6.masksize < b->target.ipv6.masksize) return -1; break; case GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4: if (a->target.ipv4.masksize > b->target.ipv4.masksize) return +1; if (a->target.ipv4.masksize < b->target.ipv4.masksize) return -1; break; } return 0; } grub_err_t grub_net_route_address (grub_net_network_level_address_t addr, grub_net_network_level_address_t *gateway, struct grub_net_network_level_interface **interf) { struct grub_net_route *route; unsigned int depth = 0; unsigned int routecnt = 0; struct grub_net_network_level_protocol *prot = NULL; grub_net_network_level_address_t curtarget = addr; *gateway = addr; FOR_NET_ROUTES(route) routecnt++; for (depth = 0; depth < routecnt + 2 && depth < GRUB_UINT_MAX; depth++) { struct grub_net_route *bestroute = NULL; FOR_NET_ROUTES(route) { if (depth && prot != route->prot) continue; if (!match_net (&route->target, &curtarget)) continue; if (route_cmp (route, bestroute) > 0) bestroute = route; } if (bestroute == NULL) return grub_error (GRUB_ERR_NET_NO_ROUTE, N_("destination unreachable")); if (!bestroute->is_gateway) { *interf = bestroute->interface; return GRUB_ERR_NONE; } if (depth == 0) *gateway = bestroute->gw; curtarget = bestroute->gw; } return grub_error (GRUB_ERR_NET_ROUTE_LOOP, /* TRANSLATORS: route loop is a condition when e.g. to contact server A you need to go through B and to contact B you need to go through A. */ N_("route loop detected")); } static grub_err_t grub_cmd_deladdr (struct grub_command *cmd __attribute__ ((unused)), int argc, char **args) { struct grub_net_network_level_interface *inter; if (argc != 1) return grub_error (GRUB_ERR_BAD_ARGUMENT, N_("one argument expected")); FOR_NET_NETWORK_LEVEL_INTERFACES (inter) if (grub_strcmp (inter->name, args[0]) == 0) break; if (inter == NULL) return grub_error (GRUB_ERR_BAD_ARGUMENT, N_("address not found")); if (inter->flags & GRUB_NET_INTERFACE_PERMANENT) return grub_error (GRUB_ERR_IO, N_("you can't delete this address")); grub_net_network_level_interface_unregister (inter); grub_free (inter->name); grub_free (inter); return GRUB_ERR_NONE; } void grub_net_addr_to_str (const grub_net_network_level_address_t *target, char *buf) { switch (target->type) { case GRUB_NET_NETWORK_LEVEL_PROTOCOL_DHCP_RECV: COMPILE_TIME_ASSERT (sizeof ("temporary") < GRUB_NET_MAX_STR_ADDR_LEN); grub_strcpy (buf, "temporary"); return; case GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6: { char *ptr = buf; grub_uint64_t n = grub_be_to_cpu64 (target->ipv6[0]); int i; for (i = 0; i < 4; i++) { grub_snprintf (ptr, 6, "%" PRIxGRUB_UINT64_T ":", (n >> (48 - 16 * i)) & 0xffff); ptr += grub_strlen (ptr); } n = grub_be_to_cpu64 (target->ipv6[1]); for (i = 0; i < 3; i++) { grub_snprintf (ptr, 6, "%" PRIxGRUB_UINT64_T ":", (n >> (48 - 16 * i)) & 0xffff); ptr += grub_strlen (ptr); } grub_snprintf (ptr, 5, "%" PRIxGRUB_UINT64_T, n & 0xffff); return; } case GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4: { grub_uint32_t n = grub_be_to_cpu32 (target->ipv4); grub_snprintf (buf, GRUB_NET_MAX_STR_ADDR_LEN, "%d.%d.%d.%d", ((n >> 24) & 0xff), ((n >> 16) & 0xff), ((n >> 8) & 0xff), ((n >> 0) & 0xff)); } return; } grub_snprintf (buf, GRUB_NET_MAX_STR_ADDR_LEN, "Unknown address type %d", target->type); } void grub_net_hwaddr_to_str (const grub_net_link_level_address_t *addr, char *str) { str[0] = 0; switch (addr->type) { case GRUB_NET_LINK_LEVEL_PROTOCOL_ETHERNET: { char *ptr; unsigned i; for (ptr = str, i = 0; i < ARRAY_SIZE (addr->mac); i++) { grub_snprintf (ptr, GRUB_NET_MAX_STR_HWADDR_LEN - (ptr - str), "%02x:", addr->mac[i] & 0xff); ptr += (sizeof ("XX:") - 1); } return; } } grub_printf (_("Unsupported hw address type %d\n"), addr->type); } int grub_net_hwaddr_cmp (const grub_net_link_level_address_t *a, const grub_net_link_level_address_t *b) { if (a->type < b->type) return -1; if (a->type > b->type) return +1; switch (a->type) { case GRUB_NET_LINK_LEVEL_PROTOCOL_ETHERNET: return grub_memcmp (a->mac, b->mac, sizeof (a->mac)); } grub_printf (_("Unsupported hw address type %d\n"), a->type); return 1; } int grub_net_addr_cmp (const grub_net_network_level_address_t *a, const grub_net_network_level_address_t *b) { if (a->type < b->type) return -1; if (a->type > b->type) return +1; switch (a->type) { case GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4: return grub_memcmp (&a->ipv4, &b->ipv4, sizeof (a->ipv4)); case GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6: return grub_memcmp (&a->ipv6, &b->ipv6, sizeof (a->ipv6)); case GRUB_NET_NETWORK_LEVEL_PROTOCOL_DHCP_RECV: return 0; } grub_printf (_("Unsupported address type %d\n"), a->type); return 1; } /* FIXME: implement this. */ static char * hwaddr_set_env (struct grub_env_var *var __attribute__ ((unused)), const char *val __attribute__ ((unused))) { return NULL; } /* FIXME: implement this. */ static char * addr_set_env (struct grub_env_var *var __attribute__ ((unused)), const char *val __attribute__ ((unused))) { return NULL; } static char * defserver_set_env (struct grub_env_var *var __attribute__ ((unused)), const char *val) { grub_free (grub_net_default_server); grub_net_default_server = grub_strdup (val); return grub_strdup (val); } static const char * defserver_get_env (struct grub_env_var *var __attribute__ ((unused)), const char *val __attribute__ ((unused))) { return grub_net_default_server ? : ""; } static const char * defip_get_env (struct grub_env_var *var __attribute__ ((unused)), const char *val __attribute__ ((unused))) { const char *intf = grub_env_get ("net_default_interface"); const char *ret = NULL; if (intf) { char *buf = grub_xasprintf ("net_%s_ip", intf); if (buf) ret = grub_env_get (buf); grub_free (buf); } return ret; } static char * defip_set_env (struct grub_env_var *var __attribute__ ((unused)), const char *val) { const char *intf = grub_env_get ("net_default_interface"); if (intf) { char *buf = grub_xasprintf ("net_%s_ip", intf); if (buf) grub_env_set (buf, val); grub_free (buf); } return NULL; } static const char * defmac_get_env (struct grub_env_var *var __attribute__ ((unused)), const char *val __attribute__ ((unused))) { const char *intf = grub_env_get ("net_default_interface"); const char *ret = NULL; if (intf) { char *buf = grub_xasprintf ("net_%s_mac", intf); if (buf) ret = grub_env_get (buf); grub_free (buf); } return ret; } static char * defmac_set_env (struct grub_env_var *var __attribute__ ((unused)), const char *val) { const char *intf = grub_env_get ("net_default_interface"); if (intf) { char *buf = grub_xasprintf ("net_%s_mac", intf); if (buf) grub_env_set (buf, val); grub_free (buf); } return NULL; } static void grub_net_network_level_interface_register (struct grub_net_network_level_interface *inter) { { char buf[GRUB_NET_MAX_STR_HWADDR_LEN]; char *name; char *ptr; grub_net_hwaddr_to_str (&inter->hwaddress, buf); name = grub_xasprintf ("net_%s_mac", inter->name); if (!name) return; for (ptr = name; *ptr; ptr++) if (*ptr == ':') *ptr = '_'; grub_env_set (name, buf); grub_register_variable_hook (name, 0, hwaddr_set_env); grub_env_export (name); grub_free (name); } { char buf[GRUB_NET_MAX_STR_ADDR_LEN]; char *name; char *ptr; grub_net_addr_to_str (&inter->address, buf); name = grub_xasprintf ("net_%s_ip", inter->name); if (!name) return; for (ptr = name; *ptr; ptr++) if (*ptr == ':') *ptr = '_'; grub_env_set (name, buf); grub_register_variable_hook (name, 0, addr_set_env); grub_env_export (name); grub_free (name); } inter->card->num_ifaces++; inter->prev = &grub_net_network_level_interfaces; inter->next = grub_net_network_level_interfaces; if (inter->next) inter->next->prev = &inter->next; grub_net_network_level_interfaces = inter; } grub_err_t grub_net_add_ipv4_local (struct grub_net_network_level_interface *inter, int mask) { grub_uint32_t ip_cpu; struct grub_net_route *route; if (inter->address.type != GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4) return 0; ip_cpu = grub_be_to_cpu32 (inter->address.ipv4); if (mask == -1) { if (!(ip_cpu & 0x80000000)) mask = 8; else if (!(ip_cpu & 0x40000000)) mask = 16; else if (!(ip_cpu & 0x20000000)) mask = 24; } if (mask == -1) return 0; route = grub_zalloc (sizeof (*route)); if (!route) return grub_errno; route->name = grub_xasprintf ("%s:local", inter->name); if (!route->name) { grub_free (route); return grub_errno; } route->target.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4; route->target.ipv4.base = grub_cpu_to_be32 (ip_cpu & (0xffffffff << (32 - mask))); route->target.ipv4.masksize = mask; route->is_gateway = 0; route->interface = inter; grub_net_route_register (route); return 0; } /* FIXME: support MAC specifying. */ static grub_err_t grub_cmd_addaddr (struct grub_command *cmd __attribute__ ((unused)), int argc, char **args) { struct grub_net_card *card; grub_net_network_level_address_t addr; grub_err_t err; grub_net_interface_flags_t flags = 0; struct grub_net_network_level_interface *inf; if (argc != 3) return grub_error (GRUB_ERR_BAD_ARGUMENT, N_("three arguments expected")); FOR_NET_CARDS (card) if (grub_strcmp (card->name, args[1]) == 0) break; if (card == NULL) return grub_error (GRUB_ERR_BAD_ARGUMENT, N_("card not found")); err = grub_net_resolve_address (args[2], &addr); if (err) return err; if (card->flags & GRUB_NET_CARD_NO_MANUAL_INTERFACES) return grub_error (GRUB_ERR_IO, "this card doesn't support address addition"); if (card->flags & GRUB_NET_CARD_HWADDRESS_IMMUTABLE) flags |= GRUB_NET_INTERFACE_HWADDRESS_IMMUTABLE; inf = grub_net_add_addr (args[0], card, &addr, &card->default_address, flags); if (inf) grub_net_add_ipv4_local (inf, -1); return grub_errno; } static grub_err_t grub_cmd_delroute (struct grub_command *cmd __attribute__ ((unused)), int argc, char **args) { struct grub_net_route *route; struct grub_net_route **prev; if (argc != 1) return grub_error (GRUB_ERR_BAD_ARGUMENT, N_("one argument expected")); for (prev = &grub_net_routes, route = *prev; route; prev = &((*prev)->next), route = *prev) if (grub_strcmp (route->name, args[0]) == 0) { *prev = route->next; grub_free (route->name); grub_free (route); if (!*prev) break; } return GRUB_ERR_NONE; } grub_err_t grub_net_add_route (const char *name, grub_net_network_level_netaddress_t target, struct grub_net_network_level_interface *inter) { struct grub_net_route *route; route = grub_zalloc (sizeof (*route)); if (!route) return grub_errno; route->name = grub_strdup (name); if (!route->name) { grub_free (route); return grub_errno; } route->target = target; route->is_gateway = 0; route->interface = inter; grub_net_route_register (route); return GRUB_ERR_NONE; } grub_err_t grub_net_add_route_gw (const char *name, grub_net_network_level_netaddress_t target, grub_net_network_level_address_t gw) { struct grub_net_route *route; route = grub_zalloc (sizeof (*route)); if (!route) return grub_errno; route->name = grub_strdup (name); if (!route->name) { grub_free (route); return grub_errno; } route->target = target; route->is_gateway = 1; route->gw = gw; grub_net_route_register (route); return GRUB_ERR_NONE; } static grub_err_t grub_cmd_addroute (struct grub_command *cmd __attribute__ ((unused)), int argc, char **args) { grub_net_network_level_netaddress_t target; if (argc < 3) return grub_error (GRUB_ERR_BAD_ARGUMENT, N_("three arguments expected")); grub_net_resolve_net_address (args[1], &target); if (grub_strcmp (args[2], "gw") == 0 && argc >= 4) { grub_err_t err; grub_net_network_level_address_t gw; err = grub_net_resolve_address (args[3], &gw); if (err) return err; return grub_net_add_route_gw (args[0], target, gw); } else { struct grub_net_network_level_interface *inter; FOR_NET_NETWORK_LEVEL_INTERFACES (inter) if (grub_strcmp (inter->name, args[2]) == 0) break; if (!inter) return grub_error (GRUB_ERR_BAD_ARGUMENT, N_("unrecognised network interface `%s'"), args[2]); return grub_net_add_route (args[0], target, inter); } } static void print_net_address (const grub_net_network_level_netaddress_t *target) { switch (target->type) { case GRUB_NET_NETWORK_LEVEL_PROTOCOL_DHCP_RECV: /* TRANSLATORS: it refers to the network address. */ grub_printf ("%s\n", _("temporary")); return; case GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4: { grub_uint32_t n = grub_be_to_cpu32 (target->ipv4.base); grub_printf ("%d.%d.%d.%d/%d ", ((n >> 24) & 0xff), ((n >> 16) & 0xff), ((n >> 8) & 0xff), ((n >> 0) & 0xff), target->ipv4.masksize); } return; case GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6: { char buf[GRUB_NET_MAX_STR_ADDR_LEN]; struct grub_net_network_level_address base; base.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV6; grub_memcpy (&base.ipv6, &target->ipv6, 16); grub_net_addr_to_str (&base, buf); grub_printf ("%s/%d ", buf, target->ipv6.masksize); } return; } grub_printf (_("Unknown address type %d\n"), target->type); } static void print_address (const grub_net_network_level_address_t *target) { char buf[GRUB_NET_MAX_STR_ADDR_LEN]; grub_net_addr_to_str (target, buf); grub_xputs (buf); } static grub_err_t grub_cmd_listroutes (struct grub_command *cmd __attribute__ ((unused)), int argc __attribute__ ((unused)), char **args __attribute__ ((unused))) { struct grub_net_route *route; FOR_NET_ROUTES(route) { grub_printf ("%s ", route->name); print_net_address (&route->target); if (route->is_gateway) { grub_printf ("gw "); print_address (&route->gw); } else grub_printf ("%s", route->interface->name); grub_printf ("\n"); } return GRUB_ERR_NONE; } static grub_err_t grub_cmd_listcards (struct grub_command *cmd __attribute__ ((unused)), int argc __attribute__ ((unused)), char **args __attribute__ ((unused))) { struct grub_net_card *card; FOR_NET_CARDS(card) { char buf[GRUB_NET_MAX_STR_HWADDR_LEN]; grub_net_hwaddr_to_str (&card->default_address, buf); grub_printf ("%s %s\n", card->name, buf); } return GRUB_ERR_NONE; } static grub_err_t grub_cmd_listaddrs (struct grub_command *cmd __attribute__ ((unused)), int argc __attribute__ ((unused)), char **args __attribute__ ((unused))) { struct grub_net_network_level_interface *inf; FOR_NET_NETWORK_LEVEL_INTERFACES (inf) { char bufh[GRUB_NET_MAX_STR_HWADDR_LEN]; char bufn[GRUB_NET_MAX_STR_ADDR_LEN]; grub_net_hwaddr_to_str (&inf->hwaddress, bufh); grub_net_addr_to_str (&inf->address, bufn); grub_printf ("%s %s %s\n", inf->name, bufh, bufn); } return GRUB_ERR_NONE; } grub_net_app_level_t grub_net_app_level_list; struct grub_net_socket *grub_net_sockets; static grub_net_t grub_net_open_real (const char *name) { grub_net_app_level_t proto; const char *protname, *server; grub_size_t protnamelen; int try; if (grub_strncmp (name, "pxe:", sizeof ("pxe:") - 1) == 0) { protname = "tftp"; protnamelen = sizeof ("tftp") - 1; server = name + sizeof ("pxe:") - 1; } else if (grub_strcmp (name, "pxe") == 0) { protname = "tftp"; protnamelen = sizeof ("tftp") - 1; server = grub_net_default_server; } else { const char *comma; comma = grub_strchr (name, ','); if (comma) { protnamelen = comma - name; server = comma + 1; protname = name; } else { protnamelen = grub_strlen (name); server = grub_net_default_server; protname = name; } } if (!server) { grub_error (GRUB_ERR_NET_BAD_ADDRESS, N_("no server is specified")); return NULL; } for (try = 0; try < 2; try++) { FOR_NET_APP_LEVEL (proto) { if (grub_memcmp (proto->name, protname, protnamelen) == 0 && proto->name[protnamelen] == 0) { grub_net_t ret = grub_zalloc (sizeof (*ret)); if (!ret) return NULL; ret->protocol = proto; ret->server = grub_strdup (server); if (!ret->server) { grub_free (ret); return NULL; } ret->fs = &grub_net_fs; return ret; } } if (try == 0) { const char *prefix, *root; char *prefdev, *comma; int skip = 0; grub_size_t devlen; /* Do not attempt to load module if it requires protocol provided by this module - it results in infinite recursion. Just continue, fail and cleanup on next iteration. */ prefix = grub_env_get ("prefix"); if (!prefix) continue; prefdev = grub_file_get_device_name (prefix); if (!prefdev) { root = grub_env_get ("root"); if (!root) continue; prefdev = grub_strdup (root); if (!prefdev) continue; } comma = grub_strchr (prefdev, ','); if (comma) *comma = '\0'; devlen = grub_strlen (prefdev); if (protnamelen == devlen && grub_memcmp (prefdev, protname, devlen) == 0) skip = 1; grub_free (prefdev); if (skip) continue; if (sizeof ("http") - 1 == protnamelen && grub_memcmp ("http", protname, protnamelen) == 0) { grub_dl_load ("http"); grub_errno = GRUB_ERR_NONE; continue; } if (sizeof ("tftp") - 1 == protnamelen && grub_memcmp ("tftp", protname, protnamelen) == 0) { grub_dl_load ("tftp"); grub_errno = GRUB_ERR_NONE; continue; } } break; } /* Restore original error. */ grub_error (GRUB_ERR_UNKNOWN_DEVICE, N_("disk `%s' not found"), name); return NULL; } static grub_err_t grub_net_fs_dir (grub_device_t device, const char *path __attribute__ ((unused)), grub_fs_dir_hook_t hook __attribute__ ((unused)), void *hook_data __attribute__ ((unused))) { if (!device->net) return grub_error (GRUB_ERR_BUG, "invalid net device"); return GRUB_ERR_NONE; } static grub_err_t grub_net_fs_open (struct grub_file *file_out, const char *name) { grub_err_t err; struct grub_file *file, *bufio; file = grub_malloc (sizeof (*file)); if (!file) return grub_errno; grub_memcpy (file, file_out, sizeof (struct grub_file)); file->device->net->packs.first = NULL; file->device->net->packs.last = NULL; file->device->net->name = grub_strdup (name); if (!file->device->net->name) return grub_errno; err = file->device->net->protocol->open (file, name); if (err) { while (file->device->net->packs.first) { grub_netbuff_free (file->device->net->packs.first->nb); grub_net_remove_packet (file->device->net->packs.first); } grub_free (file->device->net->name); grub_free (file); return err; } bufio = grub_bufio_open (file, 32768); if (! bufio) { while (file->device->net->packs.first) { grub_netbuff_free (file->device->net->packs.first->nb); grub_net_remove_packet (file->device->net->packs.first); } file->device->net->protocol->close (file); grub_free (file->device->net->name); grub_free (file); return grub_errno; } grub_memcpy (file_out, bufio, sizeof (struct grub_file)); grub_free (bufio); return GRUB_ERR_NONE; } static grub_err_t grub_net_fs_close (grub_file_t file) { while (file->device->net->packs.first) { grub_netbuff_free (file->device->net->packs.first->nb); grub_net_remove_packet (file->device->net->packs.first); } file->device->net->protocol->close (file); grub_free (file->device->net->name); return GRUB_ERR_NONE; } static void receive_packets (struct grub_net_card *card, int *stop_condition) { int received = 0; if (card->num_ifaces == 0) return; if (!card->opened) { grub_err_t err = GRUB_ERR_NONE; if (card->driver->open) err = card->driver->open (card); if (err) { grub_errno = GRUB_ERR_NONE; return; } card->opened = 1; } while (received < 100) { /* Maybe should be better have a fixed number of packets for each card and just mark them as used and not used. */ struct grub_net_buff *nb; if (received > 10 && stop_condition && *stop_condition) break; nb = card->driver->recv (card); if (!nb) { card->last_poll = grub_get_time_ms (); break; } received++; grub_net_recv_ethernet_packet (nb, card); if (grub_errno) { grub_dprintf ("net", "error receiving: %d: %s\n", grub_errno, grub_errmsg); grub_errno = GRUB_ERR_NONE; } } grub_print_error (); } static char * grub_env_write_readonly (struct grub_env_var *var __attribute__ ((unused)), const char *val __attribute__ ((unused))) { return NULL; } grub_err_t grub_env_set_net_property (const char *intername, const char *suffix, const char *value, grub_size_t len) { char *varname, *varvalue; char *ptr; varname = grub_xasprintf ("net_%s_%s", intername, suffix); if (!varname) return grub_errno; for (ptr = varname; *ptr; ptr++) if (*ptr == ':') *ptr = '_'; varvalue = grub_malloc (len + 1); if (!varvalue) { grub_free (varname); return grub_errno; } grub_memcpy (varvalue, value, len); varvalue[len] = 0; grub_err_t ret = grub_env_set (varname, varvalue); grub_register_variable_hook (varname, 0, grub_env_write_readonly); grub_env_export (varname); grub_free (varname); grub_free (varvalue); return ret; } void grub_net_poll_cards (unsigned time, int *stop_condition) { struct grub_net_card *card; grub_uint64_t start_time; start_time = grub_get_time_ms (); while ((grub_get_time_ms () - start_time) < time && (!stop_condition || !*stop_condition)) FOR_NET_CARDS (card) receive_packets (card, stop_condition); grub_net_tcp_retransmit (); } static void grub_net_poll_cards_idle_real (void) { struct grub_net_card *card; FOR_NET_CARDS (card) { grub_uint64_t ctime = grub_get_time_ms (); if (ctime < card->last_poll || ctime >= card->last_poll + card->idle_poll_delay_ms) receive_packets (card, 0); } grub_net_tcp_retransmit (); } /* Read from the packets list*/ static grub_ssize_t grub_net_fs_read_real (grub_file_t file, char *buf, grub_size_t len) { grub_net_t net = file->device->net; struct grub_net_buff *nb; char *ptr = buf; grub_size_t amount, total = 0; int try = 0; while (try <= GRUB_NET_TRIES) { while (net->packs.first) { try = 0; nb = net->packs.first->nb; amount = nb->tail - nb->data; if (amount > len) amount = len; len -= amount; total += amount; file->device->net->offset += amount; if (grub_file_progress_hook) grub_file_progress_hook (0, 0, amount, file); if (buf) { grub_memcpy (ptr, nb->data, amount); ptr += amount; } if (amount == (grub_size_t) (nb->tail - nb->data)) { grub_netbuff_free (nb); grub_net_remove_packet (net->packs.first); } else nb->data += amount; if (!len) { if (net->protocol->packets_pulled) net->protocol->packets_pulled (file); return total; } } if (net->protocol->packets_pulled) net->protocol->packets_pulled (file); if (!net->eof) { try++; grub_net_poll_cards (GRUB_NET_INTERVAL + (try * GRUB_NET_INTERVAL_ADDITION), &net->stall); } else return total; } grub_error (GRUB_ERR_TIMEOUT, N_("timeout reading `%s'"), net->name); return -1; } static grub_off_t have_ahead (struct grub_file *file) { grub_net_t net = file->device->net; grub_off_t ret = net->offset; struct grub_net_packet *pack; for (pack = net->packs.first; pack; pack = pack->next) ret += pack->nb->tail - pack->nb->data; return ret; } static grub_err_t grub_net_seek_real (struct grub_file *file, grub_off_t offset) { if (offset == file->device->net->offset) return GRUB_ERR_NONE; if (offset > file->device->net->offset) { if (!file->device->net->protocol->seek || have_ahead (file) >= offset) { grub_net_fs_read_real (file, NULL, offset - file->device->net->offset); return grub_errno; } return file->device->net->protocol->seek (file, offset); } { grub_err_t err; if (file->device->net->protocol->seek) return file->device->net->protocol->seek (file, offset); while (file->device->net->packs.first) { grub_netbuff_free (file->device->net->packs.first->nb); grub_net_remove_packet (file->device->net->packs.first); } file->device->net->protocol->close (file); file->device->net->packs.first = NULL; file->device->net->packs.last = NULL; file->device->net->offset = 0; file->device->net->eof = 0; err = file->device->net->protocol->open (file, file->device->net->name); if (err) return err; grub_net_fs_read_real (file, NULL, offset); return grub_errno; } } static grub_ssize_t grub_net_fs_read (grub_file_t file, char *buf, grub_size_t len) { if (file->offset != file->device->net->offset) { grub_err_t err; err = grub_net_seek_real (file, file->offset); if (err) return err; } return grub_net_fs_read_real (file, buf, len); } static struct grub_fs grub_net_fs = { .name = "netfs", .dir = grub_net_fs_dir, .open = grub_net_fs_open, .read = grub_net_fs_read, .close = grub_net_fs_close, .label = NULL, .uuid = NULL, .mtime = NULL, }; static grub_err_t grub_net_fini_hw (int noreturn __attribute__ ((unused))) { struct grub_net_card *card; FOR_NET_CARDS (card) if (card->opened) { if (card->driver->close) card->driver->close (card); card->opened = 0; } return GRUB_ERR_NONE; } static grub_err_t grub_net_restore_hw (void) { return GRUB_ERR_NONE; } static struct grub_preboot *fini_hnd; static grub_command_t cmd_addaddr, cmd_deladdr, cmd_addroute, cmd_delroute; static grub_command_t cmd_lsroutes, cmd_lscards; static grub_command_t cmd_lsaddr, cmd_slaac; GRUB_MOD_INIT(net) { grub_register_variable_hook ("net_default_server", defserver_get_env, defserver_set_env); grub_env_export ("net_default_server"); grub_register_variable_hook ("pxe_default_server", defserver_get_env, defserver_set_env); grub_env_export ("pxe_default_server"); grub_register_variable_hook ("net_default_ip", defip_get_env, defip_set_env); grub_env_export ("net_default_ip"); grub_register_variable_hook ("net_default_mac", defmac_get_env, defmac_set_env); grub_env_export ("net_default_mac"); cmd_addaddr = grub_register_command ("net_add_addr", grub_cmd_addaddr, /* TRANSLATORS: HWADDRESS stands for "hardware address". */ N_("SHORTNAME CARD ADDRESS [HWADDRESS]"), N_("Add a network address.")); cmd_slaac = grub_register_command ("net_ipv6_autoconf", grub_cmd_ipv6_autoconf, N_("[CARD [HWADDRESS]]"), N_("Perform an IPV6 autoconfiguration")); cmd_deladdr = grub_register_command ("net_del_addr", grub_cmd_deladdr, N_("SHORTNAME"), N_("Delete a network address.")); cmd_addroute = grub_register_command ("net_add_route", grub_cmd_addroute, /* TRANSLATORS: "gw" is a keyword. */ N_("SHORTNAME NET [INTERFACE| gw GATEWAY]"), N_("Add a network route.")); cmd_delroute = grub_register_command ("net_del_route", grub_cmd_delroute, N_("SHORTNAME"), N_("Delete a network route.")); cmd_lsroutes = grub_register_command ("net_ls_routes", grub_cmd_listroutes, "", N_("list network routes")); cmd_lscards = grub_register_command ("net_ls_cards", grub_cmd_listcards, "", N_("list network cards")); cmd_lsaddr = grub_register_command ("net_ls_addr", grub_cmd_listaddrs, "", N_("list network addresses")); grub_bootp_init (); grub_dns_init (); grub_net_open = grub_net_open_real; fini_hnd = grub_loader_register_preboot_hook (grub_net_fini_hw, grub_net_restore_hw, GRUB_LOADER_PREBOOT_HOOK_PRIO_DISK); grub_net_poll_cards_idle = grub_net_poll_cards_idle_real; } GRUB_MOD_FINI(net) { grub_register_variable_hook ("net_default_server", 0, 0); grub_register_variable_hook ("pxe_default_server", 0, 0); grub_bootp_fini (); grub_dns_fini (); grub_unregister_command (cmd_addaddr); grub_unregister_command (cmd_deladdr); grub_unregister_command (cmd_addroute); grub_unregister_command (cmd_delroute); grub_unregister_command (cmd_lsroutes); grub_unregister_command (cmd_lscards); grub_unregister_command (cmd_lsaddr); grub_unregister_command (cmd_slaac); grub_fs_unregister (&grub_net_fs); grub_net_open = NULL; grub_net_fini_hw (0); grub_loader_unregister_preboot_hook (fini_hnd); grub_net_poll_cards_idle = grub_net_poll_cards_idle_real; }