slstatus

status monitor
git clone git://git.suckless.org/slstatus
Log | Files | Refs | README | LICENSE

commit ceb13206a4d9f410ab49c3824c487658c341d8c8
parent 10dbc9543edd5b5a7929dd8fa87ed33dd545375c
Author: Laslo Hunhold <dev@frign.de>
Date:   Mon, 21 May 2018 20:08:56 +0200

Refactor fmt_human() and fix a bug

It is not necessary to copy memory or anything. Just keep a pointer to
the active prefix-array and assign the length of the arrays to a
variable.

Make the code more readable by using a switch, be more strict when an
invalid base is passed to it and fix a small oversight in the bottom of
the code where the base 1024 was forgotten to generalized.

Diffstat:
Mutil.c | 30+++++++++++++++++++-----------
Mutil.h | 2+-
2 files changed, 20 insertions(+), 12 deletions(-)

diff --git a/util.c b/util.c @@ -89,21 +89,29 @@ bprintf(const char *fmt, ...) const char * fmt_human(size_t num, int base) { - size_t i; double scaled; - const char *siprefix[] = { "", "k", "M", "G", "T", "P", "E", "Z", "Y" }; - const char *iecprefix[] = { "", "Ki", "Mi", "Gi", "Ti", "Pi", "Ei", - "Zi", "Yi" }; - char *prefix[9]; - - if (base == 1000) { - memcpy(prefix, siprefix, sizeof(prefix)); - } else if (base == 1024) { - memcpy(prefix, iecprefix, sizeof(prefix)); + size_t i, prefixlen; + const char **prefix; + const char *prefix_1000[] = { "", "k", "M", "G", "T", "P", "E", "Z", "Y" }; + const char *prefix_1024[] = { "", "Ki", "Mi", "Gi", "Ti", "Pi", "Ei", + "Zi", "Yi" }; + + switch (base) { + case 1000: + prefix = prefix_1000; + prefixlen = LEN(prefix_1000); + break; + case 1024: + prefix = prefix_1024; + prefixlen = LEN(prefix_1024); + break; + default: + warn("fmt_human: Invalid base"); + return NULL; } scaled = num; - for (i = 0; i < LEN(prefix) && scaled >= 1024; i++) { + for (i = 0; i < prefixlen && scaled >= base; i++) { scaled /= base; } diff --git a/util.h b/util.h @@ -10,5 +10,5 @@ void die(const char *, ...); int esnprintf(char *str, size_t size, const char *fmt, ...); const char *bprintf(const char *fmt, ...); -const char *fmt_human(size_t num, int iec); +const char *fmt_human(size_t num, int base); int pscanf(const char *path, const char *fmt, ...);