=== modified file 'grub-core/Makefile.core.def' --- grub-core/Makefile.core.def 2013-04-07 00:41:07 +0000 +++ grub-core/Makefile.core.def 2013-04-09 22:05:01 +0000 @@ -1831,6 +1831,12 @@ }; module = { + name = ubootnet; + common = net/drivers/uboot/ubootnet.c; + enable = uboot; +}; + +module = { name = efinet; common = net/drivers/efi/efinet.c; enable = efi; === added directory 'grub-core/net/drivers/uboot' === added file 'grub-core/net/drivers/uboot/ubootnet.c' --- grub-core/net/drivers/uboot/ubootnet.c 1970-01-01 00:00:00 +0000 +++ grub-core/net/drivers/uboot/ubootnet.c 2013-04-09 22:28:48 +0000 @@ -0,0 +1,182 @@ +/* + * GRUB -- GRand Unified Bootloader + * Copyright (C) 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 + +GRUB_MOD_LICENSE ("GPLv3+"); + +struct ubootnet_data +{ + void *cookie; + int handle; +}; + +static grub_err_t +card_open (struct grub_net_card *dev) +{ + int status; + struct ubootnet_data *data = dev->data; + + status = uboot_dev_open (data->handle); + if (status) + return grub_error (GRUB_ERR_IO, "Couldn't open network card."); + + return GRUB_ERR_NONE; +} + +static void +card_close (struct grub_net_card *dev) +{ + struct ubootnet_data *data = dev->data; + + uboot_dev_close (data->handle); +} + +static grub_err_t +send_card_buffer (struct grub_net_card *dev, struct grub_net_buff *pack) +{ + int status; + struct ubootnet_data *data = dev->data; + grub_size_t len; + + len = (pack->tail - pack->data); + if (len > dev->mtu) + len = dev->mtu; + + grub_memcpy (dev->txbuf, pack->data, len); + status = uboot_dev_send (data->handle, dev->txbuf, + len); + + if (status) + return grub_error (GRUB_ERR_IO, N_("couldn't send network packet")); + return GRUB_ERR_NONE; +} + +static struct grub_net_buff * +get_card_packet (struct grub_net_card *dev) +{ + int rc; + struct ubootnet_data *data = dev->data; + grub_uint64_t start_time; + struct grub_net_buff *nb; + int actual; + + nb = grub_netbuff_alloc (dev->mtu + 64 + 2); + if (!nb) + { + grub_netbuff_free (nb); + return NULL; + } + /* Reserve 2 bytes so that 2 + 14/18 bytes of ethernet header is divisible + by 4. So that IP header is aligned on 4 bytes. */ + grub_netbuff_reserve (nb, 2); + + start_time = grub_get_time_ms (); + do + rc = uboot_dev_recv (data->handle, nb->data, dev->mtu + 64, &actual); + while ((actual <= 0 || rc < 0) && (grub_get_time_ms () - start_time < 200)); + if (actual > 0) + { + grub_netbuff_put (nb, actual); + return nb; + } + grub_netbuff_free (nb); + return NULL; +} + +static struct grub_net_card_driver ubootnet = + { + .name = "ubnet", + .open = card_open, + .close = card_close, + .send = send_card_buffer, + .recv = get_card_packet + }; + +GRUB_MOD_INIT (ubootnet) +{ + int devcount, i; + int nfound = 0; + + devcount = uboot_dev_enum (); + + for (i = 0; i < devcount; i++) + { + struct device_info *devinfo = uboot_dev_get (i); + struct ubootnet_data *ubdata; + struct grub_net_card *card; + + if (!(devinfo->type & DEV_TYP_NET)) + continue; + + ubdata = grub_malloc (sizeof (struct ubootnet_data)); + if (!ubdata) + { + grub_print_error (); + return; + } + card = grub_zalloc (sizeof (struct grub_net_card)); + if (!card) + { + grub_free (ubdata); + grub_print_error (); + return; + } + + ubdata->handle = i; + ubdata->cookie = devinfo->cookie; + + /* FIXME: Any way to check this? */ + card->mtu = 1500; + + grub_memcpy (&(card->default_address.mac), &devinfo->di_net.hwaddr, 6); + card->default_address.type = GRUB_NET_LINK_LEVEL_PROTOCOL_ETHERNET; + + card->txbufsize = ALIGN_UP (card->mtu, 64) + 256; + card->txbuf = grub_zalloc (card->txbufsize); + if (!card->txbuf) + { + grub_print_error (); + continue; + } + + card->data = ubdata; + card->flags = 0; + card->name = grub_xasprintf ("ubnet_%d", ++nfound); + card->idle_poll_delay_ms = 10; + + card->driver = &ubootnet; + grub_net_card_register (card); + } +} + +void +grub_ubootdisk_fini (void) +{ + struct grub_net_card *card, *next; + + FOR_NET_CARDS_SAFE (card, next) + if (card->driver && grub_strcmp (card->driver->name, "ubnet") == 0) + grub_net_card_unregister (card); +} === modified file 'include/grub/uboot/uboot.h' --- include/grub/uboot/uboot.h 2013-04-07 00:41:07 +0000 +++ include/grub/uboot/uboot.h 2013-04-09 22:21:07 +0000 @@ -100,19 +100,19 @@ /* * int API_dev_enum(struct device_info *) */ -int uboot_dev_enum (void); +int EXPORT_FUNC(uboot_dev_enum) (void); -struct device_info *uboot_dev_get (int handle); +struct device_info *EXPORT_FUNC(uboot_dev_get) (int handle); /* * int API_dev_open(struct device_info *) */ -int uboot_dev_open (int handle); +int EXPORT_FUNC(uboot_dev_open) (int handle); /* * int API_dev_close(struct device_info *) */ -int uboot_dev_close (int handle); +int EXPORT_FUNC(uboot_dev_close) (int handle); /* * Notice: this is for sending network packets only, as U-Boot does not @@ -134,8 +134,8 @@ int uboot_dev_read (int handle, void *buf, lbasize_t blocks, lbastart_t start, lbasize_t * real_blocks); -int uboot_dev_recv (int handle, void *buf, int size, int *real_size); -int uboot_dev_send (int handle, void *buf, int size); +int EXPORT_FUNC(uboot_dev_recv) (int handle, void *buf, int size, int *real_size); +int EXPORT_FUNC(uboot_dev_send) (int handle, void *buf, int size); /* * int API_env_get(const char *name, char **value)