/*      $NetBSD: efinet.c,v 1.8 2018/08/18 15:55:19 kre Exp $   */

/*-
* Copyright (c) 2001 Doug Rabson
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
*    notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
*    notice, this list of conditions and the following disclaimer in the
*    documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/

#include <sys/cdefs.h>
/* __FBSDID("$FreeBSD: src/sys/boot/efi/libefi/efinet.c,v 1.6 2004/01/04 23:28:16 obrien Exp $"); */

#include <sys/param.h>

#include <lib/libsa/stand.h>
#include <lib/libsa/loadfile.h>
#include <lib/libsa/net.h>
#include <lib/libsa/netif.h>

#ifdef EFINET_DEBUG
#include <lib/libsa/ether_sprintf.c>
#endif

#include <efi.h>
#include <efilib.h>

extern struct netif_driver efi_net;

int  efinet_match(struct netif *, void *);
int  efinet_probe(struct netif *, void *);
void efinet_init(struct iodesc *, void *);
int  efinet_get(struct iodesc *, void *, size_t, saseconds_t);
int  efinet_put(struct iodesc *, void *, size_t);
void efinet_end(struct netif *);

struct netif_driver efi_net = {
       "net",                  /* netif_bname */
       efinet_match,           /* netif_match */
       efinet_probe,           /* netif_probe */
       efinet_init,            /* netif_init */
       efinet_get,             /* netif_get */
       efinet_put,             /* netif_put */
       efinet_end,             /* netif_end */
       0,                      /* netif_ifs */
       0                       /* netif_nifs */
};

#ifdef EFINET_DEBUG
static void
dump_mode(EFI_SIMPLE_NETWORK_MODE *mode)
{
       int i;

       printf("State                 = %x\n", mode->State);
       printf("HwAddressSize         = %u\n", mode->HwAddressSize);
       printf("MediaHeaderSize       = %u\n", mode->MediaHeaderSize);
       printf("MaxPacketSize         = %u\n", mode->MaxPacketSize);
       printf("NvRamSize             = %u\n", mode->NvRamSize);
       printf("NvRamAccessSize       = %u\n", mode->NvRamAccessSize);
       printf("ReceiveFilterMask     = %x\n", mode->ReceiveFilterMask);
       printf("ReceiveFilterSetting  = %u\n", mode->ReceiveFilterSetting);
       printf("MaxMCastFilterCount   = %u\n", mode->MaxMCastFilterCount);
       printf("MCastFilterCount      = %u\n", mode->MCastFilterCount);
       printf("MCastFilter           = {");
       for (i = 0; i < mode->MCastFilterCount; i++)
               printf(" %s", ether_sprintf(mode->MCastFilter[i].Addr));
       printf(" }\n");
       printf("CurrentAddress        = %s\n",
           ether_sprintf(mode->CurrentAddress.Addr));
       printf("BroadcastAddress      = %s\n",
           ether_sprintf(mode->BroadcastAddress.Addr));
       printf("PermanentAddress      = %s\n",
           ether_sprintf(mode->PermanentAddress.Addr));
       printf("IfType                = %u\n", mode->IfType);
       printf("MacAddressChangeable  = %d\n", mode->MacAddressChangeable);
       printf("MultipleTxSupported   = %d\n", mode->MultipleTxSupported);
       printf("MediaPresentSupported = %d\n", mode->MediaPresentSupported);
       printf("MediaPresent          = %d\n", mode->MediaPresent);
}
#endif

int
efinet_match(struct netif *nif, void *machdep_hint)
{

       return (1);
}

int
efinet_probe(struct netif *nif, void *machdep_hint)
{

       return (0);
}

int
efinet_put(struct iodesc *desc, void *pkt, size_t len)
{
       struct netif *nif = desc->io_netif;
       EFI_SIMPLE_NETWORK *net;
       EFI_STATUS status;
       void *buf;

       net = nif->nif_devdata;

       status = net->Transmit(net, 0, len, pkt, 0, 0, 0);
       if (status != EFI_SUCCESS)
               return -1;

       /* Wait for the buffer to be transmitted */
       do {
               buf = 0;        /* XXX Is this needed? */
               status = net->GetStatus(net, 0, &buf);
               /*
                * XXX EFI1.1 and the E1000 card returns a different
                * address than we gave.  Sigh.
                */
       } while (status == EFI_SUCCESS && buf == 0);

       /* XXX How do we deal with status != EFI_SUCCESS now? */
       return (status == EFI_SUCCESS) ? len : -1;
}


