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/fileutil.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/fileutil.cpp')
| -rw-r--r-- | src/dutil/fileutil.cpp | 265 |
1 files changed, 140 insertions, 125 deletions
diff --git a/src/dutil/fileutil.cpp b/src/dutil/fileutil.cpp index 6191ec06..c76017de 100644 --- a/src/dutil/fileutil.cpp +++ b/src/dutil/fileutil.cpp | |||
| @@ -2,6 +2,21 @@ | |||
| 2 | 2 | ||
| 3 | #include "precomp.h" | 3 | #include "precomp.h" |
| 4 | 4 | ||
| 5 | |||
| 6 | // Exit macros | ||
| 7 | #define FileExitOnLastError(x, s, ...) ExitOnLastErrorSource(DUTIL_SOURCE_FILEUTIL, x, s, __VA_ARGS__) | ||
| 8 | #define FileExitOnLastErrorDebugTrace(x, s, ...) ExitOnLastErrorDebugTraceSource(DUTIL_SOURCE_FILEUTIL, x, s, __VA_ARGS__) | ||
| 9 | #define FileExitWithLastError(x, s, ...) ExitWithLastErrorSource(DUTIL_SOURCE_FILEUTIL, x, s, __VA_ARGS__) | ||
| 10 | #define FileExitOnFailure(x, s, ...) ExitOnFailureSource(DUTIL_SOURCE_FILEUTIL, x, s, __VA_ARGS__) | ||
| 11 | #define FileExitOnRootFailure(x, s, ...) ExitOnRootFailureSource(DUTIL_SOURCE_FILEUTIL, x, s, __VA_ARGS__) | ||
| 12 | #define FileExitOnFailureDebugTrace(x, s, ...) ExitOnFailureDebugTraceSource(DUTIL_SOURCE_FILEUTIL, x, s, __VA_ARGS__) | ||
| 13 | #define FileExitOnNull(p, x, e, s, ...) ExitOnNullSource(DUTIL_SOURCE_FILEUTIL, p, x, e, s, __VA_ARGS__) | ||
| 14 | #define FileExitOnNullWithLastError(p, x, s, ...) ExitOnNullWithLastErrorSource(DUTIL_SOURCE_FILEUTIL, p, x, s, __VA_ARGS__) | ||
| 15 | #define FileExitOnNullDebugTrace(p, x, e, s, ...) ExitOnNullDebugTraceSource(DUTIL_SOURCE_FILEUTIL, p, x, e, s, __VA_ARGS__) | ||
| 16 | #define FileExitOnInvalidHandleWithLastError(p, x, s, ...) ExitOnInvalidHandleWithLastErrorSource(DUTIL_SOURCE_FILEUTIL, p, x, s, __VA_ARGS__) | ||
| 17 | #define FileExitOnWin32Error(e, x, s, ...) ExitOnWin32ErrorSource(DUTIL_SOURCE_FILEUTIL, e, x, s, __VA_ARGS__) | ||
| 18 | #define FileExitOnGdipFailure(g, x, s, ...) ExitOnGdipFailureSource(DUTIL_SOURCE_FILEUTIL, g, x, s, __VA_ARGS__) | ||
| 19 | |||
| 5 | // constants | 20 | // constants |
| 6 | 21 | ||
| 7 | const BYTE UTF8BOM[] = {0xEF, 0xBB, 0xBF}; | 22 | const BYTE UTF8BOM[] = {0xEF, 0xBB, 0xBF}; |
| @@ -15,7 +30,7 @@ const LPCWSTR REGISTRY_PENDING_FILE_RENAME_VALUE = L"PendingFileRenameOperations | |||
| 15 | 30 | ||
| 16 | ********************************************************************/ | 31 | ********************************************************************/ |
| 17 | extern "C" LPWSTR DAPI FileFromPath( | 32 | extern "C" LPWSTR DAPI FileFromPath( |
| 18 | __in LPCWSTR wzPath | 33 | __in_z LPCWSTR wzPath |
| 19 | ) | 34 | ) |
| 20 | { | 35 | { |
| 21 | if (!wzPath) | 36 | if (!wzPath) |
| @@ -42,7 +57,7 @@ extern "C" LPWSTR DAPI FileFromPath( | |||
| 42 | 57 | ||
| 43 | ********************************************************************/ | 58 | ********************************************************************/ |
| 44 | extern "C" HRESULT DAPI FileResolvePath( | 59 | extern "C" HRESULT DAPI FileResolvePath( |
| 45 | __in LPCWSTR wzRelativePath, | 60 | __in_z LPCWSTR wzRelativePath, |
| 46 | __out LPWSTR *ppwzFullPath | 61 | __out LPWSTR *ppwzFullPath |
| 47 | ) | 62 | ) |
| 48 | { | 63 | { |
| @@ -63,28 +78,28 @@ extern "C" HRESULT DAPI FileResolvePath( | |||
| 63 | // | 78 | // |
| 64 | cchExpandedPath = MAX_PATH; | 79 | cchExpandedPath = MAX_PATH; |
| 65 | hr = StrAlloc(&pwzExpandedPath, cchExpandedPath); | 80 | hr = StrAlloc(&pwzExpandedPath, cchExpandedPath); |
| 66 | ExitOnFailure(hr, "Failed to allocate space for expanded path."); | 81 | FileExitOnFailure(hr, "Failed to allocate space for expanded path."); |
| 67 | 82 | ||
| 68 | cch = ::ExpandEnvironmentStringsW(wzRelativePath, pwzExpandedPath, cchExpandedPath); | 83 | cch = ::ExpandEnvironmentStringsW(wzRelativePath, pwzExpandedPath, cchExpandedPath); |
| 69 | if (0 == cch) | 84 | if (0 == cch) |
| 70 | { | 85 | { |
| 71 | ExitWithLastError(hr, "Failed to expand environment variables in string: %ls", wzRelativePath); | 86 | FileExitWithLastError(hr, "Failed to expand environment variables in string: %ls", wzRelativePath); |
| 72 | } | 87 | } |
| 73 | else if (cchExpandedPath < cch) | 88 | else if (cchExpandedPath < cch) |
| 74 | { | 89 | { |
| 75 | cchExpandedPath = cch; | 90 | cchExpandedPath = cch; |
| 76 | hr = StrAlloc(&pwzExpandedPath, cchExpandedPath); | 91 | hr = StrAlloc(&pwzExpandedPath, cchExpandedPath); |
| 77 | ExitOnFailure(hr, "Failed to re-allocate more space for expanded path."); | 92 | FileExitOnFailure(hr, "Failed to re-allocate more space for expanded path."); |
| 78 | 93 | ||
| 79 | cch = ::ExpandEnvironmentStringsW(wzRelativePath, pwzExpandedPath, cchExpandedPath); | 94 | cch = ::ExpandEnvironmentStringsW(wzRelativePath, pwzExpandedPath, cchExpandedPath); |
| 80 | if (0 == cch) | 95 | if (0 == cch) |
| 81 | { | 96 | { |
| 82 | ExitWithLastError(hr, "Failed to expand environment variables in string: %ls", wzRelativePath); | 97 | FileExitWithLastError(hr, "Failed to expand environment variables in string: %ls", wzRelativePath); |
| 83 | } | 98 | } |
| 84 | else if (cchExpandedPath < cch) | 99 | else if (cchExpandedPath < cch) |
| 85 | { | 100 | { |
| 86 | hr = HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER); | 101 | hr = HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER); |
| 87 | ExitOnRootFailure(hr, "Failed to allocate buffer for expanded path."); | 102 | FileExitOnRootFailure(hr, "Failed to allocate buffer for expanded path."); |
| 88 | } | 103 | } |
| 89 | } | 104 | } |
| 90 | 105 | ||
| @@ -93,28 +108,28 @@ extern "C" HRESULT DAPI FileResolvePath( | |||
| 93 | // | 108 | // |
| 94 | cchFullPath = MAX_PATH; | 109 | cchFullPath = MAX_PATH; |
| 95 | hr = StrAlloc(&pwzFullPath, cchFullPath); | 110 | hr = StrAlloc(&pwzFullPath, cchFullPath); |
| 96 | ExitOnFailure(hr, "Failed to allocate space for full path."); | 111 | FileExitOnFailure(hr, "Failed to allocate space for full path."); |
| 97 | 112 | ||
| 98 | cch = ::GetFullPathNameW(pwzExpandedPath, cchFullPath, pwzFullPath, &wzFileName); | 113 | cch = ::GetFullPathNameW(pwzExpandedPath, cchFullPath, pwzFullPath, &wzFileName); |
| 99 | if (0 == cch) | 114 | if (0 == cch) |
| 100 | { | 115 | { |
| 101 | ExitWithLastError(hr, "Failed to get full path for string: %ls", pwzExpandedPath); | 116 | FileExitWithLastError(hr, "Failed to get full path for string: %ls", pwzExpandedPath); |
| 102 | } | 117 | } |
| 103 | else if (cchFullPath < cch) | 118 | else if (cchFullPath < cch) |
| 104 | { | 119 | { |
| 105 | cchFullPath = cch; | 120 | cchFullPath = cch; |
| 106 | hr = StrAlloc(&pwzFullPath, cchFullPath); | 121 | hr = StrAlloc(&pwzFullPath, cchFullPath); |
| 107 | ExitOnFailure(hr, "Failed to re-allocate more space for full path."); | 122 | FileExitOnFailure(hr, "Failed to re-allocate more space for full path."); |
| 108 | 123 | ||
| 109 | cch = ::GetFullPathNameW(pwzExpandedPath, cchFullPath, pwzFullPath, &wzFileName); | 124 | cch = ::GetFullPathNameW(pwzExpandedPath, cchFullPath, pwzFullPath, &wzFileName); |
| 110 | if (0 == cch) | 125 | if (0 == cch) |
| 111 | { | 126 | { |
| 112 | ExitWithLastError(hr, "Failed to get full path for string: %ls", pwzExpandedPath); | 127 | FileExitWithLastError(hr, "Failed to get full path for string: %ls", pwzExpandedPath); |
| 113 | } | 128 | } |
| 114 | else if (cchFullPath < cch) | 129 | else if (cchFullPath < cch) |
| 115 | { | 130 | { |
| 116 | hr = HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER); | 131 | hr = HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER); |
| 117 | ExitOnRootFailure(hr, "Failed to allocate buffer for full path."); | 132 | FileExitOnRootFailure(hr, "Failed to allocate buffer for full path."); |
| 118 | } | 133 | } |
| 119 | } | 134 | } |
| 120 | 135 | ||
| @@ -133,7 +148,7 @@ LExit: | |||
| 133 | FileStripExtension - Strip extension from filename | 148 | FileStripExtension - Strip extension from filename |
| 134 | ********************************************************************/ | 149 | ********************************************************************/ |
| 135 | extern "C" HRESULT DAPI FileStripExtension( | 150 | extern "C" HRESULT DAPI FileStripExtension( |
| 136 | __in LPCWSTR wzFileName, | 151 | __in_z LPCWSTR wzFileName, |
| 137 | __out LPWSTR *ppwzFileNameNoExtension | 152 | __out LPWSTR *ppwzFileNameNoExtension |
| 138 | ) | 153 | ) |
| 139 | { | 154 | { |
| @@ -158,14 +173,14 @@ __out LPWSTR *ppwzFileNameNoExtension | |||
| 158 | } | 173 | } |
| 159 | 174 | ||
| 160 | hr = StrAlloc(&pwzFileNameNoExtension, cchFileNameNoExtension); | 175 | hr = StrAlloc(&pwzFileNameNoExtension, cchFileNameNoExtension); |
| 161 | ExitOnFailure(hr, "failed to allocate space for File Name without extension"); | 176 | FileExitOnFailure(hr, "failed to allocate space for File Name without extension"); |
| 162 | 177 | ||
| 163 | // _wsplitpath_s can handle drive/path/filename/extension | 178 | // _wsplitpath_s can handle drive/path/filename/extension |
| 164 | errno_t err = _wsplitpath_s(wzFileName, NULL, NULL, NULL, NULL, pwzFileNameNoExtension, cchFileNameNoExtension, NULL, NULL); | 179 | errno_t err = _wsplitpath_s(wzFileName, NULL, NULL, NULL, NULL, pwzFileNameNoExtension, cchFileNameNoExtension, NULL, NULL); |
| 165 | if (0 != err) | 180 | if (0 != err) |
| 166 | { | 181 | { |
| 167 | hr = E_INVALIDARG; | 182 | hr = E_INVALIDARG; |
| 168 | ExitOnFailure(hr, "failed to parse filename: %ls", wzFileName); | 183 | FileExitOnFailure(hr, "failed to parse filename: %ls", wzFileName); |
| 169 | } | 184 | } |
| 170 | 185 | ||
| 171 | *ppwzFileNameNoExtension = pwzFileNameNoExtension; | 186 | *ppwzFileNameNoExtension = pwzFileNameNoExtension; |
| @@ -182,8 +197,8 @@ LExit: | |||
| 182 | FileChangeExtension - Changes the extension of a filename | 197 | FileChangeExtension - Changes the extension of a filename |
| 183 | ********************************************************************/ | 198 | ********************************************************************/ |
| 184 | extern "C" HRESULT DAPI FileChangeExtension( | 199 | extern "C" HRESULT DAPI FileChangeExtension( |
| 185 | __in LPCWSTR wzFileName, | 200 | __in_z LPCWSTR wzFileName, |
| 186 | __in LPCWSTR wzNewExtension, | 201 | __in_z LPCWSTR wzNewExtension, |
| 187 | __out LPWSTR *ppwzFileNameNewExtension | 202 | __out LPWSTR *ppwzFileNameNewExtension |
| 188 | ) | 203 | ) |
| 189 | { | 204 | { |
| @@ -193,10 +208,10 @@ extern "C" HRESULT DAPI FileChangeExtension( | |||
| 193 | LPWSTR sczFileName = NULL; | 208 | LPWSTR sczFileName = NULL; |
| 194 | 209 | ||
| 195 | hr = FileStripExtension(wzFileName, &sczFileName); | 210 | hr = FileStripExtension(wzFileName, &sczFileName); |
| 196 | ExitOnFailure(hr, "Failed to strip extension from file name: %ls", wzFileName); | 211 | FileExitOnFailure(hr, "Failed to strip extension from file name: %ls", wzFileName); |
| 197 | 212 | ||
| 198 | hr = StrAllocConcat(&sczFileName, wzNewExtension, 0); | 213 | hr = StrAllocConcat(&sczFileName, wzNewExtension, 0); |
| 199 | ExitOnFailure(hr, "Failed to add new extension."); | 214 | FileExitOnFailure(hr, "Failed to add new extension."); |
| 200 | 215 | ||
| 201 | *ppwzFileNameNewExtension = sczFileName; | 216 | *ppwzFileNameNewExtension = sczFileName; |
| 202 | sczFileName = NULL; | 217 | sczFileName = NULL; |
| @@ -238,11 +253,11 @@ extern "C" HRESULT DAPI FileAddSuffixToBaseName( | |||
| 238 | { | 253 | { |
| 239 | // no extension, so add the suffix at the end of the whole name | 254 | // no extension, so add the suffix at the end of the whole name |
| 240 | hr = StrAllocString(&sczNewFileName, wzFileName, 0); | 255 | hr = StrAllocString(&sczNewFileName, wzFileName, 0); |
| 241 | ExitOnFailure(hr, "Failed to allocate new file name."); | 256 | FileExitOnFailure(hr, "Failed to allocate new file name."); |
| 242 | 257 | ||
| 243 | hr = StrAllocConcat(&sczNewFileName, wzSuffix, 0); | 258 | hr = StrAllocConcat(&sczNewFileName, wzSuffix, 0); |
| 244 | } | 259 | } |
| 245 | ExitOnFailure(hr, "Failed to allocate new file name with suffix."); | 260 | FileExitOnFailure(hr, "Failed to allocate new file name with suffix."); |
| 246 | 261 | ||
| 247 | *psczNewFileName = sczNewFileName; | 262 | *psczNewFileName = sczNewFileName; |
| 248 | sczNewFileName = NULL; | 263 | sczNewFileName = NULL; |
| @@ -259,7 +274,7 @@ LExit: | |||
| 259 | 274 | ||
| 260 | ********************************************************************/ | 275 | ********************************************************************/ |
| 261 | extern "C" HRESULT DAPI FileVersion( | 276 | extern "C" HRESULT DAPI FileVersion( |
| 262 | __in LPCWSTR wzFilename, | 277 | __in_z LPCWSTR wzFilename, |
| 263 | __out DWORD *pdwVerMajor, | 278 | __out DWORD *pdwVerMajor, |
| 264 | __out DWORD* pdwVerMinor | 279 | __out DWORD* pdwVerMinor |
| 265 | ) | 280 | ) |
| @@ -274,20 +289,20 @@ extern "C" HRESULT DAPI FileVersion( | |||
| 274 | 289 | ||
| 275 | if (0 == (cbVerBuffer = ::GetFileVersionInfoSizeW(wzFilename, &dwHandle))) | 290 | if (0 == (cbVerBuffer = ::GetFileVersionInfoSizeW(wzFilename, &dwHandle))) |
| 276 | { | 291 | { |
| 277 | ExitOnLastErrorDebugTrace(hr, "failed to get version info for file: %ls", wzFilename); | 292 | FileExitOnLastErrorDebugTrace(hr, "failed to get version info for file: %ls", wzFilename); |
| 278 | } | 293 | } |
| 279 | 294 | ||
| 280 | pVerBuffer = ::GlobalAlloc(GMEM_FIXED, cbVerBuffer); | 295 | pVerBuffer = ::GlobalAlloc(GMEM_FIXED, cbVerBuffer); |
| 281 | ExitOnNullDebugTrace(pVerBuffer, hr, E_OUTOFMEMORY, "failed to allocate version info for file: %ls", wzFilename); | 296 | FileExitOnNullDebugTrace(pVerBuffer, hr, E_OUTOFMEMORY, "failed to allocate version info for file: %ls", wzFilename); |
| 282 | 297 | ||
| 283 | if (!::GetFileVersionInfoW(wzFilename, dwHandle, cbVerBuffer, pVerBuffer)) | 298 | if (!::GetFileVersionInfoW(wzFilename, dwHandle, cbVerBuffer, pVerBuffer)) |
| 284 | { | 299 | { |
| 285 | ExitOnLastErrorDebugTrace(hr, "failed to get version info for file: %ls", wzFilename); | 300 | FileExitOnLastErrorDebugTrace(hr, "failed to get version info for file: %ls", wzFilename); |
| 286 | } | 301 | } |
| 287 | 302 | ||
| 288 | if (!::VerQueryValueW(pVerBuffer, L"\\", (void**)&pvsFileInfo, &cbFileInfo)) | 303 | if (!::VerQueryValueW(pVerBuffer, L"\\", (void**)&pvsFileInfo, &cbFileInfo)) |
| 289 | { | 304 | { |
| 290 | ExitOnLastErrorDebugTrace(hr, "failed to get version value for file: %ls", wzFilename); | 305 | FileExitOnLastErrorDebugTrace(hr, "failed to get version value for file: %ls", wzFilename); |
| 291 | } | 306 | } |
| 292 | 307 | ||
| 293 | *pdwVerMajor = pvsFileInfo->dwFileVersionMS; | 308 | *pdwVerMajor = pvsFileInfo->dwFileVersionMS; |
| @@ -307,7 +322,7 @@ LExit: | |||
| 307 | 322 | ||
| 308 | *******************************************************************/ | 323 | *******************************************************************/ |
| 309 | extern "C" HRESULT DAPI FileVersionFromString( | 324 | extern "C" HRESULT DAPI FileVersionFromString( |
| 310 | __in LPCWSTR wzVersion, | 325 | __in_z LPCWSTR wzVersion, |
| 311 | __out DWORD* pdwVerMajor, | 326 | __out DWORD* pdwVerMajor, |
| 312 | __out DWORD* pdwVerMinor | 327 | __out DWORD* pdwVerMinor |
| 313 | ) | 328 | ) |
| @@ -394,7 +409,7 @@ LExit: | |||
| 394 | 409 | ||
| 395 | *******************************************************************/ | 410 | *******************************************************************/ |
| 396 | extern "C" HRESULT DAPI FileVersionFromStringEx( | 411 | extern "C" HRESULT DAPI FileVersionFromStringEx( |
| 397 | __in LPCWSTR wzVersion, | 412 | __in_z LPCWSTR wzVersion, |
| 398 | __in DWORD cchVersion, | 413 | __in DWORD cchVersion, |
| 399 | __out DWORD64* pqwVersion | 414 | __out DWORD64* pqwVersion |
| 400 | ) | 415 | ) |
| @@ -453,11 +468,11 @@ extern "C" HRESULT DAPI FileVersionFromStringEx( | |||
| 453 | 468 | ||
| 454 | DWORD cchPart; | 469 | DWORD cchPart; |
| 455 | hr = ::PtrdiffTToDWord(wzPartEnd - wzPartBegin, &cchPart); | 470 | hr = ::PtrdiffTToDWord(wzPartEnd - wzPartBegin, &cchPart); |
| 456 | ExitOnFailure(hr, "Version number part was too long."); | 471 | FileExitOnFailure(hr, "Version number part was too long."); |
| 457 | 472 | ||
| 458 | // parse version part | 473 | // parse version part |
| 459 | hr = StrStringToUInt16(wzPartBegin, cchPart, &us); | 474 | hr = StrStringToUInt16(wzPartBegin, cchPart, &us); |
| 460 | ExitOnFailure(hr, "Failed to parse version number part."); | 475 | FileExitOnFailure(hr, "Failed to parse version number part."); |
| 461 | 476 | ||
| 462 | // add part to qword version | 477 | // add part to qword version |
| 463 | qwVersion |= (DWORD64)us << ((3 - iPart) * 16); | 478 | qwVersion |= (DWORD64)us << ((3 - iPart) * 16); |
| @@ -501,7 +516,7 @@ extern "C" HRESULT DAPI FileVersionToStringEx( | |||
| 501 | 516 | ||
| 502 | // Format and return the version string. | 517 | // Format and return the version string. |
| 503 | hr = StrAllocFormatted(psczVersion, L"%u.%u.%u.%u", wMajor, wMinor, wBuild, wRevision); | 518 | hr = StrAllocFormatted(psczVersion, L"%u.%u.%u.%u", wMajor, wMinor, wBuild, wRevision); |
| 504 | ExitOnFailure(hr, "Failed to allocate and format the version number."); | 519 | FileExitOnFailure(hr, "Failed to allocate and format the version number."); |
| 505 | 520 | ||
| 506 | LExit: | 521 | LExit: |
| 507 | return hr; | 522 | return hr; |
| @@ -527,7 +542,7 @@ extern "C" HRESULT DAPI FileSetPointer( | |||
| 527 | liMove.QuadPart = dw64Move; | 542 | liMove.QuadPart = dw64Move; |
| 528 | if (!::SetFilePointerEx(hFile, liMove, &liNewPosition, dwMoveMethod)) | 543 | if (!::SetFilePointerEx(hFile, liMove, &liNewPosition, dwMoveMethod)) |
| 529 | { | 544 | { |
| 530 | ExitWithLastError(hr, "Failed to set file pointer."); | 545 | FileExitWithLastError(hr, "Failed to set file pointer."); |
| 531 | } | 546 | } |
| 532 | 547 | ||
| 533 | if (pdw64NewPosition) | 548 | if (pdw64NewPosition) |
| @@ -545,23 +560,23 @@ LExit: | |||
| 545 | 560 | ||
| 546 | ********************************************************************/ | 561 | ********************************************************************/ |
| 547 | extern "C" HRESULT DAPI FileSize( | 562 | extern "C" HRESULT DAPI FileSize( |
| 548 | __in LPCWSTR pwzFileName, | 563 | __in_z LPCWSTR pwzFileName, |
| 549 | __out LONGLONG* pllSize | 564 | __out LONGLONG* pllSize |
| 550 | ) | 565 | ) |
| 551 | { | 566 | { |
| 552 | HRESULT hr = S_OK; | 567 | HRESULT hr = S_OK; |
| 553 | HANDLE hFile = INVALID_HANDLE_VALUE; | 568 | HANDLE hFile = INVALID_HANDLE_VALUE; |
| 554 | 569 | ||
| 555 | ExitOnNull(pwzFileName, hr, E_INVALIDARG, "Attempted to check filename, but no filename was provided"); | 570 | FileExitOnNull(pwzFileName, hr, E_INVALIDARG, "Attempted to check filename, but no filename was provided"); |
| 556 | 571 | ||
| 557 | hFile = ::CreateFileW(pwzFileName, FILE_READ_ATTRIBUTES, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0); | 572 | hFile = ::CreateFileW(pwzFileName, FILE_READ_ATTRIBUTES, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0); |
| 558 | if (INVALID_HANDLE_VALUE == hFile) | 573 | if (INVALID_HANDLE_VALUE == hFile) |
| 559 | { | 574 | { |
| 560 | ExitWithLastError(hr, "Failed to open file %ls while checking file size", pwzFileName); | 575 | FileExitWithLastError(hr, "Failed to open file %ls while checking file size", pwzFileName); |
| 561 | } | 576 | } |
| 562 | 577 | ||
| 563 | hr = FileSizeByHandle(hFile, pllSize); | 578 | hr = FileSizeByHandle(hFile, pllSize); |
| 564 | ExitOnFailure(hr, "Failed to check size of file %ls by handle", pwzFileName); | 579 | FileExitOnFailure(hr, "Failed to check size of file %ls by handle", pwzFileName); |
| 565 | 580 | ||
| 566 | LExit: | 581 | LExit: |
| 567 | ReleaseFileHandle(hFile); | 582 | ReleaseFileHandle(hFile); |
| @@ -587,7 +602,7 @@ extern "C" HRESULT DAPI FileSizeByHandle( | |||
| 587 | 602 | ||
| 588 | if (!::GetFileSizeEx(hFile, &li)) | 603 | if (!::GetFileSizeEx(hFile, &li)) |
| 589 | { | 604 | { |
| 590 | ExitWithLastError(hr, "Failed to get size of file."); | 605 | FileExitWithLastError(hr, "Failed to get size of file."); |
| 591 | } | 606 | } |
| 592 | 607 | ||
| 593 | *pllSize = li.QuadPart; | 608 | *pllSize = li.QuadPart; |
| @@ -602,7 +617,7 @@ LExit: | |||
| 602 | 617 | ||
| 603 | ********************************************************************/ | 618 | ********************************************************************/ |
| 604 | extern "C" BOOL DAPI FileExistsEx( | 619 | extern "C" BOOL DAPI FileExistsEx( |
| 605 | __in LPCWSTR wzPath, | 620 | __in_z LPCWSTR wzPath, |
| 606 | __out_opt DWORD *pdwAttributes | 621 | __out_opt DWORD *pdwAttributes |
| 607 | ) | 622 | ) |
| 608 | { | 623 | { |
| @@ -655,14 +670,14 @@ extern "C" BOOL DAPI FileExistsAfterRestart( | |||
| 655 | { | 670 | { |
| 656 | ExitFunction1(hr = S_OK); | 671 | ExitFunction1(hr = S_OK); |
| 657 | } | 672 | } |
| 658 | ExitOnFailure(hr, "Failed to open pending file rename registry key."); | 673 | FileExitOnFailure(hr, "Failed to open pending file rename registry key."); |
| 659 | 674 | ||
| 660 | hr = RegReadStringArray(hkPendingFileRename, REGISTRY_PENDING_FILE_RENAME_VALUE, &rgsczRenames, &cRenames); | 675 | hr = RegReadStringArray(hkPendingFileRename, REGISTRY_PENDING_FILE_RENAME_VALUE, &rgsczRenames, &cRenames); |
| 661 | if (E_FILENOTFOUND == hr) | 676 | if (E_FILENOTFOUND == hr) |
| 662 | { | 677 | { |
| 663 | ExitFunction1(hr = S_OK); | 678 | ExitFunction1(hr = S_OK); |
| 664 | } | 679 | } |
| 665 | ExitOnFailure(hr, "Failed to read pending file renames."); | 680 | FileExitOnFailure(hr, "Failed to read pending file renames."); |
| 666 | 681 | ||
| 667 | // The pending file renames array is pairs of source and target paths. We only care | 682 | // The pending file renames array is pairs of source and target paths. We only care |
| 668 | // about checking the source paths so skip the target paths (i += 2). | 683 | // about checking the source paths so skip the target paths (i += 2). |
| @@ -678,7 +693,7 @@ extern "C" BOOL DAPI FileExistsAfterRestart( | |||
| 678 | } | 693 | } |
| 679 | 694 | ||
| 680 | hr = PathCompare(wzPath, wzRename, &nCompare); | 695 | hr = PathCompare(wzPath, wzRename, &nCompare); |
| 681 | ExitOnFailure(hr, "Failed to compare path from pending file rename to check path."); | 696 | FileExitOnFailure(hr, "Failed to compare path from pending file rename to check path."); |
| 682 | 697 | ||
| 683 | if (CSTR_EQUAL == nCompare) | 698 | if (CSTR_EQUAL == nCompare) |
| 684 | { | 699 | { |
| @@ -719,14 +734,14 @@ extern "C" HRESULT DAPI FileRemoveFromPendingRename( | |||
| 719 | { | 734 | { |
| 720 | ExitFunction1(hr = S_OK); | 735 | ExitFunction1(hr = S_OK); |
| 721 | } | 736 | } |
| 722 | ExitOnFailure(hr, "Failed to open pending file rename registry key."); | 737 | FileExitOnFailure(hr, "Failed to open pending file rename registry key."); |
| 723 | 738 | ||
| 724 | hr = RegReadStringArray(hkPendingFileRename, REGISTRY_PENDING_FILE_RENAME_VALUE, &rgsczRenames, &cRenames); | 739 | hr = RegReadStringArray(hkPendingFileRename, REGISTRY_PENDING_FILE_RENAME_VALUE, &rgsczRenames, &cRenames); |
| 725 | if (E_FILENOTFOUND == hr) | 740 | if (E_FILENOTFOUND == hr) |
| 726 | { | 741 | { |
| 727 | ExitFunction1(hr = S_OK); | 742 | ExitFunction1(hr = S_OK); |
| 728 | } | 743 | } |
| 729 | ExitOnFailure(hr, "Failed to read pending file renames."); | 744 | FileExitOnFailure(hr, "Failed to read pending file renames."); |
| 730 | 745 | ||
| 731 | // The pending file renames array is pairs of source and target paths. We only care | 746 | // The pending file renames array is pairs of source and target paths. We only care |
| 732 | // about checking the source paths so skip the target paths (i += 2). | 747 | // about checking the source paths so skip the target paths (i += 2). |
| @@ -742,7 +757,7 @@ extern "C" HRESULT DAPI FileRemoveFromPendingRename( | |||
| 742 | } | 757 | } |
| 743 | 758 | ||
| 744 | hr = PathCompare(wzPath, wzRename, &nCompare); | 759 | hr = PathCompare(wzPath, wzRename, &nCompare); |
| 745 | ExitOnFailure(hr, "Failed to compare path from pending file rename to check path."); | 760 | FileExitOnFailure(hr, "Failed to compare path from pending file rename to check path."); |
| 746 | 761 | ||
| 747 | // If we find our path in the list, null out the source and target slot and | 762 | // If we find our path in the list, null out the source and target slot and |
| 748 | // we'll compact the array next. | 763 | // we'll compact the array next. |
| @@ -772,7 +787,7 @@ extern "C" HRESULT DAPI FileRemoveFromPendingRename( | |||
| 772 | 787 | ||
| 773 | // Write the new array back to the pending file rename key. | 788 | // Write the new array back to the pending file rename key. |
| 774 | hr = RegWriteStringArray(hkPendingFileRename, REGISTRY_PENDING_FILE_RENAME_VALUE, rgsczRenames, cRenames); | 789 | hr = RegWriteStringArray(hkPendingFileRename, REGISTRY_PENDING_FILE_RENAME_VALUE, rgsczRenames, cRenames); |
| 775 | ExitOnFailure(hr, "Failed to update pending file renames."); | 790 | FileExitOnFailure(hr, "Failed to update pending file renames."); |
| 776 | } | 791 | } |
| 777 | 792 | ||
| 778 | LExit: | 793 | LExit: |
| @@ -790,7 +805,7 @@ LExit: | |||
| 790 | extern "C" HRESULT DAPI FileRead( | 805 | extern "C" HRESULT DAPI FileRead( |
| 791 | __deref_out_bcount_full(*pcbDest) LPBYTE* ppbDest, | 806 | __deref_out_bcount_full(*pcbDest) LPBYTE* ppbDest, |
| 792 | __out SIZE_T* pcbDest, | 807 | __out SIZE_T* pcbDest, |
| 793 | __in LPCWSTR wzSrcPath | 808 | __in_z LPCWSTR wzSrcPath |
| 794 | ) | 809 | ) |
| 795 | { | 810 | { |
| 796 | HRESULT hr = FileReadPartial(ppbDest, pcbDest, wzSrcPath, FALSE, 0, 0xFFFFFFFF, FALSE); | 811 | HRESULT hr = FileReadPartial(ppbDest, pcbDest, wzSrcPath, FALSE, 0, 0xFFFFFFFF, FALSE); |
| @@ -819,7 +834,7 @@ extern "C" HRESULT DAPI FileReadEx( | |||
| 819 | extern "C" HRESULT DAPI FileReadUntil( | 834 | extern "C" HRESULT DAPI FileReadUntil( |
| 820 | __deref_out_bcount_full(*pcbDest) LPBYTE* ppbDest, | 835 | __deref_out_bcount_full(*pcbDest) LPBYTE* ppbDest, |
| 821 | __out_range(<=, cbMaxRead) SIZE_T* pcbDest, | 836 | __out_range(<=, cbMaxRead) SIZE_T* pcbDest, |
| 822 | __in LPCWSTR wzSrcPath, | 837 | __in_z LPCWSTR wzSrcPath, |
| 823 | __in DWORD cbMaxRead | 838 | __in DWORD cbMaxRead |
| 824 | ) | 839 | ) |
| 825 | { | 840 | { |
| @@ -835,7 +850,7 @@ extern "C" HRESULT DAPI FileReadUntil( | |||
| 835 | extern "C" HRESULT DAPI FileReadPartial( | 850 | extern "C" HRESULT DAPI FileReadPartial( |
| 836 | __deref_out_bcount_full(*pcbDest) LPBYTE* ppbDest, | 851 | __deref_out_bcount_full(*pcbDest) LPBYTE* ppbDest, |
| 837 | __out_range(<=, cbMaxRead) SIZE_T* pcbDest, | 852 | __out_range(<=, cbMaxRead) SIZE_T* pcbDest, |
| 838 | __in LPCWSTR wzSrcPath, | 853 | __in_z LPCWSTR wzSrcPath, |
| 839 | __in BOOL fSeek, | 854 | __in BOOL fSeek, |
| 840 | __in DWORD cbStartPosition, | 855 | __in DWORD cbStartPosition, |
| 841 | __in DWORD cbMaxRead, | 856 | __in DWORD cbMaxRead, |
| @@ -850,7 +865,7 @@ extern "C" HRESULT DAPI FileReadPartial( | |||
| 850 | (with specified share mode) | 865 | (with specified share mode) |
| 851 | ********************************************************************/ | 866 | ********************************************************************/ |
| 852 | extern "C" HRESULT DAPI FileReadPartialEx( | 867 | extern "C" HRESULT DAPI FileReadPartialEx( |
| 853 | __deref_out_bcount_full(*pcbDest) LPBYTE* ppbDest, | 868 | __deref_inout_bcount_full(*pcbDest) LPBYTE* ppbDest, |
| 854 | __out_range(<=, cbMaxRead) SIZE_T* pcbDest, | 869 | __out_range(<=, cbMaxRead) SIZE_T* pcbDest, |
| 855 | __in_z LPCWSTR wzSrcPath, | 870 | __in_z LPCWSTR wzSrcPath, |
| 856 | __in BOOL fSeek, | 871 | __in BOOL fSeek, |
| @@ -868,10 +883,10 @@ extern "C" HRESULT DAPI FileReadPartialEx( | |||
| 868 | DWORD cbData = 0; | 883 | DWORD cbData = 0; |
| 869 | BYTE* pbData = NULL; | 884 | BYTE* pbData = NULL; |
| 870 | 885 | ||
| 871 | ExitOnNull(pcbDest, hr, E_INVALIDARG, "Invalid argument pcbDest"); | 886 | FileExitOnNull(pcbDest, hr, E_INVALIDARG, "Invalid argument pcbDest"); |
| 872 | ExitOnNull(ppbDest, hr, E_INVALIDARG, "Invalid argument ppbDest"); | 887 | FileExitOnNull(ppbDest, hr, E_INVALIDARG, "Invalid argument ppbDest"); |
| 873 | ExitOnNull(wzSrcPath, hr, E_INVALIDARG, "Invalid argument wzSrcPath"); | 888 | FileExitOnNull(wzSrcPath, hr, E_INVALIDARG, "Invalid argument wzSrcPath"); |
| 874 | ExitOnNull(*wzSrcPath, hr, E_INVALIDARG, "*wzSrcPath is null"); | 889 | FileExitOnNull(*wzSrcPath, hr, E_INVALIDARG, "*wzSrcPath is null"); |
| 875 | 890 | ||
| 876 | hFile = ::CreateFileW(wzSrcPath, GENERIC_READ, dwShareMode, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN, NULL); | 891 | hFile = ::CreateFileW(wzSrcPath, GENERIC_READ, dwShareMode, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN, NULL); |
| 877 | if (INVALID_HANDLE_VALUE == hFile) | 892 | if (INVALID_HANDLE_VALUE == hFile) |
| @@ -881,12 +896,12 @@ extern "C" HRESULT DAPI FileReadPartialEx( | |||
| 881 | { | 896 | { |
| 882 | ExitFunction1(hr = E_FILENOTFOUND); | 897 | ExitFunction1(hr = E_FILENOTFOUND); |
| 883 | } | 898 | } |
| 884 | ExitOnWin32Error(er, hr, "Failed to open file: %ls", wzSrcPath); | 899 | FileExitOnWin32Error(er, hr, "Failed to open file: %ls", wzSrcPath); |
| 885 | } | 900 | } |
| 886 | 901 | ||
| 887 | if (!::GetFileSizeEx(hFile, &liFileSize)) | 902 | if (!::GetFileSizeEx(hFile, &liFileSize)) |
| 888 | { | 903 | { |
| 889 | ExitWithLastError(hr, "Failed to get size of file: %ls", wzSrcPath); | 904 | FileExitWithLastError(hr, "Failed to get size of file: %ls", wzSrcPath); |
| 890 | } | 905 | } |
| 891 | 906 | ||
| 892 | if (fSeek) | 907 | if (fSeek) |
| @@ -894,13 +909,13 @@ extern "C" HRESULT DAPI FileReadPartialEx( | |||
| 894 | if (cbStartPosition > liFileSize.QuadPart) | 909 | if (cbStartPosition > liFileSize.QuadPart) |
| 895 | { | 910 | { |
| 896 | hr = E_INVALIDARG; | 911 | hr = E_INVALIDARG; |
| 897 | ExitOnFailure(hr, "Start position %d bigger than file '%ls' size %d", cbStartPosition, wzSrcPath, liFileSize.QuadPart); | 912 | FileExitOnFailure(hr, "Start position %d bigger than file '%ls' size %llu", cbStartPosition, wzSrcPath, liFileSize.QuadPart); |
| 898 | } | 913 | } |
| 899 | 914 | ||
| 900 | DWORD dwErr = ::SetFilePointer(hFile, cbStartPosition, NULL, FILE_CURRENT); | 915 | DWORD dwErr = ::SetFilePointer(hFile, cbStartPosition, NULL, FILE_CURRENT); |
| 901 | if (INVALID_SET_FILE_POINTER == dwErr) | 916 | if (INVALID_SET_FILE_POINTER == dwErr) |
| 902 | { | 917 | { |
| 903 | ExitOnLastError(hr, "Failed to seek position %d", cbStartPosition); | 918 | FileExitOnLastError(hr, "Failed to seek position %d", cbStartPosition); |
| 904 | } | 919 | } |
| 905 | } | 920 | } |
| 906 | else | 921 | else |
| @@ -918,7 +933,7 @@ extern "C" HRESULT DAPI FileReadPartialEx( | |||
| 918 | if (cbMaxRead < liFileSize.QuadPart - cbStartPosition) | 933 | if (cbMaxRead < liFileSize.QuadPart - cbStartPosition) |
| 919 | { | 934 | { |
| 920 | hr = HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER); | 935 | hr = HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER); |
| 921 | ExitOnRootFailure(hr, "Failed to load file: %ls, too large.", wzSrcPath); | 936 | FileExitOnRootFailure(hr, "Failed to load file: %ls, too large.", wzSrcPath); |
| 922 | } | 937 | } |
| 923 | } | 938 | } |
| 924 | 939 | ||
| @@ -932,7 +947,7 @@ extern "C" HRESULT DAPI FileReadPartialEx( | |||
| 932 | } | 947 | } |
| 933 | 948 | ||
| 934 | LPVOID pv = MemReAlloc(*ppbDest, cbData, TRUE); | 949 | LPVOID pv = MemReAlloc(*ppbDest, cbData, TRUE); |
| 935 | ExitOnNull(pv, hr, E_OUTOFMEMORY, "Failed to re-allocate memory to read in file: %ls", wzSrcPath); | 950 | FileExitOnNull(pv, hr, E_OUTOFMEMORY, "Failed to re-allocate memory to read in file: %ls", wzSrcPath); |
| 936 | 951 | ||
| 937 | pbData = static_cast<BYTE*>(pv); | 952 | pbData = static_cast<BYTE*>(pv); |
| 938 | } | 953 | } |
| @@ -945,7 +960,7 @@ extern "C" HRESULT DAPI FileReadPartialEx( | |||
| 945 | } | 960 | } |
| 946 | 961 | ||
| 947 | pbData = static_cast<BYTE*>(MemAlloc(cbData, TRUE)); | 962 | pbData = static_cast<BYTE*>(MemAlloc(cbData, TRUE)); |
| 948 | ExitOnNull(pbData, hr, E_OUTOFMEMORY, "Failed to allocate memory to read in file: %ls", wzSrcPath); | 963 | FileExitOnNull(pbData, hr, E_OUTOFMEMORY, "Failed to allocate memory to read in file: %ls", wzSrcPath); |
| 949 | } | 964 | } |
| 950 | 965 | ||
| 951 | DWORD cbTotalRead = 0; | 966 | DWORD cbTotalRead = 0; |
| @@ -954,11 +969,11 @@ extern "C" HRESULT DAPI FileReadPartialEx( | |||
| 954 | { | 969 | { |
| 955 | DWORD cbRemaining = 0; | 970 | DWORD cbRemaining = 0; |
| 956 | hr = ::ULongSub(cbData, cbTotalRead, &cbRemaining); | 971 | hr = ::ULongSub(cbData, cbTotalRead, &cbRemaining); |
| 957 | ExitOnFailure(hr, "Underflow calculating remaining buffer size."); | 972 | FileExitOnFailure(hr, "Underflow calculating remaining buffer size."); |
| 958 | 973 | ||
| 959 | if (!::ReadFile(hFile, pbData + cbTotalRead, cbRemaining, &cbRead, NULL)) | 974 | if (!::ReadFile(hFile, pbData + cbTotalRead, cbRemaining, &cbRead, NULL)) |
| 960 | { | 975 | { |
| 961 | ExitWithLastError(hr, "Failed to read from file: %ls", wzSrcPath); | 976 | FileExitWithLastError(hr, "Failed to read from file: %ls", wzSrcPath); |
| 962 | } | 977 | } |
| 963 | 978 | ||
| 964 | cbTotalRead += cbRead; | 979 | cbTotalRead += cbRead; |
| @@ -967,7 +982,7 @@ extern "C" HRESULT DAPI FileReadPartialEx( | |||
| 967 | if (cbTotalRead != cbData) | 982 | if (cbTotalRead != cbData) |
| 968 | { | 983 | { |
| 969 | hr = E_UNEXPECTED; | 984 | hr = E_UNEXPECTED; |
| 970 | ExitOnFailure(hr, "Failed to completely read file: %ls", wzSrcPath); | 985 | FileExitOnFailure(hr, "Failed to completely read file: %ls", wzSrcPath); |
| 971 | } | 986 | } |
| 972 | 987 | ||
| 973 | *ppbDest = pbData; | 988 | *ppbDest = pbData; |
| @@ -999,10 +1014,10 @@ extern "C" HRESULT DAPI FileWrite( | |||
| 999 | 1014 | ||
| 1000 | // Open the file | 1015 | // Open the file |
| 1001 | hFile = ::CreateFileW(pwzFileName, GENERIC_WRITE, FILE_SHARE_READ, NULL, CREATE_ALWAYS, dwFlagsAndAttributes, NULL); | 1016 | hFile = ::CreateFileW(pwzFileName, GENERIC_WRITE, FILE_SHARE_READ, NULL, CREATE_ALWAYS, dwFlagsAndAttributes, NULL); |
| 1002 | ExitOnInvalidHandleWithLastError(hFile, hr, "Failed to open file: %ls", pwzFileName); | 1017 | FileExitOnInvalidHandleWithLastError(hFile, hr, "Failed to open file: %ls", pwzFileName); |
| 1003 | 1018 | ||
| 1004 | hr = FileWriteHandle(hFile, pbData, cbData); | 1019 | hr = FileWriteHandle(hFile, pbData, cbData); |
| 1005 | ExitOnFailure(hr, "Failed to write to file: %ls", pwzFileName); | 1020 | FileExitOnFailure(hr, "Failed to write to file: %ls", pwzFileName); |
| 1006 | 1021 | ||
| 1007 | if (pHandle) | 1022 | if (pHandle) |
| 1008 | { | 1023 | { |
| @@ -1036,7 +1051,7 @@ extern "C" HRESULT DAPI FileWriteHandle( | |||
| 1036 | { | 1051 | { |
| 1037 | if (!::WriteFile(hFile, pbData + cbTotal, (DWORD)(cbData - cbTotal), &cbDataWritten, NULL)) | 1052 | if (!::WriteFile(hFile, pbData + cbTotal, (DWORD)(cbData - cbTotal), &cbDataWritten, NULL)) |
| 1038 | { | 1053 | { |
| 1039 | ExitOnLastError(hr, "Failed to write data to file handle."); | 1054 | FileExitOnLastError(hr, "Failed to write data to file handle."); |
| 1040 | } | 1055 | } |
| 1041 | 1056 | ||
| 1042 | cbTotal += cbDataWritten; | 1057 | cbTotal += cbDataWritten; |
| @@ -1068,13 +1083,13 @@ extern "C" HRESULT DAPI FileCopyUsingHandles( | |||
| 1068 | cbRead = static_cast<DWORD>((0 == cbCopy) ? countof(rgbData) : min(countof(rgbData), cbCopy - cbTotalCopied)); | 1083 | cbRead = static_cast<DWORD>((0 == cbCopy) ? countof(rgbData) : min(countof(rgbData), cbCopy - cbTotalCopied)); |
| 1069 | if (!::ReadFile(hSource, rgbData, cbRead, &cbRead, NULL)) | 1084 | if (!::ReadFile(hSource, rgbData, cbRead, &cbRead, NULL)) |
| 1070 | { | 1085 | { |
| 1071 | ExitWithLastError(hr, "Failed to read from source."); | 1086 | FileExitWithLastError(hr, "Failed to read from source."); |
| 1072 | } | 1087 | } |
| 1073 | 1088 | ||
| 1074 | if (cbRead) | 1089 | if (cbRead) |
| 1075 | { | 1090 | { |
| 1076 | hr = FileWriteHandle(hTarget, rgbData, cbRead); | 1091 | hr = FileWriteHandle(hTarget, rgbData, cbRead); |
| 1077 | ExitOnFailure(hr, "Failed to write to target."); | 1092 | FileExitOnFailure(hr, "Failed to write to target."); |
| 1078 | } | 1093 | } |
| 1079 | 1094 | ||
| 1080 | cbTotalCopied += cbRead; | 1095 | cbTotalCopied += cbRead; |
| @@ -1095,8 +1110,8 @@ LExit: | |||
| 1095 | 1110 | ||
| 1096 | *******************************************************************/ | 1111 | *******************************************************************/ |
| 1097 | extern "C" HRESULT DAPI FileEnsureCopy( | 1112 | extern "C" HRESULT DAPI FileEnsureCopy( |
| 1098 | __in LPCWSTR wzSource, | 1113 | __in_z LPCWSTR wzSource, |
| 1099 | __in LPCWSTR wzTarget, | 1114 | __in_z LPCWSTR wzTarget, |
| 1100 | __in BOOL fOverwrite | 1115 | __in BOOL fOverwrite |
| 1101 | ) | 1116 | ) |
| 1102 | { | 1117 | { |
| @@ -1132,12 +1147,12 @@ extern "C" HRESULT DAPI FileEnsureCopy( | |||
| 1132 | *pwzLastSlash = L'\0'; // null terminate | 1147 | *pwzLastSlash = L'\0'; // null terminate |
| 1133 | hr = DirEnsureExists(wzTarget, NULL); | 1148 | hr = DirEnsureExists(wzTarget, NULL); |
| 1134 | *pwzLastSlash = L'\\'; // now put the slash back | 1149 | *pwzLastSlash = L'\\'; // now put the slash back |
| 1135 | ExitOnFailureDebugTrace(hr, "failed to create directory while copying file: '%ls' to: '%ls'", wzSource, wzTarget); | 1150 | FileExitOnFailureDebugTrace(hr, "failed to create directory while copying file: '%ls' to: '%ls'", wzSource, wzTarget); |
| 1136 | 1151 | ||
| 1137 | // try to copy again | 1152 | // try to copy again |
| 1138 | if (!::CopyFileW(wzSource, wzTarget, fOverwrite)) | 1153 | if (!::CopyFileW(wzSource, wzTarget, fOverwrite)) |
| 1139 | { | 1154 | { |
| 1140 | ExitOnLastErrorDebugTrace(hr, "failed to copy file: '%ls' to: '%ls'", wzSource, wzTarget); | 1155 | FileExitOnLastErrorDebugTrace(hr, "failed to copy file: '%ls' to: '%ls'", wzSource, wzTarget); |
| 1141 | } | 1156 | } |
| 1142 | } | 1157 | } |
| 1143 | else // no path was specified so just return the error | 1158 | else // no path was specified so just return the error |
| @@ -1186,7 +1201,7 @@ extern "C" HRESULT DAPI FileEnsureCopyWithRetry( | |||
| 1186 | break; // no reason to retry these errors. | 1201 | break; // no reason to retry these errors. |
| 1187 | } | 1202 | } |
| 1188 | } | 1203 | } |
| 1189 | ExitOnFailure(hr, "Failed to copy file: '%ls' to: '%ls' after %u retries.", wzSource, wzTarget, i); | 1204 | FileExitOnFailure(hr, "Failed to copy file: '%ls' to: '%ls' after %u retries.", wzSource, wzTarget, i); |
| 1190 | 1205 | ||
| 1191 | LExit: | 1206 | LExit: |
| 1192 | return hr; | 1207 | return hr; |
| @@ -1198,8 +1213,8 @@ LExit: | |||
| 1198 | 1213 | ||
| 1199 | *******************************************************************/ | 1214 | *******************************************************************/ |
| 1200 | extern "C" HRESULT DAPI FileEnsureMove( | 1215 | extern "C" HRESULT DAPI FileEnsureMove( |
| 1201 | __in LPCWSTR wzSource, | 1216 | __in_z LPCWSTR wzSource, |
| 1202 | __in LPCWSTR wzTarget, | 1217 | __in_z LPCWSTR wzTarget, |
| 1203 | __in BOOL fOverwrite, | 1218 | __in BOOL fOverwrite, |
| 1204 | __in BOOL fAllowCopy | 1219 | __in BOOL fAllowCopy |
| 1205 | ) | 1220 | ) |
| @@ -1260,12 +1275,12 @@ extern "C" HRESULT DAPI FileEnsureMove( | |||
| 1260 | *pwzLastSlash = L'\0'; // null terminate | 1275 | *pwzLastSlash = L'\0'; // null terminate |
| 1261 | hr = DirEnsureExists(wzTarget, NULL); | 1276 | hr = DirEnsureExists(wzTarget, NULL); |
| 1262 | *pwzLastSlash = L'\\'; // now put the slash back | 1277 | *pwzLastSlash = L'\\'; // now put the slash back |
| 1263 | ExitOnFailureDebugTrace(hr, "failed to create directory while moving file: '%ls' to: '%ls'", wzSource, wzTarget); | 1278 | FileExitOnFailureDebugTrace(hr, "failed to create directory while moving file: '%ls' to: '%ls'", wzSource, wzTarget); |
| 1264 | 1279 | ||
| 1265 | // try to move again | 1280 | // try to move again |
| 1266 | if (!::MoveFileExW(wzSource, wzTarget, dwFlags)) | 1281 | if (!::MoveFileExW(wzSource, wzTarget, dwFlags)) |
| 1267 | { | 1282 | { |
| 1268 | ExitOnLastErrorDebugTrace(hr, "failed to move file: '%ls' to: '%ls'", wzSource, wzTarget); | 1283 | FileExitOnLastErrorDebugTrace(hr, "failed to move file: '%ls' to: '%ls'", wzSource, wzTarget); |
| 1269 | } | 1284 | } |
| 1270 | } | 1285 | } |
| 1271 | else // no path was specified so just return the error | 1286 | else // no path was specified so just return the error |
| @@ -1310,7 +1325,7 @@ extern "C" HRESULT DAPI FileEnsureMoveWithRetry( | |||
| 1310 | 1325 | ||
| 1311 | hr = FileEnsureMove(wzSource, wzTarget, fOverwrite, fAllowCopy); | 1326 | hr = FileEnsureMove(wzSource, wzTarget, fOverwrite, fAllowCopy); |
| 1312 | } | 1327 | } |
| 1313 | ExitOnFailure(hr, "Failed to move file: '%ls' to: '%ls' after %u retries.", wzSource, wzTarget, i); | 1328 | FileExitOnFailure(hr, "Failed to move file: '%ls' to: '%ls' after %u retries.", wzSource, wzTarget, i); |
| 1314 | 1329 | ||
| 1315 | LExit: | 1330 | LExit: |
| 1316 | return hr; | 1331 | return hr; |
| @@ -1323,8 +1338,8 @@ LExit: | |||
| 1323 | NOTE: uses ANSI functions internally so it is Win9x safe | 1338 | NOTE: uses ANSI functions internally so it is Win9x safe |
| 1324 | ********************************************************************/ | 1339 | ********************************************************************/ |
| 1325 | extern "C" HRESULT DAPI FileCreateTemp( | 1340 | extern "C" HRESULT DAPI FileCreateTemp( |
| 1326 | __in LPCWSTR wzPrefix, | 1341 | __in_z LPCWSTR wzPrefix, |
| 1327 | __in LPCWSTR wzExtension, | 1342 | __in_z LPCWSTR wzExtension, |
| 1328 | __deref_opt_out_z LPWSTR* ppwzTempFile, | 1343 | __deref_opt_out_z LPWSTR* ppwzTempFile, |
| 1329 | __out_opt HANDLE* phTempFile | 1344 | __out_opt HANDLE* phTempFile |
| 1330 | ) | 1345 | ) |
| @@ -1340,13 +1355,13 @@ extern "C" HRESULT DAPI FileCreateTemp( | |||
| 1340 | int i = 0; | 1355 | int i = 0; |
| 1341 | 1356 | ||
| 1342 | hr = StrAnsiAlloc(&pszTempPath, cchTempPath); | 1357 | hr = StrAnsiAlloc(&pszTempPath, cchTempPath); |
| 1343 | ExitOnFailure(hr, "failed to allocate memory for the temp path"); | 1358 | FileExitOnFailure(hr, "failed to allocate memory for the temp path"); |
| 1344 | ::GetTempPathA(cchTempPath, pszTempPath); | 1359 | ::GetTempPathA(cchTempPath, pszTempPath); |
| 1345 | 1360 | ||
| 1346 | for (i = 0; i < 1000 && INVALID_HANDLE_VALUE == hTempFile; ++i) | 1361 | for (i = 0; i < 1000 && INVALID_HANDLE_VALUE == hTempFile; ++i) |
| 1347 | { | 1362 | { |
| 1348 | hr = StrAnsiAllocFormatted(&pszTempFile, "%s%ls%05d.%ls", pszTempPath, wzPrefix, i, wzExtension); | 1363 | hr = StrAnsiAllocFormatted(&pszTempFile, "%s%ls%05d.%ls", pszTempPath, wzPrefix, i, wzExtension); |
| 1349 | ExitOnFailure(hr, "failed to allocate memory for log file"); | 1364 | FileExitOnFailure(hr, "failed to allocate memory for log file"); |
| 1350 | 1365 | ||
| 1351 | hTempFile = ::CreateFileA(pszTempFile, GENERIC_WRITE, FILE_SHARE_READ, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL); | 1366 | hTempFile = ::CreateFileA(pszTempFile, GENERIC_WRITE, FILE_SHARE_READ, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL); |
| 1352 | if (INVALID_HANDLE_VALUE == hTempFile) | 1367 | if (INVALID_HANDLE_VALUE == hTempFile) |
| @@ -1358,7 +1373,7 @@ extern "C" HRESULT DAPI FileCreateTemp( | |||
| 1358 | hr = S_OK; | 1373 | hr = S_OK; |
| 1359 | continue; | 1374 | continue; |
| 1360 | } | 1375 | } |
| 1361 | ExitOnFailureDebugTrace(hr, "failed to create file: %ls", pszTempFile); | 1376 | FileExitOnFailureDebugTrace(hr, "failed to create file: %hs", pszTempFile); |
| 1362 | } | 1377 | } |
| 1363 | } | 1378 | } |
| 1364 | 1379 | ||
| @@ -1387,8 +1402,8 @@ LExit: | |||
| 1387 | 1402 | ||
| 1388 | *******************************************************************/ | 1403 | *******************************************************************/ |
| 1389 | extern "C" HRESULT DAPI FileCreateTempW( | 1404 | extern "C" HRESULT DAPI FileCreateTempW( |
| 1390 | __in LPCWSTR wzPrefix, | 1405 | __in_z LPCWSTR wzPrefix, |
| 1391 | __in LPCWSTR wzExtension, | 1406 | __in_z LPCWSTR wzExtension, |
| 1392 | __deref_opt_out_z LPWSTR* ppwzTempFile, | 1407 | __deref_opt_out_z LPWSTR* ppwzTempFile, |
| 1393 | __out_opt HANDLE* phTempFile | 1408 | __out_opt HANDLE* phTempFile |
| 1394 | ) | 1409 | ) |
| @@ -1405,13 +1420,13 @@ extern "C" HRESULT DAPI FileCreateTempW( | |||
| 1405 | 1420 | ||
| 1406 | if (!::GetTempPathW(cchTempPath, wzTempPath)) | 1421 | if (!::GetTempPathW(cchTempPath, wzTempPath)) |
| 1407 | { | 1422 | { |
| 1408 | ExitOnLastError(hr, "failed to get temp path"); | 1423 | FileExitOnLastError(hr, "failed to get temp path"); |
| 1409 | } | 1424 | } |
| 1410 | 1425 | ||
| 1411 | for (i = 0; i < 1000 && INVALID_HANDLE_VALUE == hTempFile; ++i) | 1426 | for (i = 0; i < 1000 && INVALID_HANDLE_VALUE == hTempFile; ++i) |
| 1412 | { | 1427 | { |
| 1413 | hr = StrAllocFormatted(&pwzTempFile, L"%s%s%05d.%s", wzTempPath, wzPrefix, i, wzExtension); | 1428 | hr = StrAllocFormatted(&pwzTempFile, L"%s%s%05d.%s", wzTempPath, wzPrefix, i, wzExtension); |
| 1414 | ExitOnFailure(hr, "failed to allocate memory for temp filename"); | 1429 | FileExitOnFailure(hr, "failed to allocate memory for temp filename"); |
| 1415 | 1430 | ||
| 1416 | hTempFile = ::CreateFileW(pwzTempFile, GENERIC_WRITE, FILE_SHARE_READ, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL); | 1431 | hTempFile = ::CreateFileW(pwzTempFile, GENERIC_WRITE, FILE_SHARE_READ, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL); |
| 1417 | if (INVALID_HANDLE_VALUE == hTempFile) | 1432 | if (INVALID_HANDLE_VALUE == hTempFile) |
| @@ -1423,7 +1438,7 @@ extern "C" HRESULT DAPI FileCreateTempW( | |||
| 1423 | hr = S_OK; | 1438 | hr = S_OK; |
| 1424 | continue; | 1439 | continue; |
| 1425 | } | 1440 | } |
| 1426 | ExitOnFailureDebugTrace(hr, "failed to create file: %ls", pwzTempFile); | 1441 | FileExitOnFailureDebugTrace(hr, "failed to create file: %ls", pwzTempFile); |
| 1427 | } | 1442 | } |
| 1428 | } | 1443 | } |
| 1429 | 1444 | ||
| @@ -1452,8 +1467,8 @@ LExit: | |||
| 1452 | 1467 | ||
| 1453 | ********************************************************************/ | 1468 | ********************************************************************/ |
| 1454 | extern "C" HRESULT DAPI FileIsSame( | 1469 | extern "C" HRESULT DAPI FileIsSame( |
| 1455 | __in LPCWSTR wzFile1, | 1470 | __in_z LPCWSTR wzFile1, |
| 1456 | __in LPCWSTR wzFile2, | 1471 | __in_z LPCWSTR wzFile2, |
| 1457 | __out LPBOOL lpfSameFile | 1472 | __out LPBOOL lpfSameFile |
| 1458 | ) | 1473 | ) |
| 1459 | { | 1474 | { |
| @@ -1464,19 +1479,19 @@ extern "C" HRESULT DAPI FileIsSame( | |||
| 1464 | BY_HANDLE_FILE_INFORMATION fileInfo2 = { }; | 1479 | BY_HANDLE_FILE_INFORMATION fileInfo2 = { }; |
| 1465 | 1480 | ||
| 1466 | hFile1 = ::CreateFileW(wzFile1, FILE_READ_ATTRIBUTES, FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL); | 1481 | hFile1 = ::CreateFileW(wzFile1, FILE_READ_ATTRIBUTES, FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL); |
| 1467 | ExitOnInvalidHandleWithLastError(hFile1, hr, "Failed to open file 1. File = '%ls'", wzFile1); | 1482 | FileExitOnInvalidHandleWithLastError(hFile1, hr, "Failed to open file 1. File = '%ls'", wzFile1); |
| 1468 | 1483 | ||
| 1469 | hFile2 = ::CreateFileW(wzFile2, FILE_READ_ATTRIBUTES, FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL); | 1484 | hFile2 = ::CreateFileW(wzFile2, FILE_READ_ATTRIBUTES, FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL); |
| 1470 | ExitOnInvalidHandleWithLastError(hFile2, hr, "Failed to open file 2. File = '%ls'", wzFile2); | 1485 | FileExitOnInvalidHandleWithLastError(hFile2, hr, "Failed to open file 2. File = '%ls'", wzFile2); |
| 1471 | 1486 | ||
| 1472 | if (!::GetFileInformationByHandle(hFile1, &fileInfo1)) | 1487 | if (!::GetFileInformationByHandle(hFile1, &fileInfo1)) |
| 1473 | { | 1488 | { |
| 1474 | ExitWithLastError(hr, "Failed to get information for file 1. File = '%ls'", wzFile1); | 1489 | FileExitWithLastError(hr, "Failed to get information for file 1. File = '%ls'", wzFile1); |
| 1475 | } | 1490 | } |
| 1476 | 1491 | ||
| 1477 | if (!::GetFileInformationByHandle(hFile2, &fileInfo2)) | 1492 | if (!::GetFileInformationByHandle(hFile2, &fileInfo2)) |
| 1478 | { | 1493 | { |
| 1479 | ExitWithLastError(hr, "Failed to get information for file 2. File = '%ls'", wzFile2); | 1494 | FileExitWithLastError(hr, "Failed to get information for file 2. File = '%ls'", wzFile2); |
| 1480 | } | 1495 | } |
| 1481 | 1496 | ||
| 1482 | *lpfSameFile = fileInfo1.dwVolumeSerialNumber == fileInfo2.dwVolumeSerialNumber && | 1497 | *lpfSameFile = fileInfo1.dwVolumeSerialNumber == fileInfo2.dwVolumeSerialNumber && |
| @@ -1495,7 +1510,7 @@ LExit: | |||
| 1495 | hidden, or system attributes if necessary. | 1510 | hidden, or system attributes if necessary. |
| 1496 | ********************************************************************/ | 1511 | ********************************************************************/ |
| 1497 | extern "C" HRESULT DAPI FileEnsureDelete( | 1512 | extern "C" HRESULT DAPI FileEnsureDelete( |
| 1498 | __in LPCWSTR wzFile | 1513 | __in_z LPCWSTR wzFile |
| 1499 | ) | 1514 | ) |
| 1500 | { | 1515 | { |
| 1501 | HRESULT hr = S_OK; | 1516 | HRESULT hr = S_OK; |
| @@ -1507,13 +1522,13 @@ extern "C" HRESULT DAPI FileEnsureDelete( | |||
| 1507 | { | 1522 | { |
| 1508 | if (!::SetFileAttributesW(wzFile, FILE_ATTRIBUTE_NORMAL)) | 1523 | if (!::SetFileAttributesW(wzFile, FILE_ATTRIBUTE_NORMAL)) |
| 1509 | { | 1524 | { |
| 1510 | ExitOnLastError(hr, "Failed to remove attributes from file: %ls", wzFile); | 1525 | FileExitOnLastError(hr, "Failed to remove attributes from file: %ls", wzFile); |
| 1511 | } | 1526 | } |
| 1512 | } | 1527 | } |
| 1513 | 1528 | ||
| 1514 | if (!::DeleteFileW(wzFile)) | 1529 | if (!::DeleteFileW(wzFile)) |
| 1515 | { | 1530 | { |
| 1516 | ExitOnLastError(hr, "Failed to delete file: %ls", wzFile); | 1531 | FileExitOnLastError(hr, "Failed to delete file: %ls", wzFile); |
| 1517 | } | 1532 | } |
| 1518 | } | 1533 | } |
| 1519 | 1534 | ||
| @@ -1525,7 +1540,7 @@ LExit: | |||
| 1525 | FileGetTime - Gets the file time of a specified file | 1540 | FileGetTime - Gets the file time of a specified file |
| 1526 | ********************************************************************/ | 1541 | ********************************************************************/ |
| 1527 | extern "C" HRESULT DAPI FileGetTime( | 1542 | extern "C" HRESULT DAPI FileGetTime( |
| 1528 | __in LPCWSTR wzFile, | 1543 | __in_z LPCWSTR wzFile, |
| 1529 | __out_opt LPFILETIME lpCreationTime, | 1544 | __out_opt LPFILETIME lpCreationTime, |
| 1530 | __out_opt LPFILETIME lpLastAccessTime, | 1545 | __out_opt LPFILETIME lpLastAccessTime, |
| 1531 | __out_opt LPFILETIME lpLastWriteTime | 1546 | __out_opt LPFILETIME lpLastWriteTime |
| @@ -1535,11 +1550,11 @@ extern "C" HRESULT DAPI FileGetTime( | |||
| 1535 | HANDLE hFile = NULL; | 1550 | HANDLE hFile = NULL; |
| 1536 | 1551 | ||
| 1537 | hFile = ::CreateFileW(wzFile, FILE_READ_ATTRIBUTES, FILE_SHARE_WRITE | FILE_SHARE_READ | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, 0, NULL); | 1552 | hFile = ::CreateFileW(wzFile, FILE_READ_ATTRIBUTES, FILE_SHARE_WRITE | FILE_SHARE_READ | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, 0, NULL); |
| 1538 | ExitOnInvalidHandleWithLastError(hFile, hr, "Failed to open file. File = '%ls'", wzFile); | 1553 | FileExitOnInvalidHandleWithLastError(hFile, hr, "Failed to open file. File = '%ls'", wzFile); |
| 1539 | 1554 | ||
| 1540 | if (!::GetFileTime(hFile, lpCreationTime, lpLastAccessTime, lpLastWriteTime)) | 1555 | if (!::GetFileTime(hFile, lpCreationTime, lpLastAccessTime, lpLastWriteTime)) |
| 1541 | { | 1556 | { |
| 1542 | ExitWithLastError(hr, "Failed to get file time for file. File = '%ls'", wzFile); | 1557 | FileExitWithLastError(hr, "Failed to get file time for file. File = '%ls'", wzFile); |
| 1543 | } | 1558 | } |
| 1544 | 1559 | ||
| 1545 | LExit: | 1560 | LExit: |
| @@ -1551,7 +1566,7 @@ LExit: | |||
| 1551 | FileSetTime - Sets the file time of a specified file | 1566 | FileSetTime - Sets the file time of a specified file |
| 1552 | ********************************************************************/ | 1567 | ********************************************************************/ |
| 1553 | extern "C" HRESULT DAPI FileSetTime( | 1568 | extern "C" HRESULT DAPI FileSetTime( |
| 1554 | __in LPCWSTR wzFile, | 1569 | __in_z LPCWSTR wzFile, |
| 1555 | __in_opt const FILETIME *lpCreationTime, | 1570 | __in_opt const FILETIME *lpCreationTime, |
| 1556 | __in_opt const FILETIME *lpLastAccessTime, | 1571 | __in_opt const FILETIME *lpLastAccessTime, |
| 1557 | __in_opt const FILETIME *lpLastWriteTime | 1572 | __in_opt const FILETIME *lpLastWriteTime |
| @@ -1561,11 +1576,11 @@ extern "C" HRESULT DAPI FileSetTime( | |||
| 1561 | HANDLE hFile = NULL; | 1576 | HANDLE hFile = NULL; |
| 1562 | 1577 | ||
| 1563 | hFile = ::CreateFileW(wzFile, FILE_WRITE_ATTRIBUTES, FILE_SHARE_WRITE | FILE_SHARE_READ | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, 0, NULL); | 1578 | hFile = ::CreateFileW(wzFile, FILE_WRITE_ATTRIBUTES, FILE_SHARE_WRITE | FILE_SHARE_READ | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, 0, NULL); |
| 1564 | ExitOnInvalidHandleWithLastError(hFile, hr, "Failed to open file. File = '%ls'", wzFile); | 1579 | FileExitOnInvalidHandleWithLastError(hFile, hr, "Failed to open file. File = '%ls'", wzFile); |
| 1565 | 1580 | ||
| 1566 | if (!::SetFileTime(hFile, lpCreationTime, lpLastAccessTime, lpLastWriteTime)) | 1581 | if (!::SetFileTime(hFile, lpCreationTime, lpLastAccessTime, lpLastWriteTime)) |
| 1567 | { | 1582 | { |
| 1568 | ExitWithLastError(hr, "Failed to set file time for file. File = '%ls'", wzFile); | 1583 | FileExitWithLastError(hr, "Failed to set file time for file. File = '%ls'", wzFile); |
| 1569 | } | 1584 | } |
| 1570 | 1585 | ||
| 1571 | LExit: | 1586 | LExit: |
| @@ -1578,7 +1593,7 @@ LExit: | |||
| 1578 | creation time of the file | 1593 | creation time of the file |
| 1579 | ********************************************************************/ | 1594 | ********************************************************************/ |
| 1580 | extern "C" HRESULT DAPI FileResetTime( | 1595 | extern "C" HRESULT DAPI FileResetTime( |
| 1581 | __in LPCWSTR wzFile | 1596 | __in_z LPCWSTR wzFile |
| 1582 | ) | 1597 | ) |
| 1583 | { | 1598 | { |
| 1584 | HRESULT hr = S_OK; | 1599 | HRESULT hr = S_OK; |
| @@ -1586,16 +1601,16 @@ extern "C" HRESULT DAPI FileResetTime( | |||
| 1586 | FILETIME ftCreateTime; | 1601 | FILETIME ftCreateTime; |
| 1587 | 1602 | ||
| 1588 | hFile = ::CreateFileW(wzFile, FILE_WRITE_ATTRIBUTES | FILE_READ_ATTRIBUTES, FILE_SHARE_WRITE | FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL); | 1603 | hFile = ::CreateFileW(wzFile, FILE_WRITE_ATTRIBUTES | FILE_READ_ATTRIBUTES, FILE_SHARE_WRITE | FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL); |
| 1589 | ExitOnInvalidHandleWithLastError(hFile, hr, "Failed to open file. File = '%ls'", wzFile); | 1604 | FileExitOnInvalidHandleWithLastError(hFile, hr, "Failed to open file. File = '%ls'", wzFile); |
| 1590 | 1605 | ||
| 1591 | if (!::GetFileTime(hFile, &ftCreateTime, NULL, NULL)) | 1606 | if (!::GetFileTime(hFile, &ftCreateTime, NULL, NULL)) |
| 1592 | { | 1607 | { |
| 1593 | ExitWithLastError(hr, "Failed to get file time for file. File = '%ls'", wzFile); | 1608 | FileExitWithLastError(hr, "Failed to get file time for file. File = '%ls'", wzFile); |
| 1594 | } | 1609 | } |
| 1595 | 1610 | ||
| 1596 | if (!::SetFileTime(hFile, NULL, NULL, &ftCreateTime)) | 1611 | if (!::SetFileTime(hFile, NULL, NULL, &ftCreateTime)) |
| 1597 | { | 1612 | { |
| 1598 | ExitWithLastError(hr, "Failed to reset file time for file. File = '%ls'", wzFile); | 1613 | FileExitWithLastError(hr, "Failed to reset file time for file. File = '%ls'", wzFile); |
| 1599 | } | 1614 | } |
| 1600 | 1615 | ||
| 1601 | LExit: | 1616 | LExit: |
| @@ -1609,7 +1624,7 @@ LExit: | |||
| 1609 | 1624 | ||
| 1610 | *******************************************************************/ | 1625 | *******************************************************************/ |
| 1611 | extern "C" HRESULT DAPI FileExecutableArchitecture( | 1626 | extern "C" HRESULT DAPI FileExecutableArchitecture( |
| 1612 | __in LPCWSTR wzFile, | 1627 | __in_z LPCWSTR wzFile, |
| 1613 | __out FILE_ARCHITECTURE *pArchitecture | 1628 | __out FILE_ARCHITECTURE *pArchitecture |
| 1614 | ) | 1629 | ) |
| 1615 | { | 1630 | { |
| @@ -1623,34 +1638,34 @@ extern "C" HRESULT DAPI FileExecutableArchitecture( | |||
| 1623 | hFile = ::CreateFileW(wzFile, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0); | 1638 | hFile = ::CreateFileW(wzFile, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0); |
| 1624 | if (hFile == INVALID_HANDLE_VALUE) | 1639 | if (hFile == INVALID_HANDLE_VALUE) |
| 1625 | { | 1640 | { |
| 1626 | ExitWithLastError(hr, "Failed to open file: %ls", wzFile); | 1641 | FileExitWithLastError(hr, "Failed to open file: %ls", wzFile); |
| 1627 | } | 1642 | } |
| 1628 | 1643 | ||
| 1629 | if (!::ReadFile(hFile, &DosImageHeader, sizeof(DosImageHeader), &cbRead, NULL)) | 1644 | if (!::ReadFile(hFile, &DosImageHeader, sizeof(DosImageHeader), &cbRead, NULL)) |
| 1630 | { | 1645 | { |
| 1631 | ExitWithLastError(hr, "Failed to read DOS header from file: %ls", wzFile); | 1646 | FileExitWithLastError(hr, "Failed to read DOS header from file: %ls", wzFile); |
| 1632 | } | 1647 | } |
| 1633 | 1648 | ||
| 1634 | if (DosImageHeader.e_magic != IMAGE_DOS_SIGNATURE) | 1649 | if (DosImageHeader.e_magic != IMAGE_DOS_SIGNATURE) |
| 1635 | { | 1650 | { |
| 1636 | hr = HRESULT_FROM_WIN32(ERROR_BAD_FORMAT); | 1651 | hr = HRESULT_FROM_WIN32(ERROR_BAD_FORMAT); |
| 1637 | ExitOnRootFailure(hr, "Read invalid DOS header from file: %ls", wzFile); | 1652 | FileExitOnRootFailure(hr, "Read invalid DOS header from file: %ls", wzFile); |
| 1638 | } | 1653 | } |
| 1639 | 1654 | ||
| 1640 | if (INVALID_SET_FILE_POINTER == ::SetFilePointer(hFile, DosImageHeader.e_lfanew, NULL, FILE_BEGIN)) | 1655 | if (INVALID_SET_FILE_POINTER == ::SetFilePointer(hFile, DosImageHeader.e_lfanew, NULL, FILE_BEGIN)) |
| 1641 | { | 1656 | { |
| 1642 | ExitWithLastError(hr, "Failed to seek the NT header in file: %ls", wzFile); | 1657 | FileExitWithLastError(hr, "Failed to seek the NT header in file: %ls", wzFile); |
| 1643 | } | 1658 | } |
| 1644 | 1659 | ||
| 1645 | if (!::ReadFile(hFile, &NtImageHeader, sizeof(NtImageHeader), &cbRead, NULL)) | 1660 | if (!::ReadFile(hFile, &NtImageHeader, sizeof(NtImageHeader), &cbRead, NULL)) |
| 1646 | { | 1661 | { |
| 1647 | ExitWithLastError(hr, "Failed to read NT header from file: %ls", wzFile); | 1662 | FileExitWithLastError(hr, "Failed to read NT header from file: %ls", wzFile); |
| 1648 | } | 1663 | } |
| 1649 | 1664 | ||
| 1650 | if (NtImageHeader.Signature != IMAGE_NT_SIGNATURE) | 1665 | if (NtImageHeader.Signature != IMAGE_NT_SIGNATURE) |
| 1651 | { | 1666 | { |
| 1652 | hr = HRESULT_FROM_WIN32(ERROR_BAD_FORMAT); | 1667 | hr = HRESULT_FROM_WIN32(ERROR_BAD_FORMAT); |
| 1653 | ExitOnRootFailure(hr, "Read invalid NT header from file: %ls", wzFile); | 1668 | FileExitOnRootFailure(hr, "Read invalid NT header from file: %ls", wzFile); |
| 1654 | } | 1669 | } |
| 1655 | 1670 | ||
| 1656 | if (IMAGE_SUBSYSTEM_NATIVE == NtImageHeader.OptionalHeader.Subsystem || | 1671 | if (IMAGE_SUBSYSTEM_NATIVE == NtImageHeader.OptionalHeader.Subsystem || |
| @@ -1677,7 +1692,7 @@ extern "C" HRESULT DAPI FileExecutableArchitecture( | |||
| 1677 | { | 1692 | { |
| 1678 | hr = HRESULT_FROM_WIN32(ERROR_BAD_FORMAT); | 1693 | hr = HRESULT_FROM_WIN32(ERROR_BAD_FORMAT); |
| 1679 | } | 1694 | } |
| 1680 | ExitOnFailure(hr, "Unexpected subsystem: %d machine type: %d specified in NT header from file: %ls", NtImageHeader.OptionalHeader.Subsystem, NtImageHeader.FileHeader.Machine, wzFile); | 1695 | FileExitOnFailure(hr, "Unexpected subsystem: %d machine type: %d specified in NT header from file: %ls", NtImageHeader.OptionalHeader.Subsystem, NtImageHeader.FileHeader.Machine, wzFile); |
| 1681 | 1696 | ||
| 1682 | LExit: | 1697 | LExit: |
| 1683 | if (hFile != INVALID_HANDLE_VALUE) | 1698 | if (hFile != INVALID_HANDLE_VALUE) |
| @@ -1706,7 +1721,7 @@ extern "C" HRESULT DAPI FileToString( | |||
| 1706 | 1721 | ||
| 1707 | // Check if the file is ANSI | 1722 | // Check if the file is ANSI |
| 1708 | hr = FileRead(&pbFullFileBuffer, &cbFullFileBuffer, wzFile); | 1723 | hr = FileRead(&pbFullFileBuffer, &cbFullFileBuffer, wzFile); |
| 1709 | ExitOnFailure(hr, "Failed to read file: %ls", wzFile); | 1724 | FileExitOnFailure(hr, "Failed to read file: %ls", wzFile); |
| 1710 | 1725 | ||
| 1711 | if (0 == cbFullFileBuffer) | 1726 | if (0 == cbFullFileBuffer) |
| 1712 | { | 1727 | { |
| @@ -1723,7 +1738,7 @@ extern "C" HRESULT DAPI FileToString( | |||
| 1723 | } | 1738 | } |
| 1724 | 1739 | ||
| 1725 | hr = StrAllocStringAnsi(&sczFileText, reinterpret_cast<LPCSTR>(pbFullFileBuffer + 3), cbFullFileBuffer - 3, CP_UTF8); | 1740 | hr = StrAllocStringAnsi(&sczFileText, reinterpret_cast<LPCSTR>(pbFullFileBuffer + 3), cbFullFileBuffer - 3, CP_UTF8); |
| 1726 | ExitOnFailure(hr, "Failed to convert file %ls from UTF-8 as its BOM indicated", wzFile); | 1741 | FileExitOnFailure(hr, "Failed to convert file %ls from UTF-8 as its BOM indicated", wzFile); |
| 1727 | 1742 | ||
| 1728 | *psczString = sczFileText; | 1743 | *psczString = sczFileText; |
| 1729 | sczFileText = NULL; | 1744 | sczFileText = NULL; |
| @@ -1737,7 +1752,7 @@ extern "C" HRESULT DAPI FileToString( | |||
| 1737 | } | 1752 | } |
| 1738 | 1753 | ||
| 1739 | hr = StrAllocString(psczString, reinterpret_cast<LPWSTR>(pbFullFileBuffer + 2), (cbFullFileBuffer - 2) / sizeof(WCHAR)); | 1754 | hr = StrAllocString(psczString, reinterpret_cast<LPWSTR>(pbFullFileBuffer + 2), (cbFullFileBuffer - 2) / sizeof(WCHAR)); |
| 1740 | ExitOnFailure(hr, "Failed to allocate copy of string"); | 1755 | FileExitOnFailure(hr, "Failed to allocate copy of string"); |
| 1741 | } | 1756 | } |
| 1742 | // No BOM, let's try to detect | 1757 | // No BOM, let's try to detect |
| 1743 | else | 1758 | else |
| @@ -1763,7 +1778,7 @@ extern "C" HRESULT DAPI FileToString( | |||
| 1763 | { | 1778 | { |
| 1764 | if (E_OUTOFMEMORY == hr) | 1779 | if (E_OUTOFMEMORY == hr) |
| 1765 | { | 1780 | { |
| 1766 | ExitOnFailure(hr, "Failed to convert file %ls from UTF-8", wzFile); | 1781 | FileExitOnFailure(hr, "Failed to convert file %ls from UTF-8", wzFile); |
| 1767 | } | 1782 | } |
| 1768 | } | 1783 | } |
| 1769 | else | 1784 | else |
| @@ -1780,7 +1795,7 @@ extern "C" HRESULT DAPI FileToString( | |||
| 1780 | } | 1795 | } |
| 1781 | 1796 | ||
| 1782 | hr = StrAllocString(psczString, reinterpret_cast<LPWSTR>(pbFullFileBuffer), cbFullFileBuffer / sizeof(WCHAR)); | 1797 | hr = StrAllocString(psczString, reinterpret_cast<LPWSTR>(pbFullFileBuffer), cbFullFileBuffer / sizeof(WCHAR)); |
| 1783 | ExitOnFailure(hr, "Failed to allocate copy of string"); | 1798 | FileExitOnFailure(hr, "Failed to allocate copy of string"); |
| 1784 | } | 1799 | } |
| 1785 | } | 1800 | } |
| 1786 | 1801 | ||
| @@ -1813,20 +1828,20 @@ extern "C" HRESULT DAPI FileFromString( | |||
| 1813 | { | 1828 | { |
| 1814 | case FILE_ENCODING_UTF8: | 1829 | case FILE_ENCODING_UTF8: |
| 1815 | hr = StrAnsiAllocString(&sczUtf8String, sczString, 0, CP_UTF8); | 1830 | hr = StrAnsiAllocString(&sczUtf8String, sczString, 0, CP_UTF8); |
| 1816 | ExitOnFailure(hr, "Failed to convert string to UTF-8 to write UTF-8 file"); | 1831 | FileExitOnFailure(hr, "Failed to convert string to UTF-8 to write UTF-8 file"); |
| 1817 | 1832 | ||
| 1818 | cbFullFileBuffer = lstrlenA(sczUtf8String); | 1833 | cbFullFileBuffer = lstrlenA(sczUtf8String); |
| 1819 | pcbFullFileBuffer = reinterpret_cast<BYTE *>(sczUtf8String); | 1834 | pcbFullFileBuffer = reinterpret_cast<BYTE *>(sczUtf8String); |
| 1820 | break; | 1835 | break; |
| 1821 | case FILE_ENCODING_UTF8_WITH_BOM: | 1836 | case FILE_ENCODING_UTF8_WITH_BOM: |
| 1822 | hr = StrAnsiAllocString(&sczUtf8String, sczString, 0, CP_UTF8); | 1837 | hr = StrAnsiAllocString(&sczUtf8String, sczString, 0, CP_UTF8); |
| 1823 | ExitOnFailure(hr, "Failed to convert string to UTF-8 to write UTF-8 file"); | 1838 | FileExitOnFailure(hr, "Failed to convert string to UTF-8 to write UTF-8 file"); |
| 1824 | 1839 | ||
| 1825 | cbStrLen = lstrlenA(sczUtf8String); | 1840 | cbStrLen = lstrlenA(sczUtf8String); |
| 1826 | cbFullFileBuffer = sizeof(UTF8BOM) + cbStrLen; | 1841 | cbFullFileBuffer = sizeof(UTF8BOM) + cbStrLen; |
| 1827 | 1842 | ||
| 1828 | pbFullFileBuffer = reinterpret_cast<BYTE *>(MemAlloc(cbFullFileBuffer, TRUE)); | 1843 | pbFullFileBuffer = reinterpret_cast<BYTE *>(MemAlloc(cbFullFileBuffer, TRUE)); |
| 1829 | ExitOnNull(pbFullFileBuffer, hr, E_OUTOFMEMORY, "Failed to allocate memory for output file buffer"); | 1844 | FileExitOnNull(pbFullFileBuffer, hr, E_OUTOFMEMORY, "Failed to allocate memory for output file buffer"); |
| 1830 | 1845 | ||
| 1831 | memcpy_s(pbFullFileBuffer, sizeof(UTF8BOM), UTF8BOM, sizeof(UTF8BOM)); | 1846 | memcpy_s(pbFullFileBuffer, sizeof(UTF8BOM), UTF8BOM, sizeof(UTF8BOM)); |
| 1832 | memcpy_s(pbFullFileBuffer + sizeof(UTF8BOM), cbStrLen, sczUtf8String, cbStrLen); | 1847 | memcpy_s(pbFullFileBuffer + sizeof(UTF8BOM), cbStrLen, sczUtf8String, cbStrLen); |
| @@ -1841,7 +1856,7 @@ extern "C" HRESULT DAPI FileFromString( | |||
| 1841 | cbFullFileBuffer = sizeof(UTF16BOM) + cbStrLen; | 1856 | cbFullFileBuffer = sizeof(UTF16BOM) + cbStrLen; |
| 1842 | 1857 | ||
| 1843 | pbFullFileBuffer = reinterpret_cast<BYTE *>(MemAlloc(cbFullFileBuffer, TRUE)); | 1858 | pbFullFileBuffer = reinterpret_cast<BYTE *>(MemAlloc(cbFullFileBuffer, TRUE)); |
| 1844 | ExitOnNull(pbFullFileBuffer, hr, E_OUTOFMEMORY, "Failed to allocate memory for output file buffer"); | 1859 | FileExitOnNull(pbFullFileBuffer, hr, E_OUTOFMEMORY, "Failed to allocate memory for output file buffer"); |
| 1845 | 1860 | ||
| 1846 | memcpy_s(pbFullFileBuffer, sizeof(UTF16BOM), UTF16BOM, sizeof(UTF16BOM)); | 1861 | memcpy_s(pbFullFileBuffer, sizeof(UTF16BOM), UTF16BOM, sizeof(UTF16BOM)); |
| 1847 | memcpy_s(pbFullFileBuffer + sizeof(UTF16BOM), cbStrLen, sczString, cbStrLen); | 1862 | memcpy_s(pbFullFileBuffer + sizeof(UTF16BOM), cbStrLen, sczString, cbStrLen); |
| @@ -1850,7 +1865,7 @@ extern "C" HRESULT DAPI FileFromString( | |||
| 1850 | } | 1865 | } |
| 1851 | 1866 | ||
| 1852 | hr = FileWrite(wzFile, dwFlagsAndAttributes, pcbFullFileBuffer, cbFullFileBuffer, NULL); | 1867 | hr = FileWrite(wzFile, dwFlagsAndAttributes, pcbFullFileBuffer, cbFullFileBuffer, NULL); |
| 1853 | ExitOnFailure(hr, "Failed to write file from string to: %ls", wzFile); | 1868 | FileExitOnFailure(hr, "Failed to write file from string to: %ls", wzFile); |
| 1854 | 1869 | ||
| 1855 | LExit: | 1870 | LExit: |
| 1856 | ReleaseStr(sczUtf8String); | 1871 | ReleaseStr(sczUtf8String); |
