Add ram and swap components on FreeBSD

[slstatus.git] / components / battery.c

diff --git a/components/battery.c b/components/battery.c
index 4c6b4f5..07b6ac1 100644 (file)
--- a/components/battery.c
+++ b/components/battery.c
@@ -1,5 +1,4 @@
 /* See LICENSE file for copyright and license details. */
 /* See LICENSE file for copyright and license details. */

-#include <errno.h>

 #include <stdio.h>
 #include <string.h>
 
 #include <stdio.h>
 #include <string.h>
 


@@ -7,6 +6,25 @@
 
 #if defined(__linux__)
        #include <limits.h>
 
 #if defined(__linux__)
        #include <limits.h>

+       #include <stdint.h>
+       #include <unistd.h>
+
+       static const char *
+       pick(const char *bat, const char *f1, const char *f2, char *path,
+            size_t length)
+       {
+               if (esnprintf(path, length, f1, bat) > 0 &&
+                   access(path, R_OK) == 0) {
+                       return f1;
+               }
+
+               if (esnprintf(path, length, f2, bat) > 0 &&
+                   access(path, R_OK) == 0) {
+                       return f2;
+               }
+
+               return NULL;
+       }

 
        const char *
        battery_perc(const char *bat)
 
        const char *
        battery_perc(const char *bat)


@@ -14,15 +32,21 @@
                int perc;
                char path[PATH_MAX];
 
                int perc;
                char path[PATH_MAX];
 

-               snprintf(path, sizeof(path), "%s%s%s", "/sys/class/power_supply/",
-                        bat, "/capacity");
-               return (pscanf(path, "%i", &perc) == 1) ? bprintf("%d", perc) : NULL;
+               if (esnprintf(path, sizeof(path),
+                             "/sys/class/power_supply/%s/capacity", bat) < 0) {
+                       return NULL;
+               }
+               if (pscanf(path, "%d", &perc) != 1) {
+                       return NULL;
+               }
+
+               return bprintf("%d", perc);

        }
 
        const char *
        battery_state(const char *bat)
        {
        }
 
        const char *
        battery_state(const char *bat)
        {

-               struct {
+               static struct {

                        char *state;
                        char *symbol;
                } map[] = {
                        char *state;
                        char *symbol;
                } map[] = {


@@ -32,8 +56,10 @@
                size_t i;
                char path[PATH_MAX], state[12];
 
                size_t i;
                char path[PATH_MAX], state[12];
 

-               snprintf(path, sizeof(path), "%s%s%s", "/sys/class/power_supply/",
-                        bat, "/status");
+               if (esnprintf(path, sizeof(path),
+                             "/sys/class/power_supply/%s/status", bat) < 0) {
+                       return NULL;
+               }

                if (pscanf(path, "%12s", state) != 1) {
                        return NULL;
                }
                if (pscanf(path, "%12s", state) != 1) {
                        return NULL;
                }


@@ -45,43 +71,91 @@
                }
                return (i == LEN(map)) ? "?" : map[i].symbol;
        }
                }
                return (i == LEN(map)) ? "?" : map[i].symbol;
        }

+
+       const char *
+       battery_remaining(const char *bat)
+       {
+               uintmax_t charge_now, current_now, m, h;
+               double timeleft;
+               char path[PATH_MAX], state[12];
+
+               if (esnprintf(path, sizeof(path),
+                             "/sys/class/power_supply/%s/status", bat) < 0) {
+                       return NULL;
+               }
+               if (pscanf(path, "%12s", state) != 1) {
+                       return NULL;
+               }
+
+               if (!pick(bat, "/sys/class/power_supply/%s/charge_now",
+                         "/sys/class/power_supply/%s/energy_now", path,
+                         sizeof(path)) ||
+                   pscanf(path, "%ju", &charge_now) < 0) {
+                       return NULL;
+               }
+
+               if (!strcmp(state, "Discharging")) {
+                       if (!pick(bat, "/sys/class/power_supply/%s/current_now",
+                                 "/sys/class/power_supply/%s/power_now", path,
+                                 sizeof(path)) ||
+                           pscanf(path, "%ju", &current_now) < 0) {
+                               return NULL;
+                       }
+
+                       if (current_now == 0) {
+                               return NULL;
+                       }
+
+                       timeleft = (double)charge_now / (double)current_now;
+                       h = timeleft;
+                       m = (timeleft - (double)h) * 60;
+
+                       return bprintf("%juh %jum", h, m);
+               }
+
+               return "";
+       }

 #elif defined(__OpenBSD__)
        #include <fcntl.h>
        #include <machine/apmvar.h>
        #include <sys/ioctl.h>
        #include <unistd.h>
 
 #elif defined(__OpenBSD__)
        #include <fcntl.h>
        #include <machine/apmvar.h>
        #include <sys/ioctl.h>
        #include <unistd.h>
 

