drivers/cpuquiet/driver.c

   1 /*
   2  * Copyright (c) 2012 NVIDIA CORPORATION.  All rights reserved.
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; version 2 of the License.
   7  *
   8  * This program is distributed in the hope that it will be useful, but WITHOUT
   9  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  10  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  11  * more details.
  12  *
  13  * You should have received a copy of the GNU General Public License along
  14  * with this program; if not, write to the Free Software Foundation, Inc.,
  15  * 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  16  *
  17  */
  18
  19 #include <linux/mutex.h>
  20 #include <linux/module.h>
  21 #include <linux/cpuquiet.h>
  22 #include <linux/cpu.h>
  23 #include <linux/jiffies.h>
  24 #include <linux/slab.h>
  25 #include <asm/cputime.h>
  26
  27 #include "cpuquiet.h"
  28
  29 struct cpuquiet_cpu_stat {
  30         cputime64_t time_up_total;
  31         u64 last_update;
  32         unsigned int up_down_count;
  33         struct kobject cpu_kobject;
  34 };
  35
  36 struct cpu_attribute {
  37         struct attribute attr;
  38         enum { up_down_count, time_up_total } type;
  39 };
  40
  41 static struct cpuquiet_driver *cpuquiet_curr_driver;
  42 struct cpuquiet_cpu_stat *stats;
  43
  44 #define CPU_ATTRIBUTE(_name) \
  45         static struct cpu_attribute _name ## _attr = {                  \
  46                 .attr =  {.name = __stringify(_name), .mode = 0444 },   \
  47                 .type   = _name,                                        \
  48 }
  49
  50 CPU_ATTRIBUTE(up_down_count);
  51 CPU_ATTRIBUTE(time_up_total);
  52
  53 static struct attribute *cpu_attributes[] = {
  54         &up_down_count_attr.attr,
  55         &time_up_total_attr.attr,
  56         NULL,
  57 };
  58
  59 static void stats_update(struct cpuquiet_cpu_stat *stat, bool up)
  60 {
  61         u64 cur_jiffies = get_jiffies_64();
  62         bool was_up = stat->up_down_count & 0x1;
  63
  64         if (was_up)
  65                 stat->time_up_total = cputime64_add(stat->time_up_total,
  66                         cputime64_sub(cur_jiffies, stat->last_update));
  67
  68         if (was_up != up)
  69                 stat->up_down_count++;
  70
  71         stat->last_update = cur_jiffies;
  72 }
  73
  74 int cpuquiet_quiesence_cpu(unsigned int cpunumber)
  75 {
  76         int err = -EPERM;
  77
  78         if (cpuquiet_curr_driver && cpuquiet_curr_driver->quiesence_cpu)
  79                 err = cpuquiet_curr_driver->quiesence_cpu(cpunumber);
  80
  81         stats_update(stats + cpunumber, 0);
  82
  83         return err;
  84 }
  85 EXPORT_SYMBOL(cpuquiet_quiesence_cpu);
  86
  87 int cpuquiet_wake_cpu(unsigned int cpunumber)
  88 {
  89         int err = -EPERM;
  90
  91         if (cpuquiet_curr_driver && cpuquiet_curr_driver->wake_cpu)
  92                 err = cpuquiet_curr_driver->wake_cpu(cpunumber);
  93
  94         stats_update(stats + cpunumber, 1);
  95
  96         return err;
  97 }
  98 EXPORT_SYMBOL(cpuquiet_wake_cpu);
  99
 100 static ssize_t stats_sysfs_show(struct kobject *kobj,
 101                         struct attribute *attr, char *buf)
 102 {
 103         struct cpu_attribute *cattr =
 104                 container_of(attr, struct cpu_attribute, attr);
 105         struct cpuquiet_cpu_stat *stat  =
 106                 container_of(kobj, struct cpuquiet_cpu_stat, cpu_kobject);
 107         ssize_t len = 0;
 108         bool was_up = stat->up_down_count & 0x1;
 109
 110         stats_update(stat, was_up);
 111
 112         switch (cattr->type) {
 113         case up_down_count:
 114                 len = sprintf(buf, "%u\n", stat->up_down_count);
 115                 break;
 116         case time_up_total:
 117                 len =  sprintf(buf, "%llu\n", stat->time_up_total);
 118                 break;
 119         }
 120
 121         return len;
 122 }
 123
 124 static const struct sysfs_ops stats_sysfs_ops = {
 125         .show = stats_sysfs_show,
 126 };
 127
 128 static struct kobj_type ktype_cpu_stats = {
 129         .sysfs_ops = &stats_sysfs_ops,
 130         .default_attrs = cpu_attributes,
 131 };
 132
 133 int cpuquiet_register_driver(struct cpuquiet_driver *drv)
 134 {
 135         int err = -EBUSY;
 136         unsigned int cpu;
 137         struct sys_device *sys_dev;
 138         u64 cur_jiffies;
 139
 140         if (!drv)
 141                 return -EINVAL;
 142
 143         stats = kzalloc(nr_cpu_ids * sizeof(*stats), GFP_KERNEL);
 144         if (!stats)
 145                 return -ENOMEM;
 146
 147         for_each_possible_cpu(cpu) {
 148                 cur_jiffies = get_jiffies_64();
 149                 stats[cpu].last_update = cur_jiffies;
 150                 if (cpu_online(cpu))
 151                         stats[cpu].up_down_count = 1;
 152                 sys_dev = get_cpu_sysdev(cpu);
 153                 if (sys_dev) {
 154                         cpuquiet_add_dev(sys_dev, cpu);
 155                         cpuquiet_cpu_kobject_init(&stats[cpu].cpu_kobject,
 156                                         &ktype_cpu_stats, "stats", cpu);
 157                 }
 158         }
 159
 160         mutex_lock(&cpuquiet_lock);
 161         if (!cpuquiet_curr_driver) {
 162                 err = 0;
 163                 cpuquiet_curr_driver = drv;
 164                 cpuquiet_switch_governor(cpuquiet_get_first_governor());
 165         }
 166         mutex_unlock(&cpuquiet_lock);
 167
 168         return err;
 169 }
 170 EXPORT_SYMBOL(cpuquiet_register_driver);
 171
 172 struct cpuquiet_driver *cpuquiet_get_driver(void)
 173 {
 174         return cpuquiet_curr_driver;
 175 }
 176
 177 void cpuquiet_unregister_driver(struct cpuquiet_driver *drv)
 178 {
 179         unsigned int cpu;
 180
 181         if (drv != cpuquiet_curr_driver) {
 182                 WARN(1, "invalid cpuquiet_unregister_driver(%s)\n",
 183                         drv->name);
 184                 return;
 185         }
 186
 187         /* stop current governor first */
 188         cpuquiet_switch_governor(NULL);
 189
 190         mutex_lock(&cpuquiet_lock);
 191         cpuquiet_curr_driver = NULL;
 192
 193         for_each_possible_cpu(cpu) {
 194                 kobject_put(&stats[cpu].cpu_kobject);
 195                 cpuquiet_remove_dev(cpu);
 196         }
 197
 198         mutex_unlock(&cpuquiet_lock);
 199 }
 200 EXPORT_SYMBOL(cpuquiet_unregister_driver);