diff options
Diffstat (limited to 'src/libs/dutil/WixToolset.DUtil/cabutil.cpp')
| -rw-r--r-- | src/libs/dutil/WixToolset.DUtil/cabutil.cpp | 617 |
1 files changed, 617 insertions, 0 deletions
diff --git a/src/libs/dutil/WixToolset.DUtil/cabutil.cpp b/src/libs/dutil/WixToolset.DUtil/cabutil.cpp new file mode 100644 index 00000000..5d77e483 --- /dev/null +++ b/src/libs/dutil/WixToolset.DUtil/cabutil.cpp | |||
| @@ -0,0 +1,617 @@ | |||
| 1 | // Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information. | ||
| 2 | |||
| 3 | #include "precomp.h" | ||
| 4 | |||
| 5 | |||
| 6 | // Exit macros | ||
| 7 | #define CabExitOnLastError(x, s, ...) ExitOnLastErrorSource(DUTIL_SOURCE_CABUTIL, x, s, __VA_ARGS__) | ||
| 8 | #define CabExitOnLastErrorDebugTrace(x, s, ...) ExitOnLastErrorDebugTraceSource(DUTIL_SOURCE_CABUTIL, x, s, __VA_ARGS__) | ||
| 9 | #define CabExitWithLastError(x, s, ...) ExitWithLastErrorSource(DUTIL_SOURCE_CABUTIL, x, s, __VA_ARGS__) | ||
| 10 | #define CabExitOnFailure(x, s, ...) ExitOnFailureSource(DUTIL_SOURCE_CABUTIL, x, s, __VA_ARGS__) | ||
| 11 | #define CabExitOnRootFailure(x, s, ...) ExitOnRootFailureSource(DUTIL_SOURCE_CABUTIL, x, s, __VA_ARGS__) | ||
| 12 | #define CabExitOnFailureDebugTrace(x, s, ...) ExitOnFailureDebugTraceSource(DUTIL_SOURCE_CABUTIL, x, s, __VA_ARGS__) | ||
| 13 | #define CabExitOnNull(p, x, e, s, ...) ExitOnNullSource(DUTIL_SOURCE_CABUTIL, p, x, e, s, __VA_ARGS__) | ||
| 14 | #define CabExitOnNullWithLastError(p, x, s, ...) ExitOnNullWithLastErrorSource(DUTIL_SOURCE_CABUTIL, p, x, s, __VA_ARGS__) | ||
| 15 | #define CabExitOnNullDebugTrace(p, x, e, s, ...) ExitOnNullDebugTraceSource(DUTIL_SOURCE_CABUTIL, p, x, e, s, __VA_ARGS__) | ||
| 16 | #define CabExitOnInvalidHandleWithLastError(p, x, s, ...) ExitOnInvalidHandleWithLastErrorSource(DUTIL_SOURCE_CABUTIL, p, x, s, __VA_ARGS__) | ||
| 17 | #define CabExitOnWin32Error(e, x, s, ...) ExitOnWin32ErrorSource(DUTIL_SOURCE_CABUTIL, e, x, s, __VA_ARGS__) | ||
| 18 | #define CabExitOnGdipFailure(g, x, s, ...) ExitOnGdipFailureSource(DUTIL_SOURCE_CABUTIL, g, x, s, __VA_ARGS__) | ||
| 19 | |||
| 20 | |||
| 21 | // external prototypes | ||
| 22 | typedef BOOL (FAR DIAMONDAPI *PFNFDIDESTROY)(VOID*); | ||
| 23 | typedef HFDI (FAR DIAMONDAPI *PFNFDICREATE)(PFNALLOC, PFNFREE, PFNOPEN, PFNREAD, PFNWRITE, PFNCLOSE, PFNSEEK, int, PERF); | ||
| 24 | typedef BOOL (FAR DIAMONDAPI *PFNFDIISCABINET)(HFDI, INT_PTR, PFDICABINETINFO); | ||
| 25 | typedef BOOL (FAR DIAMONDAPI *PFNFDICOPY)(HFDI, char *, char *, int, PFNFDINOTIFY, PFNFDIDECRYPT, void *); | ||
| 26 | |||
| 27 | |||
| 28 | // | ||
| 29 | // static globals | ||
| 30 | // | ||
| 31 | static HMODULE vhCabinetDll = NULL; | ||
| 32 | |||
| 33 | static HFDI vhfdi = NULL; | ||
| 34 | static PFNFDICREATE vpfnFDICreate = NULL; | ||
| 35 | static PFNFDICOPY vpfnFDICopy = NULL; | ||
| 36 | static PFNFDIISCABINET vpfnFDIIsCabinet = NULL; | ||
| 37 | static PFNFDIDESTROY vpfnFDIDestroy = NULL; | ||
| 38 | static ERF verf; | ||
| 39 | |||
| 40 | static DWORD64 vdw64EmbeddedOffset = 0; | ||
| 41 | |||
| 42 | // | ||
| 43 | // structs | ||
| 44 | // | ||
| 45 | struct CAB_CALLBACK_STRUCT | ||
| 46 | { | ||
| 47 | BOOL fStopExtracting; // flag set when no more files are needed | ||
| 48 | LPCWSTR pwzExtract; // file to extract ("*" means extract all) | ||
| 49 | LPCWSTR pwzExtractDir; // directory to extract files to | ||
| 50 | |||
| 51 | // possible user data | ||
| 52 | CAB_CALLBACK_PROGRESS pfnProgress; | ||
| 53 | LPVOID pvContext; | ||
| 54 | }; | ||
| 55 | |||
| 56 | // | ||
| 57 | // prototypes | ||
| 58 | // | ||
| 59 | static __callback LPVOID DIAMONDAPI CabExtractAlloc(__in DWORD dwSize); | ||
| 60 | static __callback void DIAMONDAPI CabExtractFree(__in LPVOID pvData); | ||
| 61 | static __callback INT_PTR FAR DIAMONDAPI CabExtractOpen(__in_z PSTR pszFile, __in int oflag, __in int pmode); | ||
| 62 | static __callback UINT FAR DIAMONDAPI CabExtractRead(__in INT_PTR hf, __out void FAR *pv, __in UINT cb); | ||
| 63 | static __callback UINT FAR DIAMONDAPI CabExtractWrite(__in INT_PTR hf, __in void FAR *pv, __in UINT cb); | ||
| 64 | static __callback int FAR DIAMONDAPI CabExtractClose(__in INT_PTR hf); | ||
| 65 | static __callback long FAR DIAMONDAPI CabExtractSeek(__in INT_PTR hf, __in long dist, __in int seektype); | ||
| 66 | static __callback INT_PTR DIAMONDAPI CabExtractCallback(__in FDINOTIFICATIONTYPE iNotification, __inout FDINOTIFICATION *pFDINotify); | ||
| 67 | static HRESULT DAPI CabOperation(__in LPCWSTR wzCabinet, __in LPCWSTR wzExtractFile, __in_opt LPCWSTR wzExtractDir, __in_opt CAB_CALLBACK_PROGRESS pfnProgress, __in_opt LPVOID pvContext, __in_opt STDCALL_PFNFDINOTIFY pfnNotify, __in DWORD64 dw64EmbeddedOffset); | ||
| 68 | |||
| 69 | static STDCALL_PFNFDINOTIFY v_pfnNetFx11Notify = NULL; | ||
| 70 | |||
| 71 | |||
| 72 | inline HRESULT LoadCabinetDll() | ||
| 73 | { | ||
| 74 | HRESULT hr = S_OK; | ||
| 75 | if (!vhCabinetDll) | ||
| 76 | { | ||
| 77 | hr = LoadSystemLibrary(L"cabinet.dll", &vhCabinetDll); | ||
| 78 | CabExitOnFailure(hr, "failed to load cabinet.dll"); | ||
| 79 | |||
| 80 | // retrieve all address functions | ||
| 81 | vpfnFDICreate = reinterpret_cast<PFNFDICREATE>(::GetProcAddress(vhCabinetDll, "FDICreate")); | ||
| 82 | CabExitOnNullWithLastError(vpfnFDICreate, hr, "failed to import FDICreate from CABINET.DLL"); | ||
| 83 | vpfnFDICopy = reinterpret_cast<PFNFDICOPY>(::GetProcAddress(vhCabinetDll, "FDICopy")); | ||
| 84 | CabExitOnNullWithLastError(vpfnFDICopy, hr, "failed to import FDICopy from CABINET.DLL"); | ||
| 85 | vpfnFDIIsCabinet = reinterpret_cast<PFNFDIISCABINET>(::GetProcAddress(vhCabinetDll, "FDIIsCabinet")); | ||
| 86 | CabExitOnNullWithLastError(vpfnFDIIsCabinet, hr, "failed to import FDIIsCabinetfrom CABINET.DLL"); | ||
| 87 | vpfnFDIDestroy = reinterpret_cast<PFNFDIDESTROY>(::GetProcAddress(vhCabinetDll, "FDIDestroy")); | ||
| 88 | CabExitOnNullWithLastError(vpfnFDIDestroy, hr, "failed to import FDIDestroyfrom CABINET.DLL"); | ||
| 89 | |||
| 90 | vhfdi = vpfnFDICreate(CabExtractAlloc, CabExtractFree, CabExtractOpen, CabExtractRead, CabExtractWrite, CabExtractClose, CabExtractSeek, cpuUNKNOWN, &verf); | ||
| 91 | CabExitOnNull(vhfdi, hr, E_FAIL, "failed to initialize cabinet.dll"); | ||
| 92 | } | ||
| 93 | |||
| 94 | LExit: | ||
| 95 | if (FAILED(hr) && vhCabinetDll) | ||
| 96 | { | ||
| 97 | ::FreeLibrary(vhCabinetDll); | ||
| 98 | vhCabinetDll = NULL; | ||
| 99 | } | ||
| 100 | |||
| 101 | return hr; | ||
| 102 | } | ||
| 103 | |||
| 104 | |||
| 105 | static HANDLE OpenFileWithRetry( | ||
| 106 | __in LPCWSTR wzPath, | ||
| 107 | __in DWORD dwDesiredAccess, | ||
| 108 | __in DWORD dwCreationDisposition | ||
| 109 | ) | ||
| 110 | { | ||
| 111 | HANDLE hFile = INVALID_HANDLE_VALUE; | ||
| 112 | |||
| 113 | for (DWORD i = 0; i < 30; ++i) | ||
| 114 | { | ||
| 115 | hFile = ::CreateFileW(wzPath, dwDesiredAccess, FILE_SHARE_READ, NULL, dwCreationDisposition, FILE_ATTRIBUTE_NORMAL, NULL); | ||
| 116 | if (INVALID_HANDLE_VALUE != hFile) | ||
| 117 | { | ||
| 118 | break; | ||
| 119 | } | ||
| 120 | |||
| 121 | ::Sleep(100); | ||
| 122 | } | ||
| 123 | |||
| 124 | return hFile; | ||
| 125 | } | ||
| 126 | |||
| 127 | |||
| 128 | /******************************************************************** | ||
| 129 | CabInitialize - initializes internal static variables | ||
| 130 | |||
| 131 | ********************************************************************/ | ||
| 132 | extern "C" HRESULT DAPI CabInitialize( | ||
| 133 | __in BOOL fDelayLoad | ||
| 134 | ) | ||
| 135 | { | ||
| 136 | HRESULT hr = S_OK; | ||
| 137 | |||
| 138 | if (!fDelayLoad) | ||
| 139 | { | ||
| 140 | hr = LoadCabinetDll(); | ||
| 141 | CabExitOnFailure(hr, "failed to load CABINET.DLL"); | ||
| 142 | } | ||
| 143 | |||
| 144 | LExit: | ||
| 145 | return hr; | ||
| 146 | } | ||
| 147 | |||
| 148 | |||
| 149 | /******************************************************************** | ||
| 150 | CabUninitialize - initializes internal static variables | ||
| 151 | |||
| 152 | ********************************************************************/ | ||
| 153 | extern "C" void DAPI CabUninitialize( | ||
| 154 | ) | ||
| 155 | { | ||
| 156 | if (vhfdi) | ||
| 157 | { | ||
| 158 | if (vpfnFDIDestroy) | ||
| 159 | { | ||
| 160 | vpfnFDIDestroy(vhfdi); | ||
| 161 | } | ||
| 162 | vhfdi = NULL; | ||
| 163 | } | ||
| 164 | |||
| 165 | vpfnFDICreate = NULL; | ||
| 166 | vpfnFDICopy =NULL; | ||
| 167 | vpfnFDIIsCabinet = NULL; | ||
| 168 | vpfnFDIDestroy = NULL; | ||
| 169 | |||
| 170 | if (vhCabinetDll) | ||
| 171 | { | ||
| 172 | ::FreeLibrary(vhCabinetDll); | ||
| 173 | vhCabinetDll = NULL; | ||
| 174 | } | ||
| 175 | } | ||
| 176 | |||
| 177 | /******************************************************************** | ||
| 178 | CabEnumerate - list files inside cabinet | ||
| 179 | |||
| 180 | NOTE: wzCabinet must be full path to cabinet file | ||
| 181 | pfnNotify is callback function to get notified for each file | ||
| 182 | in the cabinet | ||
| 183 | ********************************************************************/ | ||
| 184 | extern "C" HRESULT DAPI CabEnumerate( | ||
| 185 | __in_z LPCWSTR wzCabinet, | ||
| 186 | __in_z LPCWSTR wzEnumerateFile, | ||
| 187 | __in STDCALL_PFNFDINOTIFY pfnNotify, | ||
| 188 | __in DWORD64 dw64EmbeddedOffset | ||
| 189 | ) | ||
| 190 | { | ||
| 191 | return CabOperation(wzCabinet, wzEnumerateFile, NULL, NULL, NULL, pfnNotify, dw64EmbeddedOffset); | ||
| 192 | } | ||
| 193 | |||
| 194 | /******************************************************************** | ||
| 195 | CabExtract - extracts one or all files from a cabinet | ||
| 196 | |||
| 197 | NOTE: wzCabinet must be full path to cabinet file | ||
| 198 | wzExtractFile can be a single file id or "*" to extract all files | ||
| 199 | wzExttractDir must be normalized (end in a "\") | ||
| 200 | if pfnBeginFile is NULL pfnEndFile must be NULL and vice versa | ||
| 201 | ********************************************************************/ | ||
| 202 | extern "C" HRESULT DAPI CabExtract( | ||
| 203 | __in_z LPCWSTR wzCabinet, | ||
| 204 | __in_z LPCWSTR wzExtractFile, | ||
| 205 | __in_z LPCWSTR wzExtractDir, | ||
| 206 | __in_opt CAB_CALLBACK_PROGRESS pfnProgress, | ||
| 207 | __in_opt LPVOID pvContext, | ||
| 208 | __in DWORD64 dw64EmbeddedOffset | ||
| 209 | ) | ||
| 210 | { | ||
| 211 | return CabOperation(wzCabinet, wzExtractFile, wzExtractDir, pfnProgress, pvContext, NULL, dw64EmbeddedOffset); | ||
| 212 | } | ||
| 213 | |||
| 214 | // | ||
| 215 | // private | ||
| 216 | // | ||
| 217 | /******************************************************************** | ||
| 218 | FDINotify -- wrapper that converts call convention from __cdecl to __stdcall. | ||
| 219 | |||
| 220 | NOTE: Since netfx 1.1 supports only function pointers (delegates) | ||
| 221 | with __stdcall calling convention and cabinet api uses | ||
| 222 | __cdecl calling convention, we need this wrapper function. | ||
| 223 | netfx 2.0 will work with [UnmanagedFunctionPointer(CallingConvention.Cdecl)] attribute on the delegate. | ||
| 224 | TODO: remove this when upgrading to netfx 2.0. | ||
| 225 | ********************************************************************/ | ||
| 226 | static __callback INT_PTR DIAMONDAPI FDINotify( | ||
| 227 | __in FDINOTIFICATIONTYPE iNotification, | ||
| 228 | __inout FDINOTIFICATION *pFDINotify | ||
| 229 | ) | ||
| 230 | { | ||
| 231 | if (NULL != v_pfnNetFx11Notify) | ||
| 232 | { | ||
| 233 | return v_pfnNetFx11Notify(iNotification, pFDINotify); | ||
| 234 | } | ||
| 235 | else | ||
| 236 | { | ||
| 237 | return (INT_PTR)0; | ||
| 238 | } | ||
| 239 | } | ||
| 240 | |||
| 241 | |||
| 242 | /******************************************************************** | ||
| 243 | CabOperation - helper function that enumerates or extracts files | ||
| 244 | from cabinet | ||
| 245 | |||
| 246 | NOTE: wzCabinet must be full path to cabinet file | ||
| 247 | wzExtractFile can be a single file id or "*" to extract all files | ||
| 248 | wzExttractDir must be normalized (end in a "\") | ||
| 249 | if pfnBeginFile is NULL pfnEndFile must be NULL and vice versa | ||
| 250 | pfnNotify is callback function to get notified for each file | ||
| 251 | in the cabinet. If it's NULL, files will be extracted. | ||
| 252 | ********************************************************************/ | ||
| 253 | static HRESULT DAPI CabOperation( | ||
| 254 | __in LPCWSTR wzCabinet, | ||
| 255 | __in LPCWSTR wzExtractFile, | ||
| 256 | __in_opt LPCWSTR wzExtractDir, | ||
| 257 | __in_opt CAB_CALLBACK_PROGRESS pfnProgress, | ||
| 258 | __in_opt LPVOID pvContext, | ||
| 259 | __in_opt STDCALL_PFNFDINOTIFY pfnNotify, | ||
| 260 | __in DWORD64 dw64EmbeddedOffset | ||
| 261 | ) | ||
| 262 | { | ||
| 263 | HRESULT hr = S_OK; | ||
| 264 | BOOL fResult; | ||
| 265 | |||
| 266 | LPWSTR sczCabinet = NULL; | ||
| 267 | LPWSTR pwz = NULL; | ||
| 268 | CHAR szCabDirectory[MAX_PATH * 4]; // Make sure these are big enough for UTF-8 strings | ||
| 269 | CHAR szCabFile[MAX_PATH * 4]; | ||
| 270 | |||
| 271 | CAB_CALLBACK_STRUCT ccs; | ||
| 272 | PFNFDINOTIFY pfnFdiNotify; | ||
| 273 | |||
| 274 | // | ||
| 275 | // ensure the cabinet.dll is loaded | ||
| 276 | // | ||
| 277 | if (!vhfdi) | ||
| 278 | { | ||
| 279 | hr = LoadCabinetDll(); | ||
| 280 | CabExitOnFailure(hr, "failed to load CABINET.DLL"); | ||
| 281 | } | ||
| 282 | |||
| 283 | hr = StrAllocString(&sczCabinet, wzCabinet, 0); | ||
| 284 | CabExitOnFailure(hr, "Failed to make copy of cabinet name:%ls", wzCabinet); | ||
| 285 | |||
| 286 | // | ||
| 287 | // split the cabinet full path into directory and filename and convert to multi-byte (ick!) | ||
| 288 | // | ||
| 289 | pwz = FileFromPath(sczCabinet); | ||
| 290 | CabExitOnNull(pwz, hr, E_INVALIDARG, "failed to process cabinet path: %ls", wzCabinet); | ||
| 291 | |||
| 292 | if (!::WideCharToMultiByte(CP_UTF8, 0, pwz, -1, szCabFile, countof(szCabFile), NULL, NULL)) | ||
| 293 | { | ||
| 294 | CabExitWithLastError(hr, "failed to convert cabinet filename to ASCII: %ls", pwz); | ||
| 295 | } | ||
| 296 | |||
| 297 | *pwz = '\0'; | ||
| 298 | |||
| 299 | // If a full path was not provided, use the relative current directory. | ||
| 300 | if (wzCabinet == pwz) | ||
| 301 | { | ||
| 302 | hr = ::StringCchCopyA(szCabDirectory, countof(szCabDirectory), ".\\"); | ||
| 303 | CabExitOnFailure(hr, "Failed to copy relative current directory as cabinet directory."); | ||
| 304 | } | ||
| 305 | else | ||
| 306 | { | ||
| 307 | if (!::WideCharToMultiByte(CP_UTF8, 0, sczCabinet, -1, szCabDirectory, countof(szCabDirectory), NULL, NULL)) | ||
| 308 | { | ||
| 309 | CabExitWithLastError(hr, "failed to convert cabinet directory to ASCII: %ls", sczCabinet); | ||
| 310 | } | ||
| 311 | } | ||
| 312 | |||
| 313 | // | ||
| 314 | // iterate through files in cabinet extracting them to the callback function | ||
| 315 | // | ||
| 316 | ccs.fStopExtracting = FALSE; | ||
| 317 | ccs.pwzExtract = wzExtractFile; | ||
| 318 | ccs.pwzExtractDir = wzExtractDir; | ||
| 319 | ccs.pfnProgress = pfnProgress; | ||
| 320 | ccs.pvContext = pvContext; | ||
| 321 | |||
| 322 | vdw64EmbeddedOffset = dw64EmbeddedOffset; | ||
| 323 | |||
| 324 | // if pfnNotify is given, use it, otherwise use default callback | ||
| 325 | if (NULL == pfnNotify) | ||
| 326 | { | ||
| 327 | pfnFdiNotify = CabExtractCallback; | ||
| 328 | } | ||
| 329 | else | ||
| 330 | { | ||
| 331 | v_pfnNetFx11Notify = pfnNotify; | ||
| 332 | pfnFdiNotify = FDINotify; | ||
| 333 | } | ||
| 334 | fResult = vpfnFDICopy(vhfdi, szCabFile, szCabDirectory, 0, pfnFdiNotify, NULL, static_cast<void*>(&ccs)); | ||
| 335 | if (!fResult && !ccs.fStopExtracting) // if something went wrong and it wasn't us just stopping the extraction, then return a failure | ||
| 336 | { | ||
| 337 | CabExitWithLastError(hr, "failed to extract cabinet file: %ls", sczCabinet); | ||
| 338 | } | ||
| 339 | |||
| 340 | LExit: | ||
| 341 | ReleaseStr(sczCabinet); | ||
| 342 | v_pfnNetFx11Notify = NULL; | ||
| 343 | |||
| 344 | return hr; | ||
| 345 | } | ||
| 346 | |||
| 347 | /**************************************************************************** | ||
| 348 | default extract routines | ||
| 349 | |||
| 350 | ****************************************************************************/ | ||
| 351 | static __callback LPVOID DIAMONDAPI CabExtractAlloc(__in DWORD dwSize) | ||
| 352 | { | ||
| 353 | return MemAlloc(dwSize, FALSE); | ||
| 354 | } | ||
| 355 | |||
| 356 | |||
| 357 | static __callback void DIAMONDAPI CabExtractFree(__in LPVOID pvData) | ||
| 358 | { | ||
| 359 | MemFree(pvData); | ||
| 360 | } | ||
| 361 | |||
| 362 | |||
| 363 | static __callback INT_PTR FAR DIAMONDAPI CabExtractOpen(__in_z PSTR pszFile, __in int oflag, __in int pmode) | ||
| 364 | { | ||
| 365 | HRESULT hr = S_OK; | ||
| 366 | HANDLE hFile = INVALID_HANDLE_VALUE; | ||
| 367 | INT_PTR pFile = -1; | ||
| 368 | LPWSTR sczCabFile = NULL; | ||
| 369 | |||
| 370 | // if FDI asks for some unusual mode (in low memory situation it could ask for a scratch file) fail | ||
| 371 | if ((oflag != (/*_O_BINARY*/ 0x8000 | /*_O_RDONLY*/ 0x0000)) || (pmode != (_S_IREAD | _S_IWRITE))) | ||
| 372 | { | ||
| 373 | hr = E_OUTOFMEMORY; | ||
| 374 | CabExitOnFailure(hr, "FDI asked for a scratch file to be created, which is unsupported"); | ||
| 375 | } | ||
| 376 | |||
| 377 | hr = StrAllocStringAnsi(&sczCabFile, pszFile, 0, CP_UTF8); | ||
| 378 | CabExitOnFailure(hr, "Failed to convert UTF8 cab file name to wide character string"); | ||
| 379 | |||
| 380 | hFile = OpenFileWithRetry(sczCabFile, GENERIC_READ, OPEN_EXISTING); | ||
| 381 | if (INVALID_HANDLE_VALUE == hFile) | ||
| 382 | { | ||
| 383 | CabExitWithLastError(hr, "failed to open file: %ls", sczCabFile); | ||
| 384 | } | ||
| 385 | |||
| 386 | pFile = reinterpret_cast<INT_PTR>(hFile); | ||
| 387 | |||
| 388 | if (vdw64EmbeddedOffset) | ||
| 389 | { | ||
| 390 | hr = CabExtractSeek(pFile, 0, 0); | ||
| 391 | CabExitOnFailure(hr, "Failed to seek to embedded offset %I64d", vdw64EmbeddedOffset); | ||
| 392 | } | ||
| 393 | |||
| 394 | hFile = INVALID_HANDLE_VALUE; | ||
| 395 | |||
| 396 | LExit: | ||
| 397 | ReleaseFileHandle(hFile); | ||
| 398 | ReleaseStr(sczCabFile); | ||
| 399 | |||
| 400 | return FAILED(hr) ? -1 : pFile; | ||
| 401 | } | ||
| 402 | |||
| 403 | |||
| 404 | static __callback UINT FAR DIAMONDAPI CabExtractRead(__in INT_PTR hf, __out void FAR *pv, __in UINT cb) | ||
| 405 | { | ||
| 406 | HRESULT hr = S_OK; | ||
| 407 | DWORD cbRead = 0; | ||
| 408 | |||
| 409 | CabExitOnNull(hf, hr, E_INVALIDARG, "Failed to read file during cabinet extraction - no file given to read"); | ||
| 410 | if (!::ReadFile(reinterpret_cast<HANDLE>(hf), pv, cb, &cbRead, NULL)) | ||
| 411 | { | ||
| 412 | CabExitWithLastError(hr, "failed to read during cabinet extraction"); | ||
| 413 | } | ||
| 414 | |||
| 415 | LExit: | ||
| 416 | return FAILED(hr) ? -1 : cbRead; | ||
| 417 | } | ||
| 418 | |||
| 419 | |||
| 420 | static __callback UINT FAR DIAMONDAPI CabExtractWrite(__in INT_PTR hf, __in void FAR *pv, __in UINT cb) | ||
| 421 | { | ||
| 422 | HRESULT hr = S_OK; | ||
| 423 | DWORD cbWrite = 0; | ||
| 424 | |||
| 425 | CabExitOnNull(hf, hr, E_INVALIDARG, "Failed to write file during cabinet extraction - no file given to write"); | ||
| 426 | if (!::WriteFile(reinterpret_cast<HANDLE>(hf), pv, cb, &cbWrite, NULL)) | ||
| 427 | { | ||
| 428 | CabExitWithLastError(hr, "failed to write during cabinet extraction"); | ||
| 429 | } | ||
| 430 | |||
| 431 | LExit: | ||
| 432 | return FAILED(hr) ? -1 : cbWrite; | ||
| 433 | } | ||
| 434 | |||
| 435 | |||
| 436 | static __callback long FAR DIAMONDAPI CabExtractSeek(__in INT_PTR hf, __in long dist, __in int seektype) | ||
| 437 | { | ||
| 438 | HRESULT hr = S_OK; | ||
| 439 | DWORD dwMoveMethod; | ||
| 440 | LONG lMove = 0; | ||
| 441 | |||
| 442 | switch (seektype) | ||
| 443 | { | ||
| 444 | case 0: // SEEK_SET | ||
| 445 | dwMoveMethod = FILE_BEGIN; | ||
| 446 | dist += static_cast<long>(vdw64EmbeddedOffset); | ||
| 447 | break; | ||
| 448 | case 1: /// SEEK_CUR | ||
| 449 | dwMoveMethod = FILE_CURRENT; | ||
| 450 | break; | ||
| 451 | case 2: // SEEK_END | ||
| 452 | dwMoveMethod = FILE_END; | ||
| 453 | break; | ||
| 454 | default : | ||
| 455 | dwMoveMethod = 0; | ||
| 456 | hr = E_UNEXPECTED; | ||
| 457 | CabExitOnFailure(hr, "unexpected seektype in FDISeek(): %d", seektype); | ||
| 458 | } | ||
| 459 | |||
| 460 | // SetFilePointer returns -1 if it fails (this will cause FDI to quit with an FDIERROR_USER_ABORT error. | ||
| 461 | // (Unless this happens while working on a cabinet, in which case FDI returns FDIERROR_CORRUPT_CABINET) | ||
| 462 | lMove = ::SetFilePointer(reinterpret_cast<HANDLE>(hf), dist, NULL, dwMoveMethod); | ||
| 463 | if (0xFFFFFFFF == lMove) | ||
| 464 | { | ||
| 465 | CabExitWithLastError(hr, "failed to move file pointer %d bytes", dist); | ||
| 466 | } | ||
| 467 | |||
| 468 | LExit: | ||
| 469 | return FAILED(hr) ? -1 : lMove - static_cast<long>(vdw64EmbeddedOffset); | ||
| 470 | } | ||
| 471 | |||
| 472 | |||
| 473 | static __callback int FAR DIAMONDAPI CabExtractClose(__in INT_PTR hf) | ||
| 474 | { | ||
| 475 | HRESULT hr = S_OK; | ||
| 476 | |||
| 477 | if (!::CloseHandle(reinterpret_cast<HANDLE>(hf))) | ||
| 478 | { | ||
| 479 | CabExitWithLastError(hr, "failed to close file during cabinet extraction"); | ||
| 480 | } | ||
| 481 | |||
| 482 | LExit: | ||
| 483 | return FAILED(hr) ? -1 : 0; | ||
| 484 | } | ||
| 485 | |||
| 486 | |||
| 487 | static __callback INT_PTR DIAMONDAPI CabExtractCallback(__in FDINOTIFICATIONTYPE iNotification, __inout FDINOTIFICATION *pFDINotify) | ||
| 488 | { | ||
| 489 | Assert(pFDINotify->pv); | ||
| 490 | |||
| 491 | HRESULT hr = S_OK; | ||
| 492 | HANDLE hFile = INVALID_HANDLE_VALUE; | ||
| 493 | INT_PTR ipResult = 0; // result to return on success | ||
| 494 | |||
| 495 | CAB_CALLBACK_STRUCT* pccs = static_cast<CAB_CALLBACK_STRUCT*>(pFDINotify->pv); | ||
| 496 | LPCSTR sz; | ||
| 497 | WCHAR wz[MAX_PATH]; | ||
| 498 | FILETIME ft; | ||
| 499 | |||
| 500 | switch (iNotification) | ||
| 501 | { | ||
| 502 | case fdintCOPY_FILE: // begin extracting a resource from cabinet | ||
| 503 | CabExitOnNull(pFDINotify->psz1, hr, E_INVALIDARG, "No cabinet file ID given to convert"); | ||
| 504 | CabExitOnNull(pccs, hr, E_INVALIDARG, "Failed to call cabextract callback, because no callback struct was provided"); | ||
| 505 | |||
| 506 | if (pccs->fStopExtracting) | ||
| 507 | { | ||
| 508 | ExitFunction1(hr = S_FALSE); // no more extracting | ||
| 509 | } | ||
| 510 | |||
| 511 | // convert params to useful variables | ||
| 512 | sz = static_cast<LPCSTR>(pFDINotify->psz1); | ||
| 513 | if (!::MultiByteToWideChar(CP_ACP, 0, sz, -1, wz, countof(wz))) | ||
| 514 | { | ||
| 515 | CabExitWithLastError(hr, "failed to convert cabinet file id to unicode: %s", sz); | ||
| 516 | } | ||
| 517 | |||
| 518 | if (pccs->pfnProgress) | ||
| 519 | { | ||
| 520 | hr = pccs->pfnProgress(TRUE, wz, pccs->pvContext); | ||
| 521 | if (S_OK != hr) | ||
| 522 | { | ||
| 523 | ExitFunction(); | ||
| 524 | } | ||
| 525 | } | ||
| 526 | |||
| 527 | if (L'*' == *pccs->pwzExtract || 0 == lstrcmpW(pccs->pwzExtract, wz)) | ||
| 528 | { | ||
| 529 | // get the created date for the resource in the cabinet | ||
| 530 | FILETIME ftLocal; | ||
| 531 | if (!::DosDateTimeToFileTime(pFDINotify->date, pFDINotify->time, &ftLocal)) | ||
| 532 | { | ||
| 533 | CabExitWithLastError(hr, "failed to get time for resource: %ls", wz); | ||
| 534 | } | ||
| 535 | ::LocalFileTimeToFileTime(&ftLocal, &ft); | ||
| 536 | |||
| 537 | WCHAR wzPath[MAX_PATH]; | ||
| 538 | hr = ::StringCchCopyW(wzPath, countof(wzPath), pccs->pwzExtractDir); | ||
| 539 | CabExitOnFailure(hr, "failed to copy in extract directory: %ls for file: %ls", pccs->pwzExtractDir, wz); | ||
| 540 | hr = ::StringCchCatW(wzPath, countof(wzPath), wz); | ||
| 541 | CabExitOnFailure(hr, "failed to concat onto path: %ls file: %ls", wzPath, wz); | ||
| 542 | |||
| 543 | hFile = OpenFileWithRetry(wzPath, GENERIC_WRITE, CREATE_ALWAYS); | ||
| 544 | if (INVALID_HANDLE_VALUE == hFile) | ||
| 545 | { | ||
| 546 | CabExitWithLastError(hr, "failed to create file: %ls", wzPath); | ||
| 547 | } | ||
| 548 | |||
| 549 | ::SetFileTime(hFile, &ft, &ft, &ft); // try to set the file time (who cares if it fails) | ||
| 550 | |||
| 551 | if (::SetFilePointer(hFile, pFDINotify->cb, NULL, FILE_BEGIN)) // try to set the end of the file (don't worry if this fails) | ||
| 552 | { | ||
| 553 | if (::SetEndOfFile(hFile)) | ||
| 554 | { | ||
| 555 | ::SetFilePointer(hFile, 0, NULL, FILE_BEGIN); // reset the file pointer | ||
| 556 | } | ||
| 557 | } | ||
| 558 | |||
| 559 | ipResult = reinterpret_cast<INT_PTR>(hFile); | ||
| 560 | hFile = INVALID_HANDLE_VALUE; | ||
| 561 | } | ||
| 562 | else // resource wasn't requested, skip it | ||
| 563 | { | ||
| 564 | hr = S_OK; | ||
| 565 | ipResult = 0; | ||
| 566 | } | ||
| 567 | |||
| 568 | break; | ||
| 569 | case fdintCLOSE_FILE_INFO: // resource extraction complete | ||
| 570 | Assert(pFDINotify->hf && pFDINotify->psz1); | ||
| 571 | CabExitOnNull(pccs, hr, E_INVALIDARG, "Failed to call cabextract callback, because no callback struct was provided"); | ||
| 572 | |||
| 573 | // convert params to useful variables | ||
| 574 | sz = static_cast<LPCSTR>(pFDINotify->psz1); | ||
| 575 | CabExitOnNull(sz, hr, E_INVALIDARG, "Failed to convert cabinet file id, because no cabinet file id was provided"); | ||
| 576 | |||
| 577 | if (!::MultiByteToWideChar(CP_ACP, 0, sz, -1, wz, countof(wz))) | ||
| 578 | { | ||
| 579 | CabExitWithLastError(hr, "failed to convert cabinet file id to unicode: %s", sz); | ||
| 580 | } | ||
| 581 | |||
| 582 | if (NULL != pFDINotify->hf) // just close the file | ||
| 583 | { | ||
| 584 | ::CloseHandle(reinterpret_cast<HANDLE>(pFDINotify->hf)); | ||
| 585 | } | ||
| 586 | |||
| 587 | if (pccs->pfnProgress) | ||
| 588 | { | ||
| 589 | hr = pccs->pfnProgress(FALSE, wz, pccs->pvContext); | ||
| 590 | } | ||
| 591 | |||
| 592 | if (S_OK == hr && L'*' == *pccs->pwzExtract) // if everything is okay and we're extracting all files, keep going | ||
| 593 | { | ||
| 594 | ipResult = TRUE; | ||
| 595 | } | ||
| 596 | else // something went wrong or we only needed to extract one file | ||
| 597 | { | ||
| 598 | hr = S_OK; | ||
| 599 | ipResult = FALSE; | ||
| 600 | pccs->fStopExtracting = TRUE; | ||
| 601 | } | ||
| 602 | |||
| 603 | break; | ||
| 604 | case fdintPARTIAL_FILE: __fallthrough; // no action needed for these messages, fall through | ||
| 605 | case fdintNEXT_CABINET: __fallthrough; | ||
| 606 | case fdintENUMERATE: __fallthrough; | ||
| 607 | case fdintCABINET_INFO: | ||
| 608 | break; | ||
| 609 | default: | ||
| 610 | AssertSz(FALSE, "CabExtractCallback() - unknown FDI notification command"); | ||
| 611 | }; | ||
| 612 | |||
| 613 | LExit: | ||
| 614 | ReleaseFileHandle(hFile); | ||
| 615 | |||
| 616 | return (S_OK == hr) ? ipResult : -1; | ||
| 617 | } | ||
