/*
 * This file is subject to the terms and conditions of the GNU General Public
 * License.  See the file "COPYING" in the main directory of this archive
 * for more details.
 *
 * (C) Copyright 2020 Hewlett Packard Enterprise Development LP
 * Copyright (C) 1999-2009 Silicon Graphics, Inc. All rights reserved.
 */

/*
 * Cross Partition Network Interface (XPNET) support
 *
 *      XPNET provides a virtual network layered on top of the Cross
 *      Partition communication layer.
 *
 *      XPNET provides direct point-to-point and broadcast-like support
 *      for an ethernet-like device.  The ethernet broadcast medium is
 *      replaced with a point-to-point message structure which passes
 *      pointers to a DMA-capable block that a remote partition should
 *      retrieve and pass to the upper level networking layer.
 *
 */

#include <linux/slab.h>
#include <linux/module.h>
#include <linux/netdevice.h>
#include <linux/etherdevice.h>
#include "xp.h"

/*
 * The message payload transferred by XPC.
 *
 * buf_pa is the physical address where the DMA should pull from.
 *
 * NOTE: for performance reasons, buf_pa should _ALWAYS_ begin on a
 * cacheline boundary.  To accomplish this, we record the number of
 * bytes from the beginning of the first cacheline to the first useful
 * byte of the skb (leadin_ignore) and the number of bytes from the
 * last useful byte of the skb to the end of the last cacheline
 * (tailout_ignore).
 *
 * size is the number of bytes to transfer which includes the skb->len
 * (useful bytes of the senders skb) plus the leadin and tailout
 */
struct xpnet_message {
        u16 version;            /* Version for this message */
        u16 embedded_bytes;     /* #of bytes embedded in XPC message */
        u32 magic;              /* Special number indicating this is xpnet */
        unsigned long buf_pa;   /* phys address of buffer to retrieve */
        u32 size;               /* #of bytes in buffer */
        u8 leadin_ignore;       /* #of bytes to ignore at the beginning */
        u8 tailout_ignore;      /* #of bytes to ignore at the end */
        unsigned char data;     /* body of small packets */
};

/*
 * Determine the size of our message, the cacheline aligned size,
 * and then the number of message will request from XPC.
 *
 * XPC expects each message to exist in an individual cacheline.
 */
#define XPNET_MSG_SIZE          XPC_MSG_PAYLOAD_MAX_SIZE
#define XPNET_MSG_DATA_MAX      \
                (XPNET_MSG_SIZE - offsetof(struct xpnet_message, data))
#define XPNET_MSG_NENTRIES      (PAGE_SIZE / XPC_MSG_MAX_SIZE)

#define XPNET_MAX_KTHREADS      (XPNET_MSG_NENTRIES + 1)
#define XPNET_MAX_IDLE_KTHREADS (XPNET_MSG_NENTRIES + 1)

/*
 * Version number of XPNET implementation. XPNET can always talk to versions
 * with same major #, and never talk to versions with a different version.
 */
#define _XPNET_VERSION(_major, _minor)  (((_major) << 4) | (_minor))
#define XPNET_VERSION_MAJOR(_v)         ((_v) >> 4)
#define XPNET_VERSION_MINOR(_v)         ((_v) & 0xf)

#define XPNET_VERSION _XPNET_VERSION(1, 0)      /* version 1.0 */
#define XPNET_VERSION_EMBED _XPNET_VERSION(1, 1)        /* version 1.1 */
#define XPNET_MAGIC     0x88786984      /* "XNET" */

#define XPNET_VALID_MSG(_m)                                                  \
   ((XPNET_VERSION_MAJOR(_m->version) == XPNET_VERSION_MAJOR(XPNET_VERSION)) \
    && (msg->magic == XPNET_MAGIC))

#define XPNET_DEVICE_NAME               "xp0"

/*
 * When messages are queued with xpc_send_notify, a kmalloc'd buffer
 * of the following type is passed as a notification cookie.  When the
 * notification function is called, we use the cookie to decide
 * whether all outstanding message sends have completed.  The skb can
 * then be released.
 */
struct xpnet_pending_msg {
        struct sk_buff *skb;
        atomic_t use_count;
};

static struct net_device *xpnet_device;

/*
 * When we are notified of other partitions activating, we add them to
 * our bitmask of partitions to which we broadcast.
 */
static unsigned long *xpnet_broadcast_partitions;
/* protect above */
static DEFINE_SPINLOCK(xpnet_broadcast_lock);

