// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright 2019 Google Inc
 *
 * This program 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
 * 2 of the License, or (at your option) any later version.
 *
 * Provides a simple driver to control the ASPEED P2A interface which allows
 * the host to read and write to various regions of the BMC's memory.
 */

#include <linux/fs.h>
#include <linux/io.h>
#include <linux/mfd/syscon.h>
#include <linux/miscdevice.h>
#include <linux/mm.h>
#include <linux/module.h>
#include <linux/mutex.h>
#include <linux/of_address.h>
#include <linux/of_device.h>
#include <linux/platform_device.h>
#include <linux/regmap.h>
#include <linux/slab.h>
#include <linux/uaccess.h>

#include <linux/aspeed-p2a-ctrl.h>

#define DEVICE_NAME     "aspeed-p2a-ctrl"

/* SCU2C is a Misc. Control Register. */
#define SCU2C 0x2c
/* SCU180 is the PCIe Configuration Setting Control Register. */
#define SCU180 0x180
/* Bit 1 controls the P2A bridge, while bit 0 controls the entire VGA device
 * on the PCI bus.
 */
#define SCU180_ENP2A BIT(1)

/* The ast2400/2500 both have six ranges. */
#define P2A_REGION_COUNT 6

struct region {
        u64 min;
        u64 max;
        u32 bit;
};

struct aspeed_p2a_model_data {
        /* min, max, bit */
        struct region regions[P2A_REGION_COUNT];
};

struct aspeed_p2a_ctrl {
        struct miscdevice miscdev;
        struct regmap *regmap;

        const struct aspeed_p2a_model_data *config;

        /* Access to these needs to be locked, held via probe, mapping ioctl,
         * and release, remove.
         */
        struct mutex tracking;
        u32 readers;
        u32 readerwriters[P2A_REGION_COUNT];

        phys_addr_t mem_base;
        resource_size_t mem_size;
};

struct aspeed_p2a_user {
        struct file *file;
        struct aspeed_p2a_ctrl *parent;

        /* The entire memory space is opened for reading once the bridge is
         * enabled, therefore this needs only to be tracked once per user.
         * If any user has it open for read, the bridge must stay enabled.
         */
        u32 read;

        /* Each entry of the array corresponds to a P2A Region.  If the user
         * opens for read or readwrite, the reference goes up here.  On
         * release, this array is walked and references adjusted accordingly.
         */
        u32 readwrite[P2A_REGION_COUNT];
};

static void aspeed_p2a_enable_bridge(struct aspeed_p2a_ctrl *p2a_ctrl)
{
        regmap_update_bits(p2a_ctrl->regmap,
                SCU180, SCU180_ENP2A, SCU180_ENP2A);
}

static void aspeed_p2a_disable_bridge(struct aspeed_p2a_ctrl *p2a_ctrl)
{
        regmap_update_bits(p2a_ctrl->regmap, SCU180, SCU180_ENP2A, 0);
}

static int aspeed_p2a_mmap(struct file *file, struct vm_area_struct *vma)
{
        unsigned long vsize;
        pgprot_t prot;
        struct aspeed_p2a_user *priv = file->private_data;
        struct aspeed_p2a_ctrl *ctrl = priv->parent;

        if (ctrl->mem_base == 0 && ctrl->mem_size == 0)
                return -EINVAL;

        vsize = vma->vm_end - vma->vm_start;
        prot = vma->vm_page_prot;

        if (vma->vm_pgoff + vma_pages(vma) > ctrl->mem_size >> PAGE_SHIFT)
                return -EINVAL;

        /* ast2400/2500 AHB accesses are not cache coherent */
        prot = pgprot_noncached(prot);

        if (remap_pfn_range(vma, vma->vm_start,
                (ctrl->mem_base >> PAGE_SHIFT) + vma->vm_pgoff,
                vsize, prot))
                return -EAGAIN;

        return 0;
}

static bool aspeed_p2a_region_acquire(struct aspeed_p2a_user *priv,
                struct aspeed_p2a_ctrl *ctrl,
                struct aspeed_p2a_ctrl_mapping *map)
{
        int i;
        u64 base, end;
        bool matched = false;

        base = map->addr;
        end = map->addr + (map->length - 1);

        /* If the value is a legal u32, it will find a match. */
        for (i = 0; i < P2A_REGION_COUNT; i++) {
                const struct region *curr = &ctrl->config->regions[i];

                /* If the top of this region is lower than your base, skip it.
                 */
                if (curr->max < base)
                        continue;

                /* If the bottom of this region is higher than your end, bail.
                 */
                if (curr->min > end)
                        break;

                /* Lock this and update it, therefore it someone else is
                 * closing their file out, this'll preserve the increment.
                 */
                mutex_lock(&ctrl->tracking);
                ctrl->readerwriters[i] += 1;
                mutex_unlock(&ctrl->tracking);

                /* Track with the user, so when they close their file, we can
                 * decrement properly.
                 */
                priv->readwrite[i] += 1;

                /* Enable the region as read-write. */
                regmap_update_bits(ctrl->regmap, SCU2C, curr->bit, 0);
                matched = true;
        }

