[mirror_ubuntu-artful-kernel.git] / drivers / power / test_power.c

/*
 * Power supply driver for testing.
 *
 * Copyright 2010  Anton Vorontsov <cbouatmailru@gmail.com>
 *
 * Dynamic module parameter code from the Virtual Battery Driver
 * Copyright (C) 2008 Pylone, Inc.
 * By: Masashi YOKOTA <yokota@pylone.jp>
 * Originally found here:
 * http://downloads.pylone.jp/src/virtual_battery/virtual_battery-0.0.1.tar.bz2
 *
 * 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.
 */

#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/power_supply.h>
#include <linux/errno.h>
#include <linux/delay.h>
#include <linux/vermagic.h>

static int ac_online			= 1;
static int usb_online			= 1;
static int battery_status		= POWER_SUPPLY_STATUS_DISCHARGING;
static int battery_health		= POWER_SUPPLY_HEALTH_GOOD;
static int battery_present		= 1; /* true */
static int battery_technology		= POWER_SUPPLY_TECHNOLOGY_LION;
static int battery_capacity		= 50;
static int battery_voltage		= 3300;

static bool module_initialized;

static int test_power_get_ac_property(struct power_supply *psy,
				      enum power_supply_property psp,
				      union power_supply_propval *val)
{
	switch (psp) {
	case POWER_SUPPLY_PROP_ONLINE:
		val->intval = ac_online;
		break;
	default:
		return -EINVAL;
	}
	return 0;
}

static int test_power_get_usb_property(struct power_supply *psy,
				      enum power_supply_property psp,
				      union power_supply_propval *val)
{
	switch (psp) {
	case POWER_SUPPLY_PROP_ONLINE:
		val->intval = usb_online;
		break;
	default:
		return -EINVAL;
	}
	return 0;
}

static int test_power_get_battery_property(struct power_supply *psy,
					   enum power_supply_property psp,
					   union power_supply_propval *val)
{
	switch (psp) {
	case POWER_SUPPLY_PROP_MODEL_NAME:
		val->strval = "Test battery";
		break;
	case POWER_SUPPLY_PROP_MANUFACTURER:
		val->strval = "Linux";
		break;
	case POWER_SUPPLY_PROP_SERIAL_NUMBER:
		val->strval = UTS_RELEASE;
		break;
	case POWER_SUPPLY_PROP_STATUS:
		val->intval = battery_status;
		break;
	case POWER_SUPPLY_PROP_CHARGE_TYPE:
		val->intval = POWER_SUPPLY_CHARGE_TYPE_FAST;
		break;
	case POWER_SUPPLY_PROP_HEALTH:
		val->intval = battery_health;
		break;
	case POWER_SUPPLY_PROP_PRESENT:
		val->intval = battery_present;
		break;
	case POWER_SUPPLY_PROP_TECHNOLOGY:
		val->intval = battery_technology;
		break;
	case POWER_SUPPLY_PROP_CAPACITY_LEVEL:
		val->intval = POWER_SUPPLY_CAPACITY_LEVEL_NORMAL;
		break;
	case POWER_SUPPLY_PROP_CAPACITY:
	case POWER_SUPPLY_PROP_CHARGE_NOW:
		val->intval = battery_capacity;
		break;
	case POWER_SUPPLY_PROP_CHARGE_FULL_DESIGN:
	case POWER_SUPPLY_PROP_CHARGE_FULL:
		val->intval = 100;
		break;
	case POWER_SUPPLY_PROP_TIME_TO_EMPTY_AVG:
	case POWER_SUPPLY_PROP_TIME_TO_FULL_NOW:
		val->intval = 3600;
		break;
	case POWER_SUPPLY_PROP_TEMP:
		val->intval = 26;
		break;
	case POWER_SUPPLY_PROP_VOLTAGE_NOW:
		val->intval = battery_voltage;
		break;
	default:
		pr_info("%s: some properties deliberately report errors.\n",
			__func__);
		return -EINVAL;
	}
	return 0;
}

static enum power_supply_property test_power_ac_props[] = {
	POWER_SUPPLY_PROP_ONLINE,
};

static enum power_supply_property test_power_battery_props[] = {
	POWER_SUPPLY_PROP_STATUS,
	POWER_SUPPLY_PROP_CHARGE_TYPE,
	POWER_SUPPLY_PROP_HEALTH,
	POWER_SUPPLY_PROP_PRESENT,
	POWER_SUPPLY_PROP_TECHNOLOGY,
	POWER_SUPPLY_PROP_CHARGE_FULL_DESIGN,
	POWER_SUPPLY_PROP_CHARGE_FULL,
	POWER_SUPPLY_PROP_CHARGE_NOW,
	POWER_SUPPLY_PROP_CAPACITY,
	POWER_SUPPLY_PROP_CAPACITY_LEVEL,
	POWER_SUPPLY_PROP_TIME_TO_EMPTY_AVG,
	POWER_SUPPLY_PROP_TIME_TO_FULL_NOW,
	POWER_SUPPLY_PROP_MODEL_NAME,
	POWER_SUPPLY_PROP_MANUFACTURER,
	POWER_SUPPLY_PROP_SERIAL_NUMBER,
	POWER_SUPPLY_PROP_TEMP,
	POWER_SUPPLY_PROP_VOLTAGE_NOW,
};