/*
 * Since the Block Transfer Engine (BTE) is being used for the transfer
 * and it relies upon cache-line size transfers, we need to reserve at
 * least one cache-line for head and tail alignment.  The BTE is
 * limited to 8MB transfers.
 *
 * Testing has shown that changing MTU to greater than 64KB has no effect
 * on TCP as the two sides negotiate a Max Segment Size that is limited
 * to 64K.  Other protocols May use packets greater than this, but for
 * now, the default is 64KB.
 */
#define XPNET_MAX_MTU (0x800000UL - L1_CACHE_BYTES)
/* 68 comes from min TCP+IP+MAC header */
#define XPNET_MIN_MTU 68
/* 32KB has been determined to be the ideal */
#define XPNET_DEF_MTU (0x8000UL)

/*
 * The partid is encapsulated in the MAC address beginning in the following
 * octet and it consists of two octets.
 */
#define XPNET_PARTID_OCTET      2

/* Define the XPNET debug device structures to be used with dev_dbg() et al */

static struct device_driver xpnet_dbg_name = {
        .name = "xpnet"
};

static struct device xpnet_dbg_subname = {
        .init_name = "",        /* set to "" */
        .driver = &xpnet_dbg_name
};

static struct device *xpnet = &xpnet_dbg_subname;

/*
 * Packet was recevied by XPC and forwarded to us.
 */
static void
xpnet_receive(short partid, int channel, struct xpnet_message *msg)
{
        struct sk_buff *skb;
        void *dst;
        enum xp_retval ret;

        if (!XPNET_VALID_MSG(msg)) {
                /*
                 * Packet with a different XPC version.  Ignore.
                 */
                xpc_received(partid, channel, (void *)msg);

                xpnet_device->stats.rx_errors++;

                return;
        }
        dev_dbg(xpnet, "received 0x%lx, %d, %d, %d\n", msg->buf_pa, msg->size,
                msg->leadin_ignore, msg->tailout_ignore);

        /* reserve an extra cache line */
        skb = dev_alloc_skb(msg->size + L1_CACHE_BYTES);
        if (!skb) {
                dev_err(xpnet, "failed on dev_alloc_skb(%d)\n",
                        msg->size + L1_CACHE_BYTES);

                xpc_received(partid, channel, (void *)msg);

                xpnet_device->stats.rx_errors++;

                return;
        }

        /*
         * The allocated skb has some reserved space.
         * In order to use xp_remote_memcpy(), we need to get the
         * skb->data pointer moved forward.
         */
        skb_reserve(skb, (L1_CACHE_BYTES - ((u64)skb->data &
                                            (L1_CACHE_BYTES - 1)) +
                          msg->leadin_ignore));

        /*
         * Update the tail pointer to indicate data actually
         * transferred.
         */
        skb_put(skb, (msg->size - msg->leadin_ignore - msg->tailout_ignore));

        /*
         * Move the data over from the other side.
         */
        if ((XPNET_VERSION_MINOR(msg->version) == 1) &&
            (msg->embedded_bytes != 0)) {
                dev_dbg(xpnet, "copying embedded message. memcpy(0x%p, 0x%p, "
                        "%lu)\n", skb->data, &msg->data,
                        (size_t)msg->embedded_bytes);

                skb_copy_to_linear_data(skb, &msg->data,
                                        (size_t)msg->embedded_bytes);
        } else {
                dst = (void *)((u64)skb->data & ~(L1_CACHE_BYTES - 1));
                dev_dbg(xpnet, "transferring buffer to the skb->data area;\n\t"
                        "xp_remote_memcpy(0x%p, 0x%p, %u)\n", dst,
                                          (void *)msg->buf_pa, msg->size);

                ret = xp_remote_memcpy(xp_pa(dst), msg->buf_pa, msg->size);
                if (ret != xpSuccess) {
                        /*
                         * !!! Need better way of cleaning skb.  Currently skb
                         * !!! appears in_use and we can't just call
                         * !!! dev_kfree_skb.
                         */
                        dev_err(xpnet, "xp_remote_memcpy(0x%p, 0x%p, 0x%x) "
                                "returned error=0x%x\n", dst,
                                (void *)msg->buf_pa, msg->size, ret);

                        xpc_received(partid, channel, (void *)msg);

                        xpnet_device->stats.rx_errors++;

                        return;
                }
        }

        dev_dbg(xpnet, "<skb->head=0x%p skb->data=0x%p skb->tail=0x%p "
                "skb->end=0x%p skb->len=%d\n", (void *)skb->head,
                (void *)skb->data, skb_tail_pointer(skb), skb_end_pointer(skb),
                skb->len);

        skb->protocol = eth_type_trans(skb, xpnet_device);
        skb->ip_summed = CHECKSUM_UNNECESSARY;

        dev_dbg(xpnet, "passing skb to network layer\n"
                "\tskb->head=0x%p skb->data=0x%p skb->tail=0x%p "
                "skb->end=0x%p skb->len=%d\n",
                (void *)skb->head, (void *)skb->data, skb_tail_pointer(skb),
                skb_end_pointer(skb), skb->len);

        xpnet_device->stats.rx_packets++;
        xpnet_device->stats.rx_bytes += skb->len + ETH_HLEN;

        netif_rx(skb);
        xpc_received(partid, channel, (void *)msg);
}

