/* $Id: dir2.cpp 78187 2019-04-17 21:50:13Z vboxsync $ */ /** @file * IPRT - Directory Manipulation, Part 2. */ /* * Copyright (C) 2006-2019 Oracle Corporation * * This file is part of VirtualBox Open Source Edition (OSE), as * available from http://www.virtualbox.org. This file is free software; * you can redistribute it and/or modify it under the terms of the GNU * General Public License (GPL) as published by the Free Software * Foundation, in version 2 as it comes in the "COPYING" file of the * VirtualBox OSE distribution. VirtualBox OSE is distributed in the * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind. * * The contents of this file may alternatively be used under the terms * of the Common Development and Distribution License Version 1.0 * (CDDL) only, as it comes in the "COPYING.CDDL" file of the * VirtualBox OSE distribution, in which case the provisions of the * CDDL are applicable instead of those of the GPL. * * You may elect to license modified versions of this file under the * terms and conditions of either the GPL or the CDDL or both. */ /********************************************************************************************************************************* * Header Files * *********************************************************************************************************************************/ #define LOG_GROUP RTLOGGROUP_DIR #include #include "internal/iprt.h" #include #include #include #include #include #include #include #include #include #include "internal/path.h" /** * Recursion worker for RTDirRemoveRecursive. * * @returns IPRT status code. * @param pszBuf The path buffer. Contains the abs path to the * directory to recurse into. Trailing slash. * @param cchDir The length of the directory we're recursing into, * including the trailing slash. * @param cbBuf Size of the buffer @a pszBuf points to. * @param pDirEntry The dir entry buffer. (Shared to save stack.) * @param pObjInfo The object info buffer. (ditto) * @param fFlags RTDIRRMREC_F_XXX. */ static int rtDirRemoveRecursiveSub(char *pszBuf, size_t cchDir, size_t cbBuf, PRTDIRENTRY pDirEntry, PRTFSOBJINFO pObjInfo, uint32_t fFlags) { AssertReturn(RTPATH_IS_SLASH(pszBuf[cchDir - 1]), VERR_INTERNAL_ERROR_4); /* * Enumerate the directory content and dispose of it. */ RTDIR hDir; int rc = RTDirOpenFiltered(&hDir, pszBuf, RTDIRFILTER_NONE, fFlags & RTDIRRMREC_F_NO_ABS_PATH ? RTDIR_F_NO_ABS_PATH : 0); if (RT_FAILURE(rc)) return rc; while (RT_SUCCESS(rc = RTDirRead(hDir, pDirEntry, NULL))) { if (!RTDirEntryIsStdDotLink(pDirEntry)) { /* Construct the full name of the entry. */ if (cchDir + pDirEntry->cbName + 1 /* dir slash */ >= cbBuf) { rc = VERR_FILENAME_TOO_LONG; break; } memcpy(&pszBuf[cchDir], pDirEntry->szName, pDirEntry->cbName + 1); /* Deal with the unknown type. */ if (pDirEntry->enmType == RTDIRENTRYTYPE_UNKNOWN) { rc = RTPathQueryInfoEx(pszBuf, pObjInfo, RTFSOBJATTRADD_NOTHING, RTPATH_F_ON_LINK); if (RT_SUCCESS(rc) && RTFS_IS_DIRECTORY(pObjInfo->Attr.fMode)) pDirEntry->enmType = RTDIRENTRYTYPE_DIRECTORY; else if (RT_SUCCESS(rc) && RTFS_IS_FILE(pObjInfo->Attr.fMode)) pDirEntry->enmType = RTDIRENTRYTYPE_FILE; else if (RT_SUCCESS(rc) && RTFS_IS_SYMLINK(pObjInfo->Attr.fMode)) pDirEntry->enmType = RTDIRENTRYTYPE_SYMLINK; } /* Try the delete the fs object. */ switch (pDirEntry->enmType) { case RTDIRENTRYTYPE_FILE: rc = RTFileDelete(pszBuf); break; case RTDIRENTRYTYPE_DIRECTORY: { size_t cchSubDir = cchDir + pDirEntry->cbName; pszBuf[cchSubDir++] = '/'; pszBuf[cchSubDir] = '\0'; rc = rtDirRemoveRecursiveSub(pszBuf, cchSubDir, cbBuf, pDirEntry, pObjInfo, fFlags); if (RT_SUCCESS(rc)) { pszBuf[cchSubDir] = '\0'; rc = RTDirRemove(pszBuf); } break; } //case RTDIRENTRYTYPE_SYMLINK: // rc = RTSymlinkDelete(pszBuf, 0); // break; default: /** @todo not implemented yet. */ rc = VINF_SUCCESS; break; } if (RT_FAILURE(rc)) break; } } if (rc == VERR_NO_MORE_FILES) rc = VINF_SUCCESS; RTDirClose(hDir); return rc; } RTDECL(int) RTDirRemoveRecursive(const char *pszPath, uint32_t fFlags) { AssertReturn(!(fFlags & ~RTDIRRMREC_F_VALID_MASK), VERR_INVALID_PARAMETER); /* * Allocate path buffer. */ char *pszAbsPath; size_t cbAbsPathBuf = RTPATH_BIG_MAX; char *pszAbsPathFree = pszAbsPath = (char *)RTMemTmpAlloc(cbAbsPathBuf); if (!pszAbsPath) { cbAbsPathBuf = RTPATH_MAX; pszAbsPath = (char *)alloca(RTPATH_MAX); } /* * Get an absolute path because this is easier to work with and * eliminates any races with changing CWD. */ int rc; if (!(fFlags & RTDIRRMREC_F_NO_ABS_PATH)) rc = RTPathAbs(pszPath, pszAbsPath, cbAbsPathBuf); else if (*pszPath != '\0') rc = RTStrCopy(pszAbsPath, cbAbsPathBuf, pszPath); else rc = VERR_PATH_ZERO_LENGTH; if (RT_SUCCESS(rc)) { /* * This API is not permitted applied to the root of anything. */ union { RTPATHPARSED Parsed; uint8_t abParsed[RTPATHPARSED_MIN_SIZE]; } uBuf; RTPathParse(pszPath, &uBuf.Parsed, sizeof(uBuf), RTPATH_STR_F_STYLE_HOST); if ( uBuf.Parsed.cComps <= 1 && (uBuf.Parsed.fProps & RTPATH_PROP_ROOT_SLASH)) rc = VERR_ACCESS_DENIED; else { /* * Because of the above restriction, we never have to deal with the root * slash problem and can safely strip any trailing slashes and add a * definite one. */ RTPathStripTrailingSlash(pszAbsPath); size_t cchAbsPath = strlen(pszAbsPath); if (cchAbsPath + 1 < cbAbsPathBuf) { pszAbsPath[cchAbsPath++] = RTPATH_SLASH; pszAbsPath[cchAbsPath] = '\0'; /* * Check if it exists so we can return quietly if it doesn't. */ RTFSOBJINFO SharedObjInfoBuf; rc = RTPathQueryInfoEx(pszAbsPath, &SharedObjInfoBuf, RTFSOBJATTRADD_NOTHING, RTPATH_F_ON_LINK); if ( rc == VERR_PATH_NOT_FOUND || rc == VERR_FILE_NOT_FOUND) rc = VINF_SUCCESS; else if ( RT_SUCCESS(rc) && RTFS_IS_DIRECTORY(SharedObjInfoBuf.Attr.fMode)) { /* * We're all set for the recursion now, so get going. */ RTDIRENTRY SharedDirEntryBuf; rc = rtDirRemoveRecursiveSub(pszAbsPath, cchAbsPath, cbAbsPathBuf, &SharedDirEntryBuf, &SharedObjInfoBuf, fFlags); /* * Remove the specified directory if desired and removing the content was successful. */ if ( RT_SUCCESS(rc) && !(fFlags & RTDIRRMREC_F_CONTENT_ONLY)) { pszAbsPath[cchAbsPath] = 0; rc = RTDirRemove(pszAbsPath); } } else if (RT_SUCCESS(rc)) rc = VERR_NOT_A_DIRECTORY; } else rc = VERR_FILENAME_TOO_LONG; } } if (pszAbsPathFree) RTMemTmpFree(pszAbsPathFree); return rc; }