swap.c: Refactor getstats() and its usage

Add a return value so we are able to check any errors and actually fail
when there has been a critical error instead of carrying on uselessly.
This commit is contained in:
Laslo Hunhold 2018-05-21 23:03:34 +02:00 committed by Aaron Marcher
parent ceb13206a4
commit 7f91ce265e
1 changed files with 22 additions and 14 deletions

View File

@ -135,29 +135,27 @@
#include <sys/types.h> #include <sys/types.h>
#include <unistd.h> #include <unistd.h>
static void static int
getstats(int *total, int *used) getstats(int *total, int *used)
{ {
struct swapent *sep, *fsep; struct swapent *sep, *fsep;
int rnswap, nswap, i; int rnswap, nswap, i;
nswap = swapctl(SWAP_NSWAP, 0, 0); if ((nswap = swapctl(SWAP_NSWAP, 0, 0)) < 1) {
if (nswap < 1) {
warn("swaptctl 'SWAP_NSWAP':"); warn("swaptctl 'SWAP_NSWAP':");
return 1;
} }
if (!(fsep = sep = calloc(nswap, sizeof(*sep)))) {
fsep = sep = calloc(nswap, sizeof(*sep));
if (!sep) {
warn("calloc 'nswap':"); warn("calloc 'nswap':");
return 1;
} }
if ((rnswap = swapctl(SWAP_STATS, (void *)sep, nswap)) < 0) {
rnswap = swapctl(SWAP_STATS, (void *)sep, nswap);
if (rnswap < 0) {
warn("swapctl 'SWAP_STATA':"); warn("swapctl 'SWAP_STATA':");
return 1;
} }
if (nswap != rnswap) { if (nswap != rnswap) {
warn("getstats: SWAP_STATS != SWAP_NSWAP"); warn("getstats: SWAP_STATS != SWAP_NSWAP");
return 1;
} }
*total = 0; *total = 0;
@ -169,6 +167,8 @@
} }
free(fsep); free(fsep);
return 0;
} }
const char * const char *
@ -176,7 +176,9 @@
{ {
int total, used; int total, used;
getstats(&total, &used); if (getstats(&total, &used)) {
return NULL;
}
return fmt_human((total - used) * 1024, 1024); return fmt_human((total - used) * 1024, 1024);
} }
@ -186,7 +188,9 @@
{ {
int total, used; int total, used;
getstats(&total, &used); if (getstats(&total, &used)) {
return NULL;
}
if (total == 0) { if (total == 0) {
return NULL; return NULL;
@ -200,7 +204,9 @@
{ {
int total, used; int total, used;
getstats(&total, &used); if (getstats(&total, &used)) {
return NULL;
}
return fmt_human(total * 1024, 1024); return fmt_human(total * 1024, 1024);
} }
@ -210,7 +216,9 @@
{ {
int total, used; int total, used;
getstats(&total, &used); if (getstats(&total, &used)) {
return NULL;
}
return fmt_human(used * 1024, 1024); return fmt_human(used * 1024, 1024);
} }