/*
 * This is the handler which XPC calls during any sort of change in
 * state or message reception on a connection.
 */
static void
xpnet_connection_activity(enum xp_retval reason, short partid, int channel,
                          void *data, void *key)
{
        DBUG_ON(partid < 0 || partid >= xp_max_npartitions);
        DBUG_ON(channel != XPC_NET_CHANNEL);

        switch (reason) {
        case xpMsgReceived:     /* message received */
                DBUG_ON(data == NULL);

                xpnet_receive(partid, channel, (struct xpnet_message *)data);
                break;

        case xpConnected:       /* connection completed to a partition */
                spin_lock_bh(&xpnet_broadcast_lock);
                __set_bit(partid, xpnet_broadcast_partitions);
                spin_unlock_bh(&xpnet_broadcast_lock);

                netif_carrier_on(xpnet_device);

                dev_dbg(xpnet, "%s connected to partition %d\n",
                        xpnet_device->name, partid);
                break;

        default:
                spin_lock_bh(&xpnet_broadcast_lock);
                __clear_bit(partid, xpnet_broadcast_partitions);
                spin_unlock_bh(&xpnet_broadcast_lock);

                if (bitmap_empty(xpnet_broadcast_partitions,
                                 xp_max_npartitions)) {
                        netif_carrier_off(xpnet_device);
                }

                dev_dbg(xpnet, "%s disconnected from partition %d\n",
                        xpnet_device->name, partid);
                break;
        }
}

static int
xpnet_dev_open(struct net_device *dev)
{
        enum xp_retval ret;

        dev_dbg(xpnet, "calling xpc_connect(%d, 0x%p, NULL, %ld, %ld, %ld, "
                "%ld)\n", XPC_NET_CHANNEL, xpnet_connection_activity,
                (unsigned long)XPNET_MSG_SIZE,
                (unsigned long)XPNET_MSG_NENTRIES,
                (unsigned long)XPNET_MAX_KTHREADS,
                (unsigned long)XPNET_MAX_IDLE_KTHREADS);

        ret = xpc_connect(XPC_NET_CHANNEL, xpnet_connection_activity, NULL,
                          XPNET_MSG_SIZE, XPNET_MSG_NENTRIES,
                          XPNET_MAX_KTHREADS, XPNET_MAX_IDLE_KTHREADS);
        if (ret != xpSuccess) {
                dev_err(xpnet, "ifconfig up of %s failed on XPC connect, "
                        "ret=%d\n", dev->name, ret);

                return -ENOMEM;
        }

        dev_dbg(xpnet, "ifconfig up of %s; XPC connected\n", dev->name);

        return 0;
}

static int
xpnet_dev_stop(struct net_device *dev)
{
        xpc_disconnect(XPC_NET_CHANNEL);

        dev_dbg(xpnet, "ifconfig down of %s; XPC disconnected\n", dev->name);

        return 0;
}

/*
 * Notification that the other end has received the message and
 * DMA'd the skb information.  At this point, they are done with
 * our side.  When all recipients are done processing, we
 * release the skb and then release our pending message structure.
 */
static void
xpnet_send_completed(enum xp_retval reason, short partid, int channel,
                     void *__qm)
{
        struct xpnet_pending_msg *queued_msg = (struct xpnet_pending_msg *)__qm;

        DBUG_ON(queued_msg == NULL);

        dev_dbg(xpnet, "message to %d notified with reason %d\n",
                partid, reason);

        if (atomic_dec_return(&queued_msg->use_count) == 0) {
                dev_dbg(xpnet, "all acks for skb->head=-x%p\n",
                        (void *)queued_msg->skb->head);

                dev_kfree_skb_any(queued_msg->skb);
                kfree(queued_msg);
        }
}

