Xinqi Bao's Git

battery_remaining: check for division by zero
[slstatus.git] / components / disk.c
index 2ce260b..cf3c099 100644 (file)
@@ -12,28 +12,25 @@ disk_free(const char *mnt)
        struct statvfs fs;
 
        if (statvfs(mnt, &fs) < 0) {
-               fprintf(stderr, "statvfs '%s': %s\n", mnt, strerror(errno));
+               warn("statvfs '%s':", mnt);
                return NULL;
        }
 
-       return bprintf("%f",
-                      (float)fs.f_bsize * (float)fs.f_bfree / 1024 / 1024 / 1024);
+       return fmt_human_2(fs.f_frsize * fs.f_bavail);
 }
 
 const char *
 disk_perc(const char *mnt)
 {
-       int perc;
        struct statvfs fs;
 
        if (statvfs(mnt, &fs) < 0) {
-               fprintf(stderr, "statvfs '%s': %s\n", mnt, strerror(errno));
+               warn("statvfs '%s':", mnt);
                return NULL;
        }
 
-       perc = 100 * (1.0f - ((float)fs.f_bfree / (float)fs.f_blocks));
-
-       return bprintf("%d", perc);
+       return bprintf("%d", (int)(100 *
+                      (1.0f - ((float)fs.f_bavail / (float)fs.f_blocks))));
 }
 
 const char *
@@ -42,12 +39,11 @@ disk_total(const char *mnt)
        struct statvfs fs;
 
        if (statvfs(mnt, &fs) < 0) {
-               fprintf(stderr, "statvfs '%s': %s\n", mnt, strerror(errno));
+               warn("statvfs '%s':", mnt);
                return NULL;
        }
 
-       return bprintf("%f",
-                      (float)fs.f_bsize * (float)fs.f_blocks / 1024 / 1024 / 1024);
+       return fmt_human_2(fs.f_frsize * fs.f_blocks);
 }
 
 const char *
@@ -56,11 +52,9 @@ disk_used(const char *mnt)
        struct statvfs fs;
 
        if (statvfs(mnt, &fs) < 0) {
-               fprintf(stderr, "statvfs '%s': %s\n", mnt, strerror(errno));
+               warn("statvfs '%s':", mnt);
                return NULL;
        }
 
-       return bprintf("%f",
-                      (float)fs.f_bsize * ((float)fs.f_blocks -
-                      (float)fs.f_bfree) / 1024 / 1024 / 1024);
+       return fmt_human_2(fs.f_frsize * (fs.f_blocks - fs.f_bfree));
 }