fix: division by zero possible

some filesystem have zero Inode
shirou · Dec 23, 2016 · c73b668 · c73b668
1 parent a0f760e
commit c73b668
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/disk/disk_unix.go b/disk/disk_unix.go
@@ -24,11 +24,22 @@ func Usage(path string) (*UsageStat, error) {
 	// if could not get InodesTotal, return empty
 	if ret.InodesTotal < ret.InodesFree {
 		return ret, nil
-	}
+	}	
+
 	ret.InodesUsed = (ret.InodesTotal - ret.InodesFree)
-	ret.InodesUsedPercent = (float64(ret.InodesUsed) / float64(ret.InodesTotal)) * 100.0
 	ret.Used = (uint64(stat.Blocks) - uint64(stat.Bfree)) * uint64(bsize)
-	ret.UsedPercent = (float64(ret.Used) / float64(ret.Total)) * 100.0
+
+        if ret.InodesTotal == 0 {
+		ret.InodesUsedPercent = 0
+	} else {
+		ret.InodesUsedPercent = (float64(ret.InodesUsed) / float64(ret.InodesTotal)) * 100.0
+	}
 
+	if ret.Total == 0 {
+		ret.UsedPercent = 0
+	} else {
+		ret.UsedPercent = (float64(ret.Used) / float64(ret.Total)) * 100.0
+	}
+
 	return ret, nil
 }