drivers/power/supply/test_power.c

   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
  24 enum test_power_id {
  25         TEST_AC,
  26         TEST_BATTERY,
  27         TEST_USB,
  28         TEST_POWER_NUM,
  29 };
  30
  31 static int ac_online                    = 1;
  32 static int usb_online                   = 1;
  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;
  38 static int battery_voltage              = 3300;
  39
  40 static bool module_initialized;
  41
  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:
  48                 val->intval = ac_online;
  49                 break;
  50         default:
  51                 return -EINVAL;
  52         }
  53         return 0;
  54 }
  55
  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
  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:
  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
 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,
 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
 152 static char *test_power_ac_supplied_to[] = {
 153         "test_battery",
 154 };
 155
 156 static struct power_supply *test_power_supplies[TEST_POWER_NUM];
 157
 158 static 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
 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 };
 195
 196 static 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;
 218 failed:
 219         while (--i >= 0)
 220                 power_supply_unregister(test_power_supplies[i]);
 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... */
 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 }
 244 module_exit(test_power_exit);
 245
 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[] = {
 255         { 0,  "off"  },
 256         { 1,  "on" },
 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;
 304         if (cr < 0)
 305                 return def_val;
 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
 331 static inline void signal_power_supply_changed(struct power_supply *psy)
 332 {
 333         if (module_initialized)
 334                 power_supply_changed(psy);
 335 }
 336
 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);
 340         signal_power_supply_changed(test_power_supplies[TEST_AC]);
 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         strcat(buffer, "\n");
 348         return strlen(buffer);
 349 }
 350
 351 static int param_set_usb_online(const char *key, const struct kernel_param *kp)
 352 {
 353         usb_online = map_get_value(map_ac_online, key, usb_online);
 354         signal_power_supply_changed(test_power_supplies[TEST_USB]);
 355         return 0;
 356 }
 357
 358 static int param_get_usb_online(char *buffer, const struct kernel_param *kp)
 359 {
 360         strcpy(buffer, map_get_key(map_ac_online, usb_online, "unknown"));
 361         strcat(buffer, "\n");
 362         return strlen(buffer);
 363 }
 364
 365 static int param_set_battery_status(const char *key,
 366                                         const struct kernel_param *kp)
 367 {
 368         battery_status = map_get_value(map_status, key, battery_status);
 369         signal_power_supply_changed(test_power_supplies[TEST_BATTERY]);
 370         return 0;
 371 }
 372
 373 static int param_get_battery_status(char *buffer, const struct kernel_param *kp)
 374 {
 375         strcpy(buffer, map_get_key(map_status, battery_status, "unknown"));
 376         strcat(buffer, "\n");
 377         return strlen(buffer);
 378 }
 379
 380 static int param_set_battery_health(const char *key,
 381                                         const struct kernel_param *kp)
 382 {
 383         battery_health = map_get_value(map_health, key, battery_health);
 384         signal_power_supply_changed(test_power_supplies[TEST_BATTERY]);
 385         return 0;
 386 }
 387
 388 static int param_get_battery_health(char *buffer, const struct kernel_param *kp)
 389 {
 390         strcpy(buffer, map_get_key(map_health, battery_health, "unknown"));
 391         strcat(buffer, "\n");
 392         return strlen(buffer);
 393 }
 394
 395 static int param_set_battery_present(const char *key,
 396                                         const struct kernel_param *kp)
 397 {
 398         battery_present = map_get_value(map_present, key, battery_present);
 399         signal_power_supply_changed(test_power_supplies[TEST_AC]);
 400         return 0;
 401 }
 402
 403 static int param_get_battery_present(char *buffer,
 404                                         const struct kernel_param *kp)
 405 {
 406         strcpy(buffer, map_get_key(map_present, battery_present, "unknown"));
 407         strcat(buffer, "\n");
 408         return strlen(buffer);
 409 }
 410
 411 static int param_set_battery_technology(const char *key,
 412                                         const struct kernel_param *kp)
 413 {
 414         battery_technology = map_get_value(map_technology, key,
 415                                                 battery_technology);
 416         signal_power_supply_changed(test_power_supplies[TEST_BATTERY]);
 417         return 0;
 418 }
 419
 420 static int param_get_battery_technology(char *buffer,
 421                                         const struct kernel_param *kp)
 422 {
 423         strcpy(buffer,
 424                 map_get_key(map_technology, battery_technology, "unknown"));
 425         strcat(buffer, "\n");
 426         return strlen(buffer);
 427 }
 428
 429 static int param_set_battery_capacity(const char *key,
 430                                         const struct kernel_param *kp)
 431 {
 432         int tmp;
 433
 434         if (1 != sscanf(key, "%d", &tmp))
 435                 return -EINVAL;
 436
 437         battery_capacity = tmp;
 438         signal_power_supply_changed(test_power_supplies[TEST_BATTERY]);
 439         return 0;
 440 }
 441
 442 #define param_get_battery_capacity param_get_int
 443
 444 static int param_set_battery_voltage(const char *key,
 445                                         const struct kernel_param *kp)
 446 {
 447         int tmp;
 448
 449         if (1 != sscanf(key, "%d", &tmp))
 450                 return -EINVAL;
 451
 452         battery_voltage = tmp;
 453         signal_power_supply_changed(test_power_supplies[TEST_BATTERY]);
 454         return 0;
 455 }
 456
 457 #define param_get_battery_voltage param_get_int
 458
 459 static const struct kernel_param_ops param_ops_ac_online = {
 460         .set = param_set_ac_online,
 461         .get = param_get_ac_online,
 462 };
 463
 464 static const struct kernel_param_ops param_ops_usb_online = {
 465         .set = param_set_usb_online,
 466         .get = param_get_usb_online,
 467 };
 468
 469 static const struct kernel_param_ops param_ops_battery_status = {
 470         .set = param_set_battery_status,
 471         .get = param_get_battery_status,
 472 };
 473
 474 static const struct kernel_param_ops param_ops_battery_present = {
 475         .set = param_set_battery_present,
 476         .get = param_get_battery_present,
 477 };
 478
 479 static const struct kernel_param_ops param_ops_battery_technology = {
 480         .set = param_set_battery_technology,
 481         .get = param_get_battery_technology,
 482 };
 483
 484 static const struct kernel_param_ops param_ops_battery_health = {
 485         .set = param_set_battery_health,
 486         .get = param_get_battery_health,
 487 };
 488
 489 static const struct kernel_param_ops param_ops_battery_capacity = {
 490         .set = param_set_battery_capacity,
 491         .get = param_get_battery_capacity,
 492 };
 493
 494 static const struct kernel_param_ops param_ops_battery_voltage = {
 495         .set = param_set_battery_voltage,
 496         .get = param_get_battery_voltage,
 497 };
 498
 499 #define param_check_ac_online(name, p) __param_check(name, p, void);
 500 #define param_check_usb_online(name, p) __param_check(name, p, void);
 501 #define param_check_battery_status(name, p) __param_check(name, p, void);
 502 #define param_check_battery_present(name, p) __param_check(name, p, void);
 503 #define param_check_battery_technology(name, p) __param_check(name, p, void);
 504 #define param_check_battery_health(name, p) __param_check(name, p, void);
 505 #define param_check_battery_capacity(name, p) __param_check(name, p, void);
 506 #define param_check_battery_voltage(name, p) __param_check(name, p, void);
 507
 508
 509 module_param(ac_online, ac_online, 0644);
 510 MODULE_PARM_DESC(ac_online, "AC charging state <on|off>");
 511
 512 module_param(usb_online, usb_online, 0644);
 513 MODULE_PARM_DESC(usb_online, "USB charging state <on|off>");
 514
 515 module_param(battery_status, battery_status, 0644);
 516 MODULE_PARM_DESC(battery_status,
 517         "battery status <charging|discharging|not-charging|full>");
 518
 519 module_param(battery_present, battery_present, 0644);
 520 MODULE_PARM_DESC(battery_present,
 521         "battery presence state <good|overheat|dead|overvoltage|failure>");
 522
 523 module_param(battery_technology, battery_technology, 0644);
 524 MODULE_PARM_DESC(battery_technology,
 525         "battery technology <NiMH|LION|LIPO|LiFe|NiCd|LiMn>");
 526
 527 module_param(battery_health, battery_health, 0644);
 528 MODULE_PARM_DESC(battery_health,
 529         "battery health state <good|overheat|dead|overvoltage|failure>");
 530
 531 module_param(battery_capacity, battery_capacity, 0644);
 532 MODULE_PARM_DESC(battery_capacity, "battery capacity (percentage)");
 533
 534 module_param(battery_voltage, battery_voltage, 0644);
 535 MODULE_PARM_DESC(battery_voltage, "battery voltage (millivolts)");
 536
 537 MODULE_DESCRIPTION("Power supply driver for testing");
 538 MODULE_AUTHOR("Anton Vorontsov <cbouatmailru@gmail.com>");
 539 MODULE_LICENSE("GPL");