/******************************************************************************
 *
 *  $Id$
 *
 *  Copyright (C) 2006  Florian Pose, Ingenieurgemeinschaft IgH
 *
 *  This file is part of the IgH EtherCAT Master.
 *
 *  The IgH EtherCAT Master 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; version 2 of the License.
 *
 *  The IgH EtherCAT Master 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 the IgH EtherCAT Master; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 *
 *****************************************************************************/

/**
   \file
   Ethernet-over-EtherCAT (EoE).
*/

/*****************************************************************************/

#include "../include/ecrt.h"
#include "globals.h"
#include "master.h"
#include "slave.h"
#include "mailbox.h"
#include "ethernet.h"

/*****************************************************************************/

/**
   Contains the private data of an EoE net_device.
*/

typedef struct
{
    struct net_device_stats stats; /**< device statistics */
    ec_eoe_t *eoe; /**< pointer to parent eoe object */
}
ec_eoedev_priv_t;

/*****************************************************************************/

void ec_eoedev_init(struct net_device *);
int ec_eoedev_open(struct net_device *);
int ec_eoedev_stop(struct net_device *);
int ec_eoedev_tx(struct sk_buff *, struct net_device *);
struct net_device_stats *ec_eoedev_stats(struct net_device *);

/*****************************************************************************/

/**
   EoE constructor.
*/

int ec_eoe_init(ec_eoe_t *eoe, ec_slave_t *slave)
{
    ec_eoedev_priv_t *priv;
    int result;

    eoe->slave = slave;
    eoe->rx_state = EC_EOE_IDLE;

    if (!(eoe->dev =
          alloc_netdev(sizeof(ec_eoedev_priv_t), "eoe%d", ec_eoedev_init))) {
        EC_ERR("Unable to allocate net_device for EoE object!\n");
        goto out_return;
    }

    // set EoE object reference
    priv = netdev_priv(eoe->dev);
    priv->eoe = eoe;

    // connect the net_device to the kernel
    if ((result = register_netdev(eoe->dev))) {
        EC_ERR("Unable to register net_device: error %i\n", result);
        goto out_free;
    }

    return 0;

 out_free:
    free_netdev(eoe->dev);
    eoe->dev = NULL;
 out_return:
    return -1;
}

/*****************************************************************************/

/**
   EoE destructor.
*/

void ec_eoe_clear(ec_eoe_t *eoe)
{
    if (eoe->dev) {
        unregister_netdev(eoe->dev);
        free_netdev(eoe->dev);
    }
}

/*****************************************************************************/

/**
   Runs the EoE state machine.
*/

