/*
 *  spreadspace avr utils
 *
 *
 *  Copyright (C) 2013-2018 Christian Pointner <equinox@spreadspace.org>
 *
 *  This file is part of spreadspace avr utils.
 *
 *  spreadspace avr utils 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
 *  any later version.
 *
 *  spreadspace avr utils 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 spreadspace avr utils. If not, see <http://www.gnu.org/licenses/>.
 */

#include <avr/io.h>
#include <avr/wdt.h>
#include <avr/interrupt.h>
#include <avr/power.h>
#include <string.h>

#include "util.h"
#include "led.h"

/*
             LUFA Library
     Copyright (C) Dean Camera, 2014.

  dean [at] fourwalledcubicle [dot] com
           www.lufa-lib.org
*/
#include <LUFA/Version.h>
#include <LUFA/Drivers/USB/USB.h>
#include "lufa-descriptor-rndis.h"

#define ADAPTER_MAC_ADDR {0x02, 0x00, 0x02, 0x00, 0x02, 0x00}
#define MY_MAC_ADDR      {0x01, 0x02, 0x03, 0x04, 0x05, 0x06}

static uint8_t RNDIS_Message_Buffer[192];

/** LUFA RNDIS Class driver interface configuration and state information. This structure is
 *  passed to all RNDIS Class driver functions, so that multiple instances of the same class
 *  within a device can be differentiated from one another.
 */
USB_ClassInfo_RNDIS_Device_t Ethernet_RNDIS_Interface =
  {
    .Config =
      {
        .ControlInterfaceNumber         = INTERFACE_ID_CDC_CCI,
        .DataINEndpoint                 =
          {
            .Address                = CDC_TX_EPADDR,
            .Size                   = CDC_TXRX_EPSIZE,
            .Banks                  = 1,
          },
        .DataOUTEndpoint                =
          {
            .Address                = CDC_RX_EPADDR,
            .Size                   = CDC_TXRX_EPSIZE,
            .Banks                  = 1,
          },
        .NotificationEndpoint           =
          {
            .Address                = CDC_NOTIFICATION_EPADDR,
            .Size                   = CDC_NOTIFICATION_EPSIZE,
            .Banks                  = 1,
          },
        .AdapterVendorDescription       = "LUFA RNDIS Demo Adapter",
        .AdapterMACAddress              = {ADAPTER_MAC_ADDR},
        .MessageBuffer                  = RNDIS_Message_Buffer,
        .MessageBufferLength            = sizeof(RNDIS_Message_Buffer),
      },
  };

/** Event handler for the library USB Configuration Changed event. */
void EVENT_USB_Device_ConfigurationChanged(void)
{
  RNDIS_Device_ConfigureEndpoints(&Ethernet_RNDIS_Interface);
}

/** Event handler for the library USB Control Request reception event. */
void EVENT_USB_Device_ControlRequest(void)
{
  RNDIS_Device_ProcessControlRequest(&Ethernet_RNDIS_Interface);
}

/* end LUFA RNDIS Class driver specific definitions*/


typedef uint8_t mac_addr_t[6];
typedef struct __attribute__((__packed__))
{
  mac_addr_t dest;
  mac_addr_t src;
  uint16_t ether_type;
} ethernet_frame_header_t;

typedef struct {
  uint8_t data[1550];
  uint16_t len;
} ethernet_buf_t;

ethernet_buf_t eth_recv_buf;
ethernet_frame_header_t* eth_recv_hdr = (ethernet_frame_header_t*)eth_recv_buf.data;
mac_addr_t my_mac = MY_MAC_ADDR;


int main(void)
{
  MCUSR &= ~(1 << WDRF);
  wdt_disable();

  cpu_init();
  led_init();
  USB_Init();
  sei();

  led_off();
  for (;;) {
    if (RNDIS_Device_IsPacketReceived(&Ethernet_RNDIS_Interface))
    {
      led_on();
      RNDIS_Device_ReadPacket(&Ethernet_RNDIS_Interface, eth_recv_buf.data, &(eth_recv_buf.len));

      memcpy(eth_recv_hdr->dest, eth_recv_hdr->src, sizeof(mac_addr_t));
      memcpy(eth_recv_hdr->src, my_mac, sizeof(mac_addr_t));

      RNDIS_Device_SendPacket(&Ethernet_RNDIS_Interface, eth_recv_buf.data, eth_recv_buf.len);
      led_off();
    }

    RNDIS_Device_USBTask(&Ethernet_RNDIS_Interface);
    USB_USBTask();
  }
}