X-Git-Url: https://git.xinqibao.xyz/slstatus.git/blobdiff_plain/c1f79d741a160ea17a110ca09b3deb705636ec4a..a1ac203d16c379d7fc05765545af2977a7a3584d:/components/swap.c diff --git a/components/swap.c b/components/swap.c index caa4788..2509db1 100644 --- a/components/swap.c +++ b/components/swap.c @@ -1,80 +1,82 @@ /* See LICENSE file for copyright and license details. */ -#include +#include #include +#include #include #include "../util.h" #if defined(__linux__) - static size_t - pread(const char *path, char *buf, size_t bufsiz) + static int + get_swap_info(long *s_total, long *s_free, long *s_cached) { FILE *fp; - size_t bytes_read; - - if (!(fp = fopen(path, "r"))) { - fprintf(stderr, "fopen '%s': %s\n", path, strerror(errno)); - return 0; + struct { + const char *name; + const size_t len; + long *var; + } ent[] = { + { "SwapTotal", sizeof("SwapTotal") - 1, s_total }, + { "SwapFree", sizeof("SwapFree") - 1, s_free }, + { "SwapCached", sizeof("SwapCached") - 1, s_cached }, + }; + size_t line_len = 0, i, left; + char *line = NULL; + + /* get number of fields we want to extract */ + for (i = 0, left = 0; i < LEN(ent); i++) { + if (ent[i].var) { + left++; + } } - if (!(bytes_read = fread(buf, sizeof(char), bufsiz, fp))) { - fprintf(stderr, "fread '%s': %s\n", path, strerror(errno)); - fclose(fp); - return 0; + + if (!(fp = fopen("/proc/meminfo", "r"))) { + warn("fopen '/proc/meminfo':"); + return 1; } - fclose(fp); - buf[bytes_read] = '\0'; + /* read file line by line and extract field information */ + while (left > 0 && getline(&line, &line_len, fp) >= 0) { + for (i = 0; i < LEN(ent); i++) { + if (ent[i].var && + !strncmp(line, ent[i].name, ent[i].len)) { + sscanf(line + ent[i].len + 1, + "%ld kB\n", ent[i].var); + left--; + break; + } + } + } + free(line); + if (ferror(fp)) { + warn("getline '/proc/meminfo':"); + return 1; + } - return bytes_read; + fclose(fp); + return 0; } const char * swap_free(void) { - long total, free; - char *match; - - if (!pread("/proc/meminfo", buf, sizeof(buf) - 1)) { - return NULL; - } + long free; - if (!(match = strstr(buf, "SwapTotal"))) { + if (get_swap_info(NULL, &free, NULL)) { return NULL; } - sscanf(match, "SwapTotal: %ld kB\n", &total); - if (!(match = strstr(buf, "SwapFree"))) { - return NULL; - } - sscanf(match, "SwapFree: %ld kB\n", &free); - - return bprintf("%f", (float)free / 1024 / 1024); + return fmt_human(free * 1024, 1024); } const char * swap_perc(void) { long total, free, cached; - char *match; - - if (!pread("/proc/meminfo", buf, sizeof(buf) - 1)) { - return NULL; - } - - if (!(match = strstr(buf, "SwapTotal"))) { - return NULL; - } - sscanf(match, "SwapTotal: %ld kB\n", &total); - - if (!(match = strstr(buf, "SwapCached"))) { - return NULL; - } - sscanf(match, "SwapCached: %ld kB\n", &cached); - if (!(match = strstr(buf, "SwapFree"))) { + if (get_swap_info(&total, &free, &cached) || total == 0) { return NULL; } - sscanf(match, "SwapFree: %ld kB\n", &free); return bprintf("%d", 100 * (total - free - cached) / total); } @@ -83,77 +85,52 @@ swap_total(void) { long total; - char *match; - - if (!pread("/proc/meminfo", buf, sizeof(buf) - 1)) { - return NULL; - } - if (!(match = strstr(buf, "SwapTotal"))) { + if (get_swap_info(&total, NULL, NULL)) { return NULL; } - sscanf(match, "SwapTotal: %ld kB\n", &total); - return bprintf("%f", (float)total / 1024 / 1024); + return fmt_human(total * 1024, 1024); } const char * swap_used(void) { long total, free, cached; - char *match; - - if (!pread("/proc/meminfo", buf, sizeof(buf) - 1)) { - return NULL; - } - - if (!(match = strstr(buf, "SwapTotal"))) { - return NULL; - } - sscanf(match, "SwapTotal: %ld kB\n", &total); - if (!(match = strstr(buf, "SwapCached"))) { + if (get_swap_info(&total, &free, &cached)) { return NULL; } - sscanf(match, "SwapCached: %ld kB\n", &cached); - if (!(match = strstr(buf, "SwapFree"))) { - return NULL; - } - sscanf(match, "SwapFree: %ld kB\n", &free); - - return bprintf("%f", (float)(total - free - cached) / 1024 / 1024); + return fmt_human((total - free - cached) * 1024, 1024); } #elif defined(__OpenBSD__) #include - #include /* dbtob */ #include #include #include - static void + static int getstats(int *total, int *used) { struct swapent *sep, *fsep; int rnswap, nswap, i; - nswap = swapctl(SWAP_NSWAP, 0, 0); - if (nswap < 1) { - fprintf(stderr, "swaptctl 'SWAP_NSWAP': %s\n", strerror(errno)); + if ((nswap = swapctl(SWAP_NSWAP, 0, 0)) < 1) { + warn("swaptctl 'SWAP_NSWAP':"); + return 1; } - - fsep = sep = calloc(nswap, sizeof(*sep)); - if (!sep) { - fprintf(stderr, "calloc 'nswap': %s\n", strerror(errno)); + if (!(fsep = sep = calloc(nswap, sizeof(*sep)))) { + warn("calloc 'nswap':"); + return 1; } - - rnswap = swapctl(SWAP_STATS, (void *)sep, nswap); - if (rnswap < 0) { - fprintf(stderr, "swapctl 'SWAP_STATA': %s\n", strerror(errno)); + if ((rnswap = swapctl(SWAP_STATS, (void *)sep, nswap)) < 0) { + warn("swapctl 'SWAP_STATA':"); + return 1; } - if (nswap != rnswap) { - fprintf(stderr, "SWAP_STATS != SWAP_NSWAP\n"); + warn("getstats: SWAP_STATS != SWAP_NSWAP"); + return 1; } *total = 0; @@ -165,6 +142,8 @@ } free(fsep); + + return 0; } const char * @@ -172,9 +151,11 @@ { int total, used; - getstats(&total, &used); + if (getstats(&total, &used)) { + return NULL; + } - return bprintf("%f", (float)(total - used) / 1024 / 1024); + return fmt_human((total - used) * 1024, 1024); } const char * @@ -182,7 +163,13 @@ { int total, used; - getstats(&total, &used); + if (getstats(&total, &used)) { + return NULL; + } + + if (total == 0) { + return NULL; + } return bprintf("%d", 100 * used / total); } @@ -192,9 +179,11 @@ { int total, used; - getstats(&total, &used); + if (getstats(&total, &used)) { + return NULL; + } - return bprintf("%f", (float)total / 1024 / 1024); + return fmt_human(total * 1024, 1024); } const char * @@ -202,8 +191,94 @@ { int total, used; - getstats(&total, &used); + if (getstats(&total, &used)) { + return NULL; + } + + return fmt_human(used * 1024, 1024); + } +#elif defined(__FreeBSD__) + #include + #include + #include + #include + #include + + static int getswapinfo(struct kvm_swap *swap_info, size_t size) + { + kvm_t *kd; + + kd = kvm_openfiles(NULL, "/dev/null", NULL, 0, NULL); + if(kd == NULL) { + warn("kvm_openfiles '/dev/null':"); + return 0; + } + + if(kvm_getswapinfo(kd, swap_info, size, 0 /* Unused flags */) == -1) { + warn("kvm_getswapinfo:"); + kvm_close(kd); + return 0; + } + + kvm_close(kd); + return 1; + } + + const char * + swap_free(void) + { + struct kvm_swap swap_info[1]; + long used, total; + + if(!getswapinfo(swap_info, 1)) + return NULL; + + total = swap_info[0].ksw_total; + used = swap_info[0].ksw_used; + + return fmt_human((total - used) * getpagesize(), 1024); + } + + const char * + swap_perc(void) + { + struct kvm_swap swap_info[1]; + long used, total; + + if(!getswapinfo(swap_info, 1)) + return NULL; + + total = swap_info[0].ksw_total; + used = swap_info[0].ksw_used; + + return bprintf("%d", used * 100 / total); + } + + const char * + swap_total(void) + { + struct kvm_swap swap_info[1]; + long total; + + if(!getswapinfo(swap_info, 1)) + return NULL; + + total = swap_info[0].ksw_total; + + return fmt_human(total * getpagesize(), 1024); + } + + const char * + swap_used(void) + { + struct kvm_swap swap_info[1]; + long used; + + if(!getswapinfo(swap_info, 1)) + return NULL; + + used = swap_info[0].ksw_used; - return bprintf("%f", (float)used / 1024 / 1024); + return fmt_human(used * getpagesize(), 1024); } #endif