soc: rockchip: pm_test: add driver handling Rockchip pm_test
authorFinley Xiao <finley.xiao@rock-chips.com>
Tue, 8 Nov 2016 10:05:53 +0000 (18:05 +0800)
committerHuang, Tao <huangtao@rock-chips.com>
Thu, 10 Nov 2016 06:29:29 +0000 (14:29 +0800)
Add /sys/pm_tests/ node, only for internal testing.

Change-Id: I1fe9f09e6d410c9ed939666255111d2a8865466c
Signed-off-by: Finley Xiao <finley.xiao@rock-chips.com>
drivers/soc/rockchip/Kconfig
drivers/soc/rockchip/Makefile
drivers/soc/rockchip/pm_test.c [new file with mode: 0644]

index 7140ff8..049dbe1 100644 (file)
@@ -3,6 +3,13 @@ if ARCH_ROCKCHIP || COMPILE_TEST
 #
 # Rockchip Soc drivers
 #
+config ROCKCHIP_PM_TEST
+       bool "Rockchip pm_test support"
+       default n
+       help
+         Say y here to enable pm_test support.
+         Add /sys/pm_tests/ node, only for internal testing.
+
 config ROCKCHIP_PM_DOMAINS
         bool "Rockchip generic power domain"
         depends on PM
index 286cda9..53a4bf8 100644 (file)
@@ -1,6 +1,7 @@
 #
 # Rockchip Soc drivers
 #
+obj-$(CONFIG_ROCKCHIP_PM_TEST) += pm_test.o
 obj-$(CONFIG_ROCKCHIP_PM_DOMAINS) += pm_domains.o
 obj-$(CONFIG_FIQ_DEBUGGER) += rk_fiq_debugger.o
 obj-$(CONFIG_MMC_DW_ROCKCHIP)  += sdmmc_vendor_storage.o
