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 battery_status               = POWER_SUPPLY_STATUS_DISCHARGING;
  26 static int battery_health               = POWER_SUPPLY_HEALTH_GOOD;
  27 static int battery_present              = 1; /* true */
  28 static int battery_technology           = POWER_SUPPLY_TECHNOLOGY_LION;
  29 static int battery_capacity             = 50;
  30
  31 static int test_power_get_ac_property(struct power_supply *psy,
  32                                       enum power_supply_property psp,
  33                                       union power_supply_propval *val)
  34 {
  35         switch (psp) {
  36         case POWER_SUPPLY_PROP_ONLINE:
  37                 val->intval = ac_online;
  38                 break;
  39         default:
  40                 return -EINVAL;
  41         }
  42         return 0;
  43 }
  44
  45 static int test_power_get_battery_property(struct power_supply *psy,
  46                                            enum power_supply_property psp,
  47                                            union power_supply_propval *val)
  48 {
  49         switch (psp) {
  50         case POWER_SUPPLY_PROP_MODEL_NAME:
  51                 val->strval = "Test battery";
  52                 break;
  53         case POWER_SUPPLY_PROP_MANUFACTURER:
  54                 val->strval = "Linux";
  55                 break;
  56         case POWER_SUPPLY_PROP_SERIAL_NUMBER:
  57                 val->strval = UTS_RELEASE;
  58                 break;
  59         case POWER_SUPPLY_PROP_STATUS:
  60                 val->intval = battery_status;
  61                 break;
  62         case POWER_SUPPLY_PROP_CHARGE_TYPE:
  63                 val->intval = POWER_SUPPLY_CHARGE_TYPE_FAST;
  64                 break;
  65         case POWER_SUPPLY_PROP_HEALTH:
  66                 val->intval = battery_health;
  67                 break;
  68         case POWER_SUPPLY_PROP_PRESENT:
  69                 val->intval = battery_present;
  70                 break;
  71         case POWER_SUPPLY_PROP_TECHNOLOGY:
  72                 val->intval = battery_technology;
  73                 break;
  74         case POWER_SUPPLY_PROP_CAPACITY_LEVEL:
  75                 val->intval = POWER_SUPPLY_CAPACITY_LEVEL_NORMAL;
  76                 break;
  77         case POWER_SUPPLY_PROP_CAPACITY:
  78         case POWER_SUPPLY_PROP_CHARGE_NOW:
  79                 val->intval = battery_capacity;
  80                 break;
  81         case POWER_SUPPLY_PROP_CHARGE_FULL_DESIGN:
  82         case POWER_SUPPLY_PROP_CHARGE_FULL:
  83                 val->intval = 100;
  84                 break;
  85         case POWER_SUPPLY_PROP_TIME_TO_EMPTY_AVG:
  86         case POWER_SUPPLY_PROP_TIME_TO_FULL_NOW:
  87                 val->intval = 3600;
  88                 break;
  89         default:
  90                 pr_info("%s: some properties deliberately report errors.\n",
  91                         __func__);
  92                 return -EINVAL;
  93         }
  94         return 0;
  95 }
  96
  97 static enum power_supply_property test_power_ac_props[] = {
  98         POWER_SUPPLY_PROP_ONLINE,
  99 };
 100
 101 static enum power_supply_property test_power_battery_props[] = {
 102         POWER_SUPPLY_PROP_STATUS,
 103         POWER_SUPPLY_PROP_CHARGE_TYPE,
 104         POWER_SUPPLY_PROP_HEALTH,
 105         POWER_SUPPLY_PROP_PRESENT,
 106         POWER_SUPPLY_PROP_TECHNOLOGY,
 107         POWER_SUPPLY_PROP_CHARGE_FULL_DESIGN,
 108         POWER_SUPPLY_PROP_CHARGE_FULL,
 109         POWER_SUPPLY_PROP_CHARGE_NOW,
 110         POWER_SUPPLY_PROP_CAPACITY,
 111         POWER_SUPPLY_PROP_CAPACITY_LEVEL,
 112         POWER_SUPPLY_PROP_TIME_TO_EMPTY_AVG,
 113         POWER_SUPPLY_PROP_TIME_TO_FULL_NOW,
 114         POWER_SUPPLY_PROP_MODEL_NAME,
 115         POWER_SUPPLY_PROP_MANUFACTURER,
 116         POWER_SUPPLY_PROP_SERIAL_NUMBER,
 117 };
 118
 119 static char *test_power_ac_supplied_to[] = {
 120         "test_battery",
 121 };
 122
 123 static struct power_supply test_power_supplies[] = {
 124         {
 125                 .name = "test_ac",
 126                 .type = POWER_SUPPLY_TYPE_MAINS,
 127                 .supplied_to = test_power_ac_supplied_to,
 128                 .num_supplicants = ARRAY_SIZE(test_power_ac_supplied_to),
 129                 .properties = test_power_ac_props,
 130                 .num_properties = ARRAY_SIZE(test_power_ac_props),
 131                 .get_property = test_power_get_ac_property,
 132         }, {
 133                 .name = "test_battery",
 134                 .type = POWER_SUPPLY_TYPE_BATTERY,
 135                 .properties = test_power_battery_props,
 136                 .num_properties = ARRAY_SIZE(test_power_battery_props),
 137                 .get_property = test_power_get_battery_property,
 138         },
 139 };
 140
 141
 142 static int __init test_power_init(void)
 143 {
 144         int i;
 145         int ret;
 146
 147         for (i = 0; i < ARRAY_SIZE(test_power_supplies); i++) {
 148                 ret = power_supply_register(NULL, &test_power_supplies[i]);
 149                 if (ret) {
 150                         pr_err("%s: failed to register %s\n", __func__,
 151                                 test_power_supplies[i].name);
 152                         goto failed;
 153                 }
 154         }
 155
 156         return 0;
 157 failed:
 158         while (--i >= 0)
 159                 power_supply_unregister(&test_power_supplies[i]);
 160         return ret;
 161 }
 162 module_init(test_power_init);
 163
 164 static void __exit test_power_exit(void)
 165 {
 166         int i;
 167
 168         /* Let's see how we handle changes... */
 169         ac_online = 0;
 170         battery_status = POWER_SUPPLY_STATUS_DISCHARGING;
 171         for (i = 0; i < ARRAY_SIZE(test_power_supplies); i++)
 172                 power_supply_changed(&test_power_supplies[i]);
 173         pr_info("%s: 'changed' event sent, sleeping for 10 seconds...\n",
 174                 __func__);
 175         ssleep(10);
 176
 177         for (i = 0; i < ARRAY_SIZE(test_power_supplies); i++)
 178                 power_supply_unregister(&test_power_supplies[i]);
 179 }
 180 module_exit(test_power_exit);
 181
 182
 183
 184 #define MAX_KEYLENGTH 256
 185 struct battery_property_map {
 186         int value;
 187         char const *key;
 188 };
 189
 190 static struct battery_property_map map_ac_online[] = {
 191         { 0,  "on"  },
 192         { 1,  "off" },
 193         { -1, NULL  },
 194 };
 195
 196 static struct battery_property_map map_status[] = {
 197         { POWER_SUPPLY_STATUS_CHARGING,     "charging"     },
 198         { POWER_SUPPLY_STATUS_DISCHARGING,  "discharging"  },
 199         { POWER_SUPPLY_STATUS_NOT_CHARGING, "not-charging" },
 200         { POWER_SUPPLY_STATUS_FULL,         "full"         },
 201         { -1,                               NULL           },
 202 };
 203
 204 static struct battery_property_map map_health[] = {
 205         { POWER_SUPPLY_HEALTH_GOOD,           "good"        },
 206         { POWER_SUPPLY_HEALTH_OVERHEAT,       "overheat"    },
 207         { POWER_SUPPLY_HEALTH_DEAD,           "dead"        },
 208         { POWER_SUPPLY_HEALTH_OVERVOLTAGE,    "overvoltage" },
 209         { POWER_SUPPLY_HEALTH_UNSPEC_FAILURE, "failure"     },
 210         { -1,                                 NULL          },
 211 };
 212
 213 static struct battery_property_map map_present[] = {
 214         { 0,  "false" },
 215         { 1,  "true"  },
 216         { -1, NULL    },
 217 };
 218
 219 static struct battery_property_map map_technology[] = {
 220         { POWER_SUPPLY_TECHNOLOGY_NiMH, "NiMH" },
 221         { POWER_SUPPLY_TECHNOLOGY_LION, "LION" },
 222         { POWER_SUPPLY_TECHNOLOGY_LIPO, "LIPO" },
 223         { POWER_SUPPLY_TECHNOLOGY_LiFe, "LiFe" },
 224         { POWER_SUPPLY_TECHNOLOGY_NiCd, "NiCd" },
 225         { POWER_SUPPLY_TECHNOLOGY_LiMn, "LiMn" },
 226         { -1,                           NULL   },
 227 };
 228
 229
 230 static int map_get_value(struct battery_property_map *map, const char *key,
 231                                 int def_val)
 232 {
 233         char buf[MAX_KEYLENGTH];
 234         int cr;
 235
 236         strncpy(buf, key, MAX_KEYLENGTH);
 237         buf[MAX_KEYLENGTH-1] = '\0';
 238
 239         cr = strnlen(buf, MAX_KEYLENGTH) - 1;
 240         if (cr < 0)
 241                 return def_val;
 242         if (buf[cr] == '\n')
 243                 buf[cr] = '\0';
 244
 245         while (map->key) {
 246                 if (strncasecmp(map->key, buf, MAX_KEYLENGTH) == 0)
 247                         return map->value;
 248                 map++;
 249         }
 250
 251         return def_val;
 252 }
 253
 254
 255 static const char *map_get_key(struct battery_property_map *map, int value,
 256                                 const char *def_key)
 257 {
 258         while (map->key) {
 259                 if (map->value == value)
 260                         return map->key;
 261                 map++;
 262         }
 263
 264         return def_key;
 265 }
 266
 267 static int param_set_ac_online(const char *key, const struct kernel_param *kp)
 268 {
 269         ac_online = map_get_value(map_ac_online, key, ac_online);
 270         power_supply_changed(&test_power_supplies[0]);
 271         return 0;
 272 }
 273
 274 static int param_get_ac_online(char *buffer, const struct kernel_param *kp)
 275 {
 276         strcpy(buffer, map_get_key(map_ac_online, ac_online, "unknown"));
 277         return strlen(buffer);
 278 }
 279
 280 static int param_set_battery_status(const char *key,
 281                                         const struct kernel_param *kp)
 282 {
 283         battery_status = map_get_value(map_status, key, battery_status);
 284         power_supply_changed(&test_power_supplies[1]);
 285         return 0;
 286 }
 287
 288 static int param_get_battery_status(char *buffer, const struct kernel_param *kp)
 289 {
 290         strcpy(buffer, map_get_key(map_status, battery_status, "unknown"));
 291         return strlen(buffer);
 292 }
 293
 294 static int param_set_battery_health(const char *key,
 295                                         const struct kernel_param *kp)
 296 {
 297         battery_health = map_get_value(map_health, key, battery_health);
 298         power_supply_changed(&test_power_supplies[1]);
 299         return 0;
 300 }
 301
 302 static int param_get_battery_health(char *buffer, const struct kernel_param *kp)
 303 {
 304         strcpy(buffer, map_get_key(map_health, battery_health, "unknown"));
 305         return strlen(buffer);
 306 }
 307
 308 static int param_set_battery_present(const char *key,
 309                                         const struct kernel_param *kp)
 310 {
 311         battery_present = map_get_value(map_present, key, battery_present);
 312         power_supply_changed(&test_power_supplies[0]);
 313         return 0;
 314 }
 315
 316 static int param_get_battery_present(char *buffer,
 317                                         const struct kernel_param *kp)
 318 {
 319         strcpy(buffer, map_get_key(map_present, battery_present, "unknown"));
 320         return strlen(buffer);
 321 }
 322
 323 static int param_set_battery_technology(const char *key,
 324                                         const struct kernel_param *kp)
 325 {
 326         battery_technology = map_get_value(map_technology, key,
 327                                                 battery_technology);
 328         power_supply_changed(&test_power_supplies[1]);
 329         return 0;
 330 }
 331
 332 static int param_get_battery_technology(char *buffer,
 333                                         const struct kernel_param *kp)
 334 {
 335         strcpy(buffer,
 336                 map_get_key(map_technology, battery_technology, "unknown"));
 337         return strlen(buffer);
 338 }
 339
 340 static int param_set_battery_capacity(const char *key,
 341                                         const struct kernel_param *kp)
 342 {
 343         int tmp;
 344
 345         if (1 != sscanf(key, "%d", &tmp))
 346                 return -EINVAL;
 347
 348         battery_capacity = tmp;
 349         power_supply_changed(&test_power_supplies[1]);
 350         return 0;
 351 }
 352
 353 #define param_get_battery_capacity param_get_int
 354
 355
 356
 357 static struct kernel_param_ops param_ops_ac_online = {
 358         .set = param_set_ac_online,
 359         .get = param_get_ac_online,
 360 };
 361
 362 static struct kernel_param_ops param_ops_battery_status = {
 363         .set = param_set_battery_status,
 364         .get = param_get_battery_status,
 365 };
 366
 367 static struct kernel_param_ops param_ops_battery_present = {
 368         .set = param_set_battery_present,
 369         .get = param_get_battery_present,
 370 };
 371
 372 static struct kernel_param_ops param_ops_battery_technology = {
 373         .set = param_set_battery_technology,
 374         .get = param_get_battery_technology,
 375 };
 376
 377 static struct kernel_param_ops param_ops_battery_health = {
 378         .set = param_set_battery_health,
 379         .get = param_get_battery_health,
 380 };
 381
 382 static struct kernel_param_ops param_ops_battery_capacity = {
 383         .set = param_set_battery_capacity,
 384         .get = param_get_battery_capacity,
 385 };
 386
 387
 388 #define param_check_ac_online(name, p) __param_check(name, p, void);
 389 #define param_check_battery_status(name, p) __param_check(name, p, void);
 390 #define param_check_battery_present(name, p) __param_check(name, p, void);
 391 #define param_check_battery_technology(name, p) __param_check(name, p, void);
 392 #define param_check_battery_health(name, p) __param_check(name, p, void);
 393 #define param_check_battery_capacity(name, p) __param_check(name, p, void);
 394
 395
 396 module_param(ac_online, ac_online, 0644);
 397 MODULE_PARM_DESC(ac_online, "AC charging state <on|off>");
 398
 399 module_param(battery_status, battery_status, 0644);
 400 MODULE_PARM_DESC(battery_status,
 401         "battery status <charging|discharging|not-charging|full>");
 402
 403 module_param(battery_present, battery_present, 0644);
 404 MODULE_PARM_DESC(battery_present,
 405         "battery presence state <good|overheat|dead|overvoltage|failure>");
 406
 407 module_param(battery_technology, battery_technology, 0644);
 408 MODULE_PARM_DESC(battery_technology,
 409         "battery technology <NiMH|LION|LIPO|LiFe|NiCd|LiMn>");
 410
 411 module_param(battery_health, battery_health, 0644);
 412 MODULE_PARM_DESC(battery_health,
 413         "battery health state <good|overheat|dead|overvoltage|failure>");
 414
 415 module_param(battery_capacity, battery_capacity, 0644);
 416 MODULE_PARM_DESC(battery_capacity, "battery capacity (percentage)");
 417
 418
 419 MODULE_DESCRIPTION("Power supply driver for testing");
 420 MODULE_AUTHOR("Anton Vorontsov <cbouatmailru@gmail.com>");
 421 MODULE_LICENSE("GPL");