void ec_eoe_run(ec_eoe_t *eoe)
{
    uint8_t *data;
    ec_master_t *master;
    size_t rec_size;
    unsigned int i;
    uint8_t fragment_number;
    uint8_t complete_size;
    uint8_t frame_number;
    uint8_t last_fragment;

    master = eoe->slave->master;

    if (eoe->rx_state == EC_EOE_IDLE) {
        ec_slave_mbox_prepare_check(eoe->slave);
        ec_master_queue_command(master, &eoe->slave->mbox_command);
        eoe->rx_state = EC_EOE_CHECKING;
        return;
    }

    if (eoe->rx_state == EC_EOE_CHECKING) {
        if (eoe->slave->mbox_command.state != EC_CMD_RECEIVED) {
            master->stats.eoe_errors++;
            eoe->rx_state = EC_EOE_IDLE;
            return;
        }
        if (!ec_slave_mbox_check(eoe->slave)) {
            eoe->rx_state = EC_EOE_IDLE;
            return;
        }
        ec_slave_mbox_prepare_fetch(eoe->slave);
        ec_master_queue_command(master, &eoe->slave->mbox_command);
        eoe->rx_state = EC_EOE_FETCHING;
        return;
    }

    if (eoe->rx_state == EC_EOE_FETCHING) {
        if (eoe->slave->mbox_command.state != EC_CMD_RECEIVED) {
            master->stats.eoe_errors++;
            eoe->rx_state = EC_EOE_IDLE;
            return;
        }
        if (!(data = ec_slave_mbox_fetch(eoe->slave, 0x02, &rec_size))) {
            master->stats.eoe_errors++;
            eoe->rx_state = EC_EOE_IDLE;
            return;
        }

        fragment_number = EC_READ_U16(data + 2) & 0x003F;
        complete_size = (EC_READ_U16(data + 2) >> 6) & 0x003F;
        frame_number = (EC_READ_U16(data + 2) >> 12) & 0x0003;
        last_fragment = (EC_READ_U16(data + 2) >> 15) & 0x0001;

        EC_DBG("EOE %s received, fragment: %i, complete size: %i (0x%02X),"
               " frame %i%s\n",
               fragment_number ? "fragment" : "initiate", fragment_number,
               (complete_size - 31) / 32, complete_size, frame_number,
               last_fragment ? ", last fragment" : "");
        EC_DBG("");
        for (i = 0; i < rec_size - 2; i++) {
            printk("%02X ", data[i + 2]);
            if ((i + 1) % 16 == 0) {
                printk("\n");
                EC_DBG("");
            }
        }
        printk("\n");

        eoe->rx_state = EC_EOE_IDLE;
        return;
    }
}

/*****************************************************************************/

/**
   Prints EoE object information.
*/

void ec_eoe_print(const ec_eoe_t *eoe)
{
    EC_INFO("  EoE slave %i\n", eoe->slave->ring_position);
    EC_INFO("    RX State %i\n", eoe->rx_state);
}

/*****************************************************************************/

/**
   Initializes a net_device structure for an EoE object.
*/

void ec_eoedev_init(struct net_device *dev /**< pointer to the net_device */)
{
    ec_eoedev_priv_t *priv;

    // initialize net_device
    ether_setup(dev);
    dev->open = ec_eoedev_open;
    dev->stop = ec_eoedev_stop;
    dev->hard_start_xmit = ec_eoedev_tx;
    dev->get_stats = ec_eoedev_stats;

    // initialize private data
    priv = netdev_priv(dev);
    memset(priv, 0, sizeof(ec_eoedev_priv_t));
}

/*****************************************************************************/

/**
   Opens the virtual network device.
*/

int ec_eoedev_open(struct net_device *dev /**< EoE net_device */)
{
    ec_eoedev_priv_t *priv = netdev_priv(dev);
    netif_start_queue(dev);
    EC_INFO("%s (slave %i) opened.\n", dev->name,
            priv->eoe->slave->ring_position);
    return 0;
}

/*****************************************************************************/

/**
   Stops the virtual network device.
*/

int ec_eoedev_stop(struct net_device *dev /**< EoE net_device */)
{
    ec_eoedev_priv_t *priv = netdev_priv(dev);
    netif_stop_queue(dev);
    EC_INFO("%s (slave %i) stopped.\n", dev->name,
            priv->eoe->slave->ring_position);
    return 0;
}

/*****************************************************************************/

/**
   Transmits data via the virtual network device.
*/

int ec_eoedev_tx(struct sk_buff *skb, /**< transmit socket buffer */
                 struct net_device *dev /**< EoE net_device */
                 )
{
    ec_eoedev_priv_t *priv = netdev_priv(dev);
    priv->stats.tx_packets++;
    dev_kfree_skb(skb);
    EC_INFO("EoE device sent %i octets.\n", skb->len);
    return 0;
}

/*****************************************************************************/

/**
   Gets statistics about the virtual network device.
*/

struct net_device_stats *ec_eoedev_stats(struct net_device *dev
                                         /**< EoE net_device */)
{
    ec_eoedev_priv_t *priv = netdev_priv(dev);
    return &priv->stats;
}

/*****************************************************************************/