[linux.git] / drivers / power / test_power.c

/*
 * Power supply driver for testing.
 *
 * Copyright 2010  Anton Vorontsov <[email protected]>
 *
 * Dynamic module parameter code from the Virtual Battery Driver
 * Copyright (C) 2008 Pylone, Inc.
 * By: Masashi YOKOTA <[email protected]>
 * 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>

enum test_power_id {
	TEST_AC,
	TEST_BATTERY,
	TEST_USB,
	TEST_POWER_NUM,
};

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[TEST_POWER_NUM];

static const struct power_supply_desc test_power_desc[] = {
	[TEST_AC] = {
		.name = "test_ac",
		.type = POWER_SUPPLY_TYPE_MAINS,
		.properties = test_power_ac_props,
		.num_properties = ARRAY_SIZE(test_power_ac_props),
		.get_property = test_power_get_ac_property,
	},
	[TEST_BATTERY] = {
		.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,
	},
	[TEST_USB] = {
		.name = "test_usb",
		.type = POWER_SUPPLY_TYPE_USB,
		.properties = test_power_ac_props,
		.num_properties = ARRAY_SIZE(test_power_ac_props),
		.get_property = test_power_get_usb_property,
	},
};

static const struct power_supply_config test_power_configs[] = {
	{
		/* test_ac */
		.supplied_to = test_power_ac_supplied_to,
		.num_supplicants = ARRAY_SIZE(test_power_ac_supplied_to),
	}, {
		/* test_battery */
	}, {
		/* test_usb */
		.supplied_to = test_power_ac_supplied_to,
		.num_supplicants = ARRAY_SIZE(test_power_ac_supplied_to),
	},
};

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

	BUILD_BUG_ON(TEST_POWER_NUM != ARRAY_SIZE(test_power_supplies));
	BUILD_BUG_ON(TEST_POWER_NUM != ARRAY_SIZE(test_power_configs));

	for (i = 0; i < ARRAY_SIZE(test_power_supplies); i++) {
		test_power_supplies[i] = power_supply_register(NULL,
						&test_power_desc[i],
						&test_power_configs[i]);
		if (IS_ERR(test_power_supplies[i])) {
			pr_err("%s: failed to register %s\n", __func__,
				test_power_desc[i].name);
			ret = PTR_ERR(test_power_supplies[i]);
			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 (cr < 0)
		return def_val;
	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[TEST_AC]);
	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[TEST_USB]);
	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[TEST_BATTERY]);
	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[TEST_BATTERY]);
	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[TEST_AC]);
	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[TEST_BATTERY]);
	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[TEST_BATTERY]);
	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[TEST_BATTERY]);
	return 0;
}

#define param_get_battery_voltage param_get_int

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

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

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

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

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

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

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

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