Merge pull request #532 from GRFreire/folder-size-crash

Fix panic when trying to get folder size without needed permissions
yorukot · Dec 30, 2024 · 85f7f6f · 85f7f6f
2 parents c6370eb + 945aeab
commit 85f7f6f
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/internal/function.go b/src/internal/function.go
@@ -415,12 +415,15 @@ func calculateMD5Checksum(filePath string) (string, error) {
 // Get directory total size
 func dirSize(path string) int64 {
 	var size int64
-	filepath.Walk(path, func(_ string, info os.FileInfo, err error) error {
+	filepath.WalkDir(path, func(_ string, entry os.DirEntry, err error) error {
 		if err != nil {
 			outPutLog("Dir size function error", err)
 		}
-		if !info.IsDir() {
-			size += info.Size()
+		if !entry.IsDir() {
+			info, err := entry.Info()
+			if err == nil {
+				size += info.Size()
+			}
 		}
 		return err
 	})