static char *test_power_ac_supplied_to[] = {
	"test_battery",
};

static struct power_supply test_power_supplies[] = {
	{
		.name = "test_ac",
		.type = POWER_SUPPLY_TYPE_MAINS,
		.supplied_to = test_power_ac_supplied_to,
		.num_supplicants = ARRAY_SIZE(test_power_ac_supplied_to),
		.properties = test_power_ac_props,
		.num_properties = ARRAY_SIZE(test_power_ac_props),
		.get_property = test_power_get_ac_property,
	}, {
		.name = "test_battery",
		.type = POWER_SUPPLY_TYPE_BATTERY,
		.properties = test_power_battery_props,
		.num_properties = ARRAY_SIZE(test_power_battery_props),
		.get_property = test_power_get_battery_property,
	}, {
		.name = "test_usb",
		.type = POWER_SUPPLY_TYPE_USB,
		.supplied_to = test_power_ac_supplied_to,
		.num_supplicants = ARRAY_SIZE(test_power_ac_supplied_to),
		.properties = test_power_ac_props,
		.num_properties = ARRAY_SIZE(test_power_ac_props),
		.get_property = test_power_get_usb_property,
	},
};


static int __init test_power_init(void)
{
	int i;
	int ret;

	for (i = 0; i < ARRAY_SIZE(test_power_supplies); i++) {
		ret = power_supply_register(NULL, &test_power_supplies[i]);
		if (ret) {
			pr_err("%s: failed to register %s\n", __func__,
				test_power_supplies[i].name);
			goto failed;
		}
	}

	module_initialized = true;
	return 0;
failed:
	while (--i >= 0)
		power_supply_unregister(&test_power_supplies[i]);
	return ret;
}
module_init(test_power_init);

static void __exit test_power_exit(void)
{
	int i;

	/* Let's see how we handle changes... */
	ac_online = 0;
	usb_online = 0;
	battery_status = POWER_SUPPLY_STATUS_DISCHARGING;
	for (i = 0; i < ARRAY_SIZE(test_power_supplies); i++)
		power_supply_changed(&test_power_supplies[i]);
	pr_info("%s: 'changed' event sent, sleeping for 10 seconds...\n",
		__func__);
	ssleep(10);

	for (i = 0; i < ARRAY_SIZE(test_power_supplies); i++)
		power_supply_unregister(&test_power_supplies[i]);

	module_initialized = false;
}
module_exit(test_power_exit);


#define MAX_KEYLENGTH 256
struct battery_property_map {
	int value;
	char const *key;
};

static struct battery_property_map map_ac_online[] = {
	{ 0,  "off"  },
	{ 1,  "on" },
	{ -1, NULL  },
};

static struct battery_property_map map_status[] = {
	{ POWER_SUPPLY_STATUS_CHARGING,     "charging"     },
	{ POWER_SUPPLY_STATUS_DISCHARGING,  "discharging"  },
	{ POWER_SUPPLY_STATUS_NOT_CHARGING, "not-charging" },
	{ POWER_SUPPLY_STATUS_FULL,         "full"         },
	{ -1,                               NULL           },
};

static struct battery_property_map map_health[] = {
	{ POWER_SUPPLY_HEALTH_GOOD,           "good"        },
	{ POWER_SUPPLY_HEALTH_OVERHEAT,       "overheat"    },
	{ POWER_SUPPLY_HEALTH_DEAD,           "dead"        },
	{ POWER_SUPPLY_HEALTH_OVERVOLTAGE,    "overvoltage" },
	{ POWER_SUPPLY_HEALTH_UNSPEC_FAILURE, "failure"     },
	{ -1,                                 NULL          },
};

static struct battery_property_map map_present[] = {
	{ 0,  "false" },
	{ 1,  "true"  },
	{ -1, NULL    },
};

static struct battery_property_map map_technology[] = {
	{ POWER_SUPPLY_TECHNOLOGY_NiMH, "NiMH" },
	{ POWER_SUPPLY_TECHNOLOGY_LION, "LION" },
	{ POWER_SUPPLY_TECHNOLOGY_LIPO, "LIPO" },
	{ POWER_SUPPLY_TECHNOLOGY_LiFe, "LiFe" },
	{ POWER_SUPPLY_TECHNOLOGY_NiCd, "NiCd" },
	{ POWER_SUPPLY_TECHNOLOGY_LiMn, "LiMn" },
	{ -1,				NULL   },
};


