/* * SHFileOperation * * Copyright 2000 Juergen Schmied * Copyright 2002 Andriy Palamarchuk * Copyright 2004 Dietrich Teickner (from Odin) * Copyright 2004 Rolf Kalbermatter * * 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ #include "config.h" #include "wine/port.h" #include #include #include #include "windef.h" #include "winbase.h" #include "winreg.h" #include "shellapi.h" #include "wingdi.h" #include "winuser.h" #include "shlobj.h" #include "shresdef.h" #define NO_SHLWAPI_STREAM #include "shlwapi.h" #include "shell32_main.h" #include "undocshell.h" #include "wine/unicode.h" #include "wine/debug.h" WINE_DEFAULT_DEBUG_CHANNEL(shell); #define IsAttrib(x, y) ((INVALID_FILE_ATTRIBUTES != (x)) && ((x) & (y))) #define IsAttribFile(x) (!((x) & FILE_ATTRIBUTE_DIRECTORY)) #define IsAttribDir(x) IsAttrib(x, FILE_ATTRIBUTE_DIRECTORY) #define IsDotDir(x) ((x[0] == '.') && ((x[1] == 0) || ((x[1] == '.') && (x[2] == 0)))) #define FO_MASK 0xF static const WCHAR wWildcardFile[] = {'*',0}; static const WCHAR wWildcardChars[] = {'*','?',0}; static BOOL SHELL_DeleteDirectoryW(LPCWSTR path, BOOL bShowUI); static DWORD SHNotifyCreateDirectoryA(LPCSTR path, LPSECURITY_ATTRIBUTES sec); static DWORD SHNotifyCreateDirectoryW(LPCWSTR path, LPSECURITY_ATTRIBUTES sec); static DWORD SHNotifyRemoveDirectoryA(LPCSTR path); static DWORD SHNotifyRemoveDirectoryW(LPCWSTR path); static DWORD SHNotifyDeleteFileA(LPCSTR path); static DWORD SHNotifyDeleteFileW(LPCWSTR path); static DWORD SHNotifyMoveFileW(LPCWSTR src, LPCWSTR dest); static DWORD SHNotifyCopyFileW(LPCWSTR src, LPCWSTR dest, BOOL bFailIfExists); static DWORD SHFindAttrW(LPCWSTR pName, BOOL fileOnly); typedef struct { UINT caption_resource_id, text_resource_id; } SHELL_ConfirmIDstruc; static BOOL SHELL_ConfirmIDs(int nKindOfDialog, SHELL_ConfirmIDstruc *ids) { switch (nKindOfDialog) { case ASK_DELETE_FILE: ids->caption_resource_id = IDS_DELETEITEM_CAPTION; ids->text_resource_id = IDS_DELETEITEM_TEXT; return TRUE; case ASK_DELETE_FOLDER: ids->caption_resource_id = IDS_DELETEFOLDER_CAPTION; ids->text_resource_id = IDS_DELETEITEM_TEXT; return TRUE; case ASK_DELETE_MULTIPLE_ITEM: ids->caption_resource_id = IDS_DELETEITEM_CAPTION; ids->text_resource_id = IDS_DELETEMULTIPLE_TEXT; return TRUE; case ASK_OVERWRITE_FILE: ids->caption_resource_id = IDS_OVERWRITEFILE_CAPTION; ids->text_resource_id = IDS_OVERWRITEFILE_TEXT; return TRUE; default: FIXME(" Unhandled nKindOfDialog %d stub\n", nKindOfDialog); } return FALSE; } BOOL SHELL_ConfirmDialog(int nKindOfDialog, LPCSTR szDir) { CHAR szCaption[255], szText[255], szBuffer[MAX_PATH + 256]; SHELL_ConfirmIDstruc ids; if (!SHELL_ConfirmIDs(nKindOfDialog, &ids)) return FALSE; LoadStringA(shell32_hInstance, ids.caption_resource_id, szCaption, sizeof(szCaption)); LoadStringA(shell32_hInstance, ids.text_resource_id, szText, sizeof(szText)); FormatMessageA(FORMAT_MESSAGE_FROM_STRING|FORMAT_MESSAGE_ARGUMENT_ARRAY, szText, 0, 0, szBuffer, sizeof(szBuffer), (va_list*)&szDir); return (IDOK == MessageBoxA(GetActiveWindow(), szBuffer, szCaption, MB_OKCANCEL | MB_ICONEXCLAMATION)); } BOOL SHELL_ConfirmDialogW(int nKindOfDialog, LPCWSTR szDir) { WCHAR szCaption[255], szText[255], szBuffer[MAX_PATH + 256]; SHELL_ConfirmIDstruc ids; if (!SHELL_ConfirmIDs(nKindOfDialog, &ids)) return FALSE; LoadStringW(shell32_hInstance, ids.caption_resource_id, szCaption, sizeof(szCaption)); LoadStringW(shell32_hInstance, ids.text_resource_id, szText, sizeof(szText)); FormatMessageW(FORMAT_MESSAGE_FROM_STRING|FORMAT_MESSAGE_ARGUMENT_ARRAY, szText, 0, 0, szBuffer, sizeof(szBuffer), (va_list*)&szDir); return (IDOK == MessageBoxW(GetActiveWindow(), szBuffer, szCaption, MB_OKCANCEL | MB_ICONEXCLAMATION)); } static DWORD SHELL32_AnsiToUnicodeBuf(LPCSTR aPath, LPWSTR *wPath, DWORD minChars) { DWORD len = MultiByteToWideChar(CP_ACP, 0, aPath, -1, NULL, 0); if (len < minChars) len = minChars; *wPath = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR)); if (*wPath) { MultiByteToWideChar(CP_ACP, 0, aPath, -1, *wPath, len); return NO_ERROR; } return E_OUTOFMEMORY; } static void SHELL32_FreeUnicodeBuf(LPWSTR wPath) { HeapFree(GetProcessHeap(), 0, wPath); } /************************************************************************** * SHELL_DeleteDirectory() [internal] * * Asks for confirmation when bShowUI is true and deletes the directory and * all its subdirectories and files if necessary. */ BOOL SHELL_DeleteDirectoryA(LPCSTR pszDir, BOOL bShowUI) { LPWSTR wPath; BOOL ret = FALSE; if (!SHELL32_AnsiToUnicodeBuf(pszDir, &wPath, 0)) { ret = SHELL_DeleteDirectoryW(wPath, bShowUI); SHELL32_FreeUnicodeBuf(wPath); } return ret; } static BOOL SHELL_DeleteDirectoryW(LPCWSTR pszDir, BOOL bShowUI) { BOOL ret = TRUE; HANDLE hFind; WIN32_FIND_DATAW wfd; WCHAR szTemp[MAX_PATH]; /* Make sure the directory exists before eventually prompting the user */ PathCombineW(szTemp, pszDir, wWildcardFile); hFind = FindFirstFileW(szTemp, &wfd); if (hFind == INVALID_HANDLE_VALUE) return FALSE; if (!bShowUI || (ret = SHELL_ConfirmDialogW(ASK_DELETE_FOLDER, pszDir))) { do { LPWSTR lp = wfd.cAlternateFileName; if (!lp[0]) lp = wfd.cFileName; if (IsDotDir(lp)) continue; PathCombineW(szTemp, pszDir, lp); if (FILE_ATTRIBUTE_DIRECTORY & wfd.dwFileAttributes) ret = SHELL_DeleteDirectoryW(szTemp, FALSE); else ret = (SHNotifyDeleteFileW(szTemp) == ERROR_SUCCESS); } while (ret && FindNextFileW(hFind, &wfd)); } FindClose(hFind); if (ret) ret = (SHNotifyRemoveDirectoryW(pszDir) == ERROR_SUCCESS); return ret; } /************************************************************************** * SHELL_DeleteFileA() [internal] */ BOOL SHELL_DeleteFileA(LPCSTR pszFile, BOOL bShowUI) { if (bShowUI && !SHELL_ConfirmDialog(ASK_DELETE_FILE, pszFile)) return FALSE; return (SHNotifyDeleteFileA(pszFile) == ERROR_SUCCESS); } BOOL SHELL_DeleteFileW(LPCWSTR pszFile, BOOL bShowUI) { if (bShowUI && !SHELL_ConfirmDialogW(ASK_DELETE_FILE, pszFile)) return FALSE; return (SHNotifyDeleteFileW(pszFile) == ERROR_SUCCESS); } /************************************************************************** * Win32CreateDirectory [SHELL32.93] * * Creates a directory. Also triggers a change notify if one exists. * * PARAMS * path [I] path to directory to create * * RETURNS * TRUE if successful, FALSE otherwise * * NOTES * Verified on Win98 / IE 5 (SHELL32 4.72, March 1999 build) to be ANSI. * This is Unicode on NT/2000 */ static DWORD SHNotifyCreateDirectoryA(LPCSTR path, LPSECURITY_ATTRIBUTES sec) { LPWSTR wPath; DWORD retCode; TRACE("(%s, %p)\n", debugstr_a(path), sec); retCode = SHELL32_AnsiToUnicodeBuf(path, &wPath, 0); if (!retCode) { retCode = SHNotifyCreateDirectoryW(wPath, sec); SHELL32_FreeUnicodeBuf(wPath); } return retCode; } /**********************************************************************/ static DWORD SHNotifyCreateDirectoryW(LPCWSTR path, LPSECURITY_ATTRIBUTES sec) { TRACE("(%s, %p)\n", debugstr_w(path), sec); if (CreateDirectoryW(path, sec)) { SHChangeNotify(SHCNE_MKDIR, SHCNF_PATHW, path, NULL); return ERROR_SUCCESS; } return GetLastError(); } /**********************************************************************/ BOOL WINAPI Win32CreateDirectoryAW(LPCVOID path, LPSECURITY_ATTRIBUTES sec) { if (SHELL_OsIsUnicode()) return (SHNotifyCreateDirectoryW(path, sec) == ERROR_SUCCESS); return (SHNotifyCreateDirectoryA(path, sec) == ERROR_SUCCESS); } /************************************************************************ * Win32RemoveDirectory [SHELL32.94] * * Deletes a directory. Also triggers a change notify if one exists. * * PARAMS * path [I] path to directory to delete * * RETURNS * TRUE if successful, FALSE otherwise * * NOTES * Verified on Win98 / IE 5 (SHELL32 4.72, March 1999 build) to be ANSI. * This is Unicode on NT/2000 */ static DWORD SHNotifyRemoveDirectoryA(LPCSTR path) { LPWSTR wPath; DWORD retCode; TRACE("(%s)\n", debugstr_a(path)); retCode = SHELL32_AnsiToUnicodeBuf(path, &wPath, 0); if (!retCode) { retCode = SHNotifyRemoveDirectoryW(wPath); SHELL32_FreeUnicodeBuf(wPath); } return retCode; } /***********************************************************************/ static DWORD SHNotifyRemoveDirectoryW(LPCWSTR path) { BOOL ret; TRACE("(%s)\n", debugstr_w(path)); ret = RemoveDirectoryW(path); if (!ret) { /* Directory may be write protected */ DWORD dwAttr = GetFileAttributesW(path); if (IsAttrib(dwAttr, FILE_ATTRIBUTE_READONLY)) if (SetFileAttributesW(path, dwAttr & ~FILE_ATTRIBUTE_READONLY)) ret = RemoveDirectoryW(path); } if (ret) { SHChangeNotify(SHCNE_RMDIR, SHCNF_PATHW, path, NULL); return ERROR_SUCCESS; } return GetLastError(); } /***********************************************************************/ BOOL WINAPI Win32RemoveDirectoryAW(LPCVOID path) { if (SHELL_OsIsUnicode()) return (SHNotifyRemoveDirectoryW(path) == ERROR_SUCCESS); return (SHNotifyRemoveDirectoryA(path) == ERROR_SUCCESS); } /************************************************************************ * Win32DeleteFile [SHELL32.164] * * Deletes a file. Also triggers a change notify if one exists. * * PARAMS * path [I] path to file to delete * * RETURNS * TRUE if successful, FALSE otherwise * * NOTES * Verified on Win98 / IE 5 (SHELL32 4.72, March 1999 build) to be ANSI. * This is Unicode on NT/2000 */ static DWORD SHNotifyDeleteFileA(LPCSTR path) { LPWSTR wPath; DWORD retCode; TRACE("(%s)\n", debugstr_a(path)); retCode = SHELL32_AnsiToUnicodeBuf(path, &wPath, 0); if (!retCode) { retCode = SHNotifyDeleteFileW(wPath); SHELL32_FreeUnicodeBuf(wPath); } return retCode; } /***********************************************************************/ static DWORD SHNotifyDeleteFileW(LPCWSTR path) { BOOL ret; TRACE("(%s)\n", debugstr_w(path)); ret = DeleteFileW(path); if (!ret) { /* File may be write protected or a system file */ DWORD dwAttr = GetFileAttributesW(path); if (IsAttrib(dwAttr, FILE_ATTRIBUTE_READONLY | FILE_ATTRIBUTE_SYSTEM)) if (SetFileAttributesW(path, dwAttr & ~(FILE_ATTRIBUTE_READONLY | FILE_ATTRIBUTE_SYSTEM))) ret = DeleteFileW(path); } if (ret) { SHChangeNotify(SHCNE_DELETE, SHCNF_PATHW, path, NULL); return ERROR_SUCCESS; } return GetLastError(); } /***********************************************************************/ DWORD WINAPI Win32DeleteFileAW(LPCVOID path) { if (SHELL_OsIsUnicode()) return (SHNotifyDeleteFileW(path) == ERROR_SUCCESS); return (SHNotifyDeleteFileA(path) == ERROR_SUCCESS); } /************************************************************************ * SHNotifyMoveFile [internal] * * Moves a file. Also triggers a change notify if one exists. * * PARAMS * src [I] path to source file to move * dest [I] path to target file to move to * * RETURNS * ERORR_SUCCESS if successful */ static DWORD SHNotifyMoveFileW(LPCWSTR src, LPCWSTR dest) { BOOL ret; TRACE("(%s %s)\n", debugstr_w(src), debugstr_w(dest)); ret = MoveFileExW(src, dest, MOVEFILE_REPLACE_EXISTING); /* MOVEFILE_REPLACE_EXISTING fails with dirs, so try MoveFile */ if (!ret) ret = MoveFileW(src, dest); if (!ret) { DWORD dwAttr; dwAttr = SHFindAttrW(dest, FALSE); if (INVALID_FILE_ATTRIBUTES == dwAttr) { /* Source file may be write protected or a system file */ dwAttr = GetFileAttributesW(src); if (IsAttrib(dwAttr, FILE_ATTRIBUTE_READONLY | FILE_ATTRIBUTE_SYSTEM)) if (SetFileAttributesW(src, dwAttr & ~(FILE_ATTRIBUTE_READONLY | FILE_ATTRIBUTE_SYSTEM))) ret = MoveFileW(src, dest); } } if (ret) { SHChangeNotify(SHCNE_RENAMEITEM, SHCNF_PATHW, src, dest); return ERROR_SUCCESS; } return GetLastError(); } /************************************************************************ * SHNotifyCopyFile [internal] * * Copies a file. Also triggers a change notify if one exists. * * PARAMS * src [I] path to source file to move * dest [I] path to target file to move to * bFailIfExists [I] if TRUE, the target file will not be overwritten if * a file with this name already exists * * RETURNS * ERROR_SUCCESS if successful */ static DWORD SHNotifyCopyFileW(LPCWSTR src, LPCWSTR dest, BOOL bFailIfExists) { BOOL ret; TRACE("(%s %s %s)\n", debugstr_w(src), debugstr_w(dest), bFailIfExists ? "failIfExists" : ""); ret = CopyFileW(src, dest, bFailIfExists); if (ret) { SHChangeNotify(SHCNE_CREATE, SHCNF_PATHW, dest, NULL); return ERROR_SUCCESS; } return GetLastError(); } /************************************************************************* * SHCreateDirectory [SHELL32.165] * * This function creates a file system folder whose fully qualified path is * given by path. If one or more of the intermediate folders do not exist, * they will be created as well. * * PARAMS * hWnd [I] * path [I] path of directory to create * * RETURNS * ERRROR_SUCCESS or one of the following values: * ERROR_BAD_PATHNAME if the path is relative * ERROR_FILE_EXISTS when a file with that name exists * ERROR_PATH_NOT_FOUND can't find the path, probably invalid * ERROR_INVLID_NAME if the path contains invalid chars * ERROR_ALREADY_EXISTS when the directory already exists * ERROR_FILENAME_EXCED_RANGE if the filename was to long to process * * NOTES * exported by ordinal * Win9x exports ANSI * WinNT/2000 exports Unicode */ DWORD WINAPI SHCreateDirectory(HWND hWnd, LPCVOID path) { if (SHELL_OsIsUnicode()) return SHCreateDirectoryExW(hWnd, path, NULL); return SHCreateDirectoryExA(hWnd, path, NULL); } /************************************************************************* * SHCreateDirectoryExA [SHELL32.@] * * This function creates a file system folder whose fully qualified path is * given by path. If one or more of the intermediate folders do not exist, * they will be created as well. * * PARAMS * hWnd [I] * path [I] path of directory to create * sec [I] security attributes to use or NULL * * RETURNS * ERRROR_SUCCESS or one of the following values: * ERROR_BAD_PATHNAME or ERROR_PATH_NOT_FOUND if the path is relative * ERROR_INVLID_NAME if the path contains invalid chars * ERROR_FILE_EXISTS when a file with that name exists * ERROR_ALREADY_EXISTS when the directory already exists * ERROR_FILENAME_EXCED_RANGE if the filename was to long to process * * FIXME: Not implemented yet; * SHCreateDirectoryEx also verifies that the files in the directory will be visible * if the path is a network path to deal with network drivers which might have a limited * but unknown maximum path length. If not: * * If hWnd is set to a valid window handle, a message box is displayed warning * the user that the files may not be accessible. If the user chooses not to * proceed, the function returns ERROR_CANCELLED. * * If hWnd is set to NULL, no user interface is displayed and the function * returns ERROR_CANCELLED. */ int WINAPI SHCreateDirectoryExA(HWND hWnd, LPCSTR path, LPSECURITY_ATTRIBUTES sec) { LPWSTR wPath; DWORD retCode; TRACE("(%s, %p)\n", debugstr_a(path), sec); retCode = SHELL32_AnsiToUnicodeBuf(path, &wPath, 0); if (!retCode) { retCode = SHCreateDirectoryExW(hWnd, wPath, sec); SHELL32_FreeUnicodeBuf(wPath); } return retCode; } /************************************************************************* * SHCreateDirectoryExW [SHELL32.@] * * See SHCreateDirectoryExA. */ int WINAPI SHCreateDirectoryExW(HWND hWnd, LPCWSTR path, LPSECURITY_ATTRIBUTES sec) { int ret = ERROR_BAD_PATHNAME; TRACE("(%p, %s, %p)\n", hWnd, debugstr_w(path), sec); if (PathIsRelativeW(path)) { SetLastError(ret); } else { ret = SHNotifyCreateDirectoryW(path, sec); /* Refuse to work on certain error codes before trying to create directories recursively */ if (ret != ERROR_FILE_EXISTS && ret != ERROR_ALREADY_EXISTS && ret != ERROR_FILENAME_EXCED_RANGE) { WCHAR *pEnd, *pSlash, szTemp[MAX_PATH + 1]; /* extra for PathAddBackslash() */ lstrcpynW(szTemp, path, MAX_PATH); pEnd = PathAddBackslashW(szTemp); pSlash = szTemp + 3; while (*pSlash) { while (*pSlash && *pSlash != '\\') pSlash = CharNextW(pSlash); if (*pSlash) { *pSlash = 0; /* terminate path at separator */ ret = SHNotifyCreateDirectoryW(szTemp, pSlash + 1 == pEnd ? sec : NULL); } *pSlash++ = '\\'; /* put the separator back */ } } if (ret && hWnd && (ERROR_CANCELLED != ret)) { /* We failed and should show a dialog box */ FIXME("Show system error message, creating path %s, failed with error %d\n", debugstr_w(path), ret); ret = ERROR_CANCELLED; /* Error has been already presented to user (not really yet!) */ } } return ret; } /************************************************************************* * SHFindAttrW [internal] * * Get the Attributes for a file or directory. The difference to GetAttributes() * is that this function will also work for paths containing wildcard characters * in its filename. * PARAMS * path [I] path of directory or file to check * fileOnly [I] TRUE if only files should be found * * RETURNS * INVALID_FILE_ATTRIBUTES if the path does not exist, the actual attributes of * the first file or directory found otherwise */ static DWORD SHFindAttrW(LPCWSTR pName, BOOL fileOnly) { WIN32_FIND_DATAW wfd; BOOL b_FileMask = fileOnly && (NULL != StrPBrkW(pName, wWildcardChars)); DWORD dwAttr = INVALID_FILE_ATTRIBUTES; HANDLE hFind = FindFirstFileW(pName, &wfd); TRACE("%s %d\n", debugstr_w(pName), fileOnly); if (INVALID_HANDLE_VALUE != hFind) { do { if (b_FileMask && IsAttribDir(wfd.dwFileAttributes)) continue; dwAttr = wfd.dwFileAttributes; break; } while (FindNextFileW(hFind, &wfd)); FindClose(hFind); } return dwAttr; } /************************************************************************* * * SHNameTranslate HelperFunction for SHFileOperationA * * Translates a list of 0 terminated ASCII strings into Unicode. If *wString * is NULL, only the necessary size of the string is determined and returned, * otherwise the ASCII strings are copied into it and the buffer is increased * to point to the location after the final 0 termination char. */ DWORD SHNameTranslate(LPWSTR* wString, LPCWSTR* pWToFrom, BOOL more) { DWORD size = 0, aSize = 0; LPCSTR aString = (LPCSTR)*pWToFrom; if (aString) { do { size = lstrlenA(aString) + 1; aSize += size; aString += size; } while ((size != 1) && more); /* The two sizes might be different in the case of multibyte chars */ size = MultiByteToWideChar(CP_ACP, 0, aString, aSize, *wString, 0); if (*wString) /* only in the second loop */ { MultiByteToWideChar(CP_ACP, 0, (LPCSTR)*pWToFrom, aSize, *wString, size); *pWToFrom = *wString; *wString += size; } } return size; } /************************************************************************* * SHFileOperationA [SHELL32.@] * * Function to copy, move, delete and create one or more files with optional * user prompts. * * PARAMS * lpFileOp [I/O] pointer to a structure containing all the necessary information * * RETURNS * Success: ERROR_SUCCESS. * Failure: ERROR_CANCELLED. * * NOTES * exported by name */ int WINAPI SHFileOperationA(LPSHFILEOPSTRUCTA lpFileOp) { SHFILEOPSTRUCTW nFileOp = *((LPSHFILEOPSTRUCTW)lpFileOp); int retCode = 0; DWORD size; LPWSTR ForFree = NULL, /* we change wString in SHNameTranslate and can't use it for freeing */ wString = NULL; /* we change this in SHNameTranslate */ TRACE("\n"); if (FO_DELETE == (nFileOp.wFunc & FO_MASK)) nFileOp.pTo = NULL; /* we need a NULL or a valid pointer for translation */ if (!(nFileOp.fFlags & FOF_SIMPLEPROGRESS)) nFileOp.lpszProgressTitle = NULL; /* we need a NULL or a valid pointer for translation */ while (1) /* every loop calculate size, second translate also, if we have storage for this */ { size = SHNameTranslate(&wString, &nFileOp.lpszProgressTitle, FALSE); /* no loop */ size += SHNameTranslate(&wString, &nFileOp.pFrom, TRUE); /* internal loop */ size += SHNameTranslate(&wString, &nFileOp.pTo, TRUE); /* internal loop */ if (ForFree) { retCode = SHFileOperationW(&nFileOp); HeapFree(GetProcessHeap(), 0, ForFree); /* we cannot use wString, it was changed */ break; } else { wString = ForFree = HeapAlloc(GetProcessHeap(), 0, size * sizeof(WCHAR)); if (ForFree) continue; retCode = ERROR_OUTOFMEMORY; nFileOp.fAnyOperationsAborted = TRUE; SetLastError(retCode); return retCode; } } lpFileOp->hNameMappings = nFileOp.hNameMappings; lpFileOp->fAnyOperationsAborted = nFileOp.fAnyOperationsAborted; return retCode; } #define ERROR_SHELL_INTERNAL_FILE_NOT_FOUND 1026 typedef struct { WIN32_FIND_DATAW wfd; LPWSTR szDirectory; LPWSTR szFilename; LPWSTR szFullPath; BOOL bFromWildcard; BOOL bFromRelative; BOOL bExists; } FILE_ENTRY; typedef struct { FILE_ENTRY *feFiles; DWORD dwNumFiles; BOOL bAnyFromWildcard; BOOL bAnyDirectories; BOOL bAnyDontExist; } FILE_LIST; /* count the number of expanded files from a given wildcard */ static DWORD count_wildcard_files(LPCWSTR szWildFile) { WIN32_FIND_DATAW wfd; HANDLE hFile = FindFirstFileW(szWildFile, &wfd); BOOL res = TRUE; DWORD dwCount = 0; if (hFile == INVALID_HANDLE_VALUE) return 0; while (res) { dwCount++; res = FindNextFileW(hFile, &wfd); } FindClose(hFile); return dwCount; } /* counts the number of files in a file list including expanded wildcard files */ static DWORD count_files(LPCWSTR szFileList) { DWORD dwCount = 0; LPCWSTR p = szFileList; LPCWSTR q = p + 1; LPCWSTR str = p; /* test empty list */ if (!szFileList[0] && !szFileList[1]) return -1; /* p,q search: stop when we reach double null terminator */ while (*p || *q) { if (!*q) { if (StrPBrkW(str, wWildcardChars)) dwCount += count_wildcard_files(str); else dwCount++; str = q + 1; } p++; q++; } return dwCount; } /* adds a file to the FILE_ENTRY struct * returns TRUE if szFile is a directory, FALSE otherwise */ static BOOL add_file_to_entry(FILE_ENTRY *feFile, LPWSTR szFile, BOOL bFromWildcard) { DWORD dwLen = strlenW(szFile) + 1; LPWSTR ptr; feFile->szFullPath = HeapAlloc(GetProcessHeap(), 0, dwLen * sizeof(WCHAR)); strcpyW(feFile->szFullPath, szFile); ptr = StrRChrW(szFile, NULL, '\\'); if (ptr) { dwLen = ptr - szFile + 1; feFile->szDirectory = HeapAlloc(GetProcessHeap(), 0, dwLen * sizeof(WCHAR)); lstrcpynW(feFile->szDirectory, szFile, dwLen); dwLen = strlenW(feFile->szFullPath) - dwLen + 1; feFile->szFilename = HeapAlloc(GetProcessHeap(), 0, dwLen * sizeof(WCHAR)); strcpyW(feFile->szFilename, ptr + 1); /* skip over backslash */ } feFile->bFromWildcard = bFromWildcard; if (FindFirstFileW(feFile->szFullPath, &feFile->wfd) != INVALID_HANDLE_VALUE) { if (IsAttribDir(feFile->wfd.dwFileAttributes)) return TRUE; } else feFile->wfd.dwFileAttributes &= ~FILE_ATTRIBUTE_DIRECTORY; return FALSE; } static LPWSTR wildcard_to_file(LPWSTR szWildCard, LPWSTR szFileName) { LPWSTR szFullPath, ptr; DWORD dwDirLen, dwFullLen; ptr = StrRChrW(szWildCard, NULL, '\\'); dwDirLen = ptr - szWildCard + 1; dwFullLen = dwDirLen + strlenW(szFileName) + 1; szFullPath = HeapAlloc(GetProcessHeap(), 0, dwFullLen * sizeof(WCHAR)); lstrcpynW(szFullPath, szWildCard, dwDirLen + 1); strcatW(szFullPath, szFileName); return szFullPath; } static void parse_wildcard_files(FILE_LIST *flList, LPWSTR szFile, LPDWORD pdwListIndex) { WIN32_FIND_DATAW wfd; HANDLE hFile = FindFirstFileW(szFile, &wfd); LPWSTR szFullPath; BOOL res = TRUE, bDir; while (res) { szFullPath = wildcard_to_file(szFile, wfd.cFileName); bDir = add_file_to_entry(&flList->feFiles[(*pdwListIndex)++], szFullPath, TRUE); HeapFree(GetProcessHeap(), 0, szFullPath); res = FindNextFileW(hFile, &wfd); if (bDir) flList->bAnyDirectories = TRUE; } CloseHandle(hFile); } /* takes the null-separated file list and fills out the FILE_LIST */ static HRESULT parse_file_list(FILE_LIST *flList, LPCWSTR szFiles) { LPCWSTR ptr = szFiles; WCHAR szCurFile[MAX_PATH]; DWORD i, dwDirLen; if (!szFiles) return ERROR_INVALID_PARAMETER; flList->bAnyFromWildcard = FALSE; flList->bAnyDirectories = FALSE; flList->bAnyDontExist = FALSE; flList->dwNumFiles = count_files(szFiles); /* empty list */ if (flList->dwNumFiles == -1) return ERROR_ACCESS_DENIED; flList->feFiles = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, flList->dwNumFiles * sizeof(FILE_ENTRY)); for (i = 0; i < flList->dwNumFiles; i++) { if (!*ptr) return ERROR_ACCESS_DENIED; /* change relative to absolute path */ if (PathIsRelativeW(ptr)) { dwDirLen = GetCurrentDirectoryW(MAX_PATH, szCurFile) + 1; PathCombineW(szCurFile, szCurFile, ptr); flList->feFiles[i].bFromRelative = TRUE; } else { strcpyW(szCurFile, ptr); flList->feFiles[i].bFromRelative = FALSE; } /* parse wildcard files if they are in the filename */ if (StrPBrkW(szCurFile, wWildcardChars)) { parse_wildcard_files(flList, szCurFile, &i); flList->bAnyFromWildcard = TRUE; i--; } else if (add_file_to_entry(&flList->feFiles[i], szCurFile, FALSE)) flList->bAnyDirectories = TRUE; /* record whether the file exists */ if (!PathFileExistsW(flList->feFiles[i].szFullPath)) { flList->feFiles[i].bExists = FALSE; flList->bAnyDontExist = TRUE; } else flList->feFiles[i].bExists = TRUE; /* advance to the next string */ ptr += strlenW(ptr) + 1; } return S_OK; } /* free the FILE_LIST */ static void destroy_file_list(FILE_LIST *flList) { DWORD i; if (!flList || !flList->feFiles) return; for (i = 0; i < flList->dwNumFiles; i++) { HeapFree(GetProcessHeap(), 0, flList->feFiles[i].szDirectory); HeapFree(GetProcessHeap(), 0, flList->feFiles[i].szFilename); HeapFree(GetProcessHeap(), 0, flList->feFiles[i].szFullPath); } HeapFree(GetProcessHeap(), 0, flList->feFiles); } static void copy_dir_to_dir(LPSHFILEOPSTRUCTW lpFileOp, FILE_ENTRY *feFrom, LPWSTR szDestPath) { WCHAR szFrom[MAX_PATH], szTo[MAX_PATH]; SHFILEOPSTRUCTW fileOp; static const WCHAR wildCardFiles[] = {'*','.','*',0}; if (IsDotDir(feFrom->szFilename)) return; if (PathFileExistsW(szDestPath)) PathCombineW(szTo, szDestPath, feFrom->szFilename); else strcpyW(szTo, szDestPath); szTo[strlenW(szTo) + 1] = '\0'; SHNotifyCreateDirectoryW(szTo, NULL); PathCombineW(szFrom, feFrom->szFullPath, wildCardFiles); szFrom[strlenW(szFrom) + 1] = '\0'; memcpy(&fileOp, lpFileOp, sizeof(fileOp)); fileOp.pFrom = szFrom; fileOp.pTo = szTo; fileOp.fFlags &= ~FOF_MULTIDESTFILES; /* we know we're copying to one dir */ SHFileOperationW(&fileOp); } /* copy a file or directory to another directory */ static void copy_to_dir(LPSHFILEOPSTRUCTW lpFileOp, FILE_ENTRY *feFrom, FILE_ENTRY *feTo) { WCHAR szDestPath[MAX_PATH]; if (!PathFileExistsW(feTo->szFullPath)) SHNotifyCreateDirectoryW(feTo->szFullPath, NULL); PathCombineW(szDestPath, feTo->szFullPath, feFrom->szFilename); if (IsAttribFile(feFrom->wfd.dwFileAttributes)) SHNotifyCopyFileW(feFrom->szFullPath, szDestPath, FALSE); else if (!(lpFileOp->fFlags & FOF_FILESONLY && feFrom->bFromWildcard)) copy_dir_to_dir(lpFileOp, feFrom, szDestPath); } static void create_dest_dirs(LPWSTR szDestDir) { WCHAR dir[MAX_PATH]; LPWSTR ptr = StrChrW(szDestDir, '\\'); /* make sure all directories up to last one are created */ while (ptr && (ptr = StrChrW(ptr + 1, '\\'))) { lstrcpynW(dir, szDestDir, ptr - szDestDir + 1); if (!PathFileExistsW(dir)) SHNotifyCreateDirectoryW(dir, NULL); } /* create last directory */ if (!PathFileExistsW(szDestDir)) SHNotifyCreateDirectoryW(szDestDir, NULL); } /* the FO_COPY operation */ static HRESULT copy_files(LPSHFILEOPSTRUCTW lpFileOp, FILE_LIST *flFrom, FILE_LIST *flTo) { DWORD i; FILE_ENTRY *entryToCopy; FILE_ENTRY *fileDest = &flTo->feFiles[0]; BOOL bCancelIfAnyDirectories = FALSE; if (flFrom->bAnyDontExist) return ERROR_SHELL_INTERNAL_FILE_NOT_FOUND; if (lpFileOp->fFlags & FOF_MULTIDESTFILES && flFrom->bAnyFromWildcard) return ERROR_CANCELLED; if (!(lpFileOp->fFlags & FOF_MULTIDESTFILES) && flTo->dwNumFiles != 1) return ERROR_CANCELLED; if (lpFileOp->fFlags & FOF_MULTIDESTFILES && flFrom->dwNumFiles != 1 && flFrom->dwNumFiles != flTo->dwNumFiles) { return ERROR_CANCELLED; } if (flFrom->dwNumFiles > 1 && flTo->dwNumFiles == 1 && !PathFileExistsW(flTo->feFiles[0].szFullPath) && IsAttribFile(fileDest->wfd.dwFileAttributes)) { bCancelIfAnyDirectories = TRUE; } if (flFrom->dwNumFiles > 1 && flTo->dwNumFiles == 1 && fileDest->bFromRelative && !PathFileExistsW(fileDest->szFullPath)) { lpFileOp->fAnyOperationsAborted = TRUE; return ERROR_CANCELLED; } if (!(lpFileOp->fFlags & FOF_MULTIDESTFILES) && flFrom->dwNumFiles != 1 && PathFileExistsW(fileDest->szFullPath) && IsAttribFile(fileDest->wfd.dwFileAttributes)) { return ERROR_CANCELLED; } for (i = 0; i < flFrom->dwNumFiles; i++) { entryToCopy = &flFrom->feFiles[i]; if (lpFileOp->fFlags & FOF_MULTIDESTFILES) fileDest = &flTo->feFiles[i]; if (IsAttribDir(entryToCopy->wfd.dwFileAttributes) && !lstrcmpW(entryToCopy->szFullPath, fileDest->szDirectory)) { return ERROR_SUCCESS; } if (IsAttribDir(entryToCopy->wfd.dwFileAttributes) && bCancelIfAnyDirectories) return ERROR_CANCELLED; create_dest_dirs(fileDest->szDirectory); if (!strcmpW(entryToCopy->szFullPath, fileDest->szFullPath)) { if (IsAttribFile(entryToCopy->wfd.dwFileAttributes)) return ERROR_NO_MORE_SEARCH_HANDLES; else return ERROR_SUCCESS; } if ((flFrom->dwNumFiles > 1 && flTo->dwNumFiles == 1) || (flFrom->dwNumFiles == 1 && IsAttribDir(fileDest->wfd.dwFileAttributes))) { copy_to_dir(lpFileOp, entryToCopy, fileDest); } else if (IsAttribDir(entryToCopy->wfd.dwFileAttributes)) { copy_dir_to_dir(lpFileOp, entryToCopy, fileDest->szFullPath); } else { if (SHNotifyCopyFileW(entryToCopy->szFullPath, fileDest->szFullPath, TRUE)) { lpFileOp->fAnyOperationsAborted = TRUE; return ERROR_CANCELLED; } } } return ERROR_SUCCESS; } /* the FO_DELETE operation */ static HRESULT delete_files(LPSHFILEOPSTRUCTW lpFileOp, FILE_LIST *flFrom) { FILE_ENTRY *fileEntry; DWORD i; BOOL bPathExists; BOOL bConfirm = !(lpFileOp->fFlags & FOF_NOCONFIRMATION); if (!flFrom->dwNumFiles) return ERROR_SUCCESS; for (i = 0; i < flFrom->dwNumFiles; i++) { bPathExists = TRUE; fileEntry = &flFrom->feFiles[i]; /* delete the file or directory */ if (IsAttribFile(fileEntry->wfd.dwFileAttributes)) bPathExists = DeleteFileW(fileEntry->szFullPath); else if (!(lpFileOp->fFlags & FOF_FILESONLY && fileEntry->bFromWildcard)) bPathExists = SHELL_DeleteDirectoryW(fileEntry->szFullPath, bConfirm); if (!bPathExists) return ERROR_PATH_NOT_FOUND; } return ERROR_SUCCESS; } static void move_dir_to_dir(LPSHFILEOPSTRUCTW lpFileOp, FILE_ENTRY *feFrom, LPWSTR szDestPath) { WCHAR szFrom[MAX_PATH], szTo[MAX_PATH]; SHFILEOPSTRUCTW fileOp; static const WCHAR wildCardFiles[] = {'*','.','*',0}; if (IsDotDir(feFrom->szFilename)) return; SHNotifyCreateDirectoryW(szDestPath, NULL); PathCombineW(szFrom, feFrom->szFullPath, wildCardFiles); szFrom[strlenW(szFrom) + 1] = '\0'; strcpyW(szTo, szDestPath); szTo[strlenW(szDestPath) + 1] = '\0'; memcpy(&fileOp, lpFileOp, sizeof(fileOp)); fileOp.pFrom = szFrom; fileOp.pTo = szTo; SHFileOperationW(&fileOp); } /* moves a file or directory to another directory */ static void move_to_dir(LPSHFILEOPSTRUCTW lpFileOp, FILE_ENTRY *feFrom, FILE_ENTRY *feTo) { WCHAR szDestPath[MAX_PATH]; PathCombineW(szDestPath, feTo->szFullPath, feFrom->szFilename); if (IsAttribFile(feFrom->wfd.dwFileAttributes)) SHNotifyMoveFileW(feFrom->szFullPath, szDestPath); else if (!(lpFileOp->fFlags & FOF_FILESONLY && feFrom->bFromWildcard)) move_dir_to_dir(lpFileOp, feFrom, szDestPath); } /* the FO_MOVE operation */ static HRESULT move_files(LPSHFILEOPSTRUCTW lpFileOp, FILE_LIST *flFrom, FILE_LIST *flTo) { DWORD i; FILE_ENTRY *entryToMove; FILE_ENTRY *fileDest; if (!flFrom->dwNumFiles || !flTo->dwNumFiles) return ERROR_CANCELLED; if (!(lpFileOp->fFlags & FOF_MULTIDESTFILES) && flTo->dwNumFiles > 1 && flFrom->dwNumFiles > 1) { return ERROR_CANCELLED; } if (!(lpFileOp->fFlags & FOF_MULTIDESTFILES) && !flFrom->bAnyDirectories && flFrom->dwNumFiles > flTo->dwNumFiles) { return ERROR_CANCELLED; } if (!PathFileExistsW(flTo->feFiles[0].szDirectory)) return ERROR_CANCELLED; if ((lpFileOp->fFlags & FOF_MULTIDESTFILES) && flFrom->dwNumFiles != flTo->dwNumFiles) { return ERROR_CANCELLED; } fileDest = &flTo->feFiles[0]; for (i = 0; i < flFrom->dwNumFiles; i++) { entryToMove = &flFrom->feFiles[i]; if (lpFileOp->fFlags & FOF_MULTIDESTFILES) fileDest = &flTo->feFiles[i]; if (!PathFileExistsW(fileDest->szDirectory)) return ERROR_CANCELLED; if (fileDest->bExists && IsAttribDir(fileDest->wfd.dwFileAttributes)) move_to_dir(lpFileOp, entryToMove, fileDest); else SHNotifyMoveFileW(entryToMove->szFullPath, fileDest->szFullPath); } return ERROR_SUCCESS; } /* the FO_RENAME files */ static HRESULT rename_files(LPSHFILEOPSTRUCTW lpFileOp, FILE_LIST *flFrom, FILE_LIST *flTo) { FILE_ENTRY *feFrom; FILE_ENTRY *feTo; if (flFrom->dwNumFiles != 1) return ERROR_GEN_FAILURE; if (flTo->dwNumFiles != 1) return ERROR_CANCELLED; feFrom = &flFrom->feFiles[0]; feTo= &flTo->feFiles[0]; /* fail if destination doesn't exist */ if (!feFrom->bExists) return ERROR_SHELL_INTERNAL_FILE_NOT_FOUND; /* fail if destination already exists */ if (feTo->bExists) return ERROR_ALREADY_EXISTS; return SHNotifyMoveFileW(feFrom->szFullPath, feTo->szFullPath); } /* alert the user if an unsupported flag is used */ static void check_flags(FILEOP_FLAGS fFlags) { WORD wUnsupportedFlags = FOF_ALLOWUNDO | FOF_NO_CONNECTED_ELEMENTS | FOF_NOCOPYSECURITYATTRIBS | FOF_NORECURSEREPARSE | FOF_WANTNUKEWARNING | FOF_RENAMEONCOLLISION | FOF_WANTMAPPINGHANDLE; if (fFlags & wUnsupportedFlags) FIXME("Unsupported flags: %04x\n", fFlags); } /************************************************************************* * SHFileOperationW [SHELL32.@] * * See SHFileOperationA */ int WINAPI SHFileOperationW(LPSHFILEOPSTRUCTW lpFileOp) { FILE_LIST flFrom, flTo; int ret = 0; if (!lpFileOp) return ERROR_INVALID_PARAMETER; check_flags(lpFileOp->fFlags); ZeroMemory(&flFrom, sizeof(FILE_LIST)); ZeroMemory(&flTo, sizeof(FILE_LIST)); if ((ret = parse_file_list(&flFrom, lpFileOp->pFrom))) return ret; if (lpFileOp->wFunc != FO_DELETE) parse_file_list(&flTo, lpFileOp->pTo); switch (lpFileOp->wFunc) { case FO_COPY: ret = copy_files(lpFileOp, &flFrom, &flTo); break; case FO_DELETE: ret = delete_files(lpFileOp, &flFrom); break; case FO_MOVE: ret = move_files(lpFileOp, &flFrom, &flTo); break; case FO_RENAME: ret = rename_files(lpFileOp, &flFrom, &flTo); break; default: ret = ERROR_INVALID_PARAMETER; break; } destroy_file_list(&flFrom); if (lpFileOp->wFunc != FO_DELETE) destroy_file_list(&flTo); if (ret == ERROR_CANCELLED) lpFileOp->fAnyOperationsAborted = TRUE; return ret; } #define SHDSA_GetItemCount(hdsa) (*(int*)(hdsa)) /************************************************************************* * SHFreeNameMappings [shell32.246] * * Free the mapping handle returned by SHFileoperation if FOF_WANTSMAPPINGHANDLE * was specified. * * PARAMS * hNameMapping [I] handle to the name mappings used during renaming of files * * RETURNS * Nothing */ void WINAPI SHFreeNameMappings(HANDLE hNameMapping) { if (hNameMapping) { int i = SHDSA_GetItemCount((HDSA)hNameMapping) - 1; for (; i>= 0; i--) { LPSHNAMEMAPPINGW lp = DSA_GetItemPtr((HDSA)hNameMapping, i); SHFree(lp->pszOldPath); SHFree(lp->pszNewPath); } DSA_Destroy((HDSA)hNameMapping); } } /************************************************************************* * SheGetDirA [SHELL32.@] * */ HRESULT WINAPI SheGetDirA(LPSTR u, LPSTR v) { FIXME("%p %p stub\n",u,v); return 0; } /************************************************************************* * SheGetDirW [SHELL32.@] * */ HRESULT WINAPI SheGetDirW(LPWSTR u, LPWSTR v) { FIXME("%p %p stub\n",u,v); return 0; } /************************************************************************* * SheChangeDirA [SHELL32.@] * */ HRESULT WINAPI SheChangeDirA(LPSTR u) { FIXME("(%s),stub\n",debugstr_a(u)); return 0; } /************************************************************************* * SheChangeDirW [SHELL32.@] * */ HRESULT WINAPI SheChangeDirW(LPWSTR u) { FIXME("(%s),stub\n",debugstr_w(u)); return 0; } /************************************************************************* * IsNetDrive [SHELL32.66] */ BOOL WINAPI IsNetDrive(DWORD drive) { char root[4]; strcpy(root, "A:\\"); root[0] += (char)drive; return (GetDriveTypeA(root) == DRIVE_REMOTE); } /************************************************************************* * RealDriveType [SHELL32.524] */ INT WINAPI RealDriveType(INT drive, BOOL bQueryNet) { char root[] = "A:\\"; root[0] += (char)drive; return GetDriveTypeA(root); }