platform_system_core/charger/charger.c

980 lines
25 KiB
C
Raw Normal View History

/*
* Copyright (C) 2011 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
//#define DEBUG_UEVENTS
#define CHARGER_KLOG_LEVEL 6
#include <dirent.h>
#include <errno.h>
#include <fcntl.h>
#include <linux/input.h>
#include <linux/netlink.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/poll.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/un.h>
#include <time.h>
#include <unistd.h>
#include <cutils/android_reboot.h>
#include <cutils/klog.h>
#include <cutils/list.h>
#include <cutils/misc.h>
#include <cutils/uevent.h>
#include "minui/minui.h"
#ifndef max
#define max(a,b) ((a) > (b) ? (a) : (b))
#endif
#ifndef min
#define min(a,b) ((a) < (b) ? (a) : (b))
#endif
#define ARRAY_SIZE(x) (sizeof(x)/sizeof(x[0]))
#define MSEC_PER_SEC (1000LL)
#define NSEC_PER_MSEC (1000000LL)
#define BATTERY_UNKNOWN_TIME (2 * MSEC_PER_SEC)
#define POWER_ON_KEY_TIME (2 * MSEC_PER_SEC)
#define UNPLUGGED_SHUTDOWN_TIME (10 * MSEC_PER_SEC)
#define BATTERY_FULL_THRESH 95
#define LAST_KMSG_PATH "/proc/last_kmsg"
#define LAST_KMSG_MAX_SZ (32 * 1024)
#define LOGE(x...) do { KLOG_ERROR("charger", x); } while (0)
#define LOGI(x...) do { KLOG_INFO("charger", x); } while (0)
#define LOGV(x...) do { KLOG_DEBUG("charger", x); } while (0)
struct key_state {
bool pending;
bool down;
int64_t timestamp;
};
struct power_supply {
struct listnode list;
char name[256];
char type[32];
bool online;
bool valid;
char cap_path[PATH_MAX];
};
struct frame {
const char *name;
int disp_time;
int min_capacity;
gr_surface surface;
};
struct animation {
bool run;
struct frame *frames;
int cur_frame;
int num_frames;
int cur_cycle;
int num_cycles;
/* current capacity being animated */
int capacity;
};
struct charger {
int64_t next_screen_transition;
int64_t next_key_check;
int64_t next_pwr_check;
struct key_state keys[KEY_MAX + 1];
int uevent_fd;
struct listnode supplies;
int num_supplies;
int num_supplies_online;
struct animation *batt_anim;
gr_surface surf_unknown;
struct power_supply *battery;
};
struct uevent {
const char *action;
const char *path;
const char *subsystem;
const char *ps_name;
const char *ps_type;
const char *ps_online;
};
static struct frame batt_anim_frames[] = {
{
.name = "charger/battery_0",
.disp_time = 750,
.min_capacity = 0,
},
{
.name = "charger/battery_1",
.disp_time = 750,
.min_capacity = 20,
},
{
.name = "charger/battery_2",
.disp_time = 750,
.min_capacity = 40,
},
{
.name = "charger/battery_3",
.disp_time = 750,
.min_capacity = 60,
},
{
.name = "charger/battery_4",
.disp_time = 750,
.min_capacity = 80,
},
{
.name = "charger/battery_5",
.disp_time = 750,
.min_capacity = BATTERY_FULL_THRESH,
},
};
static struct animation battery_animation = {
.frames = batt_anim_frames,
.num_frames = ARRAY_SIZE(batt_anim_frames),
.num_cycles = 3,
};
static struct charger charger_state = {
.batt_anim = &battery_animation,
};
static int char_width;
static int char_height;
/* current time in milliseconds */
static int64_t curr_time_ms(void)
{
struct timespec tm;
clock_gettime(CLOCK_MONOTONIC, &tm);
return tm.tv_sec * MSEC_PER_SEC + (tm.tv_nsec / NSEC_PER_MSEC);
}
static void clear_screen(void)
{
gr_color(0, 0, 0, 255);
gr_fill(0, 0, gr_fb_width(), gr_fb_height());
};
#define MAX_KLOG_WRITE_BUF_SZ 256
static void dump_last_kmsg(void)
{
char *buf;
char *ptr;
unsigned sz = 0;
int len;
LOGI("\n");
LOGI("*************** LAST KMSG ***************\n");
LOGI("\n");
buf = load_file(LAST_KMSG_PATH, &sz);
if (!buf || !sz) {
LOGI("last_kmsg not found. Cold reset?\n");
goto out;
}
len = min(sz, LAST_KMSG_MAX_SZ);
ptr = buf + (sz - len);
while (len > 0) {
int cnt = min(len, MAX_KLOG_WRITE_BUF_SZ);
char yoink;
char *nl;
nl = memrchr(ptr, '\n', cnt - 1);
if (nl)
cnt = nl - ptr + 1;
yoink = ptr[cnt];
ptr[cnt] = '\0';
klog_write(6, "<6>%s", ptr);
ptr[cnt] = yoink;
len -= cnt;
ptr += cnt;
}
free(buf);
out:
LOGI("\n");
LOGI("************* END LAST KMSG *************\n");
LOGI("\n");
}
static int read_file(const char *path, char *buf, size_t sz)
{
int fd;
size_t cnt;
fd = open(path, O_RDONLY, 0);
if (fd < 0)
goto err;
cnt = read(fd, buf, sz - 1);
if (cnt <= 0)
goto err;
buf[cnt] = '\0';
if (buf[cnt - 1] == '\n') {
cnt--;
buf[cnt] = '\0';
}
close(fd);
return cnt;
err:
if (fd >= 0)
close(fd);
return -1;
}
static int read_file_int(const char *path, int *val)
{
char buf[32];
int ret;
int tmp;
char *end;
ret = read_file(path, buf, sizeof(buf));
if (ret < 0)
return -1;
tmp = strtol(buf, &end, 0);
if (end == buf ||
((end < buf+sizeof(buf)) && (*end != '\n' && *end != '\0')))
goto err;
*val = tmp;
return 0;
err:
return -1;
}
static int get_battery_capacity(struct charger *charger)
{
int ret;
int batt_cap = -1;
if (!charger->battery)
return -1;
ret = read_file_int(charger->battery->cap_path, &batt_cap);
if (ret < 0 || batt_cap > 100) {
batt_cap = -1;
}
return batt_cap;
}
static struct power_supply *find_supply(struct charger *charger,
const char *name)
{
struct listnode *node;
struct power_supply *supply;
list_for_each(node, &charger->supplies) {
supply = node_to_item(node, struct power_supply, list);
if (!strncmp(name, supply->name, sizeof(supply->name)))
return supply;
}
return NULL;
}
static struct power_supply *add_supply(struct charger *charger,
const char *name, const char *type,
const char *path, bool online)
{
struct power_supply *supply;
supply = calloc(1, sizeof(struct power_supply));
if (!supply)
return NULL;
strlcpy(supply->name, name, sizeof(supply->name));
strlcpy(supply->type, type, sizeof(supply->type));
snprintf(supply->cap_path, sizeof(supply->cap_path),
"/sys/%s/capacity", path);
supply->online = online;
list_add_tail(&charger->supplies, &supply->list);
charger->num_supplies++;
LOGV("... added %s %s %d\n", supply->name, supply->type, online);
return supply;
}
static void remove_supply(struct charger *charger, struct power_supply *supply)
{
if (!supply)
return;
list_remove(&supply->list);
charger->num_supplies--;
free(supply);
}
static void parse_uevent(const char *msg, struct uevent *uevent)
{
uevent->action = "";
uevent->path = "";
uevent->subsystem = "";
uevent->ps_name = "";
uevent->ps_online = "";
uevent->ps_type = "";
/* currently ignoring SEQNUM */
while (*msg) {
#ifdef DEBUG_UEVENTS
LOGV("uevent str: %s\n", msg);
#endif
if (!strncmp(msg, "ACTION=", 7)) {
msg += 7;
uevent->action = msg;
} else if (!strncmp(msg, "DEVPATH=", 8)) {
msg += 8;
uevent->path = msg;
} else if (!strncmp(msg, "SUBSYSTEM=", 10)) {
msg += 10;
uevent->subsystem = msg;
} else if (!strncmp(msg, "POWER_SUPPLY_NAME=", 18)) {
msg += 18;
uevent->ps_name = msg;
} else if (!strncmp(msg, "POWER_SUPPLY_ONLINE=", 20)) {
msg += 20;
uevent->ps_online = msg;
} else if (!strncmp(msg, "POWER_SUPPLY_TYPE=", 18)) {
msg += 18;
uevent->ps_type = msg;
}
/* advance to after the next \0 */
while (*msg++)
;
}
LOGV("event { '%s', '%s', '%s', '%s', '%s', '%s' }\n",
uevent->action, uevent->path, uevent->subsystem,
uevent->ps_name, uevent->ps_type, uevent->ps_online);
}
static void process_ps_uevent(struct charger *charger, struct uevent *uevent)
{
int online;
char ps_type[32];
struct power_supply *supply = NULL;
int i;
bool was_online = false;
bool battery = false;
if (uevent->ps_type[0] == '\0') {
char *path;
int ret;
if (uevent->path[0] == '\0')
return;
ret = asprintf(&path, "/sys/%s/type", uevent->path);
if (ret <= 0)
return;
ret = read_file(path, ps_type, sizeof(ps_type));
free(path);
if (ret < 0)
return;
} else {
strlcpy(ps_type, uevent->ps_type, sizeof(ps_type));
}
if (!strncmp(ps_type, "Battery", 7))
battery = true;
online = atoi(uevent->ps_online);
supply = find_supply(charger, uevent->ps_name);
if (supply) {
was_online = supply->online;
supply->online = online;
}
if (!strcmp(uevent->action, "add")) {
if (!supply) {
supply = add_supply(charger, uevent->ps_name, ps_type, uevent->path,
online);
if (!supply) {
LOGE("cannot add supply '%s' (%s %d)\n", uevent->ps_name,
uevent->ps_type, online);
return;
}
/* only pick up the first battery for now */
if (battery && !charger->battery)
charger->battery = supply;
} else {
LOGE("supply '%s' already exists..\n", uevent->ps_name);
}
} else if (!strcmp(uevent->action, "remove")) {
if (supply) {
if (charger->battery == supply)
charger->battery = NULL;
remove_supply(charger, supply);
supply = NULL;
}
} else if (!strcmp(uevent->action, "change")) {
if (!supply) {
LOGE("power supply '%s' not found ('%s' %d)\n",
uevent->ps_name, ps_type, online);
return;
}
} else {
return;
}
/* allow battery to be managed in the supply list but make it not
* contribute to online power supplies. */
if (!battery) {
if (was_online && !online)
charger->num_supplies_online--;
else if (supply && !was_online && online)
charger->num_supplies_online++;
}
LOGI("power supply %s (%s) %s (action=%s num_online=%d num_supplies=%d)\n",
uevent->ps_name, ps_type, battery ? "" : online ? "online" : "offline",
uevent->action, charger->num_supplies_online, charger->num_supplies);
}
static void process_uevent(struct charger *charger, struct uevent *uevent)
{
if (!strcmp(uevent->subsystem, "power_supply"))
process_ps_uevent(charger, uevent);
}
#define UEVENT_MSG_LEN 1024
static int handle_uevent_fd(struct charger *charger, int fd)
{
char msg[UEVENT_MSG_LEN+2];
int n;
if (fd < 0)
return -1;
while (true) {
struct uevent uevent;
n = uevent_kernel_multicast_recv(fd, msg, UEVENT_MSG_LEN);
if (n <= 0)
break;
if (n >= UEVENT_MSG_LEN) /* overflow -- discard */
continue;
msg[n] = '\0';
msg[n+1] = '\0';
parse_uevent(msg, &uevent);
process_uevent(charger, &uevent);
}
return 0;
}
static int uevent_callback(int fd, short revents, void *data)
{
struct charger *charger = data;
if (!(revents & POLLIN))
return -1;
return handle_uevent_fd(charger, fd);
}
/* force the kernel to regenerate the change events for the existing
* devices, if valid */
static void do_coldboot(struct charger *charger, DIR *d, const char *event,
bool follow_links, int max_depth)
{
struct dirent *de;
int dfd, fd;
dfd = dirfd(d);
fd = openat(dfd, "uevent", O_WRONLY);
if (fd >= 0) {
write(fd, event, strlen(event));
close(fd);
handle_uevent_fd(charger, charger->uevent_fd);
}
while ((de = readdir(d)) && max_depth > 0) {
DIR *d2;
LOGV("looking at '%s'\n", de->d_name);
if ((de->d_type != DT_DIR && !(de->d_type == DT_LNK && follow_links)) ||
de->d_name[0] == '.') {
LOGV("skipping '%s' type %d (depth=%d follow=%d)\n",
de->d_name, de->d_type, max_depth, follow_links);
continue;
}
LOGV("can descend into '%s'\n", de->d_name);
fd = openat(dfd, de->d_name, O_RDONLY | O_DIRECTORY);
if (fd < 0) {
LOGE("cannot openat %d '%s' (%d: %s)\n", dfd, de->d_name,
errno, strerror(errno));
continue;
}
d2 = fdopendir(fd);
if (d2 == 0)
close(fd);
else {
LOGV("opened '%s'\n", de->d_name);
do_coldboot(charger, d2, event, follow_links, max_depth - 1);
closedir(d2);
}
}
}
static void coldboot(struct charger *charger, const char *path,
const char *event)
{
char str[256];
LOGV("doing coldboot '%s' in '%s'\n", event, path);
DIR *d = opendir(path);
if (d) {
snprintf(str, sizeof(str), "%s\n", event);
do_coldboot(charger, d, str, true, 1);
closedir(d);
}
}
static int draw_text(const char *str, int x, int y)
{
int str_len_px = gr_measure(str);
if (x < 0)
x = (gr_fb_width() - str_len_px) / 2;
if (y < 0)
y = (gr_fb_height() - char_height) / 2;
gr_text(x, y, str);
return y + char_height;
}
static void android_green(void)
{
gr_color(0xa4, 0xc6, 0x39, 255);
}
/* returns the last y-offset of where the surface ends */
static int draw_surface_centered(struct charger *charger, gr_surface surface)
{
int w;
int h;
int x;
int y;
w = gr_get_width(surface);
h = gr_get_height(surface);
x = (gr_fb_width() - w) / 2 ;
y = (gr_fb_height() - h) / 2 ;
LOGV("drawing surface %dx%d+%d+%d\n", w, h, x, y);
gr_blit(surface, 0, 0, w, h, x, y);
return y + h;
}
static void draw_unknown(struct charger *charger)
{
int y;
if (charger->surf_unknown) {
draw_surface_centered(charger, charger->surf_unknown);
} else {
android_green();
y = draw_text("Charging!", -1, -1);
draw_text("?\?/100", -1, y + 25);
}
}
static void draw_battery(struct charger *charger)
{
struct animation *batt_anim = charger->batt_anim;
struct frame *frame = &batt_anim->frames[batt_anim->cur_frame];
if (batt_anim->num_frames != 0) {
draw_surface_centered(charger, frame->surface);
LOGV("drawing frame #%d name=%s min_cap=%d time=%d\n",
batt_anim->cur_frame, frame->name, frame->min_capacity,
frame->disp_time);
}
}
static void redraw_screen(struct charger *charger)
{
struct animation *batt_anim = charger->batt_anim;
clear_screen();
/* try to display *something* */
if (batt_anim->capacity < 0 || batt_anim->num_frames == 0)
draw_unknown(charger);
else
draw_battery(charger);
gr_flip();
}
static void kick_animation(struct animation *anim)
{
anim->run = true;
}
static void reset_animation(struct animation *anim)
{
anim->cur_cycle = 0;
anim->cur_frame = 0;
anim->run = false;
}
static void update_screen_state(struct charger *charger, int64_t now)
{
struct animation *batt_anim = charger->batt_anim;
int cur_frame;
int disp_time;
if (!batt_anim->run || now < charger->next_screen_transition)
return;
/* animation is over, blank screen and leave */
if (batt_anim->cur_cycle == batt_anim->num_cycles) {
reset_animation(batt_anim);
charger->next_screen_transition = -1;
gr_fb_blank(true);
LOGV("[%lld] animation done\n", now);
return;
}
disp_time = batt_anim->frames[batt_anim->cur_frame].disp_time;
/* animation starting, set up the animation */
if (batt_anim->cur_frame == 0) {
int batt_cap;
int ret;
LOGV("[%lld] animation starting\n", now);
batt_cap = get_battery_capacity(charger);
if (batt_cap >= 0 && batt_anim->num_frames != 0) {
int i;
/* find first frame given current capacity */
for (i = 1; i < batt_anim->num_frames; i++) {
if (batt_cap < batt_anim->frames[i].min_capacity)
break;
}
batt_anim->cur_frame = i - 1;
/* show the first frame for twice as long */
disp_time = batt_anim->frames[batt_anim->cur_frame].disp_time * 2;
}
batt_anim->capacity = batt_cap;
}
/* unblank the screen on first cycle */
if (batt_anim->cur_cycle == 0)
gr_fb_blank(false);
/* draw the new frame (@ cur_frame) */
redraw_screen(charger);
/* if we don't have anim frames, we only have one image, so just bump
* the cycle counter and exit
*/
if (batt_anim->num_frames == 0 || batt_anim->capacity < 0) {
LOGV("[%lld] animation missing or unknown battery status\n", now);
charger->next_screen_transition = now + BATTERY_UNKNOWN_TIME;
batt_anim->cur_cycle++;
return;
}
/* schedule next screen transition */
charger->next_screen_transition = now + disp_time;
/* advance frame cntr to the next valid frame
* if necessary, advance cycle cntr, and reset frame cntr
*/
batt_anim->cur_frame++;
if (batt_anim->cur_frame == batt_anim->num_frames) {
batt_anim->cur_cycle++;
batt_anim->cur_frame = 0;
/* don't reset the cycle counter, since we use that as a signal
* in a test above to check if animation is over
*/
}
}
static int set_key_callback(int code, int value, void *data)
{
struct charger *charger = data;
int64_t now = curr_time_ms();
int down = !!value;
if (code > KEY_MAX)
return -1;
/* ignore events that don't modify our state */
if (charger->keys[code].down == down)
return -1;
/* only record the down even timestamp, as the amount
* of time the key spent not being pressed is not useful */
if (down)
charger->keys[code].timestamp = now;
charger->keys[code].down = down;
charger->keys[code].pending = true;
if (down) {
LOGV("[%lld] key[%d] down\n", now, code);
} else {
int64_t duration = now - charger->keys[code].timestamp;
int64_t secs = duration / 1000;
int64_t msecs = duration - secs * 1000;
LOGV("[%lld] key[%d] up (was down for %lld.%lldsec)\n", now,
code, secs, msecs);
}
return 0;
}
static void update_input_state(struct charger *charger,
struct input_event *ev)
{
if (ev->type != EV_KEY)
return;
set_key_callback(ev->code, ev->value, charger);
}
static void set_next_key_check(struct charger *charger,
struct key_state *key,
int64_t timeout)
{
int64_t then = key->timestamp + timeout;
if (charger->next_key_check == -1 || then < charger->next_key_check)
charger->next_key_check = then;
}
static void process_key(struct charger *charger, int code, int64_t now)
{
struct key_state *key = &charger->keys[code];
int64_t next_key_check;
if (code == KEY_POWER) {
if (key->down) {
int64_t reboot_timeout = key->timestamp + POWER_ON_KEY_TIME;
if (now >= reboot_timeout) {
LOGI("[%lld] rebooting\n", now);
android_reboot(ANDROID_RB_RESTART, 0, 0);
} else {
/* if the key is pressed but timeout hasn't expired,
* make sure we wake up at the right-ish time to check
*/
set_next_key_check(charger, key, POWER_ON_KEY_TIME);
}
} else {
/* if the power key got released, force screen state cycle */
if (key->pending)
kick_animation(charger->batt_anim);
}
}
key->pending = false;
}
static void handle_input_state(struct charger *charger, int64_t now)
{
process_key(charger, KEY_POWER, now);
if (charger->next_key_check != -1 && now > charger->next_key_check)
charger->next_key_check = -1;
}
static void handle_power_supply_state(struct charger *charger, int64_t now)
{
if (charger->num_supplies_online == 0) {
if (charger->next_pwr_check == -1) {
charger->next_pwr_check = now + UNPLUGGED_SHUTDOWN_TIME;
LOGI("[%lld] device unplugged: shutting down in %lld (@ %lld)\n",
now, UNPLUGGED_SHUTDOWN_TIME, charger->next_pwr_check);
} else if (now >= charger->next_pwr_check) {
LOGI("[%lld] shutting down\n", now);
android_reboot(ANDROID_RB_POWEROFF, 0, 0);
} else {
/* otherwise we already have a shutdown timer scheduled */
}
} else {
/* online supply present, reset shutdown timer if set */
if (charger->next_pwr_check != -1) {
LOGI("[%lld] device plugged in: shutdown cancelled\n", now);
kick_animation(charger->batt_anim);
}
charger->next_pwr_check = -1;
}
}
static void wait_next_event(struct charger *charger, int64_t now)
{
int64_t next_event = INT64_MAX;
int64_t timeout;
struct input_event ev;
int ret;
LOGV("[%lld] next screen: %lld next key: %lld next pwr: %lld\n", now,
charger->next_screen_transition, charger->next_key_check,
charger->next_pwr_check);
if (charger->next_screen_transition != -1)
next_event = charger->next_screen_transition;
if (charger->next_key_check != -1 && charger->next_key_check < next_event)
next_event = charger->next_key_check;
if (charger->next_pwr_check != -1 && charger->next_pwr_check < next_event)
next_event = charger->next_pwr_check;
if (next_event != -1 && next_event != INT64_MAX)
timeout = max(0, next_event - now);
else
timeout = -1;
LOGV("[%lld] blocking (%lld)\n", now, timeout);
ret = ev_wait((int)timeout);
if (!ret)
ev_dispatch();
}
static int input_callback(int fd, short revents, void *data)
{
struct charger *charger = data;
struct input_event ev;
int ret;
ret = ev_get_input(fd, revents, &ev);
if (ret)
return -1;
update_input_state(charger, &ev);
return 0;
}
static void event_loop(struct charger *charger)
{
int ret;
while (true) {
int64_t now = curr_time_ms();
LOGV("[%lld] event_loop()\n", now);
handle_input_state(charger, now);
handle_power_supply_state(charger, now);
/* do screen update last in case any of the above want to start
* screen transitions (animations, etc)
*/
update_screen_state(charger, now);
wait_next_event(charger, now);
}
}
int main(int argc, char **argv)
{
int ret;
struct charger *charger = &charger_state;
int64_t now = curr_time_ms() - 1;
int fd;
int i;
list_init(&charger->supplies);
klog_init();
klog_set_level(CHARGER_KLOG_LEVEL);
dump_last_kmsg();
LOGI("--------------- STARTING CHARGER MODE ---------------\n");
gr_init();
gr_font_size(&char_width, &char_height);
ev_init(input_callback, charger);
fd = uevent_open_socket(64*1024, true);
if (fd >= 0) {
fcntl(fd, F_SETFL, O_NONBLOCK);
ev_add_fd(fd, uevent_callback, charger);
}
charger->uevent_fd = fd;
coldboot(charger, "/sys/class/power_supply", "add");
ret = res_create_surface("charger/battery_fail", &charger->surf_unknown);
if (ret < 0) {
LOGE("Cannot load image\n");
charger->surf_unknown = NULL;
}
for (i = 0; i < charger->batt_anim->num_frames; i++) {
struct frame *frame = &charger->batt_anim->frames[i];
ret = res_create_surface(frame->name, &frame->surface);
if (ret < 0) {
LOGE("Cannot load image %s\n", frame->name);
/* TODO: free the already allocated surfaces... */
charger->batt_anim->num_frames = 0;
charger->batt_anim->num_cycles = 1;
break;
}
}
ev_sync_key_state(set_key_callback, charger);
gr_fb_blank(true);
charger->next_screen_transition = now - 1;
charger->next_key_check = -1;
charger->next_pwr_check = -1;
reset_animation(charger->batt_anim);
kick_animation(charger->batt_anim);
event_loop(charger);
return 0;
}