Xinqi Bao's Git

cpu_freq: Change to 64 bit integers
[slstatus.git] / components / ram.c
index 57081d2..a2fff95 100644 (file)
@@ -4,54 +4,75 @@
 #include "../util.h"
 
 #if defined(__linux__)
+       #include <inttypes.h>
+
        const char *
        ram_free(void)
        {
-               long free;
+               uintmax_t free;
+
+               if (pscanf("/proc/meminfo",
+                          "MemTotal: %" PRIuMAX " kB\n"
+                          "MemFree: %" PRIuMAX " kB\n"
+                          "MemAvailable: %" PRIuMAX " kB\n",
+                          &free, &free, &free) != 3) {
+                       return NULL;
+               }
 
-               return (pscanf("/proc/meminfo", "MemFree: %ld kB\n", &free) == 1) ?
-                      bprintf("%f", (float)free / 1024 / 1024) : NULL;
+               return fmt_human(free * 1024, 1024);
        }
 
        const char *
        ram_perc(void)
        {
-               long total, free, buffers, cached;
-
-               return (pscanf("/proc/meminfo",
-                              "MemTotal: %ld kB\n"
-                              "MemFree: %ld kB\n"
-                              "MemAvailable: %ld kB\nBuffers: %ld kB\n"
-                              "Cached: %ld kB\n",
-                              &total, &free, &buffers, &buffers, &cached) == 5) ?
-                      bprintf("%d", 100 * ((total - free) - (buffers + cached)) /
-                              total) :
-                      NULL;
+               uintmax_t total, free, buffers, cached;
+
+               if (pscanf("/proc/meminfo",
+                          "MemTotal: %" PRIuMAX " kB\n"
+                          "MemFree: %" PRIuMAX " kB\n"
+                          "MemAvailable: %" PRIuMAX " kB\nBuffers: %ld kB\n"
+                          "Cached: %" PRIuMAX " kB\n",
+                          &total, &free, &buffers, &buffers, &cached) != 5) {
+                       return NULL;
+               }
+
+               if (total == 0) {
+                       return NULL;
+               }
+
+               return bprintf("%d", 100 * ((total - free) -
+                                           (buffers + cached)) / total);
        }
 
        const char *
        ram_total(void)
        {
-               long total;
+               uintmax_t total;
 
-               return (pscanf("/proc/meminfo", "MemTotal: %ld kB\n", &total) == 1) ?
-                      bprintf("%f", (float)total / 1024 / 1024) : NULL;
+               if (pscanf("/proc/meminfo", "MemTotal: %" PRIuMAX " kB\n",
+                          &total) != 1) {
+                       return NULL;
+               }
+
+               return fmt_human(total * 1024, 1024);
        }
 
        const char *
        ram_used(void)
        {
-               long total, free, buffers, cached;
-
-               return (pscanf("/proc/meminfo",
-                              "MemTotal: %ld kB\n"
-                              "MemFree: %ld kB\n"
-                              "MemAvailable: %ld kB\nBuffers: %ld kB\n"
-                              "Cached: %ld kB\n",
-                              &total, &free, &buffers, &buffers, &cached) == 5) ?
-                      bprintf("%f", (float)(total - free - buffers - cached) /
-                              1024 / 1024) :
-                      NULL;
+               uintmax_t total, free, buffers, cached;
+
+               if (pscanf("/proc/meminfo",
+                          "MemTotal: %" PRIuMAX " kB\n"
+                          "MemFree: %" PRIuMAX " kB\n"
+                          "MemAvailable: %" PRIuMAX " kB\nBuffers: %" PRIuMAX " kB\n"
+                          "Cached: %" PRIuMAX " kB\n",
+                          &total, &free, &buffers, &buffers, &cached) != 5) {
+                       return NULL;
+               }
+
+               return fmt_human((total - free - buffers - cached) * 1024,
+                                1024);
        }
 #elif defined(__OpenBSD__)
        #include <stdlib.h>
@@ -60,7 +81,7 @@
        #include <unistd.h>
 
        #define LOG1024         10
-       #define pagetok(size, pageshift) ((size) << (pageshift - LOG1024))
+       #define pagetok(size, pageshift) (size_t)(size << (pageshift - LOG1024))
 
        inline int
        load_uvmexp(struct uvmexp *uvmexp)
 
                size = sizeof(*uvmexp);
 
-               return sysctl(uvmexp_mib, 2, uvmexp, &size, NULL, 0) >= 0 ? 1 : 0;
+               if (sysctl(uvmexp_mib, 2, uvmexp, &size, NULL, 0) >= 0) {
+                       return 1;
+               }
+
+               return 0;
        }
 
        const char *
        ram_free(void)
        {
                struct uvmexp uvmexp;
-               float free;
                int free_pages;
 
                if (load_uvmexp(&uvmexp)) {
                        free_pages = uvmexp.npages - uvmexp.active;
-                       free = (float)(pagetok(free_pages, uvmexp.pageshift)) / 1024 / 1024;
-                       return bprintf("%f", free);
+                       return fmt_human(pagetok(free_pages, uvmexp.pageshift) *
+                                        1024, 1024);
                }
 
                return NULL;
        ram_total(void)
        {
                struct uvmexp uvmexp;
-               float total;
 
                if (load_uvmexp(&uvmexp)) {
-                       total = (float)(pagetok(uvmexp.npages, uvmexp.pageshift)) / 1024 / 1024;
-                       return bprintf("%f", total);
+                       return fmt_human(pagetok(uvmexp.npages,
+                                                uvmexp.pageshift) * 1024,
+                                        1024);
                }
 
                return NULL;
        ram_used(void)
        {
                struct uvmexp uvmexp;
-               float used;
 
                if (load_uvmexp(&uvmexp)) {
-                       used = (float)(pagetok(uvmexp.active, uvmexp.pageshift)) / 1024 / 1024;
-                       return bprintf("%f", used);
+                       return fmt_human(pagetok(uvmexp.active,
+                                                uvmexp.pageshift) * 1024,
+                                        1024);
                }
 
                return NULL;