/* * cabinet.dll main * * Copyright 2002 Patrik Stridvall * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ #include "config.h" #include <assert.h> #include <stdarg.h> #include <string.h> #include "windef.h" #include "winbase.h" #include "winerror.h" #define NO_SHLWAPI_REG #include "shlwapi.h" #undef NO_SHLWAPI_REG #include "cabinet.h" #include "wine/debug.h" WINE_DEFAULT_DEBUG_CHANNEL(cabinet); /*********************************************************************** * DllGetVersion (CABINET.2) * * Retrieves version information of the 'CABINET.DLL' * * PARAMS * pdvi [O] pointer to version information structure. * * RETURNS * Success: S_OK * Failure: E_INVALIDARG * * NOTES * Supposedly returns version from IE6SP1RP1 */ HRESULT WINAPI DllGetVersion (DLLVERSIONINFO *pdvi) { WARN("hmmm... not right version number \"5.1.1106.1\"?\n"); if (pdvi->cbSize != sizeof(DLLVERSIONINFO)) return E_INVALIDARG; pdvi->dwMajorVersion = 5; pdvi->dwMinorVersion = 1; pdvi->dwBuildNumber = 1106; pdvi->dwPlatformID = 1; return S_OK; } /* FDI callback functions */ static void * CDECL mem_alloc(ULONG cb) { return HeapAlloc(GetProcessHeap(), 0, cb); } static void CDECL mem_free(void *memory) { HeapFree(GetProcessHeap(), 0, memory); } static INT_PTR CDECL fdi_open(char *pszFile, int oflag, int pmode) { HANDLE handle; DWORD dwAccess = 0; DWORD dwShareMode = 0; DWORD dwCreateDisposition; switch (oflag & _O_ACCMODE) { case _O_RDONLY: dwAccess = GENERIC_READ; dwShareMode = FILE_SHARE_READ | FILE_SHARE_DELETE; break; case _O_WRONLY: dwAccess = GENERIC_WRITE; dwShareMode = FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE; break; case _O_RDWR: dwAccess = GENERIC_READ | GENERIC_WRITE; dwShareMode = FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE; break; } if (oflag & _O_CREAT) { dwCreateDisposition = OPEN_ALWAYS; if (oflag & _O_EXCL) dwCreateDisposition = CREATE_NEW; else if (oflag & _O_TRUNC) dwCreateDisposition = CREATE_ALWAYS; } else { dwCreateDisposition = OPEN_EXISTING; if (oflag & _O_TRUNC) dwCreateDisposition = TRUNCATE_EXISTING; } handle = CreateFileA(pszFile, dwAccess, dwShareMode, NULL, dwCreateDisposition, 0, NULL); return (INT_PTR) handle; } static UINT CDECL fdi_read(INT_PTR hf, void *pv, UINT cb) { HANDLE handle = (HANDLE) hf; DWORD dwRead; if (ReadFile(handle, pv, cb, &dwRead, NULL)) return dwRead; return 0; } static UINT CDECL fdi_write(INT_PTR hf, void *pv, UINT cb) { HANDLE handle = (HANDLE) hf; DWORD dwWritten; if (WriteFile(handle, pv, cb, &dwWritten, NULL)) return dwWritten; return 0; } static int CDECL fdi_close(INT_PTR hf) { HANDLE handle = (HANDLE) hf; return CloseHandle(handle) ? 0 : -1; } static LONG CDECL fdi_seek(INT_PTR hf, LONG dist, int seektype) { HANDLE handle = (HANDLE) hf; return SetFilePointer(handle, dist, NULL, seektype); } static void fill_file_node(struct FILELIST *pNode, LPCSTR szFilename) { pNode->next = NULL; pNode->DoExtract = FALSE; pNode->FileName = HeapAlloc(GetProcessHeap(), 0, strlen(szFilename) + 1); lstrcpyA(pNode->FileName, szFilename); } static BOOL file_in_list(struct FILELIST *pNode, LPCSTR szFilename, struct FILELIST **pOut) { while (pNode) { if (!lstrcmpiA(pNode->FileName, szFilename)) { if (pOut) *pOut = pNode; return TRUE; } pNode = pNode->next; } return FALSE; } static INT_PTR CDECL fdi_notify_extract(FDINOTIFICATIONTYPE fdint, PFDINOTIFICATION pfdin) { switch (fdint) { case fdintCOPY_FILE: { struct FILELIST *fileList, *node = NULL; SESSION *pDestination = pfdin->pv; LPSTR szFullPath, szDirectory; HANDLE hFile = 0; DWORD dwSize; dwSize = lstrlenA(pDestination->Destination) + lstrlenA("\\") + lstrlenA(pfdin->psz1) + 1; szFullPath = HeapAlloc(GetProcessHeap(), 0, dwSize); lstrcpyA(szFullPath, pDestination->Destination); lstrcatA(szFullPath, "\\"); lstrcatA(szFullPath, pfdin->psz1); /* pull out the destination directory string from the full path */ dwSize = strrchr(szFullPath, '\\') - szFullPath + 1; szDirectory = HeapAlloc(GetProcessHeap(), 0, dwSize); lstrcpynA(szDirectory, szFullPath, dwSize); pDestination->FileSize += pfdin->cb; if (pDestination->Operation & EXTRACT_FILLFILELIST) { fileList = HeapAlloc(GetProcessHeap(), 0, sizeof(struct FILELIST)); fill_file_node(fileList, pfdin->psz1); fileList->DoExtract = TRUE; fileList->next = pDestination->FileList; pDestination->FileList = fileList; lstrcpyA(pDestination->CurrentFile, szFullPath); pDestination->FileCount++; } if ((pDestination->Operation & EXTRACT_EXTRACTFILES) || file_in_list(pDestination->FilterList, pfdin->psz1, NULL)) { /* find the file node */ file_in_list(pDestination->FileList, pfdin->psz1, &node); if (node && !node->DoExtract) { HeapFree(GetProcessHeap(), 0, szFullPath); HeapFree(GetProcessHeap(), 0, szDirectory); return 0; } /* create the destination directory if it doesn't exist */ if (GetFileAttributesA(szDirectory) == INVALID_FILE_ATTRIBUTES) { char *ptr; for(ptr = szDirectory + strlen(pDestination->Destination)+1; *ptr; ptr++) { if(*ptr == '\\') { *ptr = 0; CreateDirectoryA(szDirectory, NULL); *ptr = '\\'; } } CreateDirectoryA(szDirectory, NULL); } hFile = CreateFileA(szFullPath, GENERIC_READ | GENERIC_WRITE, 0, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL); if (hFile == INVALID_HANDLE_VALUE) hFile = 0; else if (node) node->DoExtract = FALSE; } HeapFree(GetProcessHeap(), 0, szFullPath); HeapFree(GetProcessHeap(), 0, szDirectory); return (INT_PTR) hFile; } case fdintCLOSE_FILE_INFO: { FILETIME ft; FILETIME ftLocal; HANDLE handle = (HANDLE) pfdin->hf; if (!DosDateTimeToFileTime(pfdin->date, pfdin->time, &ft)) return FALSE; if (!LocalFileTimeToFileTime(&ft, &ftLocal)) return FALSE; if (!SetFileTime(handle, &ftLocal, 0, &ftLocal)) return FALSE; CloseHandle(handle); return TRUE; } default: return 0; } } /*********************************************************************** * Extract (CABINET.3) * * Extracts the contents of the cabinet file to the specified * destination. * * PARAMS * dest [I/O] Controls the operation of Extract. See NOTES. * szCabName [I] Filename of the cabinet to extract. * * RETURNS * Success: S_OK. * Failure: E_FAIL. * * NOTES * The following members of the dest struct control the operation * of Extract: * FileSize [O] The size of all files extracted up to CurrentFile. * Error [O] The error in case the extract operation fails. * FileList [I] A linked list of filenames. Extract only extracts * files from the cabinet that are in this list. * FileCount [O] Contains the number of files in FileList on * completion. * Operation [I] See Operation. * Destination [I] The destination directory. * CurrentFile [O] The last file extracted. * FilterList [I] A linked list of files that should not be extracted. * * Operation * If Operation contains EXTRACT_FILLFILELIST, then FileList will be * filled with all the files in the cabinet. If Operation contains * EXTRACT_EXTRACTFILES, then only the files in the FileList will * be extracted from the cabinet. EXTRACT_FILLFILELIST can be called * by itself, but EXTRACT_EXTRACTFILES must have a valid FileList * in order to succeed. If Operation contains both EXTRACT_FILLFILELIST * and EXTRACT_EXTRACTFILES, then all the files in the cabinet * will be extracted. */ HRESULT WINAPI Extract(SESSION *dest, LPCSTR szCabName) { HRESULT res = S_OK; HFDI hfdi; char *str, *end, *path = NULL, *name = NULL; TRACE("(%p, %s)\n", dest, szCabName); hfdi = FDICreate(mem_alloc, mem_free, fdi_open, fdi_read, fdi_write, fdi_close, fdi_seek, cpuUNKNOWN, &dest->Error); if (!hfdi) return E_FAIL; if (GetFileAttributesA(dest->Destination) == INVALID_FILE_ATTRIBUTES) { res = S_OK; goto end; } /* split the cabinet name into path + name */ str = HeapAlloc(GetProcessHeap(), 0, lstrlenA(szCabName)+1); if (!str) { res = E_OUTOFMEMORY; goto end; } lstrcpyA(str, szCabName); if ((end = strrchr(str, '\\'))) { path = str; end++; name = HeapAlloc( GetProcessHeap(), 0, strlen(end) + 1 ); if (!name) { res = E_OUTOFMEMORY; goto end; } strcpy( name, end ); *end = 0; } else { name = str; path = NULL; } dest->FileSize = 0; if (!FDICopy(hfdi, name, path, 0, fdi_notify_extract, NULL, dest)) res = HRESULT_FROM_WIN32(GetLastError()); end: HeapFree(GetProcessHeap(), 0, path); HeapFree(GetProcessHeap(), 0, name); FDIDestroy(hfdi); return res; }