static void
xpnet_send(struct sk_buff *skb, struct xpnet_pending_msg *queued_msg,
           u64 start_addr, u64 end_addr, u16 embedded_bytes, int dest_partid)
{
        u8 msg_buffer[XPNET_MSG_SIZE];
        struct xpnet_message *msg = (struct xpnet_message *)&msg_buffer;
        u16 msg_size = sizeof(struct xpnet_message);
        enum xp_retval ret;

        msg->embedded_bytes = embedded_bytes;
        if (unlikely(embedded_bytes != 0)) {
                msg->version = XPNET_VERSION_EMBED;
                dev_dbg(xpnet, "calling memcpy(0x%p, 0x%p, 0x%lx)\n",
                        &msg->data, skb->data, (size_t)embedded_bytes);
                skb_copy_from_linear_data(skb, &msg->data,
                                          (size_t)embedded_bytes);
                msg_size += embedded_bytes - 1;
        } else {
                msg->version = XPNET_VERSION;
        }
        msg->magic = XPNET_MAGIC;
        msg->size = end_addr - start_addr;
        msg->leadin_ignore = (u64)skb->data - start_addr;
        msg->tailout_ignore = end_addr - (u64)skb_tail_pointer(skb);
        msg->buf_pa = xp_pa((void *)start_addr);

        dev_dbg(xpnet, "sending XPC message to %d:%d\n"
                "msg->buf_pa=0x%lx, msg->size=%u, "
                "msg->leadin_ignore=%u, msg->tailout_ignore=%u\n",
                dest_partid, XPC_NET_CHANNEL, msg->buf_pa, msg->size,
                msg->leadin_ignore, msg->tailout_ignore);

        atomic_inc(&queued_msg->use_count);

        ret = xpc_send_notify(dest_partid, XPC_NET_CHANNEL, XPC_NOWAIT, msg,
                              msg_size, xpnet_send_completed, queued_msg);
        if (unlikely(ret != xpSuccess))
                atomic_dec(&queued_msg->use_count);
}

/*
 * Network layer has formatted a packet (skb) and is ready to place it
 * "on the wire".  Prepare and send an xpnet_message to all partitions
 * which have connected with us and are targets of this packet.
 *
 * MAC-NOTE:  For the XPNET driver, the MAC address contains the
 * destination partid.  If the destination partid octets are 0xffff,
 * this packet is to be broadcast to all connected partitions.
 */
static netdev_tx_t
xpnet_dev_hard_start_xmit(struct sk_buff *skb, struct net_device *dev)
{
        struct xpnet_pending_msg *queued_msg;
        u64 start_addr, end_addr;
        short dest_partid;
        u16 embedded_bytes = 0;

        dev_dbg(xpnet, ">skb->head=0x%p skb->data=0x%p skb->tail=0x%p "
                "skb->end=0x%p skb->len=%d\n", (void *)skb->head,
                (void *)skb->data, skb_tail_pointer(skb), skb_end_pointer(skb),
                skb->len);

        if (skb->data[0] == 0x33) {
                dev_kfree_skb(skb);
                return NETDEV_TX_OK;    /* nothing needed to be done */
        }

        /*
         * The xpnet_pending_msg tracks how many outstanding
         * xpc_send_notifies are relying on this skb.  When none
         * remain, release the skb.
         */
        queued_msg = kmalloc_obj(struct xpnet_pending_msg, GFP_ATOMIC);
        if (queued_msg == NULL) {
                dev_warn(xpnet, "failed to kmalloc %ld bytes; dropping "
                         "packet\n", sizeof(struct xpnet_pending_msg));

                dev->stats.tx_errors++;
                dev_kfree_skb(skb);
                return NETDEV_TX_OK;
        }

        /* get the beginning of the first cacheline and end of last */
        start_addr = ((u64)skb->data & ~(L1_CACHE_BYTES - 1));
        end_addr = L1_CACHE_ALIGN((u64)skb_tail_pointer(skb));

        /* calculate how many bytes to embed in the XPC message */
        if (unlikely(skb->len <= XPNET_MSG_DATA_MAX)) {
                /* skb->data does fit so embed */
                embedded_bytes = skb->len;
        }

        /*
         * Since the send occurs asynchronously, we set the count to one
         * and begin sending.  Any sends that happen to complete before
         * we are done sending will not free the skb.  We will be left
         * with that task during exit.  This also handles the case of
         * a packet destined for a partition which is no longer up.
         */
        atomic_set(&queued_msg->use_count, 1);
        queued_msg->skb = skb;

        if (skb->data[0] == 0xff) {
                /* we are being asked to broadcast to all partitions */
                for_each_set_bit(dest_partid, xpnet_broadcast_partitions,
                             xp_max_npartitions) {

                        xpnet_send(skb, queued_msg, start_addr, end_addr,
                                   embedded_bytes, dest_partid);
                }
        } else {
                dest_partid = (short)skb->data[XPNET_PARTID_OCTET + 1];
                dest_partid |= (short)skb->data[XPNET_PARTID_OCTET + 0] << 8;

                if (dest_partid >= 0 &&
                    dest_partid < xp_max_npartitions &&
                    test_bit(dest_partid, xpnet_broadcast_partitions) != 0) {

                        xpnet_send(skb, queued_msg, start_addr, end_addr,
                                   embedded_bytes, dest_partid);
                }
        }

        dev->stats.tx_packets++;
        dev->stats.tx_bytes += skb->len;

        if (atomic_dec_return(&queued_msg->use_count) == 0) {
                dev_kfree_skb(skb);
                kfree(queued_msg);
        }

        return NETDEV_TX_OK;
}

/*
 * Deal with transmit timeouts coming from the network layer.
 */
static void
xpnet_dev_tx_timeout(struct net_device *dev, unsigned int txqueue)
{
        dev->stats.tx_errors++;
}

static const struct net_device_ops xpnet_netdev_ops = {
        .ndo_open               = xpnet_dev_open,
        .ndo_stop               = xpnet_dev_stop,
        .ndo_start_xmit         = xpnet_dev_hard_start_xmit,
        .ndo_tx_timeout         = xpnet_dev_tx_timeout,
        .ndo_set_mac_address    = eth_mac_addr,
        .ndo_validate_addr      = eth_validate_addr,
};

static int __init
xpnet_init(void)
{
        u8 addr[ETH_ALEN];
        int result;

        if (!is_uv_system())
                return -ENODEV;

        dev_info(xpnet, "registering network device %s\n", XPNET_DEVICE_NAME);

        xpnet_broadcast_partitions = bitmap_zalloc(xp_max_npartitions,
                                                   GFP_KERNEL);
        if (xpnet_broadcast_partitions == NULL)
                return -ENOMEM;

        /*
         * use ether_setup() to init the majority of our device
         * structure and then override the necessary pieces.
         */
        xpnet_device = alloc_netdev(0, XPNET_DEVICE_NAME, NET_NAME_UNKNOWN,
                                    ether_setup);
        if (xpnet_device == NULL) {
                bitmap_free(xpnet_broadcast_partitions);
                return -ENOMEM;
        }

        netif_carrier_off(xpnet_device);

        xpnet_device->netdev_ops = &xpnet_netdev_ops;
        xpnet_device->mtu = XPNET_DEF_MTU;
        xpnet_device->min_mtu = XPNET_MIN_MTU;
        xpnet_device->max_mtu = XPNET_MAX_MTU;

        memset(addr, 0, sizeof(addr));
        /*
         * Multicast assumes the LSB of the first octet is set for multicast
         * MAC addresses.  We chose the first octet of the MAC to be unlikely
         * to collide with any vendor's officially issued MAC.
         */
        addr[0] = 0x02;     /* locally administered, no OUI */

        addr[XPNET_PARTID_OCTET + 1] = xp_partition_id;
        addr[XPNET_PARTID_OCTET + 0] = (xp_partition_id >> 8);
        eth_hw_addr_set(xpnet_device, addr);

        /*
         * ether_setup() sets this to a multicast device.  We are
         * really not supporting multicast at this time.
         */
        xpnet_device->flags &= ~IFF_MULTICAST;

        /*
         * No need to checksum as it is a DMA transfer.  The BTE will
         * report an error if the data is not retrievable and the
         * packet will be dropped.
         */
        xpnet_device->features = NETIF_F_HW_CSUM;

        result = register_netdev(xpnet_device);
        if (result != 0) {
                free_netdev(xpnet_device);
                bitmap_free(xpnet_broadcast_partitions);
        }

        return result;
}

module_init(xpnet_init);

static void __exit
xpnet_exit(void)
{
        dev_info(xpnet, "unregistering network device %s\n",
                 xpnet_device[0].name);

        unregister_netdev(xpnet_device);
        free_netdev(xpnet_device);
        bitmap_free(xpnet_broadcast_partitions);
}

module_exit(xpnet_exit);

MODULE_AUTHOR("Silicon Graphics, Inc.");
MODULE_DESCRIPTION("Cross Partition Network adapter (XPNET)");
MODULE_LICENSE("GPL");