diff options
| author | Sean Hall <r.sean.hall@gmail.com> | 2021-03-02 14:19:14 -0600 |
|---|---|---|
| committer | Sean Hall <r.sean.hall@gmail.com> | 2021-03-02 15:40:02 -0600 |
| commit | 10ebf674da5df9224e4eddd3545518434c5b455b (patch) | |
| tree | ea1f4063edd46e9942eab94dd7adb2f75c6c589e /src/dutil/dirutil.cpp | |
| parent | 3bbf1347b900ec115a12faf8f46965c9b7649696 (diff) | |
| download | wix-10ebf674da5df9224e4eddd3545518434c5b455b.tar.gz wix-10ebf674da5df9224e4eddd3545518434c5b455b.tar.bz2 wix-10ebf674da5df9224e4eddd3545518434c5b455b.zip | |
Update rest of dutil to use their own source with the Exit* macros.
Fix some CA warnings.
Diffstat (limited to 'src/dutil/dirutil.cpp')
| -rw-r--r-- | src/dutil/dirutil.cpp | 61 |
1 files changed, 38 insertions, 23 deletions
diff --git a/src/dutil/dirutil.cpp b/src/dutil/dirutil.cpp index 5e22ee65..81130a8d 100644 --- a/src/dutil/dirutil.cpp +++ b/src/dutil/dirutil.cpp | |||
| @@ -3,6 +3,21 @@ | |||
| 3 | #include "precomp.h" | 3 | #include "precomp.h" |
| 4 | 4 | ||
| 5 | 5 | ||
| 6 | // Exit macros | ||
| 7 | #define DirExitOnLastError(x, s, ...) ExitOnLastErrorSource(DUTIL_SOURCE_DIRUTIL, x, s, __VA_ARGS__) | ||
| 8 | #define DirExitOnLastErrorDebugTrace(x, s, ...) ExitOnLastErrorDebugTraceSource(DUTIL_SOURCE_DIRUTIL, x, s, __VA_ARGS__) | ||
| 9 | #define DirExitWithLastError(x, s, ...) ExitWithLastErrorSource(DUTIL_SOURCE_DIRUTIL, x, s, __VA_ARGS__) | ||
| 10 | #define DirExitOnFailure(x, s, ...) ExitOnFailureSource(DUTIL_SOURCE_DIRUTIL, x, s, __VA_ARGS__) | ||
| 11 | #define DirExitOnRootFailure(x, s, ...) ExitOnRootFailureSource(DUTIL_SOURCE_DIRUTIL, x, s, __VA_ARGS__) | ||
| 12 | #define DirExitOnFailureDebugTrace(x, s, ...) ExitOnFailureDebugTraceSource(DUTIL_SOURCE_DIRUTIL, x, s, __VA_ARGS__) | ||
| 13 | #define DirExitOnNull(p, x, e, s, ...) ExitOnNullSource(DUTIL_SOURCE_DIRUTIL, p, x, e, s, __VA_ARGS__) | ||
| 14 | #define DirExitOnNullWithLastError(p, x, s, ...) ExitOnNullWithLastErrorSource(DUTIL_SOURCE_DIRUTIL, p, x, s, __VA_ARGS__) | ||
| 15 | #define DirExitOnNullDebugTrace(p, x, e, s, ...) ExitOnNullDebugTraceSource(DUTIL_SOURCE_DIRUTIL, p, x, e, s, __VA_ARGS__) | ||
| 16 | #define DirExitOnInvalidHandleWithLastError(p, x, s, ...) ExitOnInvalidHandleWithLastErrorSource(DUTIL_SOURCE_DIRUTIL, p, x, s, __VA_ARGS__) | ||
| 17 | #define DirExitOnWin32Error(e, x, s, ...) ExitOnWin32ErrorSource(DUTIL_SOURCE_DIRUTIL, e, x, s, __VA_ARGS__) | ||
| 18 | #define DirExitOnGdipFailure(g, x, s, ...) ExitOnGdipFailureSource(DUTIL_SOURCE_DIRUTIL, g, x, s, __VA_ARGS__) | ||
| 19 | |||
| 20 | |||
| 6 | /******************************************************************* | 21 | /******************************************************************* |
| 7 | DirExists | 22 | DirExists |
| 8 | 23 | ||
| @@ -59,12 +74,12 @@ extern "C" HRESULT DAPI DirCreateTempPath( | |||
| 59 | cch = ::GetTempPathW(countof(wzDir), wzDir); | 74 | cch = ::GetTempPathW(countof(wzDir), wzDir); |
| 60 | if (!cch || cch >= countof(wzDir)) | 75 | if (!cch || cch >= countof(wzDir)) |
| 61 | { | 76 | { |
| 62 | ExitWithLastError(hr, "Failed to GetTempPath."); | 77 | DirExitWithLastError(hr, "Failed to GetTempPath."); |
| 63 | } | 78 | } |
| 64 | 79 | ||
| 65 | if (!::GetTempFileNameW(wzDir, wzPrefix, 0, wzFile)) | 80 | if (!::GetTempFileNameW(wzDir, wzPrefix, 0, wzFile)) |
| 66 | { | 81 | { |
| 67 | ExitWithLastError(hr, "Failed to GetTempFileName."); | 82 | DirExitWithLastError(hr, "Failed to GetTempFileName."); |
| 68 | } | 83 | } |
| 69 | 84 | ||
| 70 | hr = ::StringCchCopyW(wzPath, cchPath, wzFile); | 85 | hr = ::StringCchCopyW(wzPath, cchPath, wzFile); |
| @@ -111,12 +126,12 @@ extern "C" HRESULT DAPI DirEnsureExists( | |||
| 111 | } | 126 | } |
| 112 | 127 | ||
| 113 | // if there is no parent directory fail | 128 | // if there is no parent directory fail |
| 114 | ExitOnNullDebugTrace(pwzLastSlash, hr, HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND), "cannot find parent path"); | 129 | DirExitOnNullDebugTrace(pwzLastSlash, hr, HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND), "cannot find parent path"); |
| 115 | 130 | ||
| 116 | *pwzLastSlash = L'\0'; // null terminate the parent path | 131 | *pwzLastSlash = L'\0'; // null terminate the parent path |
| 117 | hr = DirEnsureExists(wzPath, psa); // recurse! | 132 | hr = DirEnsureExists(wzPath, psa); // recurse! |
| 118 | *pwzLastSlash = L'\\'; // put the slash back | 133 | *pwzLastSlash = L'\\'; // put the slash back |
| 119 | ExitOnFailureDebugTrace(hr, "failed to create path: %ls", wzPath); | 134 | DirExitOnFailureDebugTrace(hr, "failed to create path: %ls", wzPath); |
| 120 | 135 | ||
| 121 | // try to create the directory now that all parents are created | 136 | // try to create the directory now that all parents are created |
| 122 | if (!::CreateDirectoryW(wzPath, psa)) | 137 | if (!::CreateDirectoryW(wzPath, psa)) |
| @@ -197,7 +212,7 @@ extern "C" HRESULT DAPI DirEnsureDeleteEx( | |||
| 197 | er = ERROR_PATH_NOT_FOUND; | 212 | er = ERROR_PATH_NOT_FOUND; |
| 198 | } | 213 | } |
| 199 | hr = HRESULT_FROM_WIN32(er); | 214 | hr = HRESULT_FROM_WIN32(er); |
| 200 | ExitOnRootFailure(hr, "Failed to get attributes for path: %ls", wzPath); | 215 | DirExitOnRootFailure(hr, "Failed to get attributes for path: %ls", wzPath); |
| 201 | } | 216 | } |
| 202 | 217 | ||
| 203 | if (dwAttrib & FILE_ATTRIBUTE_DIRECTORY) | 218 | if (dwAttrib & FILE_ATTRIBUTE_DIRECTORY) |
| @@ -206,7 +221,7 @@ extern "C" HRESULT DAPI DirEnsureDeleteEx( | |||
| 206 | { | 221 | { |
| 207 | if (!::SetFileAttributesW(wzPath, FILE_ATTRIBUTE_NORMAL)) | 222 | if (!::SetFileAttributesW(wzPath, FILE_ATTRIBUTE_NORMAL)) |
| 208 | { | 223 | { |
| 209 | ExitWithLastError(hr, "Failed to remove read-only attribute from path: %ls", wzPath); | 224 | DirExitWithLastError(hr, "Failed to remove read-only attribute from path: %ls", wzPath); |
| 210 | } | 225 | } |
| 211 | } | 226 | } |
| 212 | 227 | ||
| @@ -217,18 +232,18 @@ extern "C" HRESULT DAPI DirEnsureDeleteEx( | |||
| 217 | { | 232 | { |
| 218 | if (!::GetTempPathW(countof(wzTempDirectory), wzTempDirectory)) | 233 | if (!::GetTempPathW(countof(wzTempDirectory), wzTempDirectory)) |
| 219 | { | 234 | { |
| 220 | ExitWithLastError(hr, "Failed to get temp directory."); | 235 | DirExitWithLastError(hr, "Failed to get temp directory."); |
| 221 | } | 236 | } |
| 222 | } | 237 | } |
| 223 | 238 | ||
| 224 | // Delete everything in this directory. | 239 | // Delete everything in this directory. |
| 225 | hr = PathConcat(wzPath, L"*.*", &sczDelete); | 240 | hr = PathConcat(wzPath, L"*.*", &sczDelete); |
| 226 | ExitOnFailure(hr, "Failed to concat wild cards to string: %ls", wzPath); | 241 | DirExitOnFailure(hr, "Failed to concat wild cards to string: %ls", wzPath); |
| 227 | 242 | ||
| 228 | hFind = ::FindFirstFileW(sczDelete, &wfd); | 243 | hFind = ::FindFirstFileW(sczDelete, &wfd); |
| 229 | if (INVALID_HANDLE_VALUE == hFind) | 244 | if (INVALID_HANDLE_VALUE == hFind) |
| 230 | { | 245 | { |
| 231 | ExitWithLastError(hr, "failed to get first file in directory: %ls", wzPath); | 246 | DirExitWithLastError(hr, "failed to get first file in directory: %ls", wzPath); |
| 232 | } | 247 | } |
| 233 | 248 | ||
| 234 | do | 249 | do |
| @@ -243,18 +258,18 @@ extern "C" HRESULT DAPI DirEnsureDeleteEx( | |||
| 243 | wfd.cFileName[MAX_PATH - 1] = L'\0'; | 258 | wfd.cFileName[MAX_PATH - 1] = L'\0'; |
| 244 | 259 | ||
| 245 | hr = PathConcat(wzPath, wfd.cFileName, &sczDelete); | 260 | hr = PathConcat(wzPath, wfd.cFileName, &sczDelete); |
| 246 | ExitOnFailure(hr, "Failed to concat filename '%ls' to directory: %ls", wfd.cFileName, wzPath); | 261 | DirExitOnFailure(hr, "Failed to concat filename '%ls' to directory: %ls", wfd.cFileName, wzPath); |
| 247 | 262 | ||
| 248 | if (fRecurse && wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) | 263 | if (fRecurse && wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) |
| 249 | { | 264 | { |
| 250 | hr = PathBackslashTerminate(&sczDelete); | 265 | hr = PathBackslashTerminate(&sczDelete); |
| 251 | ExitOnFailure(hr, "Failed to ensure path is backslash terminated: %ls", sczDelete); | 266 | DirExitOnFailure(hr, "Failed to ensure path is backslash terminated: %ls", sczDelete); |
| 252 | 267 | ||
| 253 | hr = DirEnsureDeleteEx(sczDelete, dwFlags); // recursive call | 268 | hr = DirEnsureDeleteEx(sczDelete, dwFlags); // recursive call |
| 254 | if (FAILED(hr)) | 269 | if (FAILED(hr)) |
| 255 | { | 270 | { |
| 256 | // if we failed to delete a subdirectory, keep trying to finish any remaining files | 271 | // if we failed to delete a subdirectory, keep trying to finish any remaining files |
| 257 | ExitTraceSource(DUTIL_SOURCE_DEFAULT, hr, "Failed to delete subdirectory; continuing: %ls", sczDelete); | 272 | ExitTraceSource(DUTIL_SOURCE_DIRUTIL, hr, "Failed to delete subdirectory; continuing: %ls", sczDelete); |
| 258 | hr = S_OK; | 273 | hr = S_OK; |
| 259 | } | 274 | } |
| 260 | } | 275 | } |
| @@ -264,7 +279,7 @@ extern "C" HRESULT DAPI DirEnsureDeleteEx( | |||
| 264 | { | 279 | { |
| 265 | if (!::SetFileAttributesW(sczDelete, FILE_ATTRIBUTE_NORMAL)) | 280 | if (!::SetFileAttributesW(sczDelete, FILE_ATTRIBUTE_NORMAL)) |
| 266 | { | 281 | { |
| 267 | ExitWithLastError(hr, "Failed to remove attributes from file: %ls", sczDelete); | 282 | DirExitWithLastError(hr, "Failed to remove attributes from file: %ls", sczDelete); |
| 268 | } | 283 | } |
| 269 | } | 284 | } |
| 270 | 285 | ||
| @@ -274,7 +289,7 @@ extern "C" HRESULT DAPI DirEnsureDeleteEx( | |||
| 274 | { | 289 | { |
| 275 | if (!::GetTempFileNameW(wzTempDirectory, L"DEL", 0, wzTempPath)) | 290 | if (!::GetTempFileNameW(wzTempDirectory, L"DEL", 0, wzTempPath)) |
| 276 | { | 291 | { |
| 277 | ExitWithLastError(hr, "Failed to get temp file to move to."); | 292 | DirExitWithLastError(hr, "Failed to get temp file to move to."); |
| 278 | } | 293 | } |
| 279 | 294 | ||
| 280 | // Try to move the file to the temp directory then schedule for delete, | 295 | // Try to move the file to the temp directory then schedule for delete, |
| @@ -290,7 +305,7 @@ extern "C" HRESULT DAPI DirEnsureDeleteEx( | |||
| 290 | } | 305 | } |
| 291 | else | 306 | else |
| 292 | { | 307 | { |
| 293 | ExitWithLastError(hr, "Failed to delete file: %ls", sczDelete); | 308 | DirExitWithLastError(hr, "Failed to delete file: %ls", sczDelete); |
| 294 | } | 309 | } |
| 295 | } | 310 | } |
| 296 | } | 311 | } |
| @@ -303,7 +318,7 @@ extern "C" HRESULT DAPI DirEnsureDeleteEx( | |||
| 303 | } | 318 | } |
| 304 | else | 319 | else |
| 305 | { | 320 | { |
| 306 | ExitWithLastError(hr, "Failed while looping through files in directory: %ls", wzPath); | 321 | DirExitWithLastError(hr, "Failed while looping through files in directory: %ls", wzPath); |
| 307 | } | 322 | } |
| 308 | } | 323 | } |
| 309 | 324 | ||
| @@ -318,13 +333,13 @@ extern "C" HRESULT DAPI DirEnsureDeleteEx( | |||
| 318 | } | 333 | } |
| 319 | } | 334 | } |
| 320 | 335 | ||
| 321 | ExitOnRootFailure(hr, "Failed to remove directory: %ls", wzPath); | 336 | DirExitOnRootFailure(hr, "Failed to remove directory: %ls", wzPath); |
| 322 | } | 337 | } |
| 323 | } | 338 | } |
| 324 | else | 339 | else |
| 325 | { | 340 | { |
| 326 | hr = E_UNEXPECTED; | 341 | hr = E_UNEXPECTED; |
| 327 | ExitOnFailure(hr, "Directory delete cannot delete file: %ls", wzPath); | 342 | DirExitOnFailure(hr, "Directory delete cannot delete file: %ls", wzPath); |
| 328 | } | 343 | } |
| 329 | 344 | ||
| 330 | Assert(S_OK == hr); | 345 | Assert(S_OK == hr); |
| @@ -351,22 +366,22 @@ extern "C" HRESULT DAPI DirGetCurrent( | |||
| 351 | if (psczCurrentDirectory && *psczCurrentDirectory) | 366 | if (psczCurrentDirectory && *psczCurrentDirectory) |
| 352 | { | 367 | { |
| 353 | hr = StrMaxLength(*psczCurrentDirectory, &cch); | 368 | hr = StrMaxLength(*psczCurrentDirectory, &cch); |
| 354 | ExitOnFailure(hr, "Failed to determine size of current directory."); | 369 | DirExitOnFailure(hr, "Failed to determine size of current directory."); |
| 355 | } | 370 | } |
| 356 | 371 | ||
| 357 | DWORD cchRequired = ::GetCurrentDirectoryW(static_cast<DWORD>(cch), 0 == cch ? NULL : *psczCurrentDirectory); | 372 | DWORD cchRequired = ::GetCurrentDirectoryW(static_cast<DWORD>(cch), 0 == cch ? NULL : *psczCurrentDirectory); |
| 358 | if (0 == cchRequired) | 373 | if (0 == cchRequired) |
| 359 | { | 374 | { |
| 360 | ExitWithLastError(hr, "Failed to get current directory."); | 375 | DirExitWithLastError(hr, "Failed to get current directory."); |
| 361 | } | 376 | } |
| 362 | else if (cch < cchRequired) | 377 | else if (cch < cchRequired) |
| 363 | { | 378 | { |
| 364 | hr = StrAlloc(psczCurrentDirectory, cchRequired); | 379 | hr = StrAlloc(psczCurrentDirectory, cchRequired); |
| 365 | ExitOnFailure(hr, "Failed to allocate string for current directory."); | 380 | DirExitOnFailure(hr, "Failed to allocate string for current directory."); |
| 366 | 381 | ||
| 367 | if (!::GetCurrentDirectoryW(cchRequired, *psczCurrentDirectory)) | 382 | if (!::GetCurrentDirectoryW(cchRequired, *psczCurrentDirectory)) |
| 368 | { | 383 | { |
| 369 | ExitWithLastError(hr, "Failed to get current directory using allocated string."); | 384 | DirExitWithLastError(hr, "Failed to get current directory using allocated string."); |
| 370 | } | 385 | } |
| 371 | } | 386 | } |
| 372 | 387 | ||
| @@ -387,7 +402,7 @@ extern "C" HRESULT DAPI DirSetCurrent( | |||
| 387 | 402 | ||
| 388 | if (!::SetCurrentDirectoryW(wzDirectory)) | 403 | if (!::SetCurrentDirectoryW(wzDirectory)) |
| 389 | { | 404 | { |
| 390 | ExitWithLastError(hr, "Failed to set current directory to: %ls", wzDirectory); | 405 | DirExitWithLastError(hr, "Failed to set current directory to: %ls", wzDirectory); |
| 391 | } | 406 | } |
| 392 | 407 | ||
| 393 | LExit: | 408 | LExit: |
