vaseboot/VasEBoot-core/net/drivers/uboot/ubootnet.c

/*
 *  VAS_EBOOT  --  GRand Unified Bootloader
 *  Copyright (C) 2013  Free Software Foundation, Inc.
 *
 *  VAS_EBOOT 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.
 *
 *  VAS_EBOOT 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 VAS_EBOOT.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <VasEBoot/net/netbuff.h>
#include <VasEBoot/uboot/disk.h>
#include <VasEBoot/uboot/uboot.h>
#include <VasEBoot/uboot/api_public.h>
#include <VasEBoot/dl.h>
#include <VasEBoot/net.h>
#include <VasEBoot/time.h>
#include <VasEBoot/i18n.h>

VAS_EBOOT_MOD_LICENSE ("GPLv3+");

static VasEBoot_err_t
card_open (struct VasEBoot_net_card *dev)
{
  int status;

  status = VasEBoot_uboot_dev_open (dev->data);
  if (status)
    return VasEBoot_error (VAS_EBOOT_ERR_IO, "Couldn't open network card.");

  return VAS_EBOOT_ERR_NONE;
}

static void
card_close (struct VasEBoot_net_card *dev)
{
  VasEBoot_uboot_dev_close (dev->data);
}

static VasEBoot_err_t
send_card_buffer (struct VasEBoot_net_card *dev, struct VasEBoot_net_buff *pack)
{
  int status;
  VasEBoot_size_t len;

  len = (pack->tail - pack->data);
  if (len > dev->mtu)
    len = dev->mtu;

  VasEBoot_memcpy (dev->txbuf, pack->data, len);
  status = VasEBoot_uboot_dev_send (dev->data, dev->txbuf, len);

  if (status)
    return VasEBoot_error (VAS_EBOOT_ERR_IO, N_("couldn't send network packet"));
  return VAS_EBOOT_ERR_NONE;
}

static struct VasEBoot_net_buff *
get_card_packet (struct VasEBoot_net_card *dev)
{
  int rc;
  VasEBoot_uint64_t start_time;
  struct VasEBoot_net_buff *nb;
  int actual;

  nb = VasEBoot_netbuff_alloc (dev->mtu + 64 + 2);
  if (!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. */
  VasEBoot_netbuff_reserve (nb, 2);

  start_time = VasEBoot_get_time_ms ();
  do
    {
      rc = VasEBoot_uboot_dev_recv (dev->data, nb->data, dev->mtu + 64, &actual);
      VasEBoot_dprintf ("net", "rc=%d, actual=%d, time=%lld\n", rc, actual,
		    VasEBoot_get_time_ms () - start_time);
    }
  while ((actual <= 0 || rc < 0) && (VasEBoot_get_time_ms () - start_time < 200));
  if (actual > 0)
    {
      VasEBoot_netbuff_put (nb, actual);
      return nb;
    }
  VasEBoot_netbuff_free (nb);
  return NULL;
}

static struct VasEBoot_net_card_driver ubootnet =
  {
    .name = "ubnet",
    .open = card_open,
    .close = card_close,
    .send = send_card_buffer,
    .recv = get_card_packet
  };

VAS_EBOOT_MOD_INIT (ubootnet)
{
  int devcount, i;
  int nfound = 0;

  devcount = VasEBoot_uboot_dev_enum ();

  for (i = 0; i < devcount; i++)
    {
      struct device_info *devinfo = VasEBoot_uboot_dev_get (i);
      struct VasEBoot_net_card *card;

      if (!(devinfo->type & DEV_TYP_NET))
	continue;

      card = VasEBoot_zalloc (sizeof (struct VasEBoot_net_card));
      if (!card)
	{
	  VasEBoot_print_error ();
	  return;
	}

      /* FIXME: Any way to check this?  */
      card->mtu = 1500;

      VasEBoot_memcpy (&(card->default_address.mac), &devinfo->di_net.hwaddr, 6);
      card->default_address.type = VAS_EBOOT_NET_LINK_LEVEL_PROTOCOL_ETHERNET;

      card->txbufsize = ALIGN_UP (card->mtu, 64) + 256;
      card->txbuf = VasEBoot_zalloc (card->txbufsize);
      if (!card->txbuf)
	{
	  VasEBoot_free (card);
	  VasEBoot_print_error ();
	  continue;
	}

      card->data = devinfo;
      card->flags = 0;
      card->name = VasEBoot_xasprintf ("ubnet_%d", ++nfound);
      card->idle_poll_delay_ms = 10;

      card->driver = &ubootnet;
      VasEBoot_net_card_register (card);
    }
}

VAS_EBOOT_MOD_FINI (ubootnet)
{
  struct VasEBoot_net_card *card, *next;

  FOR_NET_CARDS_SAFE (card, next)
    if (card->driver && VasEBoot_strcmp (card->driver->name, "ubnet") == 0)
      VasEBoot_net_card_unregister (card);
}