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