X-Git-Url: https://git.xinqibao.xyz/slstatus.git/blobdiff_plain/9750a3d731cd381e832bcacf1d03e48ddb46cc16..6c6c144be62d6b2da564f0875ba5f202dbaa3183:/components/swap.c diff --git a/components/swap.c b/components/swap.c index 08a97de..031b713 100644 --- a/components/swap.c +++ b/components/swap.c @@ -1,41 +1,51 @@ /* See LICENSE file for copyright and license details. */ +#include +#include +#include + +#include "../util.h" + #if defined(__linux__) - #include - #include - #include + static size_t + pread(const char *path, char *buf, size_t bufsiz) + { + FILE *fp; + size_t bytes_read; + + if (!(fp = fopen(path, "r"))) { + fprintf(stderr, "fopen '%s': %s\n", path, strerror(errno)); + return 0; + } + if (!(bytes_read = fread(buf, sizeof(char), bufsiz, fp))) { + fprintf(stderr, "fread '%s': %s\n", path, strerror(errno)); + fclose(fp); + return 0; + } + fclose(fp); + + buf[bytes_read] = '\0'; - #include "../util.h" + return bytes_read; + } const char * swap_free(void) { long total, free; - FILE *fp; - size_t bytes_read; char *match; - fp = fopen("/proc/meminfo", "r"); - if (fp == NULL) { - fprintf(stderr, "fopen '/proc/meminfo': %s\n", - strerror(errno)); + if (!pread("/proc/meminfo", buf, sizeof(buf) - 1)) { return NULL; } - if ((bytes_read = fread(buf, sizeof(char), sizeof(buf) - 1, - fp)) == 0) { - fprintf(stderr, "fread '/proc/meminfo': %s\n", - strerror(errno)); - fclose(fp); + if (!(match = strstr(buf, "SwapTotal"))) { return NULL; } - fclose(fp); - - if ((match = strstr(buf, "SwapTotal")) == NULL) - return NULL; sscanf(match, "SwapTotal: %ld kB\n", &total); - if ((match = strstr(buf, "SwapFree")) == NULL) + if (!(match = strstr(buf, "SwapFree"))) { return NULL; + } sscanf(match, "SwapFree: %ld kB\n", &free); return bprintf("%f", (float)free / 1024 / 1024); @@ -45,36 +55,25 @@ swap_perc(void) { long total, free, cached; - FILE *fp; - size_t bytes_read; char *match; - fp = fopen("/proc/meminfo", "r"); - if (fp == NULL) { - fprintf(stderr, "fopen '/proc/meminfo': %s\n", - strerror(errno)); + if (!pread("/proc/meminfo", buf, sizeof(buf) - 1)) { return NULL; } - if ((bytes_read = fread(buf, sizeof(char), sizeof(buf) - 1, - fp)) == 0) { - fprintf(stderr, "fread '/proc/meminfo': %s\n", - strerror(errno)); - fclose(fp); + if (!(match = strstr(buf, "SwapTotal"))) { return NULL; } - fclose(fp); - - if ((match = strstr(buf, "SwapTotal")) == NULL) - return NULL; sscanf(match, "SwapTotal: %ld kB\n", &total); - if ((match = strstr(buf, "SwapCached")) == NULL) + if (!(match = strstr(buf, "SwapCached"))) { return NULL; + } sscanf(match, "SwapCached: %ld kB\n", &cached); - if ((match = strstr(buf, "SwapFree")) == NULL) + if (!(match = strstr(buf, "SwapFree"))) { return NULL; + } sscanf(match, "SwapFree: %ld kB\n", &free); return bprintf("%d", 100 * (total - free - cached) / total); @@ -84,27 +83,15 @@ swap_total(void) { long total; - FILE *fp; - size_t bytes_read; char *match; - fp = fopen("/proc/meminfo", "r"); - if (fp == NULL) { - fprintf(stderr, "fopen '/proc/meminfo': %s\n", - strerror(errno)); + if (!pread("/proc/meminfo", buf, sizeof(buf) - 1)) { return NULL; } - if ((bytes_read = fread(buf, sizeof(char), sizeof(buf) - 1, - fp)) == 0) { - fprintf(stderr, "fread '/proc/meminfo': %s\n", - strerror(errno)); - fclose(fp); - return NULL; - } - fclose(fp); - if ((match = strstr(buf, "SwapTotal")) == NULL) + if (!(match = strstr(buf, "SwapTotal"))) { return NULL; + } sscanf(match, "SwapTotal: %ld kB\n", &total); return bprintf("%f", (float)total / 1024 / 1024); @@ -114,39 +101,111 @@ swap_used(void) { long total, free, cached; - FILE *fp; - size_t bytes_read; char *match; - fp = fopen("/proc/meminfo", "r"); - if (fp == NULL) { - fprintf(stderr, "fopen '/proc/meminfo': %s\n", - strerror(errno)); - return NULL; - } - if ((bytes_read = fread(buf, sizeof(char), sizeof(buf) - 1, - fp)) == 0) { - fprintf(stderr, "fread '/proc/meminfo': %s\n", - strerror(errno)); - fclose(fp); + if (!pread("/proc/meminfo", buf, sizeof(buf) - 1)) { return NULL; } - fclose(fp); - if ((match = strstr(buf, "SwapTotal")) == NULL) + if (!(match = strstr(buf, "SwapTotal"))) { return NULL; + } sscanf(match, "SwapTotal: %ld kB\n", &total); - if ((match = strstr(buf, "SwapCached")) == NULL) + if (!(match = strstr(buf, "SwapCached"))) { return NULL; + } sscanf(match, "SwapCached: %ld kB\n", &cached); - if ((match = strstr(buf, "SwapFree")) == NULL) + if (!(match = strstr(buf, "SwapFree"))) { return NULL; + } sscanf(match, "SwapFree: %ld kB\n", &free); return bprintf("%f", (float)(total - free - cached) / 1024 / 1024); } #elif defined(__OpenBSD__) - /* unimplemented */ + #include + #include /* dbtob */ + #include + #include + #include + + #define dbtoqb(b) dbtob((int64_t)(b)) + + static void + 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)); + } + + fsep = sep = calloc(nswap, sizeof(*sep)); + if (!sep) { + fprintf(stderr, "calloc 'nswap': %s\n", strerror(errno)); + } + + rnswap = swapctl(SWAP_STATS, (void *)sep, nswap); + if (rnswap < 0) { + fprintf(stderr, "swapctl 'SWAP_STATA': %s\n", strerror(errno)); + } + + if (nswap != rnswap) { + fprintf(stderr, "SWAP_STATS != SWAP_NSWAP\n"); + } + + *total = 0; + *used = 0; + + for (i = 0; i < rnswap; i++) { + *total += dbtoqb(sep->se_nblks); + *used += dbtoqb(sep->se_inuse); + } + + free(fsep); + } + + const char * + swap_free(void) + { + int total, used; + + getstats(&total, &used); + + return bprintf("%f", (float)(total - used) / 1024 / 1024 / 1024); + } + + const char * + swap_perc(void) + { + int total, used; + + getstats(&total, &used); + + return bprintf("%d", 100 * used / total); + } + + const char * + swap_total(void) + { + int total, used; + + getstats(&total, &used); + + return bprintf("%f", (float)total / 1024 / 1024 / 1024); + } + + const char * + swap_used(void) + { + int total, used; + + getstats(&total, &used); + + return bprintf("%f", (float)used / 1024 / 1024 / 1024); + } #endif