wintrust: Return error directly from SOFTPUB_LoadFileMessage.
[wine] / dlls / shell32 / shlfileop.c
1 /*
2  * SHFileOperation
3  *
4  * Copyright 2000 Juergen Schmied
5  * Copyright 2002 Andriy Palamarchuk
6  * Copyright 2004 Dietrich Teickner (from Odin)
7  * Copyright 2004 Rolf Kalbermatter
8  *
9  * This library is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Lesser General Public
11  * License as published by the Free Software Foundation; either
12  * version 2.1 of the License, or (at your option) any later version.
13  *
14  * This library is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public
20  * License along with this library; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
22  */
23
24 #include "config.h"
25 #include "wine/port.h"
26
27 #include <stdarg.h>
28 #include <string.h>
29 #include <ctype.h>
30 #include <assert.h>
31
32 #include "windef.h"
33 #include "winbase.h"
34 #include "winreg.h"
35 #include "shellapi.h"
36 #include "wingdi.h"
37 #include "winuser.h"
38 #include "shlobj.h"
39 #include "shresdef.h"
40 #define NO_SHLWAPI_STREAM
41 #include "shlwapi.h"
42 #include "shell32_main.h"
43 #include "undocshell.h"
44 #include "wine/debug.h"
45 #include "xdg.h"
46
47 WINE_DEFAULT_DEBUG_CHANNEL(shell);
48
49 #define IsAttrib(x, y)  ((INVALID_FILE_ATTRIBUTES != (x)) && ((x) & (y)))
50 #define IsAttribFile(x) (!((x) & FILE_ATTRIBUTE_DIRECTORY))
51 #define IsAttribDir(x)  IsAttrib(x, FILE_ATTRIBUTE_DIRECTORY)
52 #define IsDotDir(x)     ((x[0] == '.') && ((x[1] == 0) || ((x[1] == '.') && (x[2] == 0))))
53
54 #define FO_MASK         0xF
55
56 static const WCHAR wWildcardFile[] = {'*',0};
57 static const WCHAR wWildcardChars[] = {'*','?',0};
58
59 static DWORD SHNotifyCreateDirectoryA(LPCSTR path, LPSECURITY_ATTRIBUTES sec);
60 static DWORD SHNotifyCreateDirectoryW(LPCWSTR path, LPSECURITY_ATTRIBUTES sec);
61 static DWORD SHNotifyRemoveDirectoryA(LPCSTR path);
62 static DWORD SHNotifyRemoveDirectoryW(LPCWSTR path);
63 static DWORD SHNotifyDeleteFileA(LPCSTR path);
64 static DWORD SHNotifyDeleteFileW(LPCWSTR path);
65 static DWORD SHNotifyMoveFileW(LPCWSTR src, LPCWSTR dest);
66 static DWORD SHNotifyCopyFileW(LPCWSTR src, LPCWSTR dest, BOOL bFailIfExists);
67 static DWORD SHFindAttrW(LPCWSTR pName, BOOL fileOnly);
68
69 typedef struct
70 {
71     SHFILEOPSTRUCTW *req;
72     DWORD dwYesToAllMask;
73     BOOL bManyItems;
74     BOOL bCancelled;
75 } FILE_OPERATION;
76
77 /* Confirm dialogs with an optional "Yes To All" as used in file operations confirmations
78  */
79 static const WCHAR CONFIRM_MSG_PROP[] = {'W','I','N','E','_','C','O','N','F','I','R','M',0};
80
81 struct confirm_msg_info
82 {
83     LPWSTR lpszText;
84     LPWSTR lpszCaption;
85     HICON hIcon;
86     BOOL bYesToAll;
87 };
88
89 /* as some buttons may be hidden and the dialog height may change we may need
90  * to move the controls */
91 static void confirm_msg_move_button(HWND hDlg, INT iId, INT *xPos, INT yOffset, BOOL bShow)
92 {
93     HWND hButton = GetDlgItem(hDlg, iId);
94     RECT r;
95
96     if (bShow) {
97         POINT pt;
98         int width;
99
100         GetWindowRect(hButton, &r);
101         width = r.right - r.left;
102         pt.x = r.left;
103         pt.y = r.top;
104         ScreenToClient(hDlg, &pt);
105         MoveWindow(hButton, *xPos - width, pt.y - yOffset, width, r.bottom - r.top, FALSE);
106         *xPos -= width + 5;
107     }
108     else
109         ShowWindow(hButton, SW_HIDE);
110 }
111
112 /* Note: we paint the text manually and don't use the static control to make
113  * sure the text has the same height as the one computed in WM_INITDIALOG
114  */
115 static INT_PTR ConfirmMsgBox_Paint(HWND hDlg)
116 {
117     PAINTSTRUCT ps;
118     HFONT hOldFont;
119     RECT r;
120     HDC hdc;
121
122     BeginPaint(hDlg, &ps);
123     hdc = ps.hdc;
124
125     GetClientRect(GetDlgItem(hDlg, IDD_MESSAGE), &r);
126     /* this will remap the rect to dialog coords */
127     MapWindowPoints(GetDlgItem(hDlg, IDD_MESSAGE), hDlg, (LPPOINT)&r, 2);
128     hOldFont = SelectObject(hdc, (HFONT)SendDlgItemMessageW(hDlg, IDD_MESSAGE, WM_GETFONT, 0, 0));
129     DrawTextW(hdc, GetPropW(hDlg, CONFIRM_MSG_PROP), -1, &r, DT_NOPREFIX | DT_PATH_ELLIPSIS | DT_WORDBREAK);
130     SelectObject(hdc, hOldFont);
131     EndPaint(hDlg, &ps);
132     return TRUE;
133 }
134
135 static INT_PTR ConfirmMsgBox_Init(HWND hDlg, LPARAM lParam)
136 {
137     struct confirm_msg_info *info = (struct confirm_msg_info *)lParam;
138     INT xPos, yOffset;
139     int width, height;
140     HFONT hOldFont;
141     HDC hdc;
142     RECT r;
143
144     SetWindowTextW(hDlg, info->lpszCaption);
145     ShowWindow(GetDlgItem(hDlg, IDD_MESSAGE), SW_HIDE);
146     SetPropW(hDlg, CONFIRM_MSG_PROP, info->lpszText);
147     SendDlgItemMessageW(hDlg, IDD_ICON, STM_SETICON, (WPARAM)info->hIcon, 0);
148
149     /* compute the text height and resize the dialog */
150     GetClientRect(GetDlgItem(hDlg, IDD_MESSAGE), &r);
151     hdc = GetDC(hDlg);
152     yOffset = r.bottom;
153     hOldFont = SelectObject(hdc, (HFONT)SendDlgItemMessageW(hDlg, IDD_MESSAGE, WM_GETFONT, 0, 0));
154     DrawTextW(hdc, info->lpszText, -1, &r, DT_NOPREFIX | DT_PATH_ELLIPSIS | DT_WORDBREAK | DT_CALCRECT);
155     SelectObject(hdc, hOldFont);
156     yOffset -= r.bottom;
157     yOffset = min(yOffset, 35);  /* don't make the dialog too small */
158     ReleaseDC(hDlg, hdc);
159
160     GetClientRect(hDlg, &r);
161     xPos = r.right - 7;
162     GetWindowRect(hDlg, &r);
163     width = r.right - r.left;
164     height = r.bottom - r.top - yOffset;
165     MoveWindow(hDlg, (GetSystemMetrics(SM_CXSCREEN) - width)/2,
166         (GetSystemMetrics(SM_CYSCREEN) - height)/2, width, height, FALSE);
167
168     confirm_msg_move_button(hDlg, IDCANCEL,     &xPos, yOffset, info->bYesToAll);
169     confirm_msg_move_button(hDlg, IDNO,         &xPos, yOffset, TRUE);
170     confirm_msg_move_button(hDlg, IDD_YESTOALL, &xPos, yOffset, info->bYesToAll);
171     confirm_msg_move_button(hDlg, IDYES,        &xPos, yOffset, TRUE);
172     return TRUE;
173 }
174
175 static INT_PTR CALLBACK ConfirmMsgBoxProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
176 {
177     switch (uMsg)
178     {
179         case WM_INITDIALOG:
180             return ConfirmMsgBox_Init(hDlg, lParam);
181         case WM_PAINT:
182             return ConfirmMsgBox_Paint(hDlg);
183         case WM_COMMAND:
184             EndDialog(hDlg, wParam);
185             break;
186         case WM_CLOSE:
187             EndDialog(hDlg, IDCANCEL);
188             break;
189     }
190     return FALSE;
191 }
192
193 static int SHELL_ConfirmMsgBox(HWND hWnd, LPWSTR lpszText, LPWSTR lpszCaption, HICON hIcon, BOOL bYesToAll)
194 {
195     static const WCHAR wszTemplate[] = {'S','H','E','L','L','_','Y','E','S','T','O','A','L','L','_','M','S','G','B','O','X',0};
196     struct confirm_msg_info info;
197
198     info.lpszText = lpszText;
199     info.lpszCaption = lpszCaption;
200     info.hIcon = hIcon;
201     info.bYesToAll = bYesToAll;
202     return DialogBoxParamW(shell32_hInstance, wszTemplate, hWnd, ConfirmMsgBoxProc, (LPARAM)&info);
203 }
204
205 /* confirmation dialogs content */
206 typedef struct
207 {
208         HINSTANCE hIconInstance;
209         UINT icon_resource_id;
210         UINT caption_resource_id, text_resource_id;
211 } SHELL_ConfirmIDstruc;
212
213 static BOOL SHELL_ConfirmIDs(int nKindOfDialog, SHELL_ConfirmIDstruc *ids)
214 {
215         ids->hIconInstance = shell32_hInstance;
216         switch (nKindOfDialog) {
217           case ASK_DELETE_FILE:
218             ids->icon_resource_id = IDI_SHELL_CONFIRM_DELETE;
219             ids->caption_resource_id  = IDS_DELETEITEM_CAPTION;
220             ids->text_resource_id  = IDS_DELETEITEM_TEXT;
221             return TRUE;
222           case ASK_DELETE_FOLDER:
223             ids->icon_resource_id = IDI_SHELL_CONFIRM_DELETE;
224             ids->caption_resource_id  = IDS_DELETEFOLDER_CAPTION;
225             ids->text_resource_id  = IDS_DELETEITEM_TEXT;
226             return TRUE;
227           case ASK_DELETE_MULTIPLE_ITEM:
228             ids->icon_resource_id = IDI_SHELL_CONFIRM_DELETE;
229             ids->caption_resource_id  = IDS_DELETEITEM_CAPTION;
230             ids->text_resource_id  = IDS_DELETEMULTIPLE_TEXT;
231             return TRUE;
232           case ASK_TRASH_FILE:
233             ids->icon_resource_id = IDI_SHELL_TRASH_FILE;
234             ids->caption_resource_id = IDS_DELETEITEM_CAPTION;
235             ids->text_resource_id = IDS_TRASHITEM_TEXT;
236             return TRUE;
237           case ASK_TRASH_FOLDER:
238             ids->icon_resource_id = IDI_SHELL_TRASH_FILE;
239             ids->caption_resource_id = IDS_DELETEFOLDER_CAPTION;
240             ids->text_resource_id = IDS_TRASHFOLDER_TEXT;
241             return TRUE;
242           case ASK_TRASH_MULTIPLE_ITEM:
243             ids->icon_resource_id = IDI_SHELL_TRASH_FILE;
244             ids->caption_resource_id = IDS_DELETEITEM_CAPTION;
245             ids->text_resource_id = IDS_TRASHMULTIPLE_TEXT;
246             return TRUE;
247           case ASK_CANT_TRASH_ITEM:
248             ids->icon_resource_id = IDI_SHELL_CONFIRM_DELETE;
249             ids->caption_resource_id  = IDS_DELETEITEM_CAPTION;
250             ids->text_resource_id  = IDS_CANTTRASH_TEXT;
251             return TRUE;
252           case ASK_DELETE_SELECTED:
253             ids->icon_resource_id = IDI_SHELL_CONFIRM_DELETE;
254             ids->caption_resource_id  = IDS_DELETEITEM_CAPTION;
255             ids->text_resource_id  = IDS_DELETESELECTED_TEXT;
256             return TRUE;
257           case ASK_OVERWRITE_FILE:
258             ids->hIconInstance = NULL;
259             ids->icon_resource_id = IDI_WARNING;
260             ids->caption_resource_id  = IDS_OVERWRITEFILE_CAPTION;
261             ids->text_resource_id  = IDS_OVERWRITEFILE_TEXT;
262             return TRUE;
263           case ASK_OVERWRITE_FOLDER:
264             ids->hIconInstance = NULL;
265             ids->icon_resource_id = IDI_WARNING;
266             ids->caption_resource_id  = IDS_OVERWRITEFILE_CAPTION;
267             ids->text_resource_id  = IDS_OVERWRITEFOLDER_TEXT;
268             return TRUE;
269           default:
270             FIXME(" Unhandled nKindOfDialog %d stub\n", nKindOfDialog);
271         }
272         return FALSE;
273 }
274
275 static BOOL SHELL_ConfirmDialogW(HWND hWnd, int nKindOfDialog, LPCWSTR szDir, FILE_OPERATION *op)
276 {
277         WCHAR szCaption[255], szText[255], szBuffer[MAX_PATH + 256];
278         SHELL_ConfirmIDstruc ids;
279         DWORD_PTR args[1];
280         HICON hIcon;
281         int ret;
282
283         assert(nKindOfDialog >= 0 && nKindOfDialog < 32);
284         if (op && (op->dwYesToAllMask & (1 << nKindOfDialog)))
285             return TRUE;
286
287         if (!SHELL_ConfirmIDs(nKindOfDialog, &ids)) return FALSE;
288
289         LoadStringW(shell32_hInstance, ids.caption_resource_id, szCaption, sizeof(szCaption)/sizeof(WCHAR));
290         LoadStringW(shell32_hInstance, ids.text_resource_id, szText, sizeof(szText)/sizeof(WCHAR));
291
292         args[0] = (DWORD_PTR)szDir;
293         FormatMessageW(FORMAT_MESSAGE_FROM_STRING|FORMAT_MESSAGE_ARGUMENT_ARRAY,
294                        szText, 0, 0, szBuffer, sizeof(szBuffer), (__ms_va_list*)args);
295         hIcon = LoadIconW(ids.hIconInstance, (LPWSTR)MAKEINTRESOURCE(ids.icon_resource_id));
296
297         ret = SHELL_ConfirmMsgBox(hWnd, szBuffer, szCaption, hIcon, op && op->bManyItems);
298         if (op) {
299             if (ret == IDD_YESTOALL) {
300                 op->dwYesToAllMask |= (1 << nKindOfDialog);
301                 ret = IDYES;
302             }
303             if (ret == IDCANCEL)
304                 op->bCancelled = TRUE;
305             if (ret != IDYES)
306                 op->req->fAnyOperationsAborted = TRUE;
307         }
308         return ret == IDYES;
309 }
310
311 BOOL SHELL_ConfirmYesNoW(HWND hWnd, int nKindOfDialog, LPCWSTR szDir)
312 {
313     return SHELL_ConfirmDialogW(hWnd, nKindOfDialog, szDir, NULL);
314 }
315
316 static DWORD SHELL32_AnsiToUnicodeBuf(LPCSTR aPath, LPWSTR *wPath, DWORD minChars)
317 {
318         DWORD len = MultiByteToWideChar(CP_ACP, 0, aPath, -1, NULL, 0);
319
320         if (len < minChars)
321           len = minChars;
322
323         *wPath = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
324         if (*wPath)
325         {
326           MultiByteToWideChar(CP_ACP, 0, aPath, -1, *wPath, len);
327           return NO_ERROR;
328         }
329         return E_OUTOFMEMORY;
330 }
331
332 static void SHELL32_FreeUnicodeBuf(LPWSTR wPath)
333 {
334         HeapFree(GetProcessHeap(), 0, wPath);
335 }
336
337 HRESULT WINAPI SHIsFileAvailableOffline(LPCWSTR path, LPDWORD status)
338 {
339     FIXME("(%s, %p) stub\n", debugstr_w(path), status);
340     return E_FAIL;
341 }
342
343 /**************************************************************************
344  * SHELL_DeleteDirectory()  [internal]
345  *
346  * Asks for confirmation when bShowUI is true and deletes the directory and
347  * all its subdirectories and files if necessary.
348  */
349 static BOOL SHELL_DeleteDirectoryW(HWND hwnd, LPCWSTR pszDir, BOOL bShowUI)
350 {
351         BOOL    ret = TRUE;
352         HANDLE  hFind;
353         WIN32_FIND_DATAW wfd;
354         WCHAR   szTemp[MAX_PATH];
355
356         /* Make sure the directory exists before eventually prompting the user */
357         PathCombineW(szTemp, pszDir, wWildcardFile);
358         hFind = FindFirstFileW(szTemp, &wfd);
359         if (hFind == INVALID_HANDLE_VALUE)
360           return FALSE;
361
362         if (!bShowUI || (ret = SHELL_ConfirmDialogW(hwnd, ASK_DELETE_FOLDER, pszDir, NULL)))
363         {
364           do
365           {
366             if (IsDotDir(wfd.cFileName))
367               continue;
368             PathCombineW(szTemp, pszDir, wfd.cFileName);
369             if (FILE_ATTRIBUTE_DIRECTORY & wfd.dwFileAttributes)
370               ret = SHELL_DeleteDirectoryW(hwnd, szTemp, FALSE);
371             else
372               ret = (SHNotifyDeleteFileW(szTemp) == ERROR_SUCCESS);
373           } while (ret && FindNextFileW(hFind, &wfd));
374         }
375         FindClose(hFind);
376         if (ret)
377           ret = (SHNotifyRemoveDirectoryW(pszDir) == ERROR_SUCCESS);
378         return ret;
379 }
380
381 /**************************************************************************
382  * Win32CreateDirectory      [SHELL32.93]
383  *
384  * Creates a directory. Also triggers a change notify if one exists.
385  *
386  * PARAMS
387  *  path       [I]   path to directory to create
388  *
389  * RETURNS
390  *  TRUE if successful, FALSE otherwise
391  *
392  * NOTES
393  *  Verified on Win98 / IE 5 (SHELL32 4.72, March 1999 build) to be ANSI.
394  *  This is Unicode on NT/2000
395  */
396 static DWORD SHNotifyCreateDirectoryA(LPCSTR path, LPSECURITY_ATTRIBUTES sec)
397 {
398         LPWSTR wPath;
399         DWORD retCode;
400
401         TRACE("(%s, %p)\n", debugstr_a(path), sec);
402
403         retCode = SHELL32_AnsiToUnicodeBuf(path, &wPath, 0);
404         if (!retCode)
405         {
406           retCode = SHNotifyCreateDirectoryW(wPath, sec);
407           SHELL32_FreeUnicodeBuf(wPath);
408         }
409         return retCode;
410 }
411
412 /**********************************************************************/
413
414 static DWORD SHNotifyCreateDirectoryW(LPCWSTR path, LPSECURITY_ATTRIBUTES sec)
415 {
416         TRACE("(%s, %p)\n", debugstr_w(path), sec);
417
418         if (CreateDirectoryW(path, sec))
419         {
420           SHChangeNotify(SHCNE_MKDIR, SHCNF_PATHW, path, NULL);
421           return ERROR_SUCCESS;
422         }
423         return GetLastError();
424 }
425
426 /**********************************************************************/
427
428 BOOL WINAPI Win32CreateDirectoryAW(LPCVOID path, LPSECURITY_ATTRIBUTES sec)
429 {
430         if (SHELL_OsIsUnicode())
431           return (SHNotifyCreateDirectoryW(path, sec) == ERROR_SUCCESS);
432         return (SHNotifyCreateDirectoryA(path, sec) == ERROR_SUCCESS);
433 }
434
435 /************************************************************************
436  * Win32RemoveDirectory      [SHELL32.94]
437  *
438  * Deletes a directory. Also triggers a change notify if one exists.
439  *
440  * PARAMS
441  *  path       [I]   path to directory to delete
442  *
443  * RETURNS
444  *  TRUE if successful, FALSE otherwise
445  *
446  * NOTES
447  *  Verified on Win98 / IE 5 (SHELL32 4.72, March 1999 build) to be ANSI.
448  *  This is Unicode on NT/2000
449  */
450 static DWORD SHNotifyRemoveDirectoryA(LPCSTR path)
451 {
452         LPWSTR wPath;
453         DWORD retCode;
454
455         TRACE("(%s)\n", debugstr_a(path));
456
457         retCode = SHELL32_AnsiToUnicodeBuf(path, &wPath, 0);
458         if (!retCode)
459         {
460           retCode = SHNotifyRemoveDirectoryW(wPath);
461           SHELL32_FreeUnicodeBuf(wPath);
462         }
463         return retCode;
464 }
465
466 /***********************************************************************/
467
468 static DWORD SHNotifyRemoveDirectoryW(LPCWSTR path)
469 {
470         BOOL ret;
471         TRACE("(%s)\n", debugstr_w(path));
472
473         ret = RemoveDirectoryW(path);
474         if (!ret)
475         {
476           /* Directory may be write protected */
477           DWORD dwAttr = GetFileAttributesW(path);
478           if (IsAttrib(dwAttr, FILE_ATTRIBUTE_READONLY))
479             if (SetFileAttributesW(path, dwAttr & ~FILE_ATTRIBUTE_READONLY))
480               ret = RemoveDirectoryW(path);
481         }
482         if (ret)
483         {
484           SHChangeNotify(SHCNE_RMDIR, SHCNF_PATHW, path, NULL);
485           return ERROR_SUCCESS;
486         }
487         return GetLastError();
488 }
489
490 /***********************************************************************/
491
492 BOOL WINAPI Win32RemoveDirectoryAW(LPCVOID path)
493 {
494         if (SHELL_OsIsUnicode())
495           return (SHNotifyRemoveDirectoryW(path) == ERROR_SUCCESS);
496         return (SHNotifyRemoveDirectoryA(path) == ERROR_SUCCESS);
497 }
498
499 /************************************************************************
500  * Win32DeleteFile           [SHELL32.164]
501  *
502  * Deletes a file. Also triggers a change notify if one exists.
503  *
504  * PARAMS
505  *  path       [I]   path to file to delete
506  *
507  * RETURNS
508  *  TRUE if successful, FALSE otherwise
509  *
510  * NOTES
511  *  Verified on Win98 / IE 5 (SHELL32 4.72, March 1999 build) to be ANSI.
512  *  This is Unicode on NT/2000
513  */
514 static DWORD SHNotifyDeleteFileA(LPCSTR path)
515 {
516         LPWSTR wPath;
517         DWORD retCode;
518
519         TRACE("(%s)\n", debugstr_a(path));
520
521         retCode = SHELL32_AnsiToUnicodeBuf(path, &wPath, 0);
522         if (!retCode)
523         {
524           retCode = SHNotifyDeleteFileW(wPath);
525           SHELL32_FreeUnicodeBuf(wPath);
526         }
527         return retCode;
528 }
529
530 /***********************************************************************/
531
532 static DWORD SHNotifyDeleteFileW(LPCWSTR path)
533 {
534         BOOL ret;
535
536         TRACE("(%s)\n", debugstr_w(path));
537
538         ret = DeleteFileW(path);
539         if (!ret)
540         {
541           /* File may be write protected or a system file */
542           DWORD dwAttr = GetFileAttributesW(path);
543           if (IsAttrib(dwAttr, FILE_ATTRIBUTE_READONLY | FILE_ATTRIBUTE_SYSTEM))
544             if (SetFileAttributesW(path, dwAttr & ~(FILE_ATTRIBUTE_READONLY | FILE_ATTRIBUTE_SYSTEM)))
545               ret = DeleteFileW(path);
546         }
547         if (ret)
548         {
549           SHChangeNotify(SHCNE_DELETE, SHCNF_PATHW, path, NULL);
550           return ERROR_SUCCESS;
551         }
552         return GetLastError();
553 }
554
555 /***********************************************************************/
556
557 DWORD WINAPI Win32DeleteFileAW(LPCVOID path)
558 {
559         if (SHELL_OsIsUnicode())
560           return (SHNotifyDeleteFileW(path) == ERROR_SUCCESS);
561         return (SHNotifyDeleteFileA(path) == ERROR_SUCCESS);
562 }
563
564 /************************************************************************
565  * SHNotifyMoveFile          [internal]
566  *
567  * Moves a file. Also triggers a change notify if one exists.
568  *
569  * PARAMS
570  *  src        [I]   path to source file to move
571  *  dest       [I]   path to target file to move to
572  *
573  * RETURNS
574  *  ERORR_SUCCESS if successful
575  */
576 static DWORD SHNotifyMoveFileW(LPCWSTR src, LPCWSTR dest)
577 {
578         BOOL ret;
579
580         TRACE("(%s %s)\n", debugstr_w(src), debugstr_w(dest));
581
582         ret = MoveFileExW(src, dest, MOVEFILE_REPLACE_EXISTING);
583
584         /* MOVEFILE_REPLACE_EXISTING fails with dirs, so try MoveFile */
585         if (!ret)
586             ret = MoveFileW(src, dest);
587
588         if (!ret)
589         {
590           DWORD dwAttr;
591
592           dwAttr = SHFindAttrW(dest, FALSE);
593           if (INVALID_FILE_ATTRIBUTES == dwAttr)
594           {
595             /* Source file may be write protected or a system file */
596             dwAttr = GetFileAttributesW(src);
597             if (IsAttrib(dwAttr, FILE_ATTRIBUTE_READONLY | FILE_ATTRIBUTE_SYSTEM))
598               if (SetFileAttributesW(src, dwAttr & ~(FILE_ATTRIBUTE_READONLY | FILE_ATTRIBUTE_SYSTEM)))
599                 ret = MoveFileW(src, dest);
600           }
601         }
602         if (ret)
603         {
604           SHChangeNotify(SHCNE_RENAMEITEM, SHCNF_PATHW, src, dest);
605           return ERROR_SUCCESS;
606         }
607         return GetLastError();
608 }
609
610 /************************************************************************
611  * SHNotifyCopyFile          [internal]
612  *
613  * Copies a file. Also triggers a change notify if one exists.
614  *
615  * PARAMS
616  *  src           [I]   path to source file to move
617  *  dest          [I]   path to target file to move to
618  *  bFailIfExists [I]   if TRUE, the target file will not be overwritten if
619  *                      a file with this name already exists
620  *
621  * RETURNS
622  *  ERROR_SUCCESS if successful
623  */
624 static DWORD SHNotifyCopyFileW(LPCWSTR src, LPCWSTR dest, BOOL bFailIfExists)
625 {
626         BOOL ret;
627
628         TRACE("(%s %s %s)\n", debugstr_w(src), debugstr_w(dest), bFailIfExists ? "failIfExists" : "");
629
630         ret = CopyFileW(src, dest, bFailIfExists);
631         if (ret)
632         {
633           SHChangeNotify(SHCNE_CREATE, SHCNF_PATHW, dest, NULL);
634           return ERROR_SUCCESS;
635         }
636
637         return GetLastError();
638 }
639
640 /*************************************************************************
641  * SHCreateDirectory         [SHELL32.165]
642  *
643  * This function creates a file system folder whose fully qualified path is
644  * given by path. If one or more of the intermediate folders do not exist,
645  * they will be created as well.
646  *
647  * PARAMS
648  *  hWnd       [I]
649  *  path       [I]   path of directory to create
650  *
651  * RETURNS
652  *  ERROR_SUCCESS or one of the following values:
653  *  ERROR_BAD_PATHNAME if the path is relative
654  *  ERROR_FILE_EXISTS when a file with that name exists
655  *  ERROR_PATH_NOT_FOUND can't find the path, probably invalid
656  *  ERROR_INVALID_NAME if the path contains invalid chars
657  *  ERROR_ALREADY_EXISTS when the directory already exists
658  *  ERROR_FILENAME_EXCED_RANGE if the filename was to long to process
659  *
660  * NOTES
661  *  exported by ordinal
662  *  Win9x exports ANSI
663  *  WinNT/2000 exports Unicode
664  */
665 DWORD WINAPI SHCreateDirectory(HWND hWnd, LPCVOID path)
666 {
667         if (SHELL_OsIsUnicode())
668           return SHCreateDirectoryExW(hWnd, path, NULL);
669         return SHCreateDirectoryExA(hWnd, path, NULL);
670 }
671
672 /*************************************************************************
673  * SHCreateDirectoryExA      [SHELL32.@]
674  *
675  * This function creates a file system folder whose fully qualified path is
676  * given by path. If one or more of the intermediate folders do not exist,
677  * they will be created as well.
678  *
679  * PARAMS
680  *  hWnd       [I]
681  *  path       [I]   path of directory to create
682  *  sec        [I]   security attributes to use or NULL
683  *
684  * RETURNS
685  *  ERROR_SUCCESS or one of the following values:
686  *  ERROR_BAD_PATHNAME or ERROR_PATH_NOT_FOUND if the path is relative
687  *  ERROR_INVALID_NAME if the path contains invalid chars
688  *  ERROR_FILE_EXISTS when a file with that name exists
689  *  ERROR_ALREADY_EXISTS when the directory already exists
690  *  ERROR_FILENAME_EXCED_RANGE if the filename was to long to process
691  *
692  *  FIXME: Not implemented yet;
693  *  SHCreateDirectoryEx also verifies that the files in the directory will be visible
694  *  if the path is a network path to deal with network drivers which might have a limited
695  *  but unknown maximum path length. If not:
696  *
697  *  If hWnd is set to a valid window handle, a message box is displayed warning
698  *  the user that the files may not be accessible. If the user chooses not to
699  *  proceed, the function returns ERROR_CANCELLED.
700  *
701  *  If hWnd is set to NULL, no user interface is displayed and the function
702  *  returns ERROR_CANCELLED.
703  */
704 int WINAPI SHCreateDirectoryExA(HWND hWnd, LPCSTR path, LPSECURITY_ATTRIBUTES sec)
705 {
706         LPWSTR wPath;
707         DWORD retCode;
708
709         TRACE("(%s, %p)\n", debugstr_a(path), sec);
710
711         retCode = SHELL32_AnsiToUnicodeBuf(path, &wPath, 0);
712         if (!retCode)
713         {
714           retCode = SHCreateDirectoryExW(hWnd, wPath, sec);
715           SHELL32_FreeUnicodeBuf(wPath);
716         }
717         return retCode;
718 }
719
720 /*************************************************************************
721  * SHCreateDirectoryExW      [SHELL32.@]
722  *
723  * See SHCreateDirectoryExA.
724  */
725 int WINAPI SHCreateDirectoryExW(HWND hWnd, LPCWSTR path, LPSECURITY_ATTRIBUTES sec)
726 {
727         int ret = ERROR_BAD_PATHNAME;
728         TRACE("(%p, %s, %p)\n", hWnd, debugstr_w(path), sec);
729
730         if (PathIsRelativeW(path))
731         {
732           SetLastError(ret);
733         }
734         else
735         {
736           ret = SHNotifyCreateDirectoryW(path, sec);
737           /* Refuse to work on certain error codes before trying to create directories recursively */
738           if (ret != ERROR_SUCCESS &&
739               ret != ERROR_FILE_EXISTS &&
740               ret != ERROR_ALREADY_EXISTS &&
741               ret != ERROR_FILENAME_EXCED_RANGE)
742           {
743             WCHAR *pEnd, *pSlash, szTemp[MAX_PATH + 1];  /* extra for PathAddBackslash() */
744
745             lstrcpynW(szTemp, path, MAX_PATH);
746             pEnd = PathAddBackslashW(szTemp);
747             pSlash = szTemp + 3;
748
749             while (*pSlash)
750             {
751               while (*pSlash && *pSlash != '\\') pSlash++;
752               if (*pSlash)
753               {
754                 *pSlash = 0;    /* terminate path at separator */
755
756                 ret = SHNotifyCreateDirectoryW(szTemp, pSlash + 1 == pEnd ? sec : NULL);
757               }
758               *pSlash++ = '\\'; /* put the separator back */
759             }
760           }
761
762           if (ret && hWnd && (ERROR_CANCELLED != ret))
763           {
764             /* We failed and should show a dialog box */
765             FIXME("Show system error message, creating path %s, failed with error %d\n", debugstr_w(path), ret);
766             ret = ERROR_CANCELLED; /* Error has been already presented to user (not really yet!) */
767           }
768         }
769         return ret;
770 }
771
772 /*************************************************************************
773  * SHFindAttrW      [internal]
774  *
775  * Get the Attributes for a file or directory. The difference to GetAttributes()
776  * is that this function will also work for paths containing wildcard characters
777  * in its filename.
778
779  * PARAMS
780  *  path       [I]   path of directory or file to check
781  *  fileOnly   [I]   TRUE if only files should be found
782  *
783  * RETURNS
784  *  INVALID_FILE_ATTRIBUTES if the path does not exist, the actual attributes of
785  *  the first file or directory found otherwise
786  */
787 static DWORD SHFindAttrW(LPCWSTR pName, BOOL fileOnly)
788 {
789         WIN32_FIND_DATAW wfd;
790         BOOL b_FileMask = fileOnly && (NULL != StrPBrkW(pName, wWildcardChars));
791         DWORD dwAttr = INVALID_FILE_ATTRIBUTES;
792         HANDLE hFind = FindFirstFileW(pName, &wfd);
793
794         TRACE("%s %d\n", debugstr_w(pName), fileOnly);
795         if (INVALID_HANDLE_VALUE != hFind)
796         {
797           do
798           {
799             if (b_FileMask && IsAttribDir(wfd.dwFileAttributes))
800                continue;
801             dwAttr = wfd.dwFileAttributes;
802             break;
803           }
804           while (FindNextFileW(hFind, &wfd));
805           FindClose(hFind);
806         }
807         return dwAttr;
808 }
809
810 /*************************************************************************
811  *
812  * SHNameTranslate HelperFunction for SHFileOperationA
813  *
814  * Translates a list of 0 terminated ASCII strings into Unicode. If *wString
815  * is NULL, only the necessary size of the string is determined and returned,
816  * otherwise the ASCII strings are copied into it and the buffer is increased
817  * to point to the location after the final 0 termination char.
818  */
819 static DWORD SHNameTranslate(LPWSTR* wString, LPCWSTR* pWToFrom, BOOL more)
820 {
821         DWORD size = 0, aSize = 0;
822         LPCSTR aString = (LPCSTR)*pWToFrom;
823
824         if (aString)
825         {
826           do
827           {
828             size = lstrlenA(aString) + 1;
829             aSize += size;
830             aString += size;
831           } while ((size != 1) && more);
832           /* The two sizes might be different in the case of multibyte chars */
833           size = MultiByteToWideChar(CP_ACP, 0, (LPCSTR)*pWToFrom, aSize, *wString, 0);
834           if (*wString) /* only in the second loop */
835           {
836             MultiByteToWideChar(CP_ACP, 0, (LPCSTR)*pWToFrom, aSize, *wString, size);
837             *pWToFrom = *wString;
838             *wString += size;
839           }
840         }
841         return size;
842 }
843 /*************************************************************************
844  * SHFileOperationA          [SHELL32.@]
845  *
846  * Function to copy, move, delete and create one or more files with optional
847  * user prompts.
848  *
849  * PARAMS
850  *  lpFileOp   [I/O] pointer to a structure containing all the necessary information
851  *
852  * RETURNS
853  *  Success: ERROR_SUCCESS.
854  *  Failure: ERROR_CANCELLED.
855  *
856  * NOTES
857  *  exported by name
858  */
859 int WINAPI SHFileOperationA(LPSHFILEOPSTRUCTA lpFileOp)
860 {
861         SHFILEOPSTRUCTW nFileOp = *((LPSHFILEOPSTRUCTW)lpFileOp);
862         int retCode = 0;
863         DWORD size;
864         LPWSTR ForFree = NULL, /* we change wString in SHNameTranslate and can't use it for freeing */
865                wString = NULL; /* we change this in SHNameTranslate */
866
867         TRACE("\n");
868         if (FO_DELETE == (nFileOp.wFunc & FO_MASK))
869           nFileOp.pTo = NULL; /* we need a NULL or a valid pointer for translation */
870         if (!(nFileOp.fFlags & FOF_SIMPLEPROGRESS))
871           nFileOp.lpszProgressTitle = NULL; /* we need a NULL or a valid pointer for translation */
872         while (1) /* every loop calculate size, second translate also, if we have storage for this */
873         {
874           size = SHNameTranslate(&wString, &nFileOp.lpszProgressTitle, FALSE); /* no loop */
875           size += SHNameTranslate(&wString, &nFileOp.pFrom, TRUE); /* internal loop */
876           size += SHNameTranslate(&wString, &nFileOp.pTo, TRUE); /* internal loop */
877
878           if (ForFree)
879           {
880             retCode = SHFileOperationW(&nFileOp);
881             HeapFree(GetProcessHeap(), 0, ForFree); /* we cannot use wString, it was changed */
882             break;
883           }
884           else
885           {
886             wString = ForFree = HeapAlloc(GetProcessHeap(), 0, size * sizeof(WCHAR));
887             if (ForFree) continue;
888             retCode = ERROR_OUTOFMEMORY;
889             nFileOp.fAnyOperationsAborted = TRUE;
890             SetLastError(retCode);
891             return retCode;
892           }
893         }
894
895         lpFileOp->hNameMappings = nFileOp.hNameMappings;
896         lpFileOp->fAnyOperationsAborted = nFileOp.fAnyOperationsAborted;
897         return retCode;
898 }
899
900 #define ERROR_SHELL_INTERNAL_FILE_NOT_FOUND 1026
901
902 typedef struct
903 {
904     DWORD attributes;
905     LPWSTR szDirectory;
906     LPWSTR szFilename;
907     LPWSTR szFullPath;
908     BOOL bFromWildcard;
909     BOOL bFromRelative;
910     BOOL bExists;
911 } FILE_ENTRY;
912
913 typedef struct
914 {
915     FILE_ENTRY *feFiles;
916     DWORD num_alloc;
917     DWORD dwNumFiles;
918     BOOL bAnyFromWildcard;
919     BOOL bAnyDirectories;
920     BOOL bAnyDontExist;
921 } FILE_LIST;
922
923
924 static inline void grow_list(FILE_LIST *list)
925 {
926     FILE_ENTRY *new = HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, list->feFiles,
927                                   list->num_alloc * 2 * sizeof(*new) );
928     list->feFiles = new;
929     list->num_alloc *= 2;
930 }
931
932 /* adds a file to the FILE_ENTRY struct
933  */
934 static void add_file_to_entry(FILE_ENTRY *feFile, LPCWSTR szFile)
935 {
936     DWORD dwLen = lstrlenW(szFile) + 1;
937     LPCWSTR ptr;
938
939     feFile->szFullPath = HeapAlloc(GetProcessHeap(), 0, dwLen * sizeof(WCHAR));
940     lstrcpyW(feFile->szFullPath, szFile);
941
942     ptr = StrRChrW(szFile, NULL, '\\');
943     if (ptr)
944     {
945         dwLen = ptr - szFile + 1;
946         feFile->szDirectory = HeapAlloc(GetProcessHeap(), 0, dwLen * sizeof(WCHAR));
947         lstrcpynW(feFile->szDirectory, szFile, dwLen);
948
949         dwLen = lstrlenW(feFile->szFullPath) - dwLen + 1;
950         feFile->szFilename = HeapAlloc(GetProcessHeap(), 0, dwLen * sizeof(WCHAR));
951         lstrcpyW(feFile->szFilename, ptr + 1); /* skip over backslash */
952     }
953     feFile->bFromWildcard = FALSE;
954 }
955
956 static LPWSTR wildcard_to_file(LPCWSTR szWildCard, LPCWSTR szFileName)
957 {
958     LPCWSTR ptr;
959     LPWSTR szFullPath;
960     DWORD dwDirLen, dwFullLen;
961
962     ptr = StrRChrW(szWildCard, NULL, '\\');
963     dwDirLen = ptr - szWildCard + 1;
964
965     dwFullLen = dwDirLen + lstrlenW(szFileName) + 1;
966     szFullPath = HeapAlloc(GetProcessHeap(), 0, dwFullLen * sizeof(WCHAR));
967
968     lstrcpynW(szFullPath, szWildCard, dwDirLen + 1);
969     lstrcatW(szFullPath, szFileName);
970
971     return szFullPath;
972 }
973
974 static void parse_wildcard_files(FILE_LIST *flList, LPCWSTR szFile, LPDWORD pdwListIndex)
975 {
976     WIN32_FIND_DATAW wfd;
977     HANDLE hFile = FindFirstFileW(szFile, &wfd);
978     FILE_ENTRY *file;
979     LPWSTR szFullPath;
980     BOOL res;
981
982     if (hFile == INVALID_HANDLE_VALUE) return;
983
984     for (res = TRUE; res; res = FindNextFileW(hFile, &wfd))
985     {
986         if (IsDotDir(wfd.cFileName)) continue;
987         if (*pdwListIndex >= flList->num_alloc) grow_list( flList );
988         szFullPath = wildcard_to_file(szFile, wfd.cFileName);
989         file = &flList->feFiles[(*pdwListIndex)++];
990         add_file_to_entry(file, szFullPath);
991         file->bFromWildcard = TRUE;
992         file->attributes = wfd.dwFileAttributes;
993         if (IsAttribDir(file->attributes)) flList->bAnyDirectories = TRUE;
994         HeapFree(GetProcessHeap(), 0, szFullPath);
995     }
996
997     FindClose(hFile);
998 }
999
1000 /* takes the null-separated file list and fills out the FILE_LIST */
1001 static HRESULT parse_file_list(FILE_LIST *flList, LPCWSTR szFiles)
1002 {
1003     LPCWSTR ptr = szFiles;
1004     WCHAR szCurFile[MAX_PATH];
1005     DWORD i = 0;
1006
1007     if (!szFiles)
1008         return ERROR_INVALID_PARAMETER;
1009
1010     flList->bAnyFromWildcard = FALSE;
1011     flList->bAnyDirectories = FALSE;
1012     flList->bAnyDontExist = FALSE;
1013     flList->num_alloc = 32;
1014     flList->dwNumFiles = 0;
1015
1016     /* empty list */
1017     if (!szFiles[0])
1018         return ERROR_ACCESS_DENIED;
1019         
1020     flList->feFiles = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
1021                                 flList->num_alloc * sizeof(FILE_ENTRY));
1022
1023     while (*ptr)
1024     {
1025         if (i >= flList->num_alloc) grow_list( flList );
1026
1027         /* change relative to absolute path */
1028         if (PathIsRelativeW(ptr))
1029         {
1030             GetCurrentDirectoryW(MAX_PATH, szCurFile);
1031             PathCombineW(szCurFile, szCurFile, ptr);
1032             flList->feFiles[i].bFromRelative = TRUE;
1033         }
1034         else
1035         {
1036             lstrcpyW(szCurFile, ptr);
1037             flList->feFiles[i].bFromRelative = FALSE;
1038         }
1039
1040         /* parse wildcard files if they are in the filename */
1041         if (StrPBrkW(szCurFile, wWildcardChars))
1042         {
1043             parse_wildcard_files(flList, szCurFile, &i);
1044             flList->bAnyFromWildcard = TRUE;
1045             i--;
1046         }
1047         else
1048         {
1049             FILE_ENTRY *file = &flList->feFiles[i];
1050             add_file_to_entry(file, szCurFile);
1051             file->attributes = GetFileAttributesW( file->szFullPath );
1052             file->bExists = (file->attributes != INVALID_FILE_ATTRIBUTES);
1053             if (!file->bExists) flList->bAnyDontExist = TRUE;
1054             if (IsAttribDir(file->attributes)) flList->bAnyDirectories = TRUE;
1055         }
1056
1057         /* advance to the next string */
1058         ptr += lstrlenW(ptr) + 1;
1059         i++;
1060     }
1061     flList->dwNumFiles = i;
1062
1063     return S_OK;
1064 }
1065
1066 /* free the FILE_LIST */
1067 static void destroy_file_list(FILE_LIST *flList)
1068 {
1069     DWORD i;
1070
1071     if (!flList || !flList->feFiles)
1072         return;
1073
1074     for (i = 0; i < flList->dwNumFiles; i++)
1075     {
1076         HeapFree(GetProcessHeap(), 0, flList->feFiles[i].szDirectory);
1077         HeapFree(GetProcessHeap(), 0, flList->feFiles[i].szFilename);
1078         HeapFree(GetProcessHeap(), 0, flList->feFiles[i].szFullPath);
1079     }
1080
1081     HeapFree(GetProcessHeap(), 0, flList->feFiles);
1082 }
1083
1084 static void copy_dir_to_dir(FILE_OPERATION *op, const FILE_ENTRY *feFrom, LPCWSTR szDestPath)
1085 {
1086     WCHAR szFrom[MAX_PATH], szTo[MAX_PATH];
1087     SHFILEOPSTRUCTW fileOp;
1088
1089     static const WCHAR wildCardFiles[] = {'*','.','*',0};
1090
1091     if (IsDotDir(feFrom->szFilename))
1092         return;
1093
1094     if (PathFileExistsW(szDestPath))
1095         PathCombineW(szTo, szDestPath, feFrom->szFilename);
1096     else
1097         lstrcpyW(szTo, szDestPath);
1098
1099     if (!(op->req->fFlags & FOF_NOCONFIRMATION) && PathFileExistsW(szTo)) {
1100         if (!SHELL_ConfirmDialogW(op->req->hwnd, ASK_OVERWRITE_FOLDER, feFrom->szFilename, op))
1101         {
1102             /* Vista returns an ERROR_CANCELLED even if user pressed "No" */
1103             if (!op->bManyItems)
1104                 op->bCancelled = TRUE;
1105             return;
1106         }
1107     }
1108
1109     szTo[lstrlenW(szTo) + 1] = '\0';
1110     SHNotifyCreateDirectoryW(szTo, NULL);
1111
1112     PathCombineW(szFrom, feFrom->szFullPath, wildCardFiles);
1113     szFrom[lstrlenW(szFrom) + 1] = '\0';
1114
1115     fileOp = *op->req;
1116     fileOp.pFrom = szFrom;
1117     fileOp.pTo = szTo;
1118     fileOp.fFlags &= ~FOF_MULTIDESTFILES; /* we know we're copying to one dir */
1119
1120     /* Don't ask the user about overwriting files when he accepted to overwrite the
1121        folder. FIXME: this is not exactly what Windows does - e.g. there would be
1122        an additional confirmation for a nested folder */
1123     fileOp.fFlags |= FOF_NOCONFIRMATION;  
1124
1125     SHFileOperationW(&fileOp);
1126 }
1127
1128 static BOOL copy_file_to_file(FILE_OPERATION *op, const WCHAR *szFrom, const WCHAR *szTo)
1129 {
1130     if (!(op->req->fFlags & FOF_NOCONFIRMATION) && PathFileExistsW(szTo))
1131     {
1132         if (!SHELL_ConfirmDialogW(op->req->hwnd, ASK_OVERWRITE_FILE, PathFindFileNameW(szTo), op))
1133             return 0;
1134     }
1135     
1136     return SHNotifyCopyFileW(szFrom, szTo, FALSE) == 0;
1137 }
1138
1139 /* copy a file or directory to another directory */
1140 static void copy_to_dir(FILE_OPERATION *op, const FILE_ENTRY *feFrom, const FILE_ENTRY *feTo)
1141 {
1142     if (!PathFileExistsW(feTo->szFullPath))
1143         SHNotifyCreateDirectoryW(feTo->szFullPath, NULL);
1144
1145     if (IsAttribFile(feFrom->attributes))
1146     {
1147         WCHAR szDestPath[MAX_PATH];
1148
1149         PathCombineW(szDestPath, feTo->szFullPath, feFrom->szFilename);
1150         copy_file_to_file(op, feFrom->szFullPath, szDestPath);
1151     }
1152     else if (!(op->req->fFlags & FOF_FILESONLY && feFrom->bFromWildcard))
1153         copy_dir_to_dir(op, feFrom, feTo->szFullPath);
1154 }
1155
1156 static void create_dest_dirs(LPCWSTR szDestDir)
1157 {
1158     WCHAR dir[MAX_PATH];
1159     LPCWSTR ptr = StrChrW(szDestDir, '\\');
1160
1161     /* make sure all directories up to last one are created */
1162     while (ptr && (ptr = StrChrW(ptr + 1, '\\')))
1163     {
1164         lstrcpynW(dir, szDestDir, ptr - szDestDir + 1);
1165
1166         if (!PathFileExistsW(dir))
1167             SHNotifyCreateDirectoryW(dir, NULL);
1168     }
1169
1170     /* create last directory */
1171     if (!PathFileExistsW(szDestDir))
1172         SHNotifyCreateDirectoryW(szDestDir, NULL);
1173 }
1174
1175 /* the FO_COPY operation */
1176 static HRESULT copy_files(FILE_OPERATION *op, const FILE_LIST *flFrom, FILE_LIST *flTo)
1177 {
1178     DWORD i;
1179     const FILE_ENTRY *entryToCopy;
1180     const FILE_ENTRY *fileDest = &flTo->feFiles[0];
1181
1182     if (flFrom->bAnyDontExist)
1183         return ERROR_SHELL_INTERNAL_FILE_NOT_FOUND;
1184
1185     if (op->req->fFlags & FOF_MULTIDESTFILES)
1186     {
1187         if (flFrom->bAnyFromWildcard)
1188             return ERROR_CANCELLED;
1189
1190         if (flFrom->dwNumFiles != flTo->dwNumFiles)
1191         {
1192             if (flFrom->dwNumFiles != 1 && !IsAttribDir(fileDest->attributes))
1193                 return ERROR_CANCELLED;
1194
1195             /* Free all but the first entry. */
1196             for (i = 1; i < flTo->dwNumFiles; i++)
1197             {
1198                 HeapFree(GetProcessHeap(), 0, flTo->feFiles[i].szDirectory);
1199                 HeapFree(GetProcessHeap(), 0, flTo->feFiles[i].szFilename);
1200                 HeapFree(GetProcessHeap(), 0, flTo->feFiles[i].szFullPath);
1201             }
1202
1203             flTo->dwNumFiles = 1;
1204         }
1205         else if (IsAttribDir(fileDest->attributes))
1206         {
1207             for (i = 1; i < flTo->dwNumFiles; i++)
1208                 if (!IsAttribDir(flTo->feFiles[i].attributes) ||
1209                     !IsAttribDir(flFrom->feFiles[i].attributes))
1210                 {
1211                     return ERROR_CANCELLED;
1212                 }
1213         }
1214     }
1215     else if (flFrom->dwNumFiles != 1)
1216     {
1217         if (flTo->dwNumFiles != 1 && !IsAttribDir(fileDest->attributes))
1218             return ERROR_CANCELLED;
1219
1220         if (PathFileExistsW(fileDest->szFullPath) &&
1221             IsAttribFile(fileDest->attributes))
1222         {
1223             return ERROR_CANCELLED;
1224         }
1225
1226         if (flTo->dwNumFiles == 1 && fileDest->bFromRelative &&
1227             !PathFileExistsW(fileDest->szFullPath))
1228         {
1229             return ERROR_CANCELLED;
1230         }
1231     }
1232
1233     for (i = 0; i < flFrom->dwNumFiles; i++)
1234     {
1235         entryToCopy = &flFrom->feFiles[i];
1236
1237         if ((op->req->fFlags & FOF_MULTIDESTFILES) &&
1238             flTo->dwNumFiles > 1)
1239         {
1240             fileDest = &flTo->feFiles[i];
1241         }
1242
1243         if (IsAttribDir(entryToCopy->attributes) &&
1244             !lstrcmpiW(entryToCopy->szFullPath, fileDest->szDirectory))
1245         {
1246             return ERROR_SUCCESS;
1247         }
1248
1249         create_dest_dirs(fileDest->szDirectory);
1250
1251         if (!lstrcmpiW(entryToCopy->szFullPath, fileDest->szFullPath))
1252         {
1253             if (IsAttribFile(entryToCopy->attributes))
1254                 return ERROR_NO_MORE_SEARCH_HANDLES;
1255             else
1256                 return ERROR_SUCCESS;
1257         }
1258
1259         if ((flFrom->dwNumFiles > 1 && flTo->dwNumFiles == 1) ||
1260             IsAttribDir(fileDest->attributes))
1261         {
1262             copy_to_dir(op, entryToCopy, fileDest);
1263         }
1264         else if (IsAttribDir(entryToCopy->attributes))
1265         {
1266             copy_dir_to_dir(op, entryToCopy, fileDest->szFullPath);
1267         }
1268         else
1269         {
1270             if (!copy_file_to_file(op, entryToCopy->szFullPath, fileDest->szFullPath))
1271             {
1272                 op->req->fAnyOperationsAborted = TRUE;
1273                 return ERROR_CANCELLED;
1274             }
1275         }
1276
1277         /* Vista return code. XP would return e.g. ERROR_FILE_NOT_FOUND, ERROR_ALREADY_EXISTS */
1278         if (op->bCancelled)
1279             return ERROR_CANCELLED;
1280     }
1281
1282     /* Vista return code. On XP if the used pressed "No" for the last item,
1283      * ERROR_ARENA_TRASHED would be returned */
1284     return ERROR_SUCCESS;
1285 }
1286
1287 static BOOL confirm_delete_list(HWND hWnd, DWORD fFlags, BOOL fTrash, const FILE_LIST *flFrom)
1288 {
1289     if (flFrom->dwNumFiles > 1)
1290     {
1291         WCHAR tmp[8];
1292         const WCHAR format[] = {'%','d',0};
1293
1294         wnsprintfW(tmp, sizeof(tmp)/sizeof(tmp[0]), format, flFrom->dwNumFiles);
1295         return SHELL_ConfirmDialogW(hWnd, (fTrash?ASK_TRASH_MULTIPLE_ITEM:ASK_DELETE_MULTIPLE_ITEM), tmp, NULL);
1296     }
1297     else
1298     {
1299         const FILE_ENTRY *fileEntry = &flFrom->feFiles[0];
1300
1301         if (IsAttribFile(fileEntry->attributes))
1302             return SHELL_ConfirmDialogW(hWnd, (fTrash?ASK_TRASH_FILE:ASK_DELETE_FILE), fileEntry->szFullPath, NULL);
1303         else if (!(fFlags & FOF_FILESONLY && fileEntry->bFromWildcard))
1304             return SHELL_ConfirmDialogW(hWnd, (fTrash?ASK_TRASH_FOLDER:ASK_DELETE_FOLDER), fileEntry->szFullPath, NULL);
1305     }
1306     return TRUE;
1307 }
1308
1309 /* the FO_DELETE operation */
1310 static HRESULT delete_files(LPSHFILEOPSTRUCTW lpFileOp, const FILE_LIST *flFrom)
1311 {
1312     const FILE_ENTRY *fileEntry;
1313     DWORD i;
1314     BOOL bPathExists;
1315     BOOL bTrash;
1316
1317     if (!flFrom->dwNumFiles)
1318         return ERROR_SUCCESS;
1319
1320     /* Windows also checks only the first item */
1321     bTrash = (lpFileOp->fFlags & FOF_ALLOWUNDO)
1322         && TRASH_CanTrashFile(flFrom->feFiles[0].szFullPath);
1323
1324     if (!(lpFileOp->fFlags & FOF_NOCONFIRMATION) || (!bTrash && lpFileOp->fFlags & FOF_WANTNUKEWARNING))
1325         if (!confirm_delete_list(lpFileOp->hwnd, lpFileOp->fFlags, bTrash, flFrom))
1326         {
1327             lpFileOp->fAnyOperationsAborted = TRUE;
1328             return 0;
1329         }
1330
1331     for (i = 0; i < flFrom->dwNumFiles; i++)
1332     {
1333         bPathExists = TRUE;
1334         fileEntry = &flFrom->feFiles[i];
1335
1336         if (!IsAttribFile(fileEntry->attributes) &&
1337             (lpFileOp->fFlags & FOF_FILESONLY && fileEntry->bFromWildcard))
1338             continue;
1339
1340         if (bTrash)
1341         {
1342             BOOL bDelete;
1343             if (TRASH_TrashFile(fileEntry->szFullPath))
1344                 continue;
1345
1346             /* Note: Windows silently deletes the file in such a situation, we show a dialog */
1347             if (!(lpFileOp->fFlags & FOF_NOCONFIRMATION) || (lpFileOp->fFlags & FOF_WANTNUKEWARNING))
1348                 bDelete = SHELL_ConfirmDialogW(lpFileOp->hwnd, ASK_CANT_TRASH_ITEM, fileEntry->szFullPath, NULL);
1349             else
1350                 bDelete = TRUE;
1351
1352             if (!bDelete)
1353             {
1354                 lpFileOp->fAnyOperationsAborted = TRUE;
1355                 break;
1356             }
1357         }
1358         
1359         /* delete the file or directory */
1360         if (IsAttribFile(fileEntry->attributes))
1361             bPathExists = DeleteFileW(fileEntry->szFullPath);
1362         else
1363             bPathExists = SHELL_DeleteDirectoryW(lpFileOp->hwnd, fileEntry->szFullPath, FALSE);
1364
1365         if (!bPathExists)
1366             return ERROR_PATH_NOT_FOUND;
1367     }
1368
1369     return ERROR_SUCCESS;
1370 }
1371
1372 static void move_dir_to_dir(LPSHFILEOPSTRUCTW lpFileOp, const FILE_ENTRY *feFrom, LPCWSTR szDestPath)
1373 {
1374     WCHAR szFrom[MAX_PATH], szTo[MAX_PATH];
1375     SHFILEOPSTRUCTW fileOp;
1376
1377     static const WCHAR wildCardFiles[] = {'*','.','*',0};
1378
1379     if (IsDotDir(feFrom->szFilename))
1380         return;
1381
1382     SHNotifyCreateDirectoryW(szDestPath, NULL);
1383
1384     PathCombineW(szFrom, feFrom->szFullPath, wildCardFiles);
1385     szFrom[lstrlenW(szFrom) + 1] = '\0';
1386
1387     lstrcpyW(szTo, szDestPath);
1388     szTo[lstrlenW(szDestPath) + 1] = '\0';
1389
1390     fileOp = *lpFileOp;
1391     fileOp.pFrom = szFrom;
1392     fileOp.pTo = szTo;
1393
1394     SHFileOperationW(&fileOp);
1395 }
1396
1397 /* moves a file or directory to another directory */
1398 static void move_to_dir(LPSHFILEOPSTRUCTW lpFileOp, const FILE_ENTRY *feFrom, const FILE_ENTRY *feTo)
1399 {
1400     WCHAR szDestPath[MAX_PATH];
1401
1402     PathCombineW(szDestPath, feTo->szFullPath, feFrom->szFilename);
1403
1404     if (IsAttribFile(feFrom->attributes))
1405         SHNotifyMoveFileW(feFrom->szFullPath, szDestPath);
1406     else if (!(lpFileOp->fFlags & FOF_FILESONLY && feFrom->bFromWildcard))
1407         move_dir_to_dir(lpFileOp, feFrom, szDestPath);
1408 }
1409
1410 /* the FO_MOVE operation */
1411 static HRESULT move_files(LPSHFILEOPSTRUCTW lpFileOp, const FILE_LIST *flFrom, const FILE_LIST *flTo)
1412 {
1413     DWORD i;
1414     const FILE_ENTRY *entryToMove;
1415     const FILE_ENTRY *fileDest;
1416
1417     if (!flFrom->dwNumFiles || !flTo->dwNumFiles)
1418         return ERROR_CANCELLED;
1419
1420     if (!(lpFileOp->fFlags & FOF_MULTIDESTFILES) &&
1421         flTo->dwNumFiles > 1 && flFrom->dwNumFiles > 1)
1422     {
1423         return ERROR_CANCELLED;
1424     }
1425
1426     if (!(lpFileOp->fFlags & FOF_MULTIDESTFILES) &&
1427         !flFrom->bAnyDirectories &&
1428         flFrom->dwNumFiles > flTo->dwNumFiles)
1429     {
1430         return ERROR_CANCELLED;
1431     }
1432
1433     if (!PathFileExistsW(flTo->feFiles[0].szDirectory))
1434         return ERROR_CANCELLED;
1435
1436     if ((lpFileOp->fFlags & FOF_MULTIDESTFILES) &&
1437         flFrom->dwNumFiles != flTo->dwNumFiles)
1438     {
1439         return ERROR_CANCELLED;
1440     }
1441
1442     fileDest = &flTo->feFiles[0];
1443     for (i = 0; i < flFrom->dwNumFiles; i++)
1444     {
1445         entryToMove = &flFrom->feFiles[i];
1446
1447         if (lpFileOp->fFlags & FOF_MULTIDESTFILES)
1448             fileDest = &flTo->feFiles[i];
1449
1450         if (!PathFileExistsW(fileDest->szDirectory))
1451             return ERROR_CANCELLED;
1452
1453         if (fileDest->bExists && IsAttribDir(fileDest->attributes))
1454             move_to_dir(lpFileOp, entryToMove, fileDest);
1455         else
1456             SHNotifyMoveFileW(entryToMove->szFullPath, fileDest->szFullPath);
1457     }
1458
1459     return ERROR_SUCCESS;
1460 }
1461
1462 /* the FO_RENAME files */
1463 static HRESULT rename_files(LPSHFILEOPSTRUCTW lpFileOp, const FILE_LIST *flFrom, const FILE_LIST *flTo)
1464 {
1465     const FILE_ENTRY *feFrom;
1466     const FILE_ENTRY *feTo;
1467
1468     if (flFrom->dwNumFiles != 1)
1469         return ERROR_GEN_FAILURE;
1470
1471     if (flTo->dwNumFiles != 1)
1472         return ERROR_CANCELLED;
1473
1474     feFrom = &flFrom->feFiles[0];
1475     feTo= &flTo->feFiles[0];
1476
1477     /* fail if destination doesn't exist */
1478     if (!feFrom->bExists)
1479         return ERROR_SHELL_INTERNAL_FILE_NOT_FOUND;
1480
1481     /* fail if destination already exists */
1482     if (feTo->bExists)
1483         return ERROR_ALREADY_EXISTS;
1484
1485     return SHNotifyMoveFileW(feFrom->szFullPath, feTo->szFullPath);
1486 }
1487
1488 /* alert the user if an unsupported flag is used */
1489 static void check_flags(FILEOP_FLAGS fFlags)
1490 {
1491     WORD wUnsupportedFlags = FOF_NO_CONNECTED_ELEMENTS |
1492         FOF_NOCOPYSECURITYATTRIBS | FOF_NORECURSEREPARSE |
1493         FOF_RENAMEONCOLLISION | FOF_WANTMAPPINGHANDLE;
1494
1495     if (fFlags & wUnsupportedFlags)
1496         FIXME("Unsupported flags: %04x\n", fFlags);
1497 }
1498
1499 /*************************************************************************
1500  * SHFileOperationW          [SHELL32.@]
1501  *
1502  * See SHFileOperationA
1503  */
1504 int WINAPI SHFileOperationW(LPSHFILEOPSTRUCTW lpFileOp)
1505 {
1506     FILE_OPERATION op;
1507     FILE_LIST flFrom, flTo;
1508     int ret = 0;
1509
1510     if (!lpFileOp)
1511         return ERROR_INVALID_PARAMETER;
1512
1513     check_flags(lpFileOp->fFlags);
1514
1515     ZeroMemory(&flFrom, sizeof(FILE_LIST));
1516     ZeroMemory(&flTo, sizeof(FILE_LIST));
1517
1518     if ((ret = parse_file_list(&flFrom, lpFileOp->pFrom)))
1519         return ret;
1520
1521     if (lpFileOp->wFunc != FO_DELETE)
1522         parse_file_list(&flTo, lpFileOp->pTo);
1523
1524     ZeroMemory(&op, sizeof(op));
1525     op.req = lpFileOp;
1526     op.bManyItems = (flFrom.dwNumFiles > 1);
1527
1528     switch (lpFileOp->wFunc)
1529     {
1530         case FO_COPY:
1531             ret = copy_files(&op, &flFrom, &flTo);
1532             break;
1533         case FO_DELETE:
1534             ret = delete_files(lpFileOp, &flFrom);
1535             break;
1536         case FO_MOVE:
1537             ret = move_files(lpFileOp, &flFrom, &flTo);
1538             break;
1539         case FO_RENAME:
1540             ret = rename_files(lpFileOp, &flFrom, &flTo);
1541             break;
1542         default:
1543             ret = ERROR_INVALID_PARAMETER;
1544             break;
1545     }
1546
1547     destroy_file_list(&flFrom);
1548
1549     if (lpFileOp->wFunc != FO_DELETE)
1550         destroy_file_list(&flTo);
1551
1552     if (ret == ERROR_CANCELLED)
1553         lpFileOp->fAnyOperationsAborted = TRUE;
1554
1555     return ret;
1556 }
1557
1558 #define SHDSA_GetItemCount(hdsa) (*(int*)(hdsa))
1559
1560 /*************************************************************************
1561  * SHFreeNameMappings      [shell32.246]
1562  *
1563  * Free the mapping handle returned by SHFileOperation if FOF_WANTSMAPPINGHANDLE
1564  * was specified.
1565  *
1566  * PARAMS
1567  *  hNameMapping [I] handle to the name mappings used during renaming of files
1568  *
1569  * RETURNS
1570  *  Nothing
1571  */
1572 void WINAPI SHFreeNameMappings(HANDLE hNameMapping)
1573 {
1574         if (hNameMapping)
1575         {
1576           int i = SHDSA_GetItemCount((HDSA)hNameMapping) - 1;
1577
1578           for (; i>= 0; i--)
1579           {
1580             LPSHNAMEMAPPINGW lp = DSA_GetItemPtr(hNameMapping, i);
1581
1582             SHFree(lp->pszOldPath);
1583             SHFree(lp->pszNewPath);
1584           }
1585           DSA_Destroy(hNameMapping);
1586         }
1587 }
1588
1589 /*************************************************************************
1590  * SheGetDirA [SHELL32.@]
1591  *
1592  * drive = 0: returns the current directory path
1593  * drive > 0: returns the current directory path of the specified drive
1594  *            drive=1 -> A:  drive=2 -> B:  ...
1595  * returns 0 if successful
1596 */
1597 DWORD WINAPI SheGetDirA(DWORD drive, LPSTR buffer)
1598 {
1599     WCHAR org_path[MAX_PATH];
1600     DWORD ret;
1601     char drv_path[3];
1602
1603     /* change current directory to the specified drive */
1604     if (drive) {
1605         strcpy(drv_path, "A:");
1606         drv_path[0] += (char)drive-1;
1607
1608         GetCurrentDirectoryW(MAX_PATH, org_path);
1609
1610         SetCurrentDirectoryA(drv_path);
1611     }
1612
1613     /* query current directory path of the specified drive */
1614     ret = GetCurrentDirectoryA(MAX_PATH, buffer);
1615
1616     /* back to the original drive */
1617     if (drive)
1618         SetCurrentDirectoryW(org_path);
1619
1620     if (!ret)
1621         return GetLastError();
1622
1623     return 0;
1624 }
1625
1626 /*************************************************************************
1627  * SheGetDirW [SHELL32.@]
1628  *
1629  * drive = 0: returns the current directory path
1630  * drive > 0: returns the current directory path of the specified drive
1631  *            drive=1 -> A:  drive=2 -> B:  ...
1632  * returns 0 if successful
1633  */
1634 DWORD WINAPI SheGetDirW(DWORD drive, LPWSTR buffer)
1635 {
1636     WCHAR org_path[MAX_PATH];
1637     DWORD ret;
1638     char drv_path[3];
1639
1640     /* change current directory to the specified drive */
1641     if (drive) {
1642         strcpy(drv_path, "A:");
1643         drv_path[0] += (char)drive-1;
1644
1645         GetCurrentDirectoryW(MAX_PATH, org_path);
1646
1647         SetCurrentDirectoryA(drv_path);
1648     }
1649
1650     /* query current directory path of the specified drive */
1651     ret = GetCurrentDirectoryW(MAX_PATH, buffer);
1652
1653     /* back to the original drive */
1654     if (drive)
1655         SetCurrentDirectoryW(org_path);
1656
1657     if (!ret)
1658         return GetLastError();
1659
1660     return 0;
1661 }
1662
1663 /*************************************************************************
1664  * SheChangeDirA [SHELL32.@]
1665  *
1666  * changes the current directory to the specified path
1667  * and returns 0 if successful
1668  */
1669 DWORD WINAPI SheChangeDirA(LPSTR path)
1670 {
1671     if (SetCurrentDirectoryA(path))
1672         return 0;
1673     else
1674         return GetLastError();
1675 }
1676
1677 /*************************************************************************
1678  * SheChangeDirW [SHELL32.@]
1679  *
1680  * changes the current directory to the specified path
1681  * and returns 0 if successful
1682  */
1683 DWORD WINAPI SheChangeDirW(LPWSTR path)
1684 {
1685     if (SetCurrentDirectoryW(path))
1686         return 0;
1687     else
1688         return GetLastError();
1689 }
1690
1691 /*************************************************************************
1692  * IsNetDrive                   [SHELL32.66]
1693  */
1694 BOOL WINAPI IsNetDrive(DWORD drive)
1695 {
1696         char root[4];
1697         strcpy(root, "A:\\");
1698         root[0] += (char)drive;
1699         return (GetDriveTypeA(root) == DRIVE_REMOTE);
1700 }
1701
1702
1703 /*************************************************************************
1704  * RealDriveType                [SHELL32.524]
1705  */
1706 INT WINAPI RealDriveType(INT drive, BOOL bQueryNet)
1707 {
1708     char root[] = "A:\\";
1709     root[0] += (char)drive;
1710     return GetDriveTypeA(root);
1711 }
1712
1713 /***********************************************************************
1714  *              SHPathPrepareForWriteA (SHELL32.@)
1715  */
1716 HRESULT WINAPI SHPathPrepareForWriteA(HWND hwnd, IUnknown *modless, LPCSTR path, DWORD flags)
1717 {
1718     WCHAR wpath[MAX_PATH];
1719     MultiByteToWideChar( CP_ACP, 0, path, -1, wpath, MAX_PATH);
1720     return SHPathPrepareForWriteW(hwnd, modless, wpath, flags);
1721 }
1722
1723 /***********************************************************************
1724  *              SHPathPrepareForWriteW (SHELL32.@)
1725  */
1726 HRESULT WINAPI SHPathPrepareForWriteW(HWND hwnd, IUnknown *modless, LPCWSTR path, DWORD flags)
1727 {
1728     DWORD res;
1729     DWORD err;
1730     LPCWSTR realpath;
1731     int len;
1732     WCHAR* last_slash;
1733     WCHAR* temppath=NULL;
1734
1735     TRACE("%p %p %s 0x%80x\n", hwnd, modless, debugstr_w(path), flags);
1736
1737     if (flags & ~(SHPPFW_DIRCREATE|SHPPFW_ASKDIRCREATE|SHPPFW_IGNOREFILENAME))
1738         FIXME("unimplemented flags 0x%08x\n", flags);
1739
1740     /* cut off filename if necessary */
1741     if (flags & SHPPFW_IGNOREFILENAME)
1742     {
1743         last_slash = StrRChrW(path, NULL, '\\');
1744         if (last_slash == NULL)
1745             len = 1;
1746         else
1747             len = last_slash - path + 1;
1748         temppath = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
1749         if (!temppath)
1750             return E_OUTOFMEMORY;
1751         StrCpyNW(temppath, path, len);
1752         realpath = temppath;
1753     }
1754     else
1755     {
1756         realpath = path;
1757     }
1758
1759     /* try to create the directory if asked to */
1760     if (flags & (SHPPFW_DIRCREATE|SHPPFW_ASKDIRCREATE))
1761     {
1762         if (flags & SHPPFW_ASKDIRCREATE)
1763             FIXME("treating SHPPFW_ASKDIRCREATE as SHPPFW_DIRCREATE\n");
1764
1765         SHCreateDirectoryExW(0, realpath, NULL);
1766     }
1767
1768     /* check if we can access the directory */
1769     res = GetFileAttributesW(realpath);
1770
1771     HeapFree(GetProcessHeap(), 0, temppath);
1772
1773     if (res == INVALID_FILE_ATTRIBUTES)
1774     {
1775         err = GetLastError();
1776         if (err == ERROR_FILE_NOT_FOUND)
1777             return HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND);
1778         return HRESULT_FROM_WIN32(err);
1779     }
1780     else if (res & FILE_ATTRIBUTE_DIRECTORY)
1781         return S_OK;
1782     else
1783         return HRESULT_FROM_WIN32(ERROR_DIRECTORY);
1784 }