// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * MPIC timer wakeup driver
 *
 * Copyright 2013 Freescale Semiconductor, Inc.
 */

#include <linux/kernel.h>
#include <linux/slab.h>
#include <linux/errno.h>
#include <linux/module.h>
#include <linux/interrupt.h>
#include <linux/device.h>

#include <asm/mpic_timer.h>
#include <asm/mpic.h>

struct fsl_mpic_timer_wakeup {
        struct mpic_timer *timer;
        struct work_struct free_work;
};

static struct fsl_mpic_timer_wakeup *fsl_wakeup;
static DEFINE_MUTEX(sysfs_lock);

static void fsl_free_resource(struct work_struct *ws)
{
        struct fsl_mpic_timer_wakeup *wakeup =
                container_of(ws, struct fsl_mpic_timer_wakeup, free_work);

        mutex_lock(&sysfs_lock);

        if (wakeup->timer) {
                disable_irq_wake(wakeup->timer->irq);
                mpic_free_timer(wakeup->timer);
        }

        wakeup->timer = NULL;
        mutex_unlock(&sysfs_lock);
}

static irqreturn_t fsl_mpic_timer_irq(int irq, void *dev_id)
{
        struct fsl_mpic_timer_wakeup *wakeup = dev_id;

        schedule_work(&wakeup->free_work);

        return wakeup->timer ? IRQ_HANDLED : IRQ_NONE;
}

static ssize_t fsl_timer_wakeup_show(struct device *dev,
                                struct device_attribute *attr,
                                char *buf)
{
        time64_t interval = 0;

        mutex_lock(&sysfs_lock);
        if (fsl_wakeup->timer) {
                mpic_get_remain_time(fsl_wakeup->timer, &interval);
                interval++;
        }
        mutex_unlock(&sysfs_lock);

        return sprintf(buf, "%lld\n", interval);
}

static ssize_t fsl_timer_wakeup_store(struct device *dev,
                                struct device_attribute *attr,
                                const char *buf,
                                size_t count)
{
        time64_t interval;
        int ret;

        if (kstrtoll(buf, 0, &interval))
                return -EINVAL;

        guard(mutex)(&sysfs_lock);

        if (fsl_wakeup->timer) {
                disable_irq_wake(fsl_wakeup->timer->irq);
                mpic_free_timer(fsl_wakeup->timer);
                fsl_wakeup->timer = NULL;
        }

        if (!interval)
                return count;

        fsl_wakeup->timer = mpic_request_timer(fsl_mpic_timer_irq,
                                                fsl_wakeup, interval);
        if (!fsl_wakeup->timer)
                return -EINVAL;

        ret = enable_irq_wake(fsl_wakeup->timer->irq);
        if (ret) {
                mpic_free_timer(fsl_wakeup->timer);
                fsl_wakeup->timer = NULL;
                return ret;
        }

        mpic_start_timer(fsl_wakeup->timer);

        return count;
}

static struct device_attribute mpic_attributes = __ATTR(timer_wakeup, 0644,
                        fsl_timer_wakeup_show, fsl_timer_wakeup_store);

static int __init fsl_wakeup_sys_init(void)
{
        struct device *dev_root;
        int ret = -EINVAL;

        fsl_wakeup = kzalloc_obj(struct fsl_mpic_timer_wakeup);
        if (!fsl_wakeup)
                return -ENOMEM;

        INIT_WORK(&fsl_wakeup->free_work, fsl_free_resource);

        dev_root = bus_get_dev_root(&mpic_subsys);
        if (dev_root) {
                ret = device_create_file(dev_root, &mpic_attributes);
                put_device(dev_root);
                if (ret)
                        kfree(fsl_wakeup);
        }

        return ret;
}

static void __exit fsl_wakeup_sys_exit(void)
{
        struct device *dev_root;

        dev_root = bus_get_dev_root(&mpic_subsys);
        if (dev_root) {
                device_remove_file(dev_root, &mpic_attributes);
                put_device(dev_root);
        }

        mutex_lock(&sysfs_lock);

        if (fsl_wakeup->timer) {
                disable_irq_wake(fsl_wakeup->timer->irq);
                mpic_free_timer(fsl_wakeup->timer);
        }

        kfree(fsl_wakeup);

        mutex_unlock(&sysfs_lock);
}

module_init(fsl_wakeup_sys_init);
module_exit(fsl_wakeup_sys_exit);

MODULE_DESCRIPTION("Freescale MPIC global timer wakeup driver");
MODULE_LICENSE("GPL v2");
MODULE_AUTHOR("Wang Dongsheng <dongsheng.wang@freescale.com>");