Changeset 19211 in vbox for trunk/src/VBox/Runtime/r3/dir.cpp

Timestamp:

Apr 27, 2009 1:56:07 PM (16 years ago)

Author:

vboxsync

Message:

IPRT: Function RTDirRemoveRecursive added. Only partially implemented yet (no symlinks etc).

File:

• trunk/src/VBox/Runtime/r3/dir.cpp (modified) (2 diffs)

trunk/src/VBox/Runtime/r3/dir.cpp

@@ -43 +43 @@
 
 #include <iprt/dir.h>
+#include <iprt/file.h>
 #include <iprt/path.h>
 #include <iprt/alloc.h>
@@ -715 +716 @@
 }
 
+
+RTDECL(int) RTDirRemoveRecursive(const char *pszPath)
+{
+    int rc;
+
+    if (!RTDirExists(pszPath))
+        return VINF_SUCCESS;
+
+    char szAbsPath[RTPATH_MAX];
+    /** @todo use RTPathReal here instead? */
+    rc = RTPathAbs(pszPath, szAbsPath, sizeof(szAbsPath));
+    if (RT_FAILURE(rc))
+        return rc;
+
+    PRTDIR pDir = NULL;
+    rc = RTDirOpen(&pDir, szAbsPath);
+    if (RT_SUCCESS(rc))
+    {
+        RTDIRENTRY dirEntry;
+        size_t cbDirEntry = sizeof(dirEntry);
+
+        while ((rc = RTDirRead(pDir, &dirEntry, NULL /* Passing an argument won't work here yet. */)) == VINF_SUCCESS)
+        {
+            char* pszEntry = NULL;
+            rc = RTStrAPrintf(&pszEntry, "%s/%s", szAbsPath, dirEntry.szName);
+            if(    RT_SUCCESS(rc)
+                && strcmp(dirEntry.szName, ".")
+                && strcmp(dirEntry.szName, ".."))
+            {
+                switch (dirEntry.enmType)
+                {
+                    case RTDIRENTRYTYPE_FILE:
+
+                        rc = RTFileDelete(pszEntry);
+                        break;
+
+                    case RTDIRENTRYTYPE_DIRECTORY:
+
+                        rc = RTDirRemoveRecursive(pszEntry);
+                        break;
+
+                    default:
+                        /** @todo not implemented yet. */
+                        break;
+                }
+
+                RTStrFree(pszEntry);
+            }
+
+            if (RT_FAILURE(rc))
+               break;
+        }
+        if (rc == VERR_NO_MORE_FILES)
+            rc = VINF_SUCCESS;
+
+        RTDirClose(pDir);
+        rc = RTDirRemove(szAbsPath);    
+    }
+
+    LogFlow(("RTDirRemoveRecursive(%p:{%s}): returns %Rrc\n", pszPath, pszPath, rc));
+    return rc;
+}
+
+