Xinqi Bao's Git

Add ram and swap components on FreeBSD
[slstatus.git] / components / ram.c
index b200882..47e6fda 100644 (file)
 #include "../util.h"
 
 #if defined(__linux__)
-const char *
-ram_free(void)
-{
-       long free;
-
-       return (pscanf("/proc/meminfo", "MemFree: %ld kB\n", &free) == 1) ?
-              bprintf("%f", (float)free / 1024 / 1024) : NULL;
-}
-
-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;
-}
-
-const char *
-ram_total(void)
-{
-       long total;
-
-       return (pscanf("/proc/meminfo", "MemTotal: %ld kB\n", &total) == 1) ?
-              bprintf("%f", (float)total / 1024 / 1024) : NULL;
-}
-
-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;
-}
+       #include <stdint.h>
+
+       const char *
+       ram_free(void)
+       {
+               uintmax_t free;
+
+               if (pscanf("/proc/meminfo",
+                          "MemTotal: %ju kB\n"
+                          "MemFree: %ju kB\n"
+                          "MemAvailable: %ju kB\n",
+                          &free, &free, &free) != 3) {
+                       return NULL;
+               }
+
+               return fmt_human(free * 1024, 1024);
+       }
+
+       const char *
+       ram_perc(void)
+       {
+               uintmax_t total, free, buffers, cached;
+
+               if (pscanf("/proc/meminfo",
+                          "MemTotal: %ju kB\n"
+                          "MemFree: %ju kB\n"
+                          "MemAvailable: %ju kB\n"
+                          "Buffers: %ju kB\n"
+                          "Cached: %ju 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)
+       {
+               uintmax_t total;
+
+               if (pscanf("/proc/meminfo", "MemTotal: %ju kB\n", &total)
+                   != 1) {
+                       return NULL;
+               }
+
+               return fmt_human(total * 1024, 1024);
+       }
+
+       const char *
+       ram_used(void)
+       {
+               uintmax_t total, free, buffers, cached;
+
+               if (pscanf("/proc/meminfo",
+                          "MemTotal: %ju kB\n"
+                          "MemFree: %ju kB\n"
+                          "MemAvailable: %ju kB\n"
+                          "Buffers: %ju kB\n"
+                          "Cached: %ju kB\n",
+                          &total, &free, &buffers, &buffers, &cached) != 5) {
+                       return NULL;
+               }
+
+               return fmt_human((total - free - buffers - cached) * 1024,
+                                1024);
+       }
 #elif defined(__OpenBSD__)
-#include <sys/types.h>
-#include <sys/sysctl.h>
-#include <stdlib.h>
-#include <unistd.h>
-
-inline int
-load_uvmexp(struct uvmexp *uvmexp)
-{
-       int uvmexp_mib[] = {CTL_VM, VM_UVMEXP};
-       size_t size;
-
-       size = sizeof(*uvmexp);
-
-       return sysctl(uvmexp_mib, 2, uvmexp, &size, NULL, 0) >= 0 ? 1 : 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 = (double) (free_pages * uvmexp.pagesize) / 1024 / 1024 / 1024;
-               return bprintf("%f", free);
+       #include <stdlib.h>
+       #include <sys/sysctl.h>
+       #include <sys/types.h>
+       #include <unistd.h>
+
+       #define LOG1024 10
+       #define pagetok(size, pageshift) (size_t)(size << (pageshift - LOG1024))
+
+       inline int
+       load_uvmexp(struct uvmexp *uvmexp)
+       {
+               int uvmexp_mib[] = {CTL_VM, VM_UVMEXP};
+               size_t size;
+
+               size = sizeof(*uvmexp);
+
+               if (sysctl(uvmexp_mib, 2, uvmexp, &size, NULL, 0) >= 0) {
+                       return 1;
+               }
+
+               return 0;
        }
 
-       return NULL;
-}
+       const char *
+       ram_free(void)
+       {
+               struct uvmexp uvmexp;
+               int free_pages;
 
-const char *
-ram_perc(void)
-{
-       struct uvmexp uvmexp;
-       int percent;
+               if (load_uvmexp(&uvmexp)) {
+                       free_pages = uvmexp.npages - uvmexp.active;
+                       return fmt_human(pagetok(free_pages, uvmexp.pageshift) *
+                                        1024, 1024);
+               }
 
-       if (load_uvmexp(&uvmexp)) {
-               percent = uvmexp.active * 100 / uvmexp.npages;
-               return bprintf("%d", percent);
+               return NULL;
        }
 
-       return NULL;
-}
+       const char *
+       ram_perc(void)
+       {
+               struct uvmexp uvmexp;
+               int percent;
 
-const char *
-ram_total(void)
-{
-       struct uvmexp uvmexp;
-       float total;
+               if (load_uvmexp(&uvmexp)) {
+                       percent = uvmexp.active * 100 / uvmexp.npages;
+                       return bprintf("%d", percent);
+               }
 
-       if (load_uvmexp(&uvmexp)) {
-               total = (double) (uvmexp.npages * uvmexp.pagesize) / 1024 / 1024 / 1024;
-               return bprintf("%f", total);
+               return NULL;
        }
 
-       return NULL;
-}
+       const char *
+       ram_total(void)
+       {
+               struct uvmexp uvmexp;
 
-const char *
-ram_used(void)
-{
-       struct uvmexp uvmexp;
-       float used;
+               if (load_uvmexp(&uvmexp)) {
+                       return fmt_human(pagetok(uvmexp.npages,
+                                                uvmexp.pageshift) * 1024,
+                                        1024);
+               }
 
-       if (load_uvmexp(&uvmexp)) {
-               used = (double) (uvmexp.active * uvmexp.pagesize) / 1024 / 1024 / 1024;
-               return bprintf("%f", used);
+               return NULL;
        }
 
-       return NULL;
-}
+       const char *
+       ram_used(void)
+       {
+               struct uvmexp uvmexp;
+
+               if (load_uvmexp(&uvmexp)) {
+                       return fmt_human(pagetok(uvmexp.active,
+                                                uvmexp.pageshift) * 1024,
+                                        1024);
+               }
+
+               return NULL;
+       }
+#elif defined(__FreeBSD__)
+       #include <sys/sysctl.h>
+       #include <sys/vmmeter.h>
+       #include <unistd.h>
+       #include <vm/vm_param.h>
+
+       const char *
+       ram_free(void) {
+               struct vmtotal vm_stats;
+               int mib[] = {CTL_VM, VM_TOTAL};
+               size_t len;
+
+               len = sizeof(struct vmtotal);
+               if (sysctl(mib, 2, &vm_stats, &len, NULL, 0) == -1
+                               || !len)
+                       return NULL;
+
+               return fmt_human(vm_stats.t_free * getpagesize(), 1024);
+       }
+
+       const char *
+       ram_total(void) {
+               long npages;
+               size_t len;
+
+               len = sizeof(npages);
+               if (sysctlbyname("vm.stats.vm.v_page_count", &npages, &len, NULL, 0) == -1
+                               || !len)
+                       return NULL;
+
+               return fmt_human(npages * getpagesize(), 1024);
+       }
+
+       const char *
+       ram_perc(void) {
+               long npages;
+               long active;
+               size_t len;
+
+               len = sizeof(npages);
+               if (sysctlbyname("vm.stats.vm.v_page_count", &npages, &len, NULL, 0) == -1
+                               || !len)
+                       return NULL;
+
+               if (sysctlbyname("vm.stats.vm.v_active_count", &active, &len, NULL, 0) == -1
+                               || !len)
+                       return NULL;
+
+               return bprintf("%d", active * 100 / npages);
+       }
+
+       const char *
+       ram_used(void) {
+               long active;
+               size_t len;
+
+               len = sizeof(active);
+               if (sysctlbyname("vm.stats.vm.v_active_count", &active, &len, NULL, 0) == -1
+                               || !len)
+                       return NULL;
+
+               return fmt_human(active * getpagesize(), 1024);
+       }
 #endif