drivers/power/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 static int ac_online                    = 1;
  25 static int usb_online                   = 1;
  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;
  31 static int battery_voltage              = 3300;
  32
  33 static bool module_initialized;
  34
  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:
  41                 val->intval = ac_online;
  42                 break;
  43         default:
  44                 return -EINVAL;
  45         }
  46         return 0;
  47 }
  48
  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
  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:
  78                 val->intval = battery_status;
  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:
  84                 val->intval = battery_health;
  85                 break;
  86         case POWER_SUPPLY_PROP_PRESENT:
  87                 val->intval = battery_present;
  88                 break;
  89         case POWER_SUPPLY_PROP_TECHNOLOGY:
  90                 val->intval = battery_technology;
  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:
  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;
 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;
 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;
 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,
 129         POWER_SUPPLY_PROP_PRESENT,
 130         POWER_SUPPLY_PROP_TECHNOLOGY,
 131         POWER_SUPPLY_PROP_CHARGE_FULL_DESIGN,
 132         POWER_SUPPLY_PROP_CHARGE_FULL,
 133         POWER_SUPPLY_PROP_CHARGE_NOW,
 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,
 141         POWER_SUPPLY_PROP_TEMP,
 142         POWER_SUPPLY_PROP_VOLTAGE_NOW,
 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,
 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,
 172         },
 173 };
 174
 175
 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
 190         module_initialized = true;
 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... */
 204         ac_online = 0;
 205         usb_online = 0;
 206         battery_status = POWER_SUPPLY_STATUS_DISCHARGING;
 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]);
 215
 216         module_initialized = false;
 217 }
 218 module_exit(test_power_exit);
 219
 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[] = {
 229         { 0,  "off"  },
 230         { 1,  "on" },
 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
 303 static inline void signal_power_supply_changed(struct power_supply *psy)
 304 {
 305         if (module_initialized)
 306                 power_supply_changed(psy);
 307 }
 308
 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);
 312         signal_power_supply_changed(&test_power_supplies[0]);
 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
 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);
 325         signal_power_supply_changed(&test_power_supplies[2]);
 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
 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);
 339         signal_power_supply_changed(&test_power_supplies[1]);
 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);
 353         signal_power_supply_changed(&test_power_supplies[1]);
 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);
 367         signal_power_supply_changed(&test_power_supplies[0]);
 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);
 383         signal_power_supply_changed(&test_power_supplies[1]);
 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;
 404         signal_power_supply_changed(&test_power_supplies[1]);
 405         return 0;
 406 }
 407
 408 #define param_get_battery_capacity param_get_int
 409
 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;
 419         signal_power_supply_changed(&test_power_supplies[1]);
 420         return 0;
 421 }
 422
 423 #define param_get_battery_voltage param_get_int
 424
 425 static struct kernel_param_ops param_ops_ac_online = {
 426         .set = param_set_ac_online,
 427         .get = param_get_ac_online,
 428 };
 429
 430 static struct kernel_param_ops param_ops_usb_online = {
 431         .set = param_set_usb_online,
 432         .get = param_get_usb_online,
 433 };
 434
 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
 460 static struct kernel_param_ops param_ops_battery_voltage = {
 461         .set = param_set_battery_voltage,
 462         .get = param_get_battery_voltage,
 463 };
 464
 465 #define param_check_ac_online(name, p) __param_check(name, p, void);
 466 #define param_check_usb_online(name, p) __param_check(name, p, void);
 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);
 472 #define param_check_battery_voltage(name, p) __param_check(name, p, void);
 473
 474
 475 module_param(ac_online, ac_online, 0644);
 476 MODULE_PARM_DESC(ac_online, "AC charging state <on|off>");
 477
 478 module_param(usb_online, usb_online, 0644);
 479 MODULE_PARM_DESC(usb_online, "USB charging state <on|off>");
 480
 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
 500 module_param(battery_voltage, battery_voltage, 0644);
 501 MODULE_PARM_DESC(battery_voltage, "battery voltage (millivolts)");
 502
 503 MODULE_DESCRIPTION("Power supply driver for testing");
 504 MODULE_AUTHOR("Anton Vorontsov <cbouatmailru@gmail.com>");
 505 MODULE_LICENSE("GPL");