1/*
2 * Power supply driver for testing.
3 *
4 * Copyright 2010  Anton Vorontsov <cbouatmailru@gmail.com>
5 *
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 *
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
24enum test_power_id {
25	TEST_AC,
26	TEST_BATTERY,
27	TEST_USB,
28	TEST_POWER_NUM,
29};
30
31static int ac_online			= 1;
32static int usb_online			= 1;
33static int battery_status		= POWER_SUPPLY_STATUS_DISCHARGING;
34static int battery_health		= POWER_SUPPLY_HEALTH_GOOD;
35static int battery_present		= 1; /* true */
36static int battery_technology		= POWER_SUPPLY_TECHNOLOGY_LION;
37static int battery_capacity		= 50;
38static int battery_voltage		= 3300;
39
40static bool module_initialized;
41
42static 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:
48		val->intval = ac_online;
49		break;
50	default:
51		return -EINVAL;
52	}
53	return 0;
54}
55
56static 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
70static 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:
85		val->intval = battery_status;
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:
91		val->intval = battery_health;
92		break;
93	case POWER_SUPPLY_PROP_PRESENT:
94		val->intval = battery_present;
95		break;
96	case POWER_SUPPLY_PROP_TECHNOLOGY:
97		val->intval = battery_technology;
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:
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;
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;
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;
120	default:
121		pr_info("%s: some properties deliberately report errors.\n",
122			__func__);
123		return -EINVAL;
124	}
125	return 0;
126}
127
128static enum power_supply_property test_power_ac_props[] = {
129	POWER_SUPPLY_PROP_ONLINE,
130};
131
132static enum power_supply_property test_power_battery_props[] = {
133	POWER_SUPPLY_PROP_STATUS,
134	POWER_SUPPLY_PROP_CHARGE_TYPE,
135	POWER_SUPPLY_PROP_HEALTH,
136	POWER_SUPPLY_PROP_PRESENT,
137	POWER_SUPPLY_PROP_TECHNOLOGY,
138	POWER_SUPPLY_PROP_CHARGE_FULL_DESIGN,
139	POWER_SUPPLY_PROP_CHARGE_FULL,
140	POWER_SUPPLY_PROP_CHARGE_NOW,
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,
148	POWER_SUPPLY_PROP_TEMP,
149	POWER_SUPPLY_PROP_VOLTAGE_NOW,
150};
151
152static char *test_power_ac_supplied_to[] = {
153	"test_battery",
154};
155
156static struct power_supply *test_power_supplies[TEST_POWER_NUM];
157
158static const struct power_supply_desc test_power_desc[] = {
159	[TEST_AC] = {
160		.name = "test_ac",
161		.type = POWER_SUPPLY_TYPE_MAINS,
162		.properties = test_power_ac_props,
163		.num_properties = ARRAY_SIZE(test_power_ac_props),
164		.get_property = test_power_get_ac_property,
165	},
166	[TEST_BATTERY] = {
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,
172	},
173	[TEST_USB] = {
174		.name = "test_usb",
175		.type = POWER_SUPPLY_TYPE_USB,
176		.properties = test_power_ac_props,
177		.num_properties = ARRAY_SIZE(test_power_ac_props),
178		.get_property = test_power_get_usb_property,
179	},
180};
181
182static 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};
195
196static int __init test_power_init(void)
197{
198	int i;
199	int ret;
200
201	BUILD_BUG_ON(TEST_POWER_NUM != ARRAY_SIZE(test_power_supplies));
202	BUILD_BUG_ON(TEST_POWER_NUM != ARRAY_SIZE(test_power_configs));
203
204	for (i = 0; i < ARRAY_SIZE(test_power_supplies); i++) {
205		test_power_supplies[i] = power_supply_register(NULL,
206						&test_power_desc[i],
207						&test_power_configs[i]);
208		if (IS_ERR(test_power_supplies[i])) {
209			pr_err("%s: failed to register %s\n", __func__,
210				test_power_desc[i].name);
211			ret = PTR_ERR(test_power_supplies[i]);
212			goto failed;
213		}
214	}
215
216	module_initialized = true;
217	return 0;
218failed:
219	while (--i >= 0)
220		power_supply_unregister(test_power_supplies[i]);
221	return ret;
222}
223module_init(test_power_init);
224
225static void __exit test_power_exit(void)
226{
227	int i;
228
229	/* Let's see how we handle changes... */
230	ac_online = 0;
231	usb_online = 0;
232	battery_status = POWER_SUPPLY_STATUS_DISCHARGING;
233	for (i = 0; i < ARRAY_SIZE(test_power_supplies); i++)
234		power_supply_changed(test_power_supplies[i]);
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++)
240		power_supply_unregister(test_power_supplies[i]);
241
242	module_initialized = false;
243}
244module_exit(test_power_exit);
245
246
247
248#define MAX_KEYLENGTH 256
249struct battery_property_map {
250	int value;
251	char const *key;
252};
253
254static struct battery_property_map map_ac_online[] = {
255	{ 0,  "off"  },
256	{ 1,  "on" },
257	{ -1, NULL  },
258};
259
260static 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
268static 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
277static struct battery_property_map map_present[] = {
278	{ 0,  "false" },
279	{ 1,  "true"  },
280	{ -1, NULL    },
281};
282
283static 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
294static 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;
304	if (buf[cr] == '\n')
305		buf[cr] = '\0';
306
307	while (map->key) {
308		if (strncasecmp(map->key, buf, MAX_KEYLENGTH) == 0)
309			return map->value;
310		map++;
311	}
312
313	return def_val;
314}
315
316
317static const char *map_get_key(struct battery_property_map *map, int value,
318				const char *def_key)
319{
320	while (map->key) {
321		if (map->value == value)
322			return map->key;
323		map++;
324	}
325
326	return def_key;
327}
328
329static inline void signal_power_supply_changed(struct power_supply *psy)
330{
331	if (module_initialized)
332		power_supply_changed(psy);
333}
334
335static int param_set_ac_online(const char *key, const struct kernel_param *kp)
336{
337	ac_online = map_get_value(map_ac_online, key, ac_online);
338	signal_power_supply_changed(test_power_supplies[TEST_AC]);
339	return 0;
340}
341
342static int param_get_ac_online(char *buffer, const struct kernel_param *kp)
343{
344	strcpy(buffer, map_get_key(map_ac_online, ac_online, "unknown"));
345	return strlen(buffer);
346}
347
348static int param_set_usb_online(const char *key, const struct kernel_param *kp)
349{
350	usb_online = map_get_value(map_ac_online, key, usb_online);
351	signal_power_supply_changed(test_power_supplies[TEST_USB]);
352	return 0;
353}
354
355static int param_get_usb_online(char *buffer, const struct kernel_param *kp)
356{
357	strcpy(buffer, map_get_key(map_ac_online, usb_online, "unknown"));
358	return strlen(buffer);
359}
360
361static int param_set_battery_status(const char *key,
362					const struct kernel_param *kp)
363{
364	battery_status = map_get_value(map_status, key, battery_status);
365	signal_power_supply_changed(test_power_supplies[TEST_BATTERY]);
366	return 0;
367}
368
369static int param_get_battery_status(char *buffer, const struct kernel_param *kp)
370{
371	strcpy(buffer, map_get_key(map_status, battery_status, "unknown"));
372	return strlen(buffer);
373}
374
375static int param_set_battery_health(const char *key,
376					const struct kernel_param *kp)
377{
378	battery_health = map_get_value(map_health, key, battery_health);
379	signal_power_supply_changed(test_power_supplies[TEST_BATTERY]);
380	return 0;
381}
382
383static int param_get_battery_health(char *buffer, const struct kernel_param *kp)
384{
385	strcpy(buffer, map_get_key(map_health, battery_health, "unknown"));
386	return strlen(buffer);
387}
388
389static int param_set_battery_present(const char *key,
390					const struct kernel_param *kp)
391{
392	battery_present = map_get_value(map_present, key, battery_present);
393	signal_power_supply_changed(test_power_supplies[TEST_AC]);
394	return 0;
395}
396
397static int param_get_battery_present(char *buffer,
398					const struct kernel_param *kp)
399{
400	strcpy(buffer, map_get_key(map_present, battery_present, "unknown"));
401	return strlen(buffer);
402}
403
404static int param_set_battery_technology(const char *key,
405					const struct kernel_param *kp)
406{
407	battery_technology = map_get_value(map_technology, key,
408						battery_technology);
409	signal_power_supply_changed(test_power_supplies[TEST_BATTERY]);
410	return 0;
411}
412
413static int param_get_battery_technology(char *buffer,
414					const struct kernel_param *kp)
415{
416	strcpy(buffer,
417		map_get_key(map_technology, battery_technology, "unknown"));
418	return strlen(buffer);
419}
420
421static int param_set_battery_capacity(const char *key,
422					const struct kernel_param *kp)
423{
424	int tmp;
425
426	if (1 != sscanf(key, "%d", &tmp))
427		return -EINVAL;
428
429	battery_capacity = tmp;
430	signal_power_supply_changed(test_power_supplies[TEST_BATTERY]);
431	return 0;
432}
433
434#define param_get_battery_capacity param_get_int
435
436static int param_set_battery_voltage(const char *key,
437					const struct kernel_param *kp)
438{
439	int tmp;
440
441	if (1 != sscanf(key, "%d", &tmp))
442		return -EINVAL;
443
444	battery_voltage = tmp;
445	signal_power_supply_changed(test_power_supplies[TEST_BATTERY]);
446	return 0;
447}
448
449#define param_get_battery_voltage param_get_int
450
451static const struct kernel_param_ops param_ops_ac_online = {
452	.set = param_set_ac_online,
453	.get = param_get_ac_online,
454};
455
456static const struct kernel_param_ops param_ops_usb_online = {
457	.set = param_set_usb_online,
458	.get = param_get_usb_online,
459};
460
461static const struct kernel_param_ops param_ops_battery_status = {
462	.set = param_set_battery_status,
463	.get = param_get_battery_status,
464};
465
466static const struct kernel_param_ops param_ops_battery_present = {
467	.set = param_set_battery_present,
468	.get = param_get_battery_present,
469};
470
471static const struct kernel_param_ops param_ops_battery_technology = {
472	.set = param_set_battery_technology,
473	.get = param_get_battery_technology,
474};
475
476static const struct kernel_param_ops param_ops_battery_health = {
477	.set = param_set_battery_health,
478	.get = param_get_battery_health,
479};
480
481static const struct kernel_param_ops param_ops_battery_capacity = {
482	.set = param_set_battery_capacity,
483	.get = param_get_battery_capacity,
484};
485
486static const struct kernel_param_ops param_ops_battery_voltage = {
487	.set = param_set_battery_voltage,
488	.get = param_get_battery_voltage,
489};
490
491#define param_check_ac_online(name, p) __param_check(name, p, void);
492#define param_check_usb_online(name, p) __param_check(name, p, void);
493#define param_check_battery_status(name, p) __param_check(name, p, void);
494#define param_check_battery_present(name, p) __param_check(name, p, void);
495#define param_check_battery_technology(name, p) __param_check(name, p, void);
496#define param_check_battery_health(name, p) __param_check(name, p, void);
497#define param_check_battery_capacity(name, p) __param_check(name, p, void);
498#define param_check_battery_voltage(name, p) __param_check(name, p, void);
499
500
501module_param(ac_online, ac_online, 0644);
502MODULE_PARM_DESC(ac_online, "AC charging state <on|off>");
503
504module_param(usb_online, usb_online, 0644);
505MODULE_PARM_DESC(usb_online, "USB charging state <on|off>");
506
507module_param(battery_status, battery_status, 0644);
508MODULE_PARM_DESC(battery_status,
509	"battery status <charging|discharging|not-charging|full>");
510
511module_param(battery_present, battery_present, 0644);
512MODULE_PARM_DESC(battery_present,
513	"battery presence state <good|overheat|dead|overvoltage|failure>");
514
515module_param(battery_technology, battery_technology, 0644);
516MODULE_PARM_DESC(battery_technology,
517	"battery technology <NiMH|LION|LIPO|LiFe|NiCd|LiMn>");
518
519module_param(battery_health, battery_health, 0644);
520MODULE_PARM_DESC(battery_health,
521	"battery health state <good|overheat|dead|overvoltage|failure>");
522
523module_param(battery_capacity, battery_capacity, 0644);
524MODULE_PARM_DESC(battery_capacity, "battery capacity (percentage)");
525
526module_param(battery_voltage, battery_voltage, 0644);
527MODULE_PARM_DESC(battery_voltage, "battery voltage (millivolts)");
528
529MODULE_DESCRIPTION("Power supply driver for testing");
530MODULE_AUTHOR("Anton Vorontsov <cbouatmailru@gmail.com>");
531MODULE_LICENSE("GPL");
532