int
efinet_get(struct iodesc *desc, void *pkt, size_t len, saseconds_t timeout)
{
       struct netif *nif = desc->io_netif;
       EFI_SIMPLE_NETWORK *net;
       EFI_STATUS status;
       UINTN bufsz;
       time_t t;
       char buf[2048];

       net = nif->nif_devdata;

       t = time(0);
       while ((time(0) - t) < timeout) {
               bufsz = sizeof(buf);
               status = net->Receive(net, 0, &bufsz, buf, 0, 0, 0);
               if (status == EFI_SUCCESS) {
                       /*
                        * XXX EFI1.1 and the E1000 card trash our
                        * workspace if we do not do this silly copy.
                        * Either they are not respecting the len
                        * value or do not like the alignment.
                        */
                       if (bufsz > len)
                               bufsz = len;
                       memcpy(pkt, buf, bufsz);
                       return bufsz;
               }
               if (status != EFI_NOT_READY)
                       return 0;
       }

       return 0;
}

void
efinet_init(struct iodesc *desc, void *machdep_hint)
{
       struct netif *nif = desc->io_netif;
       EFI_SIMPLE_NETWORK *net;
       EFI_STATUS status;

       net = nif->nif_driver->netif_ifs[nif->nif_unit].dif_private;
       nif->nif_devdata = net;

       if (net->Mode->State == EfiSimpleNetworkStopped) {
               status = net->Start(net);
               if (status != EFI_SUCCESS) {
                       printf("net%d: cannot start interface (status=%ld)\n",
                           nif->nif_unit, status);
                       return;
               }
       }

       if (net->Mode->State != EfiSimpleNetworkInitialized) {
               status = net->Initialize(net, 0, 0);
               if (status != EFI_SUCCESS) {
                       printf("net%d: cannot init. interface (status=%ld)\n",
                           nif->nif_unit, status);
                       return;
               }
       }

       if (net->Mode->ReceiveFilterSetting == 0) {
               UINT32 mask = EFI_SIMPLE_NETWORK_RECEIVE_UNICAST |
                   EFI_SIMPLE_NETWORK_RECEIVE_BROADCAST;

               status = net->ReceiveFilters(net, mask, 0, FALSE, 0, 0);
               if (status != EFI_SUCCESS) {
                       printf("net%d: cannot set rx. filters (status=%ld)\n",
                           nif->nif_unit, status);
                       return;
               }
       }

#ifdef EFINET_DEBUG
       dump_mode(net->Mode);
#endif

       memcpy(desc->myea, net->Mode->CurrentAddress.Addr, 6);
       desc->xid = 1;

       return;
}

void
efinet_init_driver(void)
{
       EFI_STATUS      status;
       UINTN           sz;
       static EFI_GUID netid = EFI_SIMPLE_NETWORK_PROTOCOL_GUID;
       EFI_HANDLE      *handles;
       int             nifs, i;
#define MAX_INTERFACES  4
       static struct netif_dif difs[MAX_INTERFACES];
       static struct netif_stats stats[MAX_INTERFACES];

       sz = 0;
       status = BS->LocateHandle(ByProtocol, &netid, 0, &sz, 0);
       if (status != EFI_BUFFER_TOO_SMALL)
               return;
       handles = (EFI_HANDLE *) alloc(sz);
       status = BS->LocateHandle(ByProtocol, &netid, 0, &sz, handles);
       if (EFI_ERROR(status)) {
               free(handles);
               return;
       }

       nifs = sz / sizeof(EFI_HANDLE);
       if (nifs > MAX_INTERFACES)
               nifs = MAX_INTERFACES;

       efi_net.netif_nifs = nifs;
       efi_net.netif_ifs = difs;

       memset(stats, 0, sizeof(stats));
       for (i = 0; i < nifs; i++) {
               struct netif_dif *dif = &efi_net.netif_ifs[i];
               dif->dif_unit = i;
               dif->dif_nsel = 1;
               dif->dif_stats = &stats[i];

               BS->HandleProtocol(handles[i], &netid,
                                  (VOID**) &dif->dif_private);
       }

       return;
}

void
efinet_end(struct netif *nif)
{
       EFI_SIMPLE_NETWORK *net = nif->nif_devdata;

       net->Shutdown(net);
}