static int map_get_value(struct battery_property_map *map, const char *key,
				int def_val)
{
	char buf[MAX_KEYLENGTH];
	int cr;

	strncpy(buf, key, MAX_KEYLENGTH);
	buf[MAX_KEYLENGTH-1] = '\0';

	cr = strnlen(buf, MAX_KEYLENGTH) - 1;
	if (buf[cr] == '\n')
		buf[cr] = '\0';

	while (map->key) {
		if (strncasecmp(map->key, buf, MAX_KEYLENGTH) == 0)
			return map->value;
		map++;
	}

	return def_val;
}


static const char *map_get_key(struct battery_property_map *map, int value,
				const char *def_key)
{
	while (map->key) {
		if (map->value == value)
			return map->key;
		map++;
	}

	return def_key;
}

static inline void signal_power_supply_changed(struct power_supply *psy)
{
	if (module_initialized)
		power_supply_changed(psy);
}

static int param_set_ac_online(const char *key, const struct kernel_param *kp)
{
	ac_online = map_get_value(map_ac_online, key, ac_online);
	signal_power_supply_changed(&test_power_supplies[0]);
	return 0;
}

static int param_get_ac_online(char *buffer, const struct kernel_param *kp)
{
	strcpy(buffer, map_get_key(map_ac_online, ac_online, "unknown"));
	return strlen(buffer);
}

static int param_set_usb_online(const char *key, const struct kernel_param *kp)
{
	usb_online = map_get_value(map_ac_online, key, usb_online);
	signal_power_supply_changed(&test_power_supplies[2]);
	return 0;
}

static int param_get_usb_online(char *buffer, const struct kernel_param *kp)
{
	strcpy(buffer, map_get_key(map_ac_online, usb_online, "unknown"));
	return strlen(buffer);
}

static int param_set_battery_status(const char *key,
					const struct kernel_param *kp)
{
	battery_status = map_get_value(map_status, key, battery_status);
	signal_power_supply_changed(&test_power_supplies[1]);
	return 0;
}

static int param_get_battery_status(char *buffer, const struct kernel_param *kp)
{
	strcpy(buffer, map_get_key(map_status, battery_status, "unknown"));
	return strlen(buffer);
}

static int param_set_battery_health(const char *key,
					const struct kernel_param *kp)
{
	battery_health = map_get_value(map_health, key, battery_health);
	signal_power_supply_changed(&test_power_supplies[1]);
	return 0;
}

static int param_get_battery_health(char *buffer, const struct kernel_param *kp)
{
	strcpy(buffer, map_get_key(map_health, battery_health, "unknown"));
	return strlen(buffer);
}

static int param_set_battery_present(const char *key,
					const struct kernel_param *kp)
{
	battery_present = map_get_value(map_present, key, battery_present);
	signal_power_supply_changed(&test_power_supplies[0]);
	return 0;
}

static int param_get_battery_present(char *buffer,
					const struct kernel_param *kp)
{
	strcpy(buffer, map_get_key(map_present, battery_present, "unknown"));
	return strlen(buffer);
}

static int param_set_battery_technology(const char *key,
					const struct kernel_param *kp)
{
	battery_technology = map_get_value(map_technology, key,
						battery_technology);
	signal_power_supply_changed(&test_power_supplies[1]);
	return 0;
}

static int param_get_battery_technology(char *buffer,
					const struct kernel_param *kp)
{
	strcpy(buffer,
		map_get_key(map_technology, battery_technology, "unknown"));
	return strlen(buffer);
}

static int param_set_battery_capacity(const char *key,
					const struct kernel_param *kp)
{
	int tmp;

	if (1 != sscanf(key, "%d", &tmp))
		return -EINVAL;

	battery_capacity = tmp;
	signal_power_supply_changed(&test_power_supplies[1]);
	return 0;
}

#define param_get_battery_capacity param_get_int

static int param_set_battery_voltage(const char *key,
					const struct kernel_param *kp)
{
	int tmp;

	if (1 != sscanf(key, "%d", &tmp))
		return -EINVAL;

	battery_voltage = tmp;
	signal_power_supply_changed(&test_power_supplies[1]);
	return 0;
}

#define param_get_battery_voltage param_get_int

static struct kernel_param_ops param_ops_ac_online = {
	.set = param_set_ac_online,
	.get = param_get_ac_online,
};

static struct kernel_param_ops param_ops_usb_online = {
	.set = param_set_usb_online,
	.get = param_get_usb_online,
};

static struct kernel_param_ops param_ops_battery_status = {
	.set = param_set_battery_status,
	.get = param_get_battery_status,
};

static struct kernel_param_ops param_ops_battery_present = {
	.set = param_set_battery_present,
	.get = param_get_battery_present,
};

static struct kernel_param_ops param_ops_battery_technology = {
	.set = param_set_battery_technology,
	.get = param_get_battery_technology,
};