-       const char *
-       battery_perc(const char *bat)
+       static int
+       load_apm_power_info(struct apm_power_info *apm_info)

        {
        {

-               struct apm_power_info apm_info;

                int fd;
 
                int fd;
 

-               UNUSED(bat); /* no way to specify battery on OpenBSD */
-

                fd = open("/dev/apm", O_RDONLY);
                if (fd < 0) {
                fd = open("/dev/apm", O_RDONLY);
                if (fd < 0) {

-                       fprintf(stderr, "open '/dev/apm': %s\n", strerror(errno));
-                       return NULL;
+                       warn("open '/dev/apm':");
+                       return 0;

                }
 
                }
 

-               if (ioctl(fd, APM_IOC_GETPOWER, &apm_info) < 0) {
-                       fprintf(stderr, "ioctl 'APM_IOC_GETPOWER': %s\n",
-                               strerror(errno));
+               memset(apm_info, 0, sizeof(struct apm_power_info));
+               if (ioctl(fd, APM_IOC_GETPOWER, apm_info) < 0) {
+                       warn("ioctl 'APM_IOC_GETPOWER':");

                        close(fd);
                        close(fd);

-                       return NULL;
+                       return 0;

                }
                }

-               close(fd);
-
-               return bprintf("%d", apm_info.battery_life);
+               return close(fd), 1;

        }
 
        const char *
        }
 
        const char *

-       battery_state(const char *bat)
+       battery_perc(const char *unused)

        {
        {

-               int fd;
-               size_t i;

                struct apm_power_info apm_info;
                struct apm_power_info apm_info;

+
+               if (load_apm_power_info(&apm_info)) {
+                       return bprintf("%d", apm_info.battery_life);
+               }
+
+               return NULL;
+       }
+
+       const char *
+       battery_state(const char *unused)
+       {

                struct {
                        unsigned int state;
                        char *symbol;
                struct {
                        unsigned int state;
                        char *symbol;


@@ -89,28 +163,89 @@
                        { APM_AC_ON,      "+" },
                        { APM_AC_OFF,     "-" },
                };
                        { APM_AC_ON,      "+" },
                        { APM_AC_OFF,     "-" },
                };

+               struct apm_power_info apm_info;
+               size_t i;
+
+               if (load_apm_power_info(&apm_info)) {
+                       for (i = 0; i < LEN(map); i++) {
+                               if (map[i].state == apm_info.ac_state) {
+                                       break;
+                               }
+                       }
+                       return (i == LEN(map)) ? "?" : map[i].symbol;
+               }

 
 

-               UNUSED(bat); /* no way to specify battery on OpenBSD */
+               return NULL;
+       }

 
 

-               fd = open("/dev/apm", O_RDONLY);
-               if (fd < 0) {
-                       fprintf(stderr, "open '/dev/apm': %s\n", strerror(errno));
-                       return NULL;
+       const char *
+       battery_remaining(const char *unused)
+       {
+               struct apm_power_info apm_info;
+
+               if (load_apm_power_info(&apm_info)) {
+                       if (apm_info.ac_state != APM_AC_ON) {
+                               return bprintf("%uh %02um",
+                                              apm_info.minutes_left / 60,
+                                              apm_info.minutes_left % 60);
+                       } else {
+                               return "";
+                       }

                }
 
                }
 

-               if (ioctl(fd, APM_IOC_GETPOWER, &apm_info) < 0) {
-                       fprintf(stderr, "ioctl 'APM_IOC_GETPOWER': %s\n",
-                               strerror(errno));
-                       close(fd);
+               return NULL;
+       }
+#elif defined(__FreeBSD__)
+       #include <sys/sysctl.h>
+
+       const char *
+       battery_perc(const char *unused)
+       {
+               int cap;
+               size_t len;
+
+               len = sizeof(cap);
+               if (sysctlbyname("hw.acpi.battery.life", &cap, &len, NULL, 0) == -1
+                               || !len)

                        return NULL;
                        return NULL;

-               }
-               close(fd);

 
 

-               for (i = 0; i < LEN(map); i++) {
-                       if (map[i].state == apm_info.ac_state) {
-                               break;
-                       }
+               return bprintf("%d", cap);
+       }
+
+       const char *
+       battery_state(const char *unused)
+       {
+               int state;
+               size_t len;
+
+               len = sizeof(state);
+               if (sysctlbyname("hw.acpi.battery.state", &state, &len, NULL, 0) == -1
+                               || !len)
+                       return NULL;
+
+               switch(state) {
+                       case 0:
+                       case 2:
+                               return "+";
+                       case 1:
+                               return "-";
+                       default:
+                               return "?";

                }
                }

-               return (i == LEN(map)) ? "?" : map[i].symbol;
+       }
+
+       const char *
+       battery_remaining(const char *unused)
+       {
+               int rem;
+               size_t len;
+
+               len = sizeof(rem);
+               if (sysctlbyname("hw.acpi.battery.time", &rem, &len, NULL, 0) == -1
+                               || !len
+                               || rem == -1)
+                       return NULL;
+
+               return bprintf("%uh %02um", rem / 60, rem % 60);

        }
 #endif
        }
 #endif