diff --git a/drivers/soc/rockchip/pm_test.c b/drivers/soc/rockchip/pm_test.c
new file mode 100644 (file)
index 0000000..4e26865
--- /dev/null
@@ -0,0 +1,624 @@
+/*
+ * Rockchip pm_test Driver
+ *
+ * Copyright (c) 2015 Rockchip Electronics Co. Ltd.
+ * Author: Finley Xiao <finley.xiao@rock-chips.com>
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of version 2 of the GNU General Public License as
+ * published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ * more details.
+ */
+
+#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
+
+#include <linux/clk.h>
+#include <linux/clk-provider.h>
+#include <linux/cpufreq.h>
+#include <linux/fs.h>
+#include <linux/module.h>
+#include <linux/regulator/driver.h>
+#include <linux/regulator/machine.h>
+#include <linux/uaccess.h>
+#include "../../../../drivers/regulator/internal.h"
+
+#define CLK_NR_CLKS 550
+
+static int cpu_usage_run;
+
+static DEFINE_PER_CPU(struct work_struct, work_cpu_usage);
+static DEFINE_PER_CPU(struct workqueue_struct *, workqueue_cpu_usage);
+
+static const char pi_result[] = "3141592653589793238462643383279528841971693993751058209749445923078164062862089986280348253421170679821480865132823664709384469555822317253594081284811174502841270193852115559644622948954930381964428810975665933446128475648233786783165271201991456485669234634861045432664821339360726024914127372458706606315588174881520920962829254917153643678925903611330530548820466521384146951941511609433057273657595919530921861173819326117931051185480744623799627495673518857527248912279381830119491298336733624406566438602139494639522473719070217986943702770539217176293176752384674818467669451320005681271452635608277857713427577896091736371787214684409012249534301465495853710579227968925892354201995611212902196864344181598136297747713099605187072113499999983729780499510597317328160963185";
+
+struct pm_attribute {
+       struct attribute attr;
+       ssize_t (*show)(struct kobject *kobj, struct kobj_attribute *attr,
+                       char *buf);
+       ssize_t (*store)(struct kobject *kobj, struct kobj_attribute *attr,
+                        const char *buf, size_t n);
+};
+
+static ssize_t clk_rate_show(struct kobject *kobj, struct kobj_attribute *attr,
+                            char *buf)
+{
+       char *str = buf;
+
+       str += sprintf(str, "get clk rate:\n");
+       str += sprintf(str,
+               "         echo get [clk_name] > /sys/pm_tests/clk_rate\n");
+       str += sprintf(str, "set clk rate:\n");
+       str += sprintf(str,
+               "         echo rawset [clk_name] [rate(Hz)] > /sys/pm_tests/clk_rate\n");
+       str += sprintf(str, "enable clk:\n");
+       str += sprintf(str,
+               "         echo open [clk_name] > /sys/pm_tests/clk_rate\n");
+       str += sprintf(str, "disable clk:\n");
+       str += sprintf(str,
+               "         echo close [clk_name] > /sys/pm_tests/clk_rate\n");
+       if (str != buf)
+               *(str - 1) = '\n';
+
+       return (str - buf);
+}
+
+static struct clk *clk_get_by_name(const char *clk_name)
+{
+       const char *name;
+       struct clk *clk;
+       struct device_node *np;
+       struct of_phandle_args clkspec;
+       int i;
+
+       np = of_find_node_by_name(NULL, "clock-controller");
+       clkspec.np = np;
+       clkspec.args_count = 1;
+       for (i = 1; i < CLK_NR_CLKS; i++) {
+               clkspec.args[0] = i;
+               clk = of_clk_get_from_provider(&clkspec);
+               if (IS_ERR_OR_NULL(clk))
+                       continue;
+               name = __clk_get_name(clk);
+               if (strlen(name) != strlen(clk_name))
+                       continue;
+               if (!strncmp(name, clk_name, strlen(clk_name)))
+                       break;
+       }
+
+       if (i == CLK_NR_CLKS)
+               clk = NULL;
+
+       return clk;
+}
+
+static int clk_rate_get(const char *buf)
+{
+       char cmd[20], clk_name[20];
+       unsigned long rate;
+       struct clk *clk;
+       int ret;
+
+       ret = sscanf(buf, "%s %s", cmd, clk_name);
+       if (ret != 2)
+               return -EINVAL;
+
+       clk = clk_get_by_name(clk_name);
+       if (IS_ERR_OR_NULL(clk)) {
+               pr_err("get clock error\n");
+               return PTR_ERR(clk);
+       }
+
+       rate = clk_get_rate(clk);
+
+       pr_info("%s %lu Hz\n", clk_name, rate);
+
+       clk_put(clk);
+
+       return 0;
+}
+
+static int clk_rate_rawset(const char *buf)
+{
+       char cmd[20], clk_name[20];
+       struct clk *clk;
+       unsigned long rate;
+       int ret;
+
+       ret = sscanf(buf, "%s %s %lu", cmd, clk_name, &rate);
+       if (ret != 3)
+               return -EINVAL;
+
+       clk = clk_get_by_name(clk_name);
+       if (IS_ERR_OR_NULL(clk)) {
+               pr_err("get %s error\n", clk_name);
+               return PTR_ERR(clk);
+       }
+
+       ret = clk_set_rate(clk, rate);
+       if (ret) {
+               pr_err("set %s rate %d error\n", clk_name, ret);
+               return ret;
+       }
+
+       pr_debug("%s %s %lu\n", cmd, clk_name, rate);
+
+       clk_put(clk);
+
+       return ret;
+}
+
+static int clk_open(const char *buf)
+{
+       char cmd[20], clk_name[20];
+       struct clk *clk;
+       int ret;
+
+       ret = sscanf(buf, "%s %s", cmd, clk_name);
+       if (ret != 2)
+               return -EINVAL;
+
+       clk = clk_get_by_name(clk_name);
+       if (IS_ERR_OR_NULL(clk)) {
+               pr_err("get clock %s error\n", clk_name);
+               return PTR_ERR(clk);
+       }
+
+       clk_prepare_enable(clk);
+
+       pr_debug("%s %s\n", cmd, clk_name);
+
+       clk_put(clk);
+
+       return 0;
+}
+
+static int clk_close(const char *buf)
+{
+       char cmd[20], clk_name[20];
+       struct clk *clk;
+       int ret;
+
+       ret = sscanf(buf, "%s %s", cmd, clk_name);
+       if (ret != 2)
+               return -EINVAL;
+
+       clk = clk_get_by_name(clk_name);
+       if (IS_ERR_OR_NULL(clk)) {
+               pr_err("get clock %s error\n", clk_name);
+               return PTR_ERR(clk);
+       }
+
+       clk_disable_unprepare(clk);
+
+       pr_debug("%s %s\n", cmd, clk_name);
+
+       clk_put(clk);
+
+       return 0;
+}
+
+static ssize_t clk_rate_store(struct kobject *kobj, struct kobj_attribute *attr,
+                             const char *buf, size_t n)
+{
+       char cmd[20];
+       int ret;
+
+       ret = sscanf(buf, "%s", cmd);
+       if (ret != 1)
+               return -EINVAL;
+
+       if (!strncmp(cmd, "get", strlen("get"))) {
+               ret = clk_rate_get(buf);
+               if (ret)
+                       pr_err("get clk err\n");
+       } else if (!strncmp(cmd, "rawset", strlen("rawset"))) {
+               ret = clk_rate_rawset(buf);
+               if (ret)
+                       pr_err("rawset clk err\n");
+       } else if (!strncmp(cmd, "open", strlen("open"))) {
+               ret = clk_open(buf);
+               if (ret)
+                       pr_err("open clk err\n");
+       } else if (!strncmp(cmd, "close", strlen("close"))) {
+               ret = clk_close(buf);
+               if (ret)
+                       pr_err("close clk err\n");
+       } else {
+               pr_info("unsupported cmd(%s)\n", cmd);
+       }
+
+       return n;
+}
+
+static ssize_t clk_volt_show(struct kobject *kobj, struct kobj_attribute *attr,
+                            char *buf)
+{
+       char *str = buf;
+
+       str += sprintf(str, "get voltage:\n");
+       str += sprintf(str,
+               "            echo get [regulaotr_name] > /sys/pm_tests/clk_volt\n");
+       str += sprintf(str, "set voltage:\n");
+       str += sprintf(str,
+               "            echo set [regulaotr_name] [voltage(uV)] > /sys/pm_tests/clk_volt\n");
+       if (str != buf)
+               *(str - 1) = '\n';
+
+       return (str - buf);
+}
+
+static int clk_volt_get(const char *buf)
+{
+       char cmd[20], reg_name[20];
+       unsigned int ret;
+       struct regulator *regulator;
+
+       ret = sscanf(buf, "%s %s", cmd, reg_name);
+       if (ret != 2)
+               return -EINVAL;
+
+       regulator = regulator_get(NULL, reg_name);
+       if (IS_ERR_OR_NULL(regulator)) {
+               pr_err("get regulator %s error\n", reg_name);
+               return PTR_ERR(regulator);
+       }
+
+       ret = regulator_get_voltage(regulator);
+
+       pr_info("%s %duV\n", reg_name, ret);
+
+       regulator_put(regulator);
+
+       return 0;
+}
+
+static void break_up_regulator_limit(struct regulator_dev *rdev, int min_uV,
+                                    int max_uV)
+{
+       struct regulator *regulator;
+
+       list_for_each_entry(regulator, &rdev->consumer_list, list) {
+               if (!regulator->min_uV && !regulator->max_uV)
+                       continue;
+               pr_debug("%s min=%d, max=%d", dev_name(regulator->dev),
+                        regulator->min_uV, regulator->max_uV);
+               regulator->min_uV = min_uV;
+               regulator->max_uV = max_uV;
+       }
+}
+
+static int clk_volt_set(const char *buf)
+{
+       char cmd[20], reg_name[20];
+       unsigned int ret;
+       struct regulator *regulator;
+       unsigned int volt;
+       struct regulator_dev *rdev;
+       int max_uV;
+
+       ret = sscanf(buf, "%s %s %u", cmd, reg_name, &volt);
+       if (ret != 3)
+               return -EINVAL;
+
+       regulator = regulator_get(NULL, reg_name);
+       if (IS_ERR_OR_NULL(regulator)) {
+               pr_info("get regulator %s error\n", reg_name);
+               return PTR_ERR(regulator);
+       }
+
+       rdev = regulator->rdev;
+       max_uV = rdev->constraints->max_uV;
+
+       if (volt > max_uV) {
+               pr_err("invalid volt, max %d uV\n", max_uV);
+               return -EINVAL;
+       }
+
+       mutex_lock(&rdev->mutex);
+       break_up_regulator_limit(rdev, volt, max_uV);
+       mutex_unlock(&rdev->mutex);
+
+       ret = regulator_set_voltage(regulator, volt, max_uV);
+       if (ret) {
+               pr_err("set voltage %d error\n", ret);
+               regulator_put(regulator);
+               return ret;
+       }
+
+       pr_debug("set %s %duV\n", reg_name, volt);
+
+       regulator_put(regulator);
+
+       return 0;
+}
+
+static ssize_t clk_volt_store(struct kobject *kobj, struct kobj_attribute *attr,
+                             const char *buf, size_t n)
+{
+       unsigned int ret;
+       char cmd[20];
+
+       ret = sscanf(buf, "%s", cmd);
+       if (ret != 1)
+               return -EINVAL;
+
+       if (!strncmp(cmd, "get", strlen("get"))) {
+               ret = clk_volt_get(buf);
+               if (ret)
+                       pr_err("get volt err\n");
+       } else if (!strncmp(cmd, "set", strlen("set"))) {
+               ret = clk_volt_set(buf);
+               if (ret)
+                       pr_err("set volt err\n");
+       } else {
+               pr_err("unsupported cmd(%s)\n", cmd);
+       }
+
+       return n;
+}
+
+static ssize_t cpu_usage_show(struct kobject *kobj, struct kobj_attribute *attr,
+                             char *buf)
+{
+       char *str = buf;
+
+       str += sprintf(str,
+               "while(1) test:\n");
+       str += sprintf(str,
+               "              cpu 0-3:  echo start 0 0 > /sys/pm_tests/cpu_usage\n");
+       str += sprintf(str,
+               "              cpu 4-7:  echo start 0 1 > /sys/pm_tests/cpu_usage\n");
+       str += sprintf(str,
+               "              cpu 0-7:  echo start 0 2 > /sys/pm_tests/cpu_usage\n");
+       str += sprintf(str,
+               "calc_pi  test:\n");
+       str += sprintf(str,
+               "              cpu 0-3:  echo start 1 0 > /sys/pm_tests/cpu_usage\n");
+       str += sprintf(str,
+               "              cpu 4-7:  echo start 1 1 > /sys/pm_tests/cpu_usage\n");
+       str += sprintf(str,
+               "              cpu 0-7:  echo start 1 2 > /sys/pm_tests/cpu_usage\n");
+       str += sprintf(str,
+               "stop     test:\n");
+       str += sprintf(str,
+               "              cpu 0-7:  echo stop -1 -1 > /sys/pm_tests/cpu_usage\n");
+       if (str != buf)
+               *(str - 1) = '\n';
+
+       return (str - buf);
+}
+
+static inline int calc_pi(void)
+{
+       int bit = 0, i = 0;
+       long a = 10000, b = 0, c = 2800, d = 0, e = 0, g = 0;
+       int *result;
+       long *f;
+       int len = 0;
+       char *pi_calc, *pi_temp;
+       char *pi_just = (char *)&pi_result[0];
+       size_t pi_just_size = sizeof(pi_result);
+
+       result = vmalloc(10000 * sizeof(int));
+       if (!result)
+               return -ENOMEM;
+
+        f = vmalloc(2801 * sizeof(long));
+       if (!f)
+               return -ENOMEM;
+
+        pi_calc = vmalloc(1000 * sizeof(char));
+       if (!pi_calc)
+               return -ENOMEM;
+
+       for (; b - c; )
+               f[b++] = a / 5;
+       for (; d = 0, g = c * 2; c -= 14, result[bit++] = e + d / a, e = d % a)
+               for (b = c; d += f[b] * a, f[b] = d % --g, d /= g--, --b;
+                    d *= b)
+                       ;
+
+       pi_temp = pi_calc;
+       for (i = 0; i < bit; i++)
+               len += sprintf(pi_temp + len, "%d", result[i]);
+
+       if (strncmp(pi_just, pi_calc, pi_just_size) == 0) {
+               vfree(result);
+               vfree(f);
+               vfree(pi_calc);
+       } else {
+               vfree(result);
+               vfree(f);
+               vfree(pi_calc);
+
+               while (1)
+                       pr_err("calc_pi error\n");
+       }
+
+       return 0;
+}
+
+static void calc_pi2(void)
+{
+       calc_pi();
+       calc_pi();
+}
+
+static void calc_pi4(void)
+{
+       calc_pi2();
+       calc_pi2();
+}
+
+static void calc_pi8(void)
+{
+       calc_pi4();
+       calc_pi4();
+}
+
+static void calc_pi16(void)
+{
+       calc_pi8();
+       calc_pi8();
+}
+
+/* 0xffffffc000097468 - 0xffffffc0000949c4 =  10916 byte*/
+static void calc_pi32(void)
+{
+       calc_pi16();
+       calc_pi16();
+}
+
+/* 0xffffffc000099ee8 - 0xffffffc0000949c4 =  21796 byte*/
+static void calc_pi64(void)
+{
+       calc_pi32();
+       calc_pi32();
+}
+
+/* 0xffffffc00009f3e8 - 0xffffffc0000949c4 =  43556 byte*/
+static void calc_pi128(void)
+{
+       calc_pi64();
+       calc_pi64();
+}
+
+/* 0xffffffc0000a9de8 - 0xffffffc0000949c4 =  87076 byte*/
+static void calc_pi256(void)
+{
+       calc_pi128();
+       calc_pi128();
+}
+
+/* 0xffffffc0000bf1e8 - 0xffffffc0000949c4 =  174116 byte*/
+static void calc_pi512(void)
+{
+       calc_pi256();
+       calc_pi256();
+}
+
+/* 0xffffffc0000ea9e8 - 0xffffffc0000949c4 =  352295 byte*/
+static void calc_pi1024(void)
+{
+       calc_pi512();
+       calc_pi512();
+}
+
+static void calc_pi_large(void)
+{
+       calc_pi1024();
+}
+
+static void handler_cpu_usage(struct work_struct *work)
+{
+       while (cpu_usage_run == 0)
+               barrier();
+
+       while (cpu_usage_run == 1)
+               calc_pi_large();
+}
+
+static ssize_t cpu_usage_store(struct kobject *kobj,
+                              struct kobj_attribute *attr,
+                              const char *buf, size_t n)
+{
+       struct workqueue_struct *workqueue;
+       struct work_struct *work;
+       char cmd[20];
+       int cpu;
+       int ret;
+       int usage = 0;
+       int cluster = 0;
+
+       ret = sscanf(buf, "%s %d %d", cmd, &usage, &cluster);
+       if (ret != 3)
+               return -EINVAL;
+
+       if (!strncmp(cmd, "start", strlen("start"))) {
+               if (usage == 0)
+                       pr_info("start while(1) test\n");
+               else if (usage == 1)
+                       pr_info("start pi test\n");
+               else
+                       return 0;
+
+               cpu_usage_run = usage;
+               for_each_online_cpu(cpu) {
+                       work = &per_cpu(work_cpu_usage, cpu);
+                       workqueue = per_cpu(workqueue_cpu_usage, cpu);
+                       if (!work || !workqueue) {
+                               pr_err("work or workqueue NULL\n");
+                               return n;
+                       }
+
+                       if (cluster == 0 && cpu < 4)
+                               queue_work_on(cpu, workqueue, work);
+                       else if (cluster == 1 && cpu >= 4)
+                               queue_work_on(cpu, workqueue, work);
+                       else if (cluster == 2)
+                               queue_work_on(cpu, workqueue, work);
+               }
+       } else if (!strncmp(cmd, "stop", strlen("stop"))) {
+               if (cpu_usage_run == 0)
+                       pr_info("stop while(1) test\n");
+               else if (cpu_usage_run == 1)
+                       pr_info("stop pi test\n");
+               cpu_usage_run = usage;
+       }
+
+       return n;
+}
+
+static struct pm_attribute pm_attrs[] = {
+       __ATTR(clk_rate, 0644, clk_rate_show, clk_rate_store),
+       __ATTR(clk_volt, 0644, clk_volt_show, clk_volt_store),
+       __ATTR(cpu_usage, 0644, cpu_usage_show, cpu_usage_store),
+};
+
+static int __init pm_test_init(void)
+{
+       struct kobject *kobject;
+       struct workqueue_struct *workqueue;
+       struct work_struct *work;
+       int i, cpu, ret = 0;
+
+       kobject = kobject_create_and_add("pm_tests", NULL);
+       if (!kobject)
+               return -ENOMEM;
+
+       for (i = 0; i < ARRAY_SIZE(pm_attrs); i++) {
+               ret = sysfs_create_file(kobject, &pm_attrs[i].attr);
+               if (ret) {
+                       pr_err("create file index %d error\n", i);
+                       return ret;
+               }
+       }
+
+       workqueue = create_workqueue("workqueue_cpu_usage");
+       if (!workqueue) {
+               pr_err("workqueue NULL\n");
+               return -EINVAL;
+       }
+
+       for_each_online_cpu(cpu) {
+               work = &per_cpu(work_cpu_usage, cpu);
+               if (!work) {
+                       pr_err("work NULL\n");
+                       return -EINVAL;
+               }
+               INIT_WORK(work, handler_cpu_usage);
+               per_cpu(workqueue_cpu_usage, cpu) = workqueue;
+       }
+
+       return ret;
+}
+
+late_initcall(pm_test_init);
+
+MODULE_DESCRIPTION("Rockchip pm_test driver");
+MODULE_AUTHOR("Finley Xiao <finley.xiao@rock-chips.com>");
+MODULE_LICENSE("GPL v2");