        return matched;
}

static long aspeed_p2a_ioctl(struct file *file, unsigned int cmd,
                unsigned long data)
{
        struct aspeed_p2a_user *priv = file->private_data;
        struct aspeed_p2a_ctrl *ctrl = priv->parent;
        void __user *arg = (void __user *)data;
        struct aspeed_p2a_ctrl_mapping map;

        if (copy_from_user(&map, arg, sizeof(map)))
                return -EFAULT;

        switch (cmd) {
        case ASPEED_P2A_CTRL_IOCTL_SET_WINDOW:
                /* If they want a region to be read-only, since the entire
                 * region is read-only once enabled, we just need to track this
                 * user wants to read from the bridge, and if it's not enabled.
                 * Enable it.
                 */
                if (map.flags == ASPEED_P2A_CTRL_READ_ONLY) {
                        mutex_lock(&ctrl->tracking);
                        ctrl->readers += 1;
                        mutex_unlock(&ctrl->tracking);

                        /* Track with the user, so when they close their file,
                         * we can decrement properly.
                         */
                        priv->read += 1;
                } else if (map.flags == ASPEED_P2A_CTRL_READWRITE) {
                        /* If we don't acquire any region return error. */
                        if (!aspeed_p2a_region_acquire(priv, ctrl, &map)) {
                                return -EINVAL;
                        }
                } else {
                        /* Invalid map flags. */
                        return -EINVAL;
                }

                aspeed_p2a_enable_bridge(ctrl);
                return 0;
        case ASPEED_P2A_CTRL_IOCTL_GET_MEMORY_CONFIG:
                /* This is a request for the memory-region and corresponding
                 * length that is used by the driver for mmap.
                 */

                map.flags = 0;
                map.addr = ctrl->mem_base;
                map.length = ctrl->mem_size;

                return copy_to_user(arg, &map, sizeof(map)) ? -EFAULT : 0;
        }

        return -EINVAL;
}


/*
 * When a user opens this file, we create a structure to track their mappings.
 *
 * A user can map a region as read-only (bridge enabled), or read-write (bit
 * flipped, and bridge enabled).  Either way, this tracking is used, s.t. when
 * they release the device references are handled.
 *
 * The bridge is not enabled until a user calls an ioctl to map a region,
 * simply opening the device does not enable it.
 */
static int aspeed_p2a_open(struct inode *inode, struct file *file)
{
        struct aspeed_p2a_user *priv;

        priv = kmalloc(sizeof(*priv), GFP_KERNEL);
        if (!priv)
                return -ENOMEM;

        priv->file = file;
        priv->read = 0;
        memset(priv->readwrite, 0, sizeof(priv->readwrite));

        /* The file's private_data is initialized to the p2a_ctrl. */
        priv->parent = file->private_data;

        /* Set the file's private_data to the user's data. */
        file->private_data = priv;

        return 0;
}

/*
 * This will close the users mappings.  It will go through what they had opened
 * for readwrite, and decrement those counts.  If at the end, this is the last
 * user, it'll close the bridge.
 */
static int aspeed_p2a_release(struct inode *inode, struct file *file)
{
        int i;
        u32 bits = 0;
        bool open_regions = false;
        struct aspeed_p2a_user *priv = file->private_data;

        /* Lock others from changing these values until everything is updated
         * in one pass.
         */
        mutex_lock(&priv->parent->tracking);

        priv->parent->readers -= priv->read;

        for (i = 0; i < P2A_REGION_COUNT; i++) {
                priv->parent->readerwriters[i] -= priv->readwrite[i];

                if (priv->parent->readerwriters[i] > 0)
                        open_regions = true;
                else
                        bits |= priv->parent->config->regions[i].bit;
        }

        /* Setting a bit to 1 disables the region, so let's just OR with the
         * above to disable any.
         */

        /* Note, if another user is trying to ioctl, they can't grab tracking,
         * and therefore can't grab either register mutex.
         * If another user is trying to close, they can't grab tracking either.
         */
        regmap_update_bits(priv->parent->regmap, SCU2C, bits, bits);

        /* If parent->readers is zero and open windows is 0, disable the
         * bridge.
         */
        if (!open_regions && priv->parent->readers == 0)
                aspeed_p2a_disable_bridge(priv->parent);

        mutex_unlock(&priv->parent->tracking);

        kfree(priv);

        return 0;
}

static const struct file_operations aspeed_p2a_ctrl_fops = {
        .owner = THIS_MODULE,
        .mmap = aspeed_p2a_mmap,
        .unlocked_ioctl = aspeed_p2a_ioctl,
        .open = aspeed_p2a_open,
        .release = aspeed_p2a_release,
};

/* The regions are controlled by SCU2C */
static void aspeed_p2a_disable_all(struct aspeed_p2a_ctrl *p2a_ctrl)
{
        int i;
        u32 value = 0;

        for (i = 0; i < P2A_REGION_COUNT; i++)
                value |= p2a_ctrl->config->regions[i].bit;

        regmap_update_bits(p2a_ctrl->regmap, SCU2C, value, value);

        /* Disable the bridge. */
        aspeed_p2a_disable_bridge(p2a_ctrl);
}

static int aspeed_p2a_ctrl_probe(struct platform_device *pdev)
{
        struct aspeed_p2a_ctrl *misc_ctrl;
        struct device *dev;
        struct resource resm;
        struct device_node *node;
        int rc = 0;

        dev = &pdev->dev;

        misc_ctrl = devm_kzalloc(dev, sizeof(*misc_ctrl), GFP_KERNEL);
        if (!misc_ctrl)
                return -ENOMEM;

        mutex_init(&misc_ctrl->tracking);

        /* optional. */
        node = of_parse_phandle(dev->of_node, "memory-region", 0);
        if (node) {
                rc = of_address_to_resource(node, 0, &resm);
                of_node_put(node);
                if (rc) {
                        dev_err(dev, "Couldn't address to resource for reserved memory\n");
                        return -ENODEV;
                }

                misc_ctrl->mem_size = resource_size(&resm);
                misc_ctrl->mem_base = resm.start;
        }

        misc_ctrl->regmap = syscon_node_to_regmap(pdev->dev.parent->of_node);
        if (IS_ERR(misc_ctrl->regmap)) {
                dev_err(dev, "Couldn't get regmap\n");
                return -ENODEV;
        }

        misc_ctrl->config = of_device_get_match_data(dev);

        dev_set_drvdata(&pdev->dev, misc_ctrl);

        aspeed_p2a_disable_all(misc_ctrl);

        misc_ctrl->miscdev.minor = MISC_DYNAMIC_MINOR;
        misc_ctrl->miscdev.name = DEVICE_NAME;
        misc_ctrl->miscdev.fops = &aspeed_p2a_ctrl_fops;
        misc_ctrl->miscdev.parent = dev;

        rc = misc_register(&misc_ctrl->miscdev);
        if (rc)
                dev_err(dev, "Unable to register device\n");

        return rc;
}

static int aspeed_p2a_ctrl_remove(struct platform_device *pdev)
{
        struct aspeed_p2a_ctrl *p2a_ctrl = dev_get_drvdata(&pdev->dev);

        misc_deregister(&p2a_ctrl->miscdev);

        return 0;
}

#define SCU2C_DRAM      BIT(25)
#define SCU2C_SPI       BIT(24)
#define SCU2C_SOC       BIT(23)
#define SCU2C_FLASH     BIT(22)

static const struct aspeed_p2a_model_data ast2400_model_data = {
        .regions = {
                {0x00000000, 0x17FFFFFF, SCU2C_FLASH},
                {0x18000000, 0x1FFFFFFF, SCU2C_SOC},
                {0x20000000, 0x2FFFFFFF, SCU2C_FLASH},
                {0x30000000, 0x3FFFFFFF, SCU2C_SPI},
                {0x40000000, 0x5FFFFFFF, SCU2C_DRAM},
                {0x60000000, 0xFFFFFFFF, SCU2C_SOC},
        }
};

static const struct aspeed_p2a_model_data ast2500_model_data = {
        .regions = {
                {0x00000000, 0x0FFFFFFF, SCU2C_FLASH},
                {0x10000000, 0x1FFFFFFF, SCU2C_SOC},
                {0x20000000, 0x3FFFFFFF, SCU2C_FLASH},
                {0x40000000, 0x5FFFFFFF, SCU2C_SOC},
                {0x60000000, 0x7FFFFFFF, SCU2C_SPI},
                {0x80000000, 0xFFFFFFFF, SCU2C_DRAM},
        }
};

static const struct of_device_id aspeed_p2a_ctrl_match[] = {
        { .compatible = "aspeed,ast2400-p2a-ctrl",
          .data = &ast2400_model_data },
        { .compatible = "aspeed,ast2500-p2a-ctrl",
          .data = &ast2500_model_data },
        { },
};

static struct platform_driver aspeed_p2a_ctrl_driver = {
        .driver = {
                .name           = DEVICE_NAME,
                .of_match_table = aspeed_p2a_ctrl_match,
        },
        .probe = aspeed_p2a_ctrl_probe,
        .remove = aspeed_p2a_ctrl_remove,
};

module_platform_driver(aspeed_p2a_ctrl_driver);

MODULE_DEVICE_TABLE(of, aspeed_p2a_ctrl_match);
MODULE_LICENSE("GPL");
MODULE_AUTHOR("Patrick Venture <venture@google.com>");
MODULE_DESCRIPTION("Control for aspeed 2400/2500 P2A VGA HOST to BMC mappings");