Xinqi Bao's Git

projects / slstatus.git / blobdiff
summary | log | commit | diff | tree
raw | inline | side by side
fix delay in a cleaner way
[slstatus.git] / slstatus.c
diff --git a/slstatus.c b/slstatus.c
index be124f3..c92da0d 100644 (file)
--- a/slstatus.c
+++ b/slstatus.c
@@ -6,6 +6,7 @@
 #include <ifaddrs.h>
 #include <limits.h>
 #include <linux/wireless.h>
 #include <ifaddrs.h>
 #include <limits.h>
 #include <linux/wireless.h>
+#include <locale.h>
 #include <netdb.h>
 #include <pwd.h>
 #include <signal.h>
 #include <netdb.h>
 #include <pwd.h>
 #include <signal.h>
@@ -19,6 +20,7 @@
 #include <sys/socket.h>
 #include <sys/sysinfo.h>
 #include <sys/types.h>
 #include <sys/socket.h>
 #include <sys/sysinfo.h>
 #include <sys/types.h>
+#include <sys/utsname.h>
 #include <time.h>
 #include <unistd.h>
 #include <X11/Xlib.h>
 #include <time.h>
 #include <unistd.h>
 #include <X11/Xlib.h>
@@ -26,51 +28,56 @@
 #undef strlcat
 #undef strlcpy
 
 #undef strlcat
 #undef strlcpy
 
-#include "arg.h"
-#include "strlcat.h"
-#include "strlcpy.h"
-#include "concat.h"
-
-char concat[];
+#include "extern/arg.h"
+#include "extern/strlcat.h"
+#include "extern/concat.h"
 
 struct arg {
        char *(*func)();
 
 struct arg {
        char *(*func)();
-       const char *format;
+       const char *fmt;
        const char *args;
 };
 
        const char *args;
 };
 
-static char *smprintf(const char *, ...);
-static char *battery_perc(const char *);
-static char *battery_state(const char *);
+static char *smprintf(const char *fmt, ...);
+static char *battery_perc(const char *bat);
+static char *battery_state(const char *bat);
 static char *cpu_perc(void);
 static char *cpu_perc(void);
-static char *datetime(const char *);
-static char *disk_free(const char *);
-static char *disk_perc(const char *);
-static char *disk_total(const char *);
-static char *disk_used(const char *);
+static char *datetime(const char *fmt);
+static char *disk_free(const char *mnt);
+static char *disk_perc(const char *mnt);
+static char *disk_total(const char *mnt);
+static char *disk_used(const char *mnt);
 static char *entropy(void);
 static char *gid(void);
 static char *hostname(void);
 static char *entropy(void);
 static char *gid(void);
 static char *hostname(void);
-static char *ip(const char *);
+static char *ip(const char *iface);
 static char *load_avg(void);
 static char *ram_free(void);
 static char *ram_perc(void);
 static char *ram_used(void);
 static char *ram_total(void);
 static char *load_avg(void);
 static char *ram_free(void);
 static char *ram_perc(void);
 static char *ram_used(void);
 static char *ram_total(void);
-static char *run_command(const char *);
-static char *temp(const char *);
+static char *run_command(const char *cmd);
+static char *swap_free(void);
+static char *swap_perc(void);
+static char *swap_used(void);
+static char *swap_total(void);
+static char *temp(const char *file);
 static char *uid(void);
 static char *uptime(void);
 static char *username(void);
 static char *uid(void);
 static char *uptime(void);
 static char *username(void);
-static char *vol_perc(const char *);
-static char *wifi_perc(const char *);
-static char *wifi_essid(const char *);
-static void set_status(const char *);
-static void sighandler(const int);
+static char *vol_perc(const char *card);
+static char *wifi_perc(const char *iface);
+static char *wifi_essid(const char *iface);
+static char *kernel_release(void);
+static void set_status(const char *str);
+static void sighandler(const int signo);
 static void usage(void);
 
 char *argv0;
 static void usage(void);
 
 char *argv0;
-static unsigned short int delay, done, dflag, oflag;
+char concat[];
+static unsigned short int delay = 0;
+static unsigned short int done;
+static unsigned short int dflag, oflag;
 static Display *dpy;
 
 #include "config.h"
 static Display *dpy;
 
 #include "config.h"
@@ -88,8 +95,7 @@ smprintf(const char *fmt, ...)
 
        ret = malloc(++len);
        if (ret == NULL) {
 
        ret = malloc(++len);
        if (ret == NULL) {
-               perror("malloc");
-               exit(1);
+               err(1, "malloc");
        }
 
        va_start(ap, fmt);
        }
 
        va_start(ap, fmt);
@@ -100,15 +106,15 @@ smprintf(const char *fmt, ...)
 }
 
 static char *
 }
 
 static char *
-battery_perc(const char *battery)
+battery_perc(const char *bat)
 {
        int perc;
        FILE *fp;
 
 {
        int perc;
        FILE *fp;
 
-       ccat(3, "/sys/class/power_supply/", battery, "/capacity");
+       ccat(3, "/sys/class/power_supply/", bat, "/capacity");
        fp = fopen(concat, "r");
        if (fp == NULL) {
        fp = fopen(concat, "r");
        if (fp == NULL) {
-               warn("Error opening battery file: %s", concat);
+               warn("Failed to open file %s", concat);
                return smprintf(UNKNOWN_STR);
        }
        fscanf(fp, "%i", &perc);
                return smprintf(UNKNOWN_STR);
        }
        fscanf(fp, "%i", &perc);
@@ -118,28 +124,29 @@ battery_perc(const char *battery)
 }
 
 static char *
 }
 
 static char *
-battery_state(const char *battery)
+battery_state(const char *bat)
 {
 {
-       char state[12]; 
+       char state[12];
        FILE *fp;
 
        FILE *fp;
 
-       ccat(3, "/sys/class/power_supply/", battery, "/status");
+       ccat(3, "/sys/class/power_supply/", bat, "/status");
        fp = fopen(concat, "r");
        if (fp == NULL) {
        fp = fopen(concat, "r");
        if (fp == NULL) {
-               warn("Error opening battery file: %s", concat);
+               warn("Failed to open file %s", concat);
                return smprintf(UNKNOWN_STR);
        }
        fscanf(fp, "%12s", state);
        fclose(fp);
 
                return smprintf(UNKNOWN_STR);
        }
        fscanf(fp, "%12s", state);
        fclose(fp);
 
-       if (strcmp(state, "Charging") == 0)
+       if (strcmp(state, "Charging") == 0) {
                return smprintf("+");
                return smprintf("+");
-       else if (strcmp(state, "Discharging") == 0)
+       } else if (strcmp(state, "Discharging") == 0) {
                return smprintf("-");
                return smprintf("-");
-       else if (strcmp(state, "Full") == 0)
+       } else if (strcmp(state, "Full") == 0) {
                return smprintf("=");
                return smprintf("=");
-       else
+       } else {
                return smprintf("?");
                return smprintf("?");
+       }
 }
 
 static char *
 }
 
 static char *
@@ -147,52 +154,52 @@ cpu_perc(void)
 {
        int perc;
        long double a[4], b[4];
 {
        int perc;
        long double a[4], b[4];
-       FILE *fp = fopen("/proc/stat","r");
+       FILE *fp;
 
 
+       fp = fopen("/proc/stat", "r");
        if (fp == NULL) {
        if (fp == NULL) {
-               warn("Error opening stat file");
+               warn("Failed to open file /proc/stat");
                return smprintf(UNKNOWN_STR);
        }
                return smprintf(UNKNOWN_STR);
        }
-
        fscanf(fp, "%*s %Lf %Lf %Lf %Lf", &a[0], &a[1], &a[2], &a[3]);
        fclose(fp);
 
        fscanf(fp, "%*s %Lf %Lf %Lf %Lf", &a[0], &a[1], &a[2], &a[3]);
        fclose(fp);
 
-       delay = (UPDATE_INTERVAL - (UPDATE_INTERVAL - 1));
+       delay++;
        sleep(delay);
 
        sleep(delay);
 
-       fp = fopen("/proc/stat","r");
+       fp = fopen("/proc/stat", "r");
        if (fp == NULL) {
        if (fp == NULL) {
-               warn("Error opening stat file");
+               warn("Failed to open file /proc/stat");
                return smprintf(UNKNOWN_STR);
        }
                return smprintf(UNKNOWN_STR);
        }
-
        fscanf(fp, "%*s %Lf %Lf %Lf %Lf", &b[0], &b[1], &b[2], &b[3]);
        fclose(fp);
        fscanf(fp, "%*s %Lf %Lf %Lf %Lf", &b[0], &b[1], &b[2], &b[3]);
        fclose(fp);
-       perc = 100 * ((b[0]+b[1]+b[2]) - (a[0]+a[1]+a[2])) / ((b[0]+b[1]+b[2]+b[3]) - (a[0]+a[1]+a[2]+a[3]));
 
 
+       perc = 100 * ((b[0]+b[1]+b[2]) - (a[0]+a[1]+a[2])) / ((b[0]+b[1]+b[2]+b[3]) - (a[0]+a[1]+a[2]+a[3]));
        return smprintf("%d%%", perc);
 }
 
 static char *
        return smprintf("%d%%", perc);
 }
 
 static char *
-datetime(const char *timeformat)
+datetime(const char *fmt)
 {
        time_t t;
 {
        time_t t;
-       char timestr[80];
+       char str[80];
 
        t = time(NULL);
 
        t = time(NULL);
-       if (strftime(timestr, sizeof(timestr), timeformat, localtime(&t)) == 0)
+       if (strftime(str, sizeof(str), fmt, localtime(&t)) == 0) {
                return smprintf(UNKNOWN_STR);
                return smprintf(UNKNOWN_STR);
+       }
 
 
-       return smprintf("%s", timestr);
+       return smprintf("%s", str);
 }
 
 static char *
 }
 
 static char *
-disk_free(const char *mountpoint)
+disk_free(const char *mnt)
 {
        struct statvfs fs;
 
 {
        struct statvfs fs;
 
-       if (statvfs(mountpoint, &fs) < 0) {
-               warn("Could not get filesystem info");
+       if (statvfs(mnt, &fs) < 0) {
+               warn("Failed to get filesystem info");
                return smprintf(UNKNOWN_STR);
        }
 
                return smprintf(UNKNOWN_STR);
        }
 
@@ -200,13 +207,13 @@ disk_free(const char *mountpoint)
 }
 
 static char *
 }
 
 static char *
-disk_perc(const char *mountpoint)
+disk_perc(const char *mnt)
 {
 {
-       int perc = 0;
+       int perc;
        struct statvfs fs;
 
        struct statvfs fs;
 
-       if (statvfs(mountpoint, &fs) < 0) {
-               warn("Could not get filesystem info");
+       if (statvfs(mnt, &fs) < 0) {
+               warn("Failed to get filesystem info");
                return smprintf(UNKNOWN_STR);
        }
 
                return smprintf(UNKNOWN_STR);
        }
 
@@ -216,12 +223,12 @@ disk_perc(const char *mountpoint)
 }
 
 static char *
 }
 
 static char *
-disk_total(const char *mountpoint)
+disk_total(const char *mnt)
 {
        struct statvfs fs;
 
 {
        struct statvfs fs;
 
-       if (statvfs(mountpoint, &fs) < 0) {
-               warn("Could not get filesystem info");
+       if (statvfs(mnt, &fs) < 0) {
+               warn("Failed to get filesystem info");
                return smprintf(UNKNOWN_STR);
        }
 
                return smprintf(UNKNOWN_STR);
        }
 
@@ -229,12 +236,12 @@ disk_total(const char *mountpoint)
 }
 
 static char *
 }
 
 static char *
-disk_used(const char *mountpoint)
+disk_used(const char *mnt)
 {
        struct statvfs fs;
 
 {
        struct statvfs fs;
 
-       if (statvfs(mountpoint, &fs) < 0) {
-               warn("Could not get filesystem info");
+       if (statvfs(mnt, &fs) < 0) {
+               warn("Failed to get filesystem info");
                return smprintf(UNKNOWN_STR);
        }
 
                return smprintf(UNKNOWN_STR);
        }
 
@@ -244,18 +251,18 @@ disk_used(const char *mountpoint)
 static char *
 entropy(void)
 {
 static char *
 entropy(void)
 {
-       int entropy = 0;
-       FILE *fp = fopen("/proc/sys/kernel/random/entropy_avail", "r");
+       int num;
+       FILE *fp;
 
 
+       fp= fopen("/proc/sys/kernel/random/entropy_avail", "r");
        if (fp == NULL) {
        if (fp == NULL) {
-               warn("Could not open entropy file");
+               warn("Failed to open file /proc/sys/kernel/random/entropy_avail");
                return smprintf(UNKNOWN_STR);
        }
                return smprintf(UNKNOWN_STR);
        }
-
-       fscanf(fp, "%d", &entropy);
+       fscanf(fp, "%d", &num);
        fclose(fp);
 
        fclose(fp);
 
-       return smprintf("%d", entropy);
+       return smprintf("%d", num);
 }
 
 static char *
 }
 
 static char *
@@ -267,43 +274,36 @@ gid(void)
 static char *
 hostname(void)
 {
 static char *
 hostname(void)
 {
-       char hostname[HOST_NAME_MAX];
-       FILE *fp = fopen("/proc/sys/kernel/hostname", "r");
+       char buf[HOST_NAME_MAX];
 
 
-       if (fp == NULL) {
-               warn("Could not open hostname file");
+       if (gethostname(buf, sizeof(buf)) == -1) {
+               warn("hostname");
                return smprintf(UNKNOWN_STR);
        }
 
                return smprintf(UNKNOWN_STR);
        }
 
-       fgets(hostname, sizeof(hostname), fp);
-       hostname[strlen(hostname)-1] = '\0';
-       fclose(fp);
-
-       return smprintf("%s", hostname);
+       return smprintf("%s", buf);
 }
 
 static char *
 }
 
 static char *
-ip(const char *interface)
+ip(const char *iface)
 {
        struct ifaddrs *ifaddr, *ifa;
        int s;
        char host[NI_MAXHOST];
 
        if (getifaddrs(&ifaddr) == -1) {
 {
        struct ifaddrs *ifaddr, *ifa;
        int s;
        char host[NI_MAXHOST];
 
        if (getifaddrs(&ifaddr) == -1) {
-               warn("Error getting IP address");
+               warn("Failed to get IP address for interface %s", iface);
                return smprintf(UNKNOWN_STR);
        }
 
        for (ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) {
                return smprintf(UNKNOWN_STR);
        }
 
        for (ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) {
-               if (ifa->ifa_addr == NULL)
+               if (ifa->ifa_addr == NULL) {
                        continue;
                        continue;
-
-               s = getnameinfo(ifa->ifa_addr, sizeof(struct sockaddr_in), host, NI_MAXHOST,
-                                                               NULL, 0, NI_NUMERICHOST);
-
-               if ((strcmp(ifa->ifa_name, interface) == 0) && (ifa->ifa_addr->sa_family == AF_INET)) {
+               }
+               s = getnameinfo(ifa->ifa_addr, sizeof(struct sockaddr_in), host, NI_MAXHOST, NULL, 0, NI_NUMERICHOST);
+               if ((strcmp(ifa->ifa_name, iface) == 0) && (ifa->ifa_addr->sa_family == AF_INET)) {
                        if (s != 0) {
                        if (s != 0) {
-                               warnx("Error getting IP address.");
+                               warnx("Failed to get IP address for interface %s", iface);
                                return smprintf(UNKNOWN_STR);
                        }
                        return smprintf("%s", host);
                                return smprintf(UNKNOWN_STR);
                        }
                        return smprintf("%s", host);
@@ -321,7 +321,7 @@ load_avg(void)
        double avgs[3];
 
        if (getloadavg(avgs, 3) < 0) {
        double avgs[3];
 
        if (getloadavg(avgs, 3) < 0) {
-               warnx("Error getting load avg.");
+               warnx("Failed to get the load avg");
                return smprintf(UNKNOWN_STR);
        }
 
                return smprintf(UNKNOWN_STR);
        }
 
@@ -332,13 +332,13 @@ static char *
 ram_free(void)
 {
        long free;
 ram_free(void)
 {
        long free;
-       FILE *fp = fopen("/proc/meminfo", "r");
+       FILE *fp;
 
 
+       fp = fopen("/proc/meminfo", "r");
        if (fp == NULL) {
        if (fp == NULL) {
-               warn("Error opening meminfo file");
+               warn("Failed to open file /proc/meminfo");
                return smprintf(UNKNOWN_STR);
        }
                return smprintf(UNKNOWN_STR);
        }
-
        fscanf(fp, "MemFree: %ld kB\n", &free);
        fclose(fp);
 
        fscanf(fp, "MemFree: %ld kB\n", &free);
        fclose(fp);
 
@@ -348,37 +348,34 @@ ram_free(void)
 static char *
 ram_perc(void)
 {
 static char *
 ram_perc(void)
 {
-       int perc;
        long total, free, buffers, cached;
        long total, free, buffers, cached;
-       FILE *fp = fopen("/proc/meminfo", "r");
+       FILE *fp;
 
 
+       fp = fopen("/proc/meminfo", "r");
        if (fp == NULL) {
        if (fp == NULL) {
-               warn("Error opening meminfo file");
+               warn("Failed to open file /proc/meminfo");
                return smprintf(UNKNOWN_STR);
        }
                return smprintf(UNKNOWN_STR);
        }
-
        fscanf(fp, "MemTotal: %ld kB\n", &total);
        fscanf(fp, "MemFree: %ld kB\n", &free);
        fscanf(fp, "MemAvailable: %ld kB\nBuffers: %ld kB\n", &buffers, &buffers);
        fscanf(fp, "Cached: %ld kB\n", &cached);
        fscanf(fp, "MemTotal: %ld kB\n", &total);
        fscanf(fp, "MemFree: %ld kB\n", &free);
        fscanf(fp, "MemAvailable: %ld kB\nBuffers: %ld kB\n", &buffers, &buffers);
        fscanf(fp, "Cached: %ld kB\n", &cached);
-
        fclose(fp);
        fclose(fp);
-       perc = 100 * ((total - free) - (buffers + cached)) / total;
 
 
-       return smprintf("%d%%", perc);
+       return smprintf("%d%%", 100 * ((total - free) - (buffers + cached)) / total);
 }
 
 static char *
 ram_total(void)
 {
        long total;
 }
 
 static char *
 ram_total(void)
 {
        long total;
-       FILE *fp = fopen("/proc/meminfo", "r");
+       FILE *fp;
 
 
+       fp = fopen("/proc/meminfo", "r");
        if (fp == NULL) {
        if (fp == NULL) {
-               warn("Error opening meminfo file");
+               warn("Failed to open file /proc/meminfo");
                return smprintf(UNKNOWN_STR);
        }
                return smprintf(UNKNOWN_STR);
        }
-
        fscanf(fp, "MemTotal: %ld kB\n", &total);
        fclose(fp);
 
        fscanf(fp, "MemTotal: %ld kB\n", &total);
        fclose(fp);
 
@@ -388,72 +385,214 @@ ram_total(void)
 static char *
 ram_used(void)
 {
 static char *
 ram_used(void)
 {
-       long free, total, buffers, cached, used;
-       FILE *fp = fopen("/proc/meminfo", "r");
+       long free, total, buffers, cached;
+       FILE *fp;
 
 
+       fp = fopen("/proc/meminfo", "r");
        if (fp == NULL) {
        if (fp == NULL) {
-               warn("Error opening meminfo file");
+               warn("Failed to open file /proc/meminfo");
                return smprintf(UNKNOWN_STR);
        }
                return smprintf(UNKNOWN_STR);
        }
-
        fscanf(fp, "MemTotal: %ld kB\n", &total);
        fscanf(fp, "MemFree: %ld kB\n", &free);
        fscanf(fp, "MemAvailable: %ld kB\nBuffers: %ld kB\n", &buffers, &buffers);
        fscanf(fp, "Cached: %ld kB\n", &cached);
        fscanf(fp, "MemTotal: %ld kB\n", &total);
        fscanf(fp, "MemFree: %ld kB\n", &free);
        fscanf(fp, "MemAvailable: %ld kB\nBuffers: %ld kB\n", &buffers, &buffers);
        fscanf(fp, "Cached: %ld kB\n", &cached);
+       fclose(fp);
+
+       return smprintf("%f", (float)(total - free - buffers - cached) / 1024 / 1024);
+}
+
+static char *
+run_command(const char *cmd)
+{
+       char *nlptr;
+       FILE *fp;
+       char buf[1024] = UNKNOWN_STR;
+
+       fp = popen(cmd, "r");
+       if (fp == NULL) {
+               warn("Failed to get command output for %s", cmd);
+               return smprintf(UNKNOWN_STR);
+       }
+       fgets(buf, sizeof(buf), fp);
+       pclose(fp);
+       buf[strlen(buf)] = '\0';
 
 
+       if ((nlptr = strstr(buf, "\n")) != NULL) {
+               nlptr[0] = '\0';
+       }
+
+       return smprintf("%s", buf);
+}
+
+static char *
+swap_free(void)
+{
+       long total, free;
+       FILE *fp;
+       char buf[2048];
+       size_t bytes_read;
+       char *match;
+
+       fp = fopen("/proc/meminfo", "r");
+       if (fp == NULL) {
+               warn("Failed to open file /proc/meminfo");
+               return smprintf(UNKNOWN_STR);
+       }
+       bytes_read = fread(buf, sizeof(char), sizeof(buf), fp);
+       buf[bytes_read] = '\0';
        fclose(fp);
        fclose(fp);
-       used = total - free - buffers - cached;
+       if (bytes_read == 0 || bytes_read == sizeof(buf)) {
+               warn("Failed to read from /proc/meminfo");
+               return smprintf(UNKNOWN_STR);
+       }
 
 
-       return smprintf("%f", (float)used / 1024 / 1024);
+       match = strstr(buf, "SwapTotal");
+       sscanf(match, "SwapTotal: %ld kB\n", &total);
+       if (total == 0) {
+               return smprintf(UNKNOWN_STR);
+       }
+
+       match = strstr(buf, "SwapFree");
+       sscanf(match, "SwapFree: %ld kB\n", &free);
+
+       return smprintf("%f", (float)free / 1024 / 1024);
 }
 
 static char *
 }
 
 static char *
-run_command(const char* command)
+swap_perc(void)
 {
 {
-       FILE *fp = popen(command, "r");
-       char buffer[64] = "\0";
+       long total, free, cached;
+       FILE *fp;
+       char buf[2048];
+       size_t bytes_read;
+       char *match;
 
 
+       fp = fopen("/proc/meminfo", "r");
        if (fp == NULL) {
        if (fp == NULL) {
-               warn("Could not get command output for: %s", command);
+               warn("Failed to open file /proc/meminfo");
+               return smprintf(UNKNOWN_STR);
+       }
+       bytes_read = fread(buf, sizeof(char), sizeof(buf), fp);
+       buf[bytes_read] = '\0';
+       fclose(fp);
+       if (bytes_read == 0 || bytes_read == sizeof(buf)) {
+               warn("Failed to read from /proc/meminfo");
                return smprintf(UNKNOWN_STR);
        }
 
                return smprintf(UNKNOWN_STR);
        }
 
-       fgets(buffer, sizeof(buffer), fp);
-       buffer[sizeof(buffer)-1] = '\0';
+       match = strstr(buf, "SwapTotal");
+       sscanf(match, "SwapTotal: %ld kB\n", &total);
+       if (total == 0) {
+               return smprintf(UNKNOWN_STR);
+       }
 
 
-       pclose(fp);
-       return smprintf("%s", buffer);
+       match = strstr(buf, "SwapCached");
+       sscanf(match, "SwapCached: %ld kB\n", &cached);
+
+       match = strstr(buf, "SwapFree");
+       sscanf(match, "SwapFree: %ld kB\n", &free);
+
+       return smprintf("%d%%", 100 * (total - free - cached) / total);
 }
 
 static char *
 }
 
 static char *
-temp(const char *file)
+swap_total(void)
 {
 {
-       int temperature;
-       FILE *fp = fopen(file, "r");
+       long total;
+       FILE *fp;
+       char buf[2048];
+       size_t bytes_read;
+       char *match;
 
 
+       fp = fopen("/proc/meminfo", "r");
        if (fp == NULL) {
        if (fp == NULL) {
-               warn("Could not open temperature file");
+               warn("Failed to open file /proc/meminfo");
+               return smprintf(UNKNOWN_STR);
+       }
+       bytes_read = fread(buf, sizeof(char), sizeof(buf), fp);
+       buf[bytes_read] = '\0';
+       fclose(fp);
+       if (bytes_read == 0 || bytes_read == sizeof(buf)) {
+               warn("Failed to read from /proc/meminfo");
+               return smprintf(UNKNOWN_STR);
+       }
+
+       match = strstr(buf, "SwapTotal");
+       sscanf(match, "SwapTotal: %ld kB\n", &total);
+       if (total == 0) {
                return smprintf(UNKNOWN_STR);
        }
 
                return smprintf(UNKNOWN_STR);
        }
 
-       fscanf(fp, "%d", &temperature);
+       return smprintf("%f", (float)total / 1024 / 1024);
+}
+
+static char *
+swap_used(void)
+{
+       long total, free, cached;
+       FILE *fp;
+       char buf[2048];
+       size_t bytes_read;
+       char *match;
+
+       fp = fopen("/proc/meminfo", "r");
+       if (fp == NULL) {
+               warn("Failed to open file /proc/meminfo");
+               return smprintf(UNKNOWN_STR);
+       }
+       bytes_read = fread(buf, sizeof(char), sizeof(buf), fp);
+       buf[bytes_read] = '\0';
        fclose(fp);
        fclose(fp);
+       if (bytes_read == 0 || bytes_read == sizeof(buf)) {
+               warn("Failed to read from /proc/meminfo");
+               return smprintf(UNKNOWN_STR);
+       }
+
+       match = strstr(buf, "SwapTotal");
+       sscanf(match, "SwapTotal: %ld kB\n", &total);
+       if (total == 0) {
+               return smprintf(UNKNOWN_STR);
+       }
+
+       match = strstr(buf, "SwapCached");
+       sscanf(match, "SwapCached: %ld kB\n", &cached);
+
+       match = strstr(buf, "SwapFree");
+       sscanf(match, "SwapFree: %ld kB\n", &free);
 
 
-       return smprintf("%d°C", temperature / 1000);
+       return smprintf("%f", (float)(total - free - cached) / 1024 / 1024);
+}
+
+static char *
+temp(const char *file)
+{
+       int temp;
+       FILE *fp;
+
+       fp = fopen(file, "r");
+       if (fp == NULL) {
+               warn("Failed to open file %s", file);
+               return smprintf(UNKNOWN_STR);
+       }
+       fscanf(fp, "%d", &temp);
+       fclose(fp);
+
+       return smprintf("%d°C", temp / 1000);
 }
 
 static char *
 uptime(void)
 {
        struct sysinfo info;
 }
 
 static char *
 uptime(void)
 {
        struct sysinfo info;
-       int hours = 0;
-       int minutes = 0;
+       int h = 0;
+       int m = 0;
 
        sysinfo(&info);
 
        sysinfo(&info);
-       hours = info.uptime / 3600;
-       minutes = (info.uptime - hours * 3600 ) / 60;
+       h = info.uptime / 3600;
+       m = (info.uptime - h * 3600 ) / 60;
 
 
-       return smprintf("%dh %dm", hours, minutes);
+       return smprintf("%dh %dm", h, m);
 }
 
 static char *
 }
 
 static char *
@@ -463,7 +602,7 @@ username(void)
        struct passwd *pw = getpwuid(uid);
 
        if (pw == NULL) {
        struct passwd *pw = getpwuid(uid);
 
        if (pw == NULL) {
-               warn("Could not get username");
+               warn("Failed to get username");
                return smprintf(UNKNOWN_STR);
        }
 
                return smprintf(UNKNOWN_STR);
        }
 
@@ -478,15 +617,16 @@ uid(void)
 
 
 static char *
 
 
 static char *
-vol_perc(const char *soundcard)
+vol_perc(const char *card)
 {
 {
+       int mute;
        long int vol, max, min;
        snd_mixer_t *handle;
        snd_mixer_elem_t *elem;
        snd_mixer_selem_id_t *s_elem;
 
        snd_mixer_open(&handle, 0);
        long int vol, max, min;
        snd_mixer_t *handle;
        snd_mixer_elem_t *elem;
        snd_mixer_selem_id_t *s_elem;
 
        snd_mixer_open(&handle, 0);
-       snd_mixer_attach(handle, soundcard);
+       snd_mixer_attach(handle, card);
        snd_mixer_selem_register(handle, NULL, NULL);
        snd_mixer_load(handle);
        snd_mixer_selem_id_malloc(&s_elem);
        snd_mixer_selem_register(handle, NULL, NULL);
        snd_mixer_load(handle);
        snd_mixer_selem_id_malloc(&s_elem);
@@ -496,67 +636,69 @@ vol_perc(const char *soundcard)
        if (elem == NULL) {
                snd_mixer_selem_id_free(s_elem);
                snd_mixer_close(handle);
        if (elem == NULL) {
                snd_mixer_selem_id_free(s_elem);
                snd_mixer_close(handle);
-               warn("Failed to get volume percentage for: %s", soundcard);
+               warn("Failed to get volume percentage for %s", card);
                return smprintf(UNKNOWN_STR);
        }
 
        snd_mixer_handle_events(handle);
        snd_mixer_selem_get_playback_volume_range(elem, &min, &max);
        snd_mixer_selem_get_playback_volume(elem, 0, &vol);
                return smprintf(UNKNOWN_STR);
        }
 
        snd_mixer_handle_events(handle);
        snd_mixer_selem_get_playback_volume_range(elem, &min, &max);
        snd_mixer_selem_get_playback_volume(elem, 0, &vol);
+       snd_mixer_selem_get_playback_switch(elem, 0, &mute);
 
        snd_mixer_selem_id_free(s_elem);
        snd_mixer_close(handle);
 
 
        snd_mixer_selem_id_free(s_elem);
        snd_mixer_close(handle);
 
-       return smprintf("%d%%", ((uint_fast16_t)(vol * 100) / max));
+       if (!mute)
+               return smprintf("mute");
+       else if (max == 0)
+               return smprintf("0%%");
+       else
+               return smprintf("%lu%%", ((uint_fast16_t)(vol * 100) / max));
 }
 
 static char *
 }
 
 static char *
-wifi_perc(const char *wificard)
+wifi_perc(const char *iface)
 {
 {
-       int strength;
+       int perc;
        char buf[255];
        char *datastart;
        char status[5];
        FILE *fp;
 
        char buf[255];
        char *datastart;
        char status[5];
        FILE *fp;
 
-       ccat(3, "/sys/class/net/", wificard, "/operstate");
-
+       ccat(3, "/sys/class/net/", iface, "/operstate");
        fp = fopen(concat, "r");
        fp = fopen(concat, "r");
-
        if (fp == NULL) {
        if (fp == NULL) {
-               warn("Error opening wifi operstate file");
+               warn("Failed to open file %s", concat);
                return smprintf(UNKNOWN_STR);
        }
                return smprintf(UNKNOWN_STR);
        }
-
        fgets(status, 5, fp);
        fclose(fp);
        fgets(status, 5, fp);
        fclose(fp);
-       if(strcmp(status, "up\n") != 0)
+       if(strcmp(status, "up\n") != 0) {
                return smprintf(UNKNOWN_STR);
                return smprintf(UNKNOWN_STR);
+       }
 
        fp = fopen("/proc/net/wireless", "r");
        if (fp == NULL) {
 
        fp = fopen("/proc/net/wireless", "r");
        if (fp == NULL) {
-               warn("Error opening wireless file");
+               warn("Failed to open file /proc/net/wireless");
                return smprintf(UNKNOWN_STR);
        }
                return smprintf(UNKNOWN_STR);
        }
-
-       ccat(2, wificard, ":");
+       ccat(2, iface, ":");
        fgets(buf, sizeof(buf), fp);
        fgets(buf, sizeof(buf), fp);
        fgets(buf, sizeof(buf), fp);
        fgets(buf, sizeof(buf), fp);
        fgets(buf, sizeof(buf), fp);
        fgets(buf, sizeof(buf), fp);
+       fclose(fp);
 
        datastart = strstr(buf, concat);
        if (datastart != NULL) {
                datastart = strstr(buf, ":");
 
        datastart = strstr(buf, concat);
        if (datastart != NULL) {
                datastart = strstr(buf, ":");
-               sscanf(datastart + 1, " %*d   %d  %*d  %*d                %*d      %*d          %*d              %*d      %*d            %*d", &strength);
+               sscanf(datastart + 1, " %*d   %d  %*d  %*d                %*d      %*d          %*d              %*d      %*d            %*d", &perc);
        }
 
        }
 
-       fclose(fp);
-
-       return smprintf("%d%%", strength);
+       return smprintf("%d%%", perc);
 }
 
 static char *
 }
 
 static char *
-wifi_essid(const char *wificard)
+wifi_essid(const char *iface)
 {
        char id[IW_ESSID_MAX_SIZE+1];
        int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
 {
        char id[IW_ESSID_MAX_SIZE+1];
        int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
@@ -564,14 +706,14 @@ wifi_essid(const char *wificard)
 
        memset(&wreq, 0, sizeof(struct iwreq));
        wreq.u.essid.length = IW_ESSID_MAX_SIZE+1;
 
        memset(&wreq, 0, sizeof(struct iwreq));
        wreq.u.essid.length = IW_ESSID_MAX_SIZE+1;
-       sprintf(wreq.ifr_name, wificard);
+       sprintf(wreq.ifr_name, iface);
        if (sockfd == -1) {
        if (sockfd == -1) {
-               warn("Cannot open socket for interface: %s", wificard);
+               warn("Failed to get ESSID for interface %s", iface);
                return smprintf(UNKNOWN_STR);
        }
        wreq.u.essid.pointer = id;
        if (ioctl(sockfd,SIOCGIWESSID, &wreq) == -1) {
                return smprintf(UNKNOWN_STR);
        }
        wreq.u.essid.pointer = id;
        if (ioctl(sockfd,SIOCGIWESSID, &wreq) == -1) {
-               warn("Get ESSID ioctl failed for interface %s", wificard);
+               warn("Failed to get ESSID for interface %s", iface);
                return smprintf(UNKNOWN_STR);
        }
 
                return smprintf(UNKNOWN_STR);
        }
 
@@ -583,6 +725,16 @@ wifi_essid(const char *wificard)
                return smprintf("%s", (char *)wreq.u.essid.pointer);
 }
 
                return smprintf("%s", (char *)wreq.u.essid.pointer);
 }
 
+static char *
+kernel_release(void)
+{
+       struct utsname udata;
+       if (uname(&udata) < 0)
+               return smprintf(UNKNOWN_STR);
+
+       return smprintf("%s", udata.release);
+}
+
 static void
 set_status(const char *str)
 {
 static void
 set_status(const char *str)
 {
@@ -593,27 +745,23 @@ set_status(const char *str)
 static void
 sighandler(const int signo)
 {
 static void
 sighandler(const int signo)
 {
-       if (signo == SIGTERM || signo == SIGINT)
+       if (signo == SIGTERM || signo == SIGINT) {
                done = 1;
                done = 1;
+       }
 }
 
 static void
 usage(void)
 {
 }
 
 static void
 usage(void)
 {
-       fprintf(stderr,
-               "slstatus (c) 2016, drkhsh\n"
-               "usage: %s [-dho]\n",
-               argv0);
+       fprintf(stderr, "usage: %s [-dhov]\n", argv0);
        exit(1);
 }
 
        exit(1);
 }
 
-
-
 int
 main(int argc, char *argv[])
 {
        unsigned short int i;
 int
 main(int argc, char *argv[])
 {
        unsigned short int i;
-       char status_string[4096];
+       char status_string[2048];
        char *res, *element;
        struct arg argument;
        struct sigaction act;
        char *res, *element;
        struct arg argument;
        struct sigaction act;
@@ -625,58 +773,75 @@ main(int argc, char *argv[])
                case 'o':
                        oflag = 1;
                        break;
                case 'o':
                        oflag = 1;
                        break;
+               case 'v':
+                       printf("slstatus %s (C) 2016 slstatus engineers\n", VERSION);
+                       return 0;
                default:
                        usage();
        } ARGEND
 
                default:
                        usage();
        } ARGEND
 
-       if (dflag && oflag)
+       if (dflag && oflag) {
                usage();
                usage();
-       if (dflag)
-               (void)daemon(1, 1);
+       }
+       if (dflag && daemon(1, 1) < 0) {
+               err(1, "daemon");
+       }
 
        memset(&act, 0, sizeof(act));
        act.sa_handler = sighandler;
        sigaction(SIGINT,  &act, 0);
        sigaction(SIGTERM, &act, 0);
 
 
        memset(&act, 0, sizeof(act));
        act.sa_handler = sighandler;
        sigaction(SIGINT,  &act, 0);
        sigaction(SIGTERM, &act, 0);
 
-       dpy = XOpenDisplay(NULL);
+       if (!oflag) {
+               dpy = XOpenDisplay(NULL);
+       }
+
+       setlocale(LC_ALL, "");
 
        while (!done) {
                status_string[0] = '\0';
 
                for (i = 0; i < sizeof(args) / sizeof(args[0]); ++i) {
                        argument = args[i];
 
        while (!done) {
                status_string[0] = '\0';
 
                for (i = 0; i < sizeof(args) / sizeof(args[0]); ++i) {
                        argument = args[i];
-                       if (argument.args == NULL)
+                       if (argument.args == NULL) {
                                res = argument.func();
                                res = argument.func();
-                       else
+                       } else {
                                res = argument.func(argument.args);
                                res = argument.func(argument.args);
-                       element = smprintf(argument.format, res);
+                       }
+                       element = smprintf(argument.fmt, res);
                        if (element == NULL) {
                                element = smprintf(UNKNOWN_STR);
                        if (element == NULL) {
                                element = smprintf(UNKNOWN_STR);
-                               warnx("Failed to format output.");
+                               warnx("Failed to format output");
                        }
                        }
-                       strlcat(status_string, element, sizeof(status_string));
+                       if (strlcat(status_string, element, sizeof(status_string)) >= sizeof(status_string))
+                               warnx("Output too long");
                        free(res);
                        free(element);
                }
 
                        free(res);
                        free(element);
                }
 
-               if (!oflag)
+               if (!oflag) {
                        set_status(status_string);
                        set_status(status_string);
-               else
+               } else {
                        printf("%s\n", status_string);
                        printf("%s\n", status_string);
+               }
 
                /*
                 * subtract delay time spend in function
                 * calls from the actual global delay time
                 */
 
                /*
                 * subtract delay time spend in function
                 * calls from the actual global delay time
                 */
-               sleep(UPDATE_INTERVAL - delay);
-               delay = 0;
+               if ((UPDATE_INTERVAL - delay) <= 0) {
+                       delay = 0;
+                       continue;
+               } else {
+                       sleep(UPDATE_INTERVAL - delay);
+                       delay = 0;
+               }
        }
 
        }
 
-       if (!oflag)
+       if (!oflag) {
                set_status(NULL);
                set_status(NULL);
-
-       XCloseDisplay(dpy);
+               XCloseDisplay(dpy);
+       }
 
        return 0;
 }
 
        return 0;
 }