static struct kernel_param_ops param_ops_battery_health = {
	.set = param_set_battery_health,
	.get = param_get_battery_health,
};

static struct kernel_param_ops param_ops_battery_capacity = {
	.set = param_set_battery_capacity,
	.get = param_get_battery_capacity,
};

static struct kernel_param_ops param_ops_battery_voltage = {
	.set = param_set_battery_voltage,
	.get = param_get_battery_voltage,
};

#define param_check_ac_online(name, p) __param_check(name, p, void);
#define param_check_usb_online(name, p) __param_check(name, p, void);
#define param_check_battery_status(name, p) __param_check(name, p, void);
#define param_check_battery_present(name, p) __param_check(name, p, void);
#define param_check_battery_technology(name, p) __param_check(name, p, void);
#define param_check_battery_health(name, p) __param_check(name, p, void);
#define param_check_battery_capacity(name, p) __param_check(name, p, void);
#define param_check_battery_voltage(name, p) __param_check(name, p, void);


module_param(ac_online, ac_online, 0644);
MODULE_PARM_DESC(ac_online, "AC charging state <on|off>");

module_param(usb_online, usb_online, 0644);
MODULE_PARM_DESC(usb_online, "USB charging state <on|off>");

module_param(battery_status, battery_status, 0644);
MODULE_PARM_DESC(battery_status,
	"battery status <charging|discharging|not-charging|full>");

module_param(battery_present, battery_present, 0644);
MODULE_PARM_DESC(battery_present,
	"battery presence state <good|overheat|dead|overvoltage|failure>");

module_param(battery_technology, battery_technology, 0644);
MODULE_PARM_DESC(battery_technology,
	"battery technology <NiMH|LION|LIPO|LiFe|NiCd|LiMn>");

module_param(battery_health, battery_health, 0644);
MODULE_PARM_DESC(battery_health,
	"battery health state <good|overheat|dead|overvoltage|failure>");

module_param(battery_capacity, battery_capacity, 0644);
MODULE_PARM_DESC(battery_capacity, "battery capacity (percentage)");

module_param(battery_voltage, battery_voltage, 0644);
MODULE_PARM_DESC(battery_voltage, "battery voltage (millivolts)");

MODULE_DESCRIPTION("Power supply driver for testing");
MODULE_AUTHOR("Anton Vorontsov <cbouatmailru@gmail.com>");
MODULE_LICENSE("GPL");
Commit	Line	Data
a1e50fd4 AV	1	/*
	2	* Power supply driver for testing.
	3	*
	4	* Copyright 2010 Anton Vorontsov <cbouatmailru@gmail.com>
	5	*
f17ef9b2 JS	6	* Dynamic module parameter code from the Virtual Battery Driver
	7	* Copyright (C) 2008 Pylone, Inc.
	8	* By: Masashi YOKOTA <yokota@pylone.jp>
	9	* Originally found here:
	10	* http://downloads.pylone.jp/src/virtual_battery/virtual_battery-0.0.1.tar.bz2
	11	*
a1e50fd4 AV	12	* This program is free software; you can redistribute it and/or modify
	13	* it under the terms of the GNU General Public License version 2 as
	14	* published by the Free Software Foundation.
	15	*/
	16
	17	#include <linux/kernel.h>
	18	#include <linux/module.h>
	19	#include <linux/power_supply.h>
	20	#include <linux/errno.h>
	21	#include <linux/delay.h>
	22	#include <linux/vermagic.h>
	23
f17ef9b2	24	static int ac_online = 1;
73847375	25	static int usb_online = 1;
f17ef9b2 JS	26	static int battery_status = POWER_SUPPLY_STATUS_DISCHARGING;
	27	static int battery_health = POWER_SUPPLY_HEALTH_GOOD;
	28	static int battery_present = 1; /* true */
	29	static int battery_technology = POWER_SUPPLY_TECHNOLOGY_LION;
	30	static int battery_capacity = 50;
85a392d4	31	static int battery_voltage = 3300;
a1e50fd4	32
9239ebcf AG	33	static bool module_initialized;
9239ebcf AG	34
a1e50fd4 AV	35	static int test_power_get_ac_property(struct power_supply *psy,
	36	enum power_supply_property psp,
	37	union power_supply_propval *val)
	38	{
	39	switch (psp) {
	40	case POWER_SUPPLY_PROP_ONLINE:
f17ef9b2	41	val->intval = ac_online;
a1e50fd4 AV	42	break;
	43	default:
	44	return -EINVAL;
	45	}
	46	return 0;
	47	}
	48
73847375 DES	49	static int test_power_get_usb_property(struct power_supply *psy,
	50	enum power_supply_property psp,
	51	union power_supply_propval *val)
	52	{
	53	switch (psp) {
	54	case POWER_SUPPLY_PROP_ONLINE:
	55	val->intval = usb_online;
	56	break;
	57	default:
	58	return -EINVAL;
	59	}
	60	return 0;
	61	}
	62
a1e50fd4 AV	63	static int test_power_get_battery_property(struct power_supply *psy,
	64	enum power_supply_property psp,
	65	union power_supply_propval *val)
	66	{
	67	switch (psp) {
	68	case POWER_SUPPLY_PROP_MODEL_NAME:
	69	val->strval = "Test battery";
	70	break;
	71	case POWER_SUPPLY_PROP_MANUFACTURER:
	72	val->strval = "Linux";
	73	break;
	74	case POWER_SUPPLY_PROP_SERIAL_NUMBER:
	75	val->strval = UTS_RELEASE;
	76	break;
	77	case POWER_SUPPLY_PROP_STATUS:
f17ef9b2	78	val->intval = battery_status;
a1e50fd4 AV	79	break;
	80	case POWER_SUPPLY_PROP_CHARGE_TYPE:
	81	val->intval = POWER_SUPPLY_CHARGE_TYPE_FAST;
	82	break;
	83	case POWER_SUPPLY_PROP_HEALTH:
f17ef9b2 JS	84	val->intval = battery_health;
	85	break;
	86	case POWER_SUPPLY_PROP_PRESENT:
	87	val->intval = battery_present;
a1e50fd4 AV	88	break;
a1e50fd4 AV	89	case POWER_SUPPLY_PROP_TECHNOLOGY:
f17ef9b2	90	val->intval = battery_technology;
a1e50fd4 AV	91	break;
	92	case POWER_SUPPLY_PROP_CAPACITY_LEVEL:
	93	val->intval = POWER_SUPPLY_CAPACITY_LEVEL_NORMAL;
	94	break;
	95	case POWER_SUPPLY_PROP_CAPACITY:
f17ef9b2 JS	96	case POWER_SUPPLY_PROP_CHARGE_NOW:
	97	val->intval = battery_capacity;
	98	break;
	99	case POWER_SUPPLY_PROP_CHARGE_FULL_DESIGN:
	100	case POWER_SUPPLY_PROP_CHARGE_FULL:
	101	val->intval = 100;
a1e50fd4 AV	102	break;
	103	case POWER_SUPPLY_PROP_TIME_TO_EMPTY_AVG:
	104	case POWER_SUPPLY_PROP_TIME_TO_FULL_NOW:
	105	val->intval = 3600;
	106	break;
85a392d4 DES	107	case POWER_SUPPLY_PROP_TEMP:
	108	val->intval = 26;
	109	break;
	110	case POWER_SUPPLY_PROP_VOLTAGE_NOW:
	111	val->intval = battery_voltage;
	112	break;
a1e50fd4 AV	113	default:
	114	pr_info("%s: some properties deliberately report errors.\n",
	115	__func__);
	116	return -EINVAL;
	117	}
	118	return 0;
	119	}
	120
	121	static enum power_supply_property test_power_ac_props[] = {
	122	POWER_SUPPLY_PROP_ONLINE,
	123	};
	124
	125	static enum power_supply_property test_power_battery_props[] = {
	126	POWER_SUPPLY_PROP_STATUS,
	127	POWER_SUPPLY_PROP_CHARGE_TYPE,
	128	POWER_SUPPLY_PROP_HEALTH,
f17ef9b2	129	POWER_SUPPLY_PROP_PRESENT,
a1e50fd4	130	POWER_SUPPLY_PROP_TECHNOLOGY,
f17ef9b2	131	POWER_SUPPLY_PROP_CHARGE_FULL_DESIGN,
a1e50fd4	132	POWER_SUPPLY_PROP_CHARGE_FULL,
f17ef9b2	133	POWER_SUPPLY_PROP_CHARGE_NOW,
a1e50fd4 AV	134	POWER_SUPPLY_PROP_CAPACITY,
	135	POWER_SUPPLY_PROP_CAPACITY_LEVEL,
	136	POWER_SUPPLY_PROP_TIME_TO_EMPTY_AVG,
	137	POWER_SUPPLY_PROP_TIME_TO_FULL_NOW,
	138	POWER_SUPPLY_PROP_MODEL_NAME,
	139	POWER_SUPPLY_PROP_MANUFACTURER,
	140	POWER_SUPPLY_PROP_SERIAL_NUMBER,
85a392d4 DES	141	POWER_SUPPLY_PROP_TEMP,
85a392d4 DES	142	POWER_SUPPLY_PROP_VOLTAGE_NOW,
a1e50fd4 AV	143	};
	144
	145	static char *test_power_ac_supplied_to[] = {
	146	"test_battery",
	147	};
	148
	149	static struct power_supply test_power_supplies[] = {
	150	{
	151	.name = "test_ac",
	152	.type = POWER_SUPPLY_TYPE_MAINS,
	153	.supplied_to = test_power_ac_supplied_to,
	154	.num_supplicants = ARRAY_SIZE(test_power_ac_supplied_to),
	155	.properties = test_power_ac_props,
	156	.num_properties = ARRAY_SIZE(test_power_ac_props),
	157	.get_property = test_power_get_ac_property,
	158	}, {
	159	.name = "test_battery",
	160	.type = POWER_SUPPLY_TYPE_BATTERY,
	161	.properties = test_power_battery_props,
	162	.num_properties = ARRAY_SIZE(test_power_battery_props),
	163	.get_property = test_power_get_battery_property,
73847375 DES	164	}, {
	165	.name = "test_usb",
	166	.type = POWER_SUPPLY_TYPE_USB,
	167	.supplied_to = test_power_ac_supplied_to,
	168	.num_supplicants = ARRAY_SIZE(test_power_ac_supplied_to),
	169	.properties = test_power_ac_props,
	170	.num_properties = ARRAY_SIZE(test_power_ac_props),
	171	.get_property = test_power_get_usb_property,
a1e50fd4 AV	172	},
	173	};
	174
f17ef9b2	175
a1e50fd4 AV	176	static int __init test_power_init(void)
	177	{
	178	int i;
	179	int ret;
	180
	181	for (i = 0; i < ARRAY_SIZE(test_power_supplies); i++) {
	182	ret = power_supply_register(NULL, &test_power_supplies[i]);
	183	if (ret) {
	184	pr_err("%s: failed to register %s\n", __func__,
	185	test_power_supplies[i].name);
	186	goto failed;
	187	}
	188	}
	189
9239ebcf	190	module_initialized = true;
a1e50fd4 AV	191	return 0;
	192	failed:
	193	while (--i >= 0)
	194	power_supply_unregister(&test_power_supplies[i]);
	195	return ret;
	196	}
	197	module_init(test_power_init);
	198
	199	static void __exit test_power_exit(void)
	200	{
	201	int i;
	202
	203	/* Let's see how we handle changes... */
f17ef9b2	204	ac_online = 0;
73847375	205	usb_online = 0;
f17ef9b2	206	battery_status = POWER_SUPPLY_STATUS_DISCHARGING;
a1e50fd4 AV	207	for (i = 0; i < ARRAY_SIZE(test_power_supplies); i++)
	208	power_supply_changed(&test_power_supplies[i]);
	209	pr_info("%s: 'changed' event sent, sleeping for 10 seconds...\n",
	210	__func__);
	211	ssleep(10);
	212
	213	for (i = 0; i < ARRAY_SIZE(test_power_supplies); i++)
	214	power_supply_unregister(&test_power_supplies[i]);
9239ebcf AG	215
9239ebcf AG	216	module_initialized = false;
a1e50fd4 AV	217	}
	218	module_exit(test_power_exit);
	219
f17ef9b2 JS	220
	221
	222	#define MAX_KEYLENGTH 256
	223	struct battery_property_map {
	224	int value;
	225	char const *key;
	226	};
	227
	228	static struct battery_property_map map_ac_online[] = {
9239ebcf AG	229	{ 0, "off" },
9239ebcf AG	230	{ 1, "on" },
f17ef9b2 JS	231	{ -1, NULL },
	232	};
	233
	234	static struct battery_property_map map_status[] = {
	235	{ POWER_SUPPLY_STATUS_CHARGING, "charging" },
	236	{ POWER_SUPPLY_STATUS_DISCHARGING, "discharging" },
	237	{ POWER_SUPPLY_STATUS_NOT_CHARGING, "not-charging" },
	238	{ POWER_SUPPLY_STATUS_FULL, "full" },
	239	{ -1, NULL },
	240	};
	241
	242	static struct battery_property_map map_health[] = {
	243	{ POWER_SUPPLY_HEALTH_GOOD, "good" },
	244	{ POWER_SUPPLY_HEALTH_OVERHEAT, "overheat" },
	245	{ POWER_SUPPLY_HEALTH_DEAD, "dead" },
	246	{ POWER_SUPPLY_HEALTH_OVERVOLTAGE, "overvoltage" },
	247	{ POWER_SUPPLY_HEALTH_UNSPEC_FAILURE, "failure" },
	248	{ -1, NULL },
	249	};
	250
	251	static struct battery_property_map map_present[] = {
	252	{ 0, "false" },
	253	{ 1, "true" },
	254	{ -1, NULL },
	255	};
	256
	257	static struct battery_property_map map_technology[] = {
	258	{ POWER_SUPPLY_TECHNOLOGY_NiMH, "NiMH" },
	259	{ POWER_SUPPLY_TECHNOLOGY_LION, "LION" },
	260	{ POWER_SUPPLY_TECHNOLOGY_LIPO, "LIPO" },
	261	{ POWER_SUPPLY_TECHNOLOGY_LiFe, "LiFe" },
	262	{ POWER_SUPPLY_TECHNOLOGY_NiCd, "NiCd" },
	263	{ POWER_SUPPLY_TECHNOLOGY_LiMn, "LiMn" },
	264	{ -1, NULL },
	265	};
	266
	267
	268	static int map_get_value(struct battery_property_map map, const char key,
	269	int def_val)
	270	{
	271	char buf[MAX_KEYLENGTH];
	272	int cr;
	273
	274	strncpy(buf, key, MAX_KEYLENGTH);
	275	buf[MAX_KEYLENGTH-1] = '\0';
	276
	277	cr = strnlen(buf, MAX_KEYLENGTH) - 1;
	278	if (buf[cr] == '\n')
	279	buf[cr] = '\0';
	280
	281	while (map->key) {
	282	if (strncasecmp(map->key, buf, MAX_KEYLENGTH) == 0)
	283	return map->value;
	284	map++;
	285	}
	286
	287	return def_val;
	288	}
	289
	290
	291	static const char map_get_key(struct battery_property_map map, int value,
	292	const char *def_key)
	293	{
	294	while (map->key) {
295	if (map->value == value)
296	return map->key;
297	map++;
298	}
299
300	return def_key;
301	}
302
9239ebcf AG	303	static inline void signal_power_supply_changed(struct power_supply *psy)
	304	{
	305	if (module_initialized)
	306	power_supply_changed(psy);
	307	}
	308
f17ef9b2 JS	309	static int param_set_ac_online(const char key, const struct kernel_param kp)
	310	{
	311	ac_online = map_get_value(map_ac_online, key, ac_online);
9239ebcf	312	signal_power_supply_changed(&test_power_supplies[0]);
f17ef9b2 JS	313	return 0;
	314	}
	315
	316	static int param_get_ac_online(char buffer, const struct kernel_param kp)
	317	{
	318	strcpy(buffer, map_get_key(map_ac_online, ac_online, "unknown"));
	319	return strlen(buffer);
	320	}
	321
73847375 DES	322	static int param_set_usb_online(const char key, const struct kernel_param kp)
	323	{
	324	usb_online = map_get_value(map_ac_online, key, usb_online);
9239ebcf	325	signal_power_supply_changed(&test_power_supplies[2]);
73847375 DES	326	return 0;
	327	}
	328
	329	static int param_get_usb_online(char buffer, const struct kernel_param kp)
	330	{
	331	strcpy(buffer, map_get_key(map_ac_online, usb_online, "unknown"));
	332	return strlen(buffer);
	333	}
	334
f17ef9b2 JS	335	static int param_set_battery_status(const char *key,
	336	const struct kernel_param *kp)
	337	{
	338	battery_status = map_get_value(map_status, key, battery_status);
9239ebcf	339	signal_power_supply_changed(&test_power_supplies[1]);
f17ef9b2 JS	340	return 0;
	341	}
	342
	343	static int param_get_battery_status(char buffer, const struct kernel_param kp)
	344	{
	345	strcpy(buffer, map_get_key(map_status, battery_status, "unknown"));
	346	return strlen(buffer);
	347	}
	348
	349	static int param_set_battery_health(const char *key,
	350	const struct kernel_param *kp)
	351	{
	352	battery_health = map_get_value(map_health, key, battery_health);
9239ebcf	353	signal_power_supply_changed(&test_power_supplies[1]);
f17ef9b2 JS	354	return 0;
	355	}
	356
	357	static int param_get_battery_health(char buffer, const struct kernel_param kp)
	358	{
	359	strcpy(buffer, map_get_key(map_health, battery_health, "unknown"));
	360	return strlen(buffer);
	361	}
	362
	363	static int param_set_battery_present(const char *key,
	364	const struct kernel_param *kp)
	365	{
	366	battery_present = map_get_value(map_present, key, battery_present);
9239ebcf	367	signal_power_supply_changed(&test_power_supplies[0]);
f17ef9b2 JS	368	return 0;
	369	}
	370
	371	static int param_get_battery_present(char *buffer,
	372	const struct kernel_param *kp)
	373	{
	374	strcpy(buffer, map_get_key(map_present, battery_present, "unknown"));
	375	return strlen(buffer);
	376	}
	377
	378	static int param_set_battery_technology(const char *key,
	379	const struct kernel_param *kp)
	380	{
	381	battery_technology = map_get_value(map_technology, key,
	382	battery_technology);
9239ebcf	383	signal_power_supply_changed(&test_power_supplies[1]);
f17ef9b2 JS	384	return 0;
	385	}
	386
	387	static int param_get_battery_technology(char *buffer,
	388	const struct kernel_param *kp)
	389	{
	390	strcpy(buffer,
	391	map_get_key(map_technology, battery_technology, "unknown"));
	392	return strlen(buffer);
	393	}
	394
	395	static int param_set_battery_capacity(const char *key,
	396	const struct kernel_param *kp)
	397	{
	398	int tmp;
	399
	400	if (1 != sscanf(key, "%d", &tmp))
	401	return -EINVAL;
	402
	403	battery_capacity = tmp;
9239ebcf	404	signal_power_supply_changed(&test_power_supplies[1]);
f17ef9b2 JS	405	return 0;
	406	}
	407
	408	#define param_get_battery_capacity param_get_int
	409
85a392d4 DES	410	static int param_set_battery_voltage(const char *key,
	411	const struct kernel_param *kp)
	412	{
	413	int tmp;
	414
	415	if (1 != sscanf(key, "%d", &tmp))
	416	return -EINVAL;
	417
	418	battery_voltage = tmp;
9239ebcf	419	signal_power_supply_changed(&test_power_supplies[1]);
85a392d4 DES	420	return 0;
	421	}
	422
	423	#define param_get_battery_voltage param_get_int
	424
f17ef9b2 JS	425	static struct kernel_param_ops param_ops_ac_online = {
	426	.set = param_set_ac_online,
	427	.get = param_get_ac_online,
	428	};
	429
73847375 DES	430	static struct kernel_param_ops param_ops_usb_online = {
	431	.set = param_set_usb_online,
	432	.get = param_get_usb_online,
	433	};
	434
f17ef9b2 JS	435	static struct kernel_param_ops param_ops_battery_status = {
	436	.set = param_set_battery_status,
	437	.get = param_get_battery_status,
	438	};
	439
	440	static struct kernel_param_ops param_ops_battery_present = {
	441	.set = param_set_battery_present,
	442	.get = param_get_battery_present,
	443	};
	444
	445	static struct kernel_param_ops param_ops_battery_technology = {
	446	.set = param_set_battery_technology,
	447	.get = param_get_battery_technology,
	448	};
	449
	450	static struct kernel_param_ops param_ops_battery_health = {
	451	.set = param_set_battery_health,
	452	.get = param_get_battery_health,
	453	};
	454
	455	static struct kernel_param_ops param_ops_battery_capacity = {
	456	.set = param_set_battery_capacity,
	457	.get = param_get_battery_capacity,
	458	};
	459
85a392d4 DES	460	static struct kernel_param_ops param_ops_battery_voltage = {
	461	.set = param_set_battery_voltage,
	462	.get = param_get_battery_voltage,
	463	};
f17ef9b2 JS	464
f17ef9b2 JS	465	#define param_check_ac_online(name, p) __param_check(name, p, void);
73847375	466	#define param_check_usb_online(name, p) __param_check(name, p, void);
f17ef9b2 JS	467	#define param_check_battery_status(name, p) __param_check(name, p, void);
	468	#define param_check_battery_present(name, p) __param_check(name, p, void);
	469	#define param_check_battery_technology(name, p) __param_check(name, p, void);
	470	#define param_check_battery_health(name, p) __param_check(name, p, void);
	471	#define param_check_battery_capacity(name, p) __param_check(name, p, void);
85a392d4	472	#define param_check_battery_voltage(name, p) __param_check(name, p, void);
f17ef9b2 JS	473
	474
	475	module_param(ac_online, ac_online, 0644);
	476	MODULE_PARM_DESC(ac_online, "AC charging state <on\|off>");
	477
73847375 DES	478	module_param(usb_online, usb_online, 0644);
	479	MODULE_PARM_DESC(usb_online, "USB charging state <on\|off>");
	480
f17ef9b2 JS	481	module_param(battery_status, battery_status, 0644);
	482	MODULE_PARM_DESC(battery_status,
	483	"battery status <charging\|discharging\|not-charging\|full>");
	484
	485	module_param(battery_present, battery_present, 0644);
	486	MODULE_PARM_DESC(battery_present,
	487	"battery presence state <good\|overheat\|dead\|overvoltage\|failure>");
	488
	489	module_param(battery_technology, battery_technology, 0644);
	490	MODULE_PARM_DESC(battery_technology,
	491	"battery technology <NiMH\|LION\|LIPO\|LiFe\|NiCd\|LiMn>");
	492
	493	module_param(battery_health, battery_health, 0644);
	494	MODULE_PARM_DESC(battery_health,
	495	"battery health state <good\|overheat\|dead\|overvoltage\|failure>");
	496
	497	module_param(battery_capacity, battery_capacity, 0644);
	498	MODULE_PARM_DESC(battery_capacity, "battery capacity (percentage)");
	499
85a392d4 DES	500	module_param(battery_voltage, battery_voltage, 0644);
85a392d4 DES	501	MODULE_PARM_DESC(battery_voltage, "battery voltage (millivolts)");
f17ef9b2	502
a1e50fd4 AV	503	MODULE_DESCRIPTION("Power supply driver for testing");
	504	MODULE_AUTHOR("Anton Vorontsov <cbouatmailru@gmail.com>");
	505	MODULE_LICENSE("GPL");