/*
 * CPUFreq governor based on scheduler-provided CPU utilization data.
 *
 * Copyright (C) 2016, Intel Corporation
 * Author: Rafael J. Wysocki <rafael.j.wysocki@intel.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */

#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt

#include <linux/cpufreq.h>
#include <linux/slab.h>
#include <trace/events/power.h>

#include "sched.h"

struct sugov_tunables {
        struct gov_attr_set attr_set;
        unsigned int rate_limit_us;
};

struct sugov_policy {
        struct cpufreq_policy *policy;

        struct sugov_tunables *tunables;
        struct list_head tunables_hook;

        raw_spinlock_t update_lock;  /* For shared policies */
        u64 last_freq_update_time;
        s64 freq_update_delay_ns;
        unsigned int next_freq;

        /* The next fields are only needed if fast switch cannot be used. */
        struct irq_work irq_work;
        struct work_struct work;
        struct mutex work_lock;
        bool work_in_progress;

        bool need_freq_update;
};

struct sugov_cpu {
        struct update_util_data update_util;
        struct sugov_policy *sg_policy;

        unsigned int cached_raw_freq;
        unsigned long iowait_boost;
        unsigned long iowait_boost_max;
        u64 last_update;

        /* The fields below are only needed when sharing a policy. */
        unsigned long util;
        unsigned long max;
        unsigned int flags;
};

static DEFINE_PER_CPU(struct sugov_cpu, sugov_cpu);

/************************ Governor internals ***********************/

static bool sugov_should_update_freq(struct sugov_policy *sg_policy, u64 time)
{
        s64 delta_ns;

        if (sg_policy->work_in_progress)
                return false;

        if (unlikely(sg_policy->need_freq_update)) {
                sg_policy->need_freq_update = false;
                /*
                 * This happens when limits change, so forget the previous
                 * next_freq value and force an update.
                 */
                sg_policy->next_freq = UINT_MAX;
                return true;
        }

        delta_ns = time - sg_policy->last_freq_update_time;
        return delta_ns >= sg_policy->freq_update_delay_ns;
}

static void sugov_update_commit(struct sugov_policy *sg_policy, u64 time,
                                unsigned int next_freq)
{
        struct cpufreq_policy *policy = sg_policy->policy;

        sg_policy->last_freq_update_time = time;

        if (policy->fast_switch_enabled) {
                if (sg_policy->next_freq == next_freq) {
                        trace_cpu_frequency(policy->cur, smp_processor_id());
                        return;
                }
                sg_policy->next_freq = next_freq;
                next_freq = cpufreq_driver_fast_switch(policy, next_freq);
                if (next_freq == CPUFREQ_ENTRY_INVALID)
                        return;

                policy->cur = next_freq;
                trace_cpu_frequency(next_freq, smp_processor_id());
        } else if (sg_policy->next_freq != next_freq) {
                sg_policy->next_freq = next_freq;
                sg_policy->work_in_progress = true;
                irq_work_queue(&sg_policy->irq_work);
        }
}

/**
 * get_next_freq - Compute a new frequency for a given cpufreq policy.
 * @sg_cpu: schedutil cpu object to compute the new frequency for.
 * @util: Current CPU utilization.
 * @max: CPU capacity.
 *
 * If the utilization is frequency-invariant, choose the new frequency to be
 * proportional to it, that is
 *
 * next_freq = C * max_freq * util / max
 *
 * Otherwise, approximate the would-be frequency-invariant utilization by
 * util_raw * (curr_freq / max_freq) which leads to
 *
 * next_freq = C * curr_freq * util_raw / max
 *
 * Take C = 1.25 for the frequency tipping point at (util / max) = 0.8.
 *
 * The lowest driver-supported frequency which is equal or greater than the raw
 * next_freq (as calculated above) is returned, subject to policy min/max and
 * cpufreq driver limitations.
 */
static unsigned int get_next_freq(struct sugov_cpu *sg_cpu, unsigned long util,
                                  unsigned long max)
{
        struct sugov_policy *sg_policy = sg_cpu->sg_policy;
        struct cpufreq_policy *policy = sg_policy->policy;
        unsigned int freq = arch_scale_freq_invariant() ?
                                policy->cpuinfo.max_freq : policy->cur;

        freq = (freq + (freq >> 2)) * util / max;

        if (freq == sg_cpu->cached_raw_freq && sg_policy->next_freq != UINT_MAX)
                return sg_policy->next_freq;
        sg_cpu->cached_raw_freq = freq;
        return cpufreq_driver_resolve_freq(policy, freq);
}

static void sugov_get_util(unsigned long *util, unsigned long *max)
{
        struct rq *rq = this_rq();
        unsigned long cfs_max;

        cfs_max = arch_scale_cpu_capacity(NULL, smp_processor_id());

        *util = min(rq->cfs.avg.util_avg, cfs_max);
        *max = cfs_max;
}

static void sugov_set_iowait_boost(struct sugov_cpu *sg_cpu, u64 time,
                                   unsigned int flags)
{
        if (flags & SCHED_CPUFREQ_IOWAIT) {
                sg_cpu->iowait_boost = sg_cpu->iowait_boost_max;
        } else if (sg_cpu->iowait_boost) {
                s64 delta_ns = time - sg_cpu->last_update;

                /* Clear iowait_boost if the CPU apprears to have been idle. */
                if (delta_ns > TICK_NSEC)
                        sg_cpu->iowait_boost = 0;
        }
}

static void sugov_iowait_boost(struct sugov_cpu *sg_cpu, unsigned long *util,
                               unsigned long *max)
{
        unsigned long boost_util = sg_cpu->iowait_boost;
        unsigned long boost_max = sg_cpu->iowait_boost_max;

        if (!boost_util)
                return;

        if (*util * boost_max < *max * boost_util) {
                *util = boost_util;
                *max = boost_max;
        }
        sg_cpu->iowait_boost >>= 1;
}

static void sugov_update_single(struct update_util_data *hook, u64 time,
                                unsigned int flags)
{
        struct sugov_cpu *sg_cpu = container_of(hook, struct sugov_cpu, update_util);
        struct sugov_policy *sg_policy = sg_cpu->sg_policy;
        struct cpufreq_policy *policy = sg_policy->policy;
        unsigned long util, max;
        unsigned int next_f;

        sugov_set_iowait_boost(sg_cpu, time, flags);
        sg_cpu->last_update = time;

        if (!sugov_should_update_freq(sg_policy, time))
                return;

        if (flags & SCHED_CPUFREQ_RT_DL) {
                next_f = policy->cpuinfo.max_freq;
        } else {
                sugov_get_util(&util, &max);
                sugov_iowait_boost(sg_cpu, &util, &max);
                next_f = get_next_freq(sg_cpu, util, max);
        }
        sugov_update_commit(sg_policy, time, next_f);
}

static unsigned int sugov_next_freq_shared(struct sugov_cpu *sg_cpu,
                                           unsigned long util, unsigned long max,
                                           unsigned int flags)
{
        struct sugov_policy *sg_policy = sg_cpu->sg_policy;
        struct cpufreq_policy *policy = sg_policy->policy;
        unsigned int max_f = policy->cpuinfo.max_freq;
        u64 last_freq_update_time = sg_policy->last_freq_update_time;
        unsigned int j;

        if (flags & SCHED_CPUFREQ_RT_DL)
                return max_f;

        sugov_iowait_boost(sg_cpu, &util, &max);

        for_each_cpu(j, policy->cpus) {
                struct sugov_cpu *j_sg_cpu;
                unsigned long j_util, j_max;
                s64 delta_ns;

                if (j == smp_processor_id())
                        continue;

                j_sg_cpu = &per_cpu(sugov_cpu, j);
                /*
                 * If the CPU utilization was last updated before the previous
                 * frequency update and the time elapsed between the last update
                 * of the CPU utilization and the last frequency update is long
                 * enough, don't take the CPU into account as it probably is
                 * idle now (and clear iowait_boost for it).
                 */
                delta_ns = last_freq_update_time - j_sg_cpu->last_update;
                if (delta_ns > TICK_NSEC) {
                        j_sg_cpu->iowait_boost = 0;
                        continue;
                }
                if (j_sg_cpu->flags & SCHED_CPUFREQ_RT_DL)
                        return max_f;

                j_util = j_sg_cpu->util;
                j_max = j_sg_cpu->max;
                if (j_util * max > j_max * util) {
                        util = j_util;
                        max = j_max;
                }

                sugov_iowait_boost(j_sg_cpu, &util, &max);
        }

        return get_next_freq(sg_cpu, util, max);
}

static void sugov_update_shared(struct update_util_data *hook, u64 time,
                                unsigned int flags)
{
        struct sugov_cpu *sg_cpu = container_of(hook, struct sugov_cpu, update_util);
        struct sugov_policy *sg_policy = sg_cpu->sg_policy;
        unsigned long util, max;
        unsigned int next_f;

        sugov_get_util(&util, &max);

        raw_spin_lock(&sg_policy->update_lock);

        sg_cpu->util = util;
        sg_cpu->max = max;
        sg_cpu->flags = flags;

        sugov_set_iowait_boost(sg_cpu, time, flags);
        sg_cpu->last_update = time;

        if (sugov_should_update_freq(sg_policy, time)) {
                next_f = sugov_next_freq_shared(sg_cpu, util, max, flags);
                sugov_update_commit(sg_policy, time, next_f);
        }

        raw_spin_unlock(&sg_policy->update_lock);
}

static void sugov_work(struct work_struct *work)
{
        struct sugov_policy *sg_policy = container_of(work, struct sugov_policy, work);

        mutex_lock(&sg_policy->work_lock);
        __cpufreq_driver_target(sg_policy->policy, sg_policy->next_freq,
                                CPUFREQ_RELATION_L);
        mutex_unlock(&sg_policy->work_lock);

        sg_policy->work_in_progress = false;
}

static void sugov_irq_work(struct irq_work *irq_work)
{
        struct sugov_policy *sg_policy;

        sg_policy = container_of(irq_work, struct sugov_policy, irq_work);
        schedule_work_on(smp_processor_id(), &sg_policy->work);
}

/************************** sysfs interface ************************/

static struct sugov_tunables *global_tunables;
static DEFINE_MUTEX(global_tunables_lock);

static inline struct sugov_tunables *to_sugov_tunables(struct gov_attr_set *attr_set)
{
        return container_of(attr_set, struct sugov_tunables, attr_set);
}

static ssize_t rate_limit_us_show(struct gov_attr_set *attr_set, char *buf)
{
        struct sugov_tunables *tunables = to_sugov_tunables(attr_set);

        return sprintf(buf, "%u\n", tunables->rate_limit_us);
}

static ssize_t rate_limit_us_store(struct gov_attr_set *attr_set, const char *buf,
                                   size_t count)
{
        struct sugov_tunables *tunables = to_sugov_tunables(attr_set);
        struct sugov_policy *sg_policy;
        unsigned int rate_limit_us;

        if (kstrtouint(buf, 10, &rate_limit_us))
                return -EINVAL;

        tunables->rate_limit_us = rate_limit_us;

        list_for_each_entry(sg_policy, &attr_set->policy_list, tunables_hook)
                sg_policy->freq_update_delay_ns = rate_limit_us * NSEC_PER_USEC;

        return count;
}

static struct governor_attr rate_limit_us = __ATTR_RW(rate_limit_us);

static struct attribute *sugov_attributes[] = {
        &rate_limit_us.attr,
        NULL
};

static struct kobj_type sugov_tunables_ktype = {
        .default_attrs = sugov_attributes,
        .sysfs_ops = &governor_sysfs_ops,
};

/********************** cpufreq governor interface *********************/

static struct cpufreq_governor schedutil_gov;

static struct sugov_policy *sugov_policy_alloc(struct cpufreq_policy *policy)
{
        struct sugov_policy *sg_policy;

        sg_policy = kzalloc(sizeof(*sg_policy), GFP_KERNEL);
        if (!sg_policy)
                return NULL;

        sg_policy->policy = policy;
        init_irq_work(&sg_policy->irq_work, sugov_irq_work);
        INIT_WORK(&sg_policy->work, sugov_work);
        mutex_init(&sg_policy->work_lock);
        raw_spin_lock_init(&sg_policy->update_lock);
        return sg_policy;
}

static void sugov_policy_free(struct sugov_policy *sg_policy)
{
        mutex_destroy(&sg_policy->work_lock);
        kfree(sg_policy);
}

static struct sugov_tunables *sugov_tunables_alloc(struct sugov_policy *sg_policy)
{
        struct sugov_tunables *tunables;

        tunables = kzalloc(sizeof(*tunables), GFP_KERNEL);
        if (tunables) {
                gov_attr_set_init(&tunables->attr_set, &sg_policy->tunables_hook);
                if (!have_governor_per_policy())
                        global_tunables = tunables;
        }
        return tunables;
}

static void sugov_tunables_free(struct sugov_tunables *tunables)
{
        if (!have_governor_per_policy())
                global_tunables = NULL;

        kfree(tunables);
}

static int sugov_init(struct cpufreq_policy *policy)
{
        struct sugov_policy *sg_policy;
        struct sugov_tunables *tunables;
        unsigned int lat;
        int ret = 0;

        /* State should be equivalent to EXIT */
        if (policy->governor_data)
                return -EBUSY;

        sg_policy = sugov_policy_alloc(policy);
        if (!sg_policy)
                return -ENOMEM;

        mutex_lock(&global_tunables_lock);

        if (global_tunables) {
                if (WARN_ON(have_governor_per_policy())) {
                        ret = -EINVAL;
                        goto free_sg_policy;
                }
                policy->governor_data = sg_policy;
                sg_policy->tunables = global_tunables;

                gov_attr_set_get(&global_tunables->attr_set, &sg_policy->tunables_hook);
                goto out;
        }

        tunables = sugov_tunables_alloc(sg_policy);
        if (!tunables) {
                ret = -ENOMEM;
                goto free_sg_policy;
        }

        tunables->rate_limit_us = LATENCY_MULTIPLIER;
        lat = policy->cpuinfo.transition_latency / NSEC_PER_USEC;
        if (lat)
                tunables->rate_limit_us *= lat;

        policy->governor_data = sg_policy;
        sg_policy->tunables = tunables;

        ret = kobject_init_and_add(&tunables->attr_set.kobj, &sugov_tunables_ktype,
                                   get_governor_parent_kobj(policy), "%s",
                                   schedutil_gov.name);
        if (ret)
                goto fail;

 out:
        mutex_unlock(&global_tunables_lock);

        cpufreq_enable_fast_switch(policy);
        return 0;

 fail:
        policy->governor_data = NULL;
        sugov_tunables_free(tunables);

 free_sg_policy:
        mutex_unlock(&global_tunables_lock);

        sugov_policy_free(sg_policy);
        pr_err("initialization failed (error %d)\n", ret);
        return ret;
}

static void sugov_exit(struct cpufreq_policy *policy)
{
        struct sugov_policy *sg_policy = policy->governor_data;
        struct sugov_tunables *tunables = sg_policy->tunables;
        unsigned int count;

        cpufreq_disable_fast_switch(policy);

        mutex_lock(&global_tunables_lock);

        count = gov_attr_set_put(&tunables->attr_set, &sg_policy->tunables_hook);
        policy->governor_data = NULL;
        if (!count)
                sugov_tunables_free(tunables);

        mutex_unlock(&global_tunables_lock);

        sugov_policy_free(sg_policy);
}

static int sugov_start(struct cpufreq_policy *policy)
{
        struct sugov_policy *sg_policy = policy->governor_data;
        unsigned int cpu;

        sg_policy->freq_update_delay_ns = sg_policy->tunables->rate_limit_us * NSEC_PER_USEC;
        sg_policy->last_freq_update_time = 0;
        sg_policy->next_freq = UINT_MAX;
        sg_policy->work_in_progress = false;
        sg_policy->need_freq_update = false;

        for_each_cpu(cpu, policy->cpus) {
                struct sugov_cpu *sg_cpu = &per_cpu(sugov_cpu, cpu);

                sg_cpu->sg_policy = sg_policy;
                if (policy_is_shared(policy)) {
                        sg_cpu->util = 0;
                        sg_cpu->max = 0;
                        sg_cpu->flags = SCHED_CPUFREQ_RT;
                        sg_cpu->last_update = 0;
                        sg_cpu->cached_raw_freq = 0;
                        sg_cpu->iowait_boost = 0;
                        sg_cpu->iowait_boost_max = policy->cpuinfo.max_freq;
                        cpufreq_add_update_util_hook(cpu, &sg_cpu->update_util,
                                                     sugov_update_shared);
                } else {
                        cpufreq_add_update_util_hook(cpu, &sg_cpu->update_util,
                                                     sugov_update_single);
                }
        }
        return 0;
}

static void sugov_stop(struct cpufreq_policy *policy)
{
        struct sugov_policy *sg_policy = policy->governor_data;
        unsigned int cpu;

        for_each_cpu(cpu, policy->cpus)
                cpufreq_remove_update_util_hook(cpu);

        synchronize_sched();

        irq_work_sync(&sg_policy->irq_work);
        cancel_work_sync(&sg_policy->work);
}

static void sugov_limits(struct cpufreq_policy *policy)
{
        struct sugov_policy *sg_policy = policy->governor_data;

        if (!policy->fast_switch_enabled) {
                mutex_lock(&sg_policy->work_lock);
                cpufreq_policy_apply_limits(policy);
                mutex_unlock(&sg_policy->work_lock);
        }

        sg_policy->need_freq_update = true;
}

static struct cpufreq_governor schedutil_gov = {
        .name = "schedutil",
        .owner = THIS_MODULE,
        .init = sugov_init,
        .exit = sugov_exit,
        .start = sugov_start,
        .stop = sugov_stop,
        .limits = sugov_limits,
};

#ifdef CONFIG_CPU_FREQ_DEFAULT_GOV_SCHEDUTIL
struct cpufreq_governor *cpufreq_default_governor(void)
{
        return &schedutil_gov;
}
#endif

static int __init sugov_register(void)
{
        return cpufreq_register_governor(&schedutil_gov);
}
fs_initcall(sugov_register);