avoid failure when checking free space if cache directory doesn't exist

vmware · Dec 1, 2023 · 6f2ca3f · 6f2ca3f
1 parent 13751a8
commit 6f2ca3f
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/client/packageutils.c b/client/packageutils.c
@@ -1000,21 +1000,28 @@ TDNFGetAvailableCacheBytes(
     )
 {
     uint32_t dwError = 0;
-    struct statfs tmpStatfsBuffer = {0};
+    struct statfs stfs = {0};
+    struct stat st = {0};
 
     if(!pConf || !pConf->pszCacheDir || !pqwAvailCacheDirBytes)
     {
         dwError = ERROR_TDNF_INVALID_PARAMETER;
         BAIL_ON_TDNF_ERROR(dwError);
     }
 
-    if (statfs(pConf->pszCacheDir, &tmpStatfsBuffer) != 0)
+    if (stat(pConf->pszCacheDir, &st) != 0) {
+        /* avoid failure when checking space, and dir doesn't exist */
+        dwError = TDNFUtilsMakeDirs(pConf->pszCacheDir);
+        BAIL_ON_TDNF_ERROR(dwError);
+    }
+
+    if (statfs(pConf->pszCacheDir, &stfs) != 0)
     {
         dwError = errno;
         BAIL_ON_TDNF_SYSTEM_ERROR(dwError);
     }
 
-    *pqwAvailCacheDirBytes = tmpStatfsBuffer.f_bsize * tmpStatfsBuffer.f_bavail;
+    *pqwAvailCacheDirBytes = stfs.f_bsize * stfs.f_bavail;
 
 cleanup:
     return dwError;