wined3d: Make SRGB write correction working with 1.x shaders in arb.
[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 CALLBACK 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, (LPWSTR)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 CALLBACK 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, (HANDLE)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), (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 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  *  ERRROR_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_INVLID_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  *  ERRROR_SUCCESS or one of the following values:
686  *  ERROR_BAD_PATHNAME or ERROR_PATH_NOT_FOUND if the path is relative
687  *  ERROR_INVLID_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, dwDirLen;
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             dwDirLen = GetCurrentDirectoryW(MAX_PATH, szCurFile) + 1;
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     memcpy(&fileOp, op->req, sizeof(fileOp));
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, const FILE_LIST *flTo)
1177 {
1178     DWORD i;
1179     const FILE_ENTRY *entryToCopy;
1180     const FILE_ENTRY *fileDest = &flTo->feFiles[0];
1181     BOOL bCancelIfAnyDirectories = FALSE;
1182
1183     if (flFrom->bAnyDontExist)
1184         return ERROR_SHELL_INTERNAL_FILE_NOT_FOUND;
1185
1186     if (op->req->fFlags & FOF_MULTIDESTFILES && flFrom->bAnyFromWildcard)
1187         return ERROR_CANCELLED;
1188
1189     if (!(op->req->fFlags & FOF_MULTIDESTFILES) && flTo->dwNumFiles != 1)
1190         return ERROR_CANCELLED;
1191
1192     if (op->req->fFlags & FOF_MULTIDESTFILES && flFrom->dwNumFiles != 1 &&
1193         flFrom->dwNumFiles != flTo->dwNumFiles)
1194     {
1195         return ERROR_CANCELLED;
1196     }
1197
1198     if (flFrom->dwNumFiles > 1 && flTo->dwNumFiles == 1 &&
1199         !PathFileExistsW(flTo->feFiles[0].szFullPath) &&
1200         IsAttribFile(fileDest->attributes))
1201     {
1202         bCancelIfAnyDirectories = TRUE;
1203     }
1204
1205     if (flFrom->dwNumFiles > 1 && flTo->dwNumFiles == 1 && fileDest->bFromRelative &&
1206         !PathFileExistsW(fileDest->szFullPath))
1207     {
1208         op->req->fAnyOperationsAborted = TRUE;
1209         return ERROR_CANCELLED;
1210     }
1211
1212     if (!(op->req->fFlags & FOF_MULTIDESTFILES) && flFrom->dwNumFiles != 1 &&
1213         PathFileExistsW(fileDest->szFullPath) &&
1214         IsAttribFile(fileDest->attributes))
1215     {
1216         return ERROR_CANCELLED;
1217     }
1218
1219     for (i = 0; i < flFrom->dwNumFiles; i++)
1220     {
1221         entryToCopy = &flFrom->feFiles[i];
1222
1223         if (op->req->fFlags & FOF_MULTIDESTFILES)
1224             fileDest = &flTo->feFiles[i];
1225
1226         if (IsAttribDir(entryToCopy->attributes) &&
1227             !lstrcmpiW(entryToCopy->szFullPath, fileDest->szDirectory))
1228         {
1229             return ERROR_SUCCESS;
1230         }
1231
1232         if (IsAttribDir(entryToCopy->attributes) && bCancelIfAnyDirectories)
1233             return ERROR_CANCELLED;
1234
1235         create_dest_dirs(fileDest->szDirectory);
1236
1237         if (!lstrcmpiW(entryToCopy->szFullPath, fileDest->szFullPath))
1238         {
1239             if (IsAttribFile(entryToCopy->attributes))
1240                 return ERROR_NO_MORE_SEARCH_HANDLES;
1241             else
1242                 return ERROR_SUCCESS;
1243         }
1244
1245         if ((flFrom->dwNumFiles > 1 && flTo->dwNumFiles == 1) ||
1246             (flFrom->dwNumFiles == 1 && IsAttribDir(fileDest->attributes)))
1247         {
1248             copy_to_dir(op, entryToCopy, fileDest);
1249         }
1250         else if (IsAttribDir(entryToCopy->attributes))
1251         {
1252             copy_dir_to_dir(op, entryToCopy, fileDest->szFullPath);
1253         }
1254         else
1255         {
1256             if (!copy_file_to_file(op, entryToCopy->szFullPath, fileDest->szFullPath))
1257             {
1258                 op->req->fAnyOperationsAborted = TRUE;
1259                 return ERROR_CANCELLED;
1260             }
1261         }
1262
1263         /* Vista return code. XP would return e.g. ERROR_FILE_NOT_FOUND, ERROR_ALREADY_EXISTS */
1264         if (op->bCancelled)
1265             return ERROR_CANCELLED;
1266     }
1267
1268     /* Vista return code. On XP if the used pressed "No" for the last item,
1269      * ERROR_ARENA_TRASHED would be returned */
1270     return ERROR_SUCCESS;
1271 }
1272
1273 static BOOL confirm_delete_list(HWND hWnd, DWORD fFlags, BOOL fTrash, const FILE_LIST *flFrom)
1274 {
1275     if (flFrom->dwNumFiles > 1)
1276     {
1277         WCHAR tmp[8];
1278         const WCHAR format[] = {'%','d',0};
1279
1280         wnsprintfW(tmp, sizeof(tmp)/sizeof(tmp[0]), format, flFrom->dwNumFiles);
1281         return SHELL_ConfirmDialogW(hWnd, (fTrash?ASK_TRASH_MULTIPLE_ITEM:ASK_DELETE_MULTIPLE_ITEM), tmp, NULL);
1282     }
1283     else
1284     {
1285         const FILE_ENTRY *fileEntry = &flFrom->feFiles[0];
1286
1287         if (IsAttribFile(fileEntry->attributes))
1288             return SHELL_ConfirmDialogW(hWnd, (fTrash?ASK_TRASH_FILE:ASK_DELETE_FILE), fileEntry->szFullPath, NULL);
1289         else if (!(fFlags & FOF_FILESONLY && fileEntry->bFromWildcard))
1290             return SHELL_ConfirmDialogW(hWnd, (fTrash?ASK_TRASH_FOLDER:ASK_DELETE_FOLDER), fileEntry->szFullPath, NULL);
1291     }
1292     return TRUE;
1293 }
1294
1295 /* the FO_DELETE operation */
1296 static HRESULT delete_files(LPSHFILEOPSTRUCTW lpFileOp, const FILE_LIST *flFrom)
1297 {
1298     const FILE_ENTRY *fileEntry;
1299     DWORD i;
1300     BOOL bPathExists;
1301     BOOL bTrash;
1302
1303     if (!flFrom->dwNumFiles)
1304         return ERROR_SUCCESS;
1305
1306     /* Windows also checks only the first item */
1307     bTrash = (lpFileOp->fFlags & FOF_ALLOWUNDO)
1308         && TRASH_CanTrashFile(flFrom->feFiles[0].szFullPath);
1309
1310     if (!(lpFileOp->fFlags & FOF_NOCONFIRMATION) || (!bTrash && lpFileOp->fFlags & FOF_WANTNUKEWARNING))
1311         if (!confirm_delete_list(lpFileOp->hwnd, lpFileOp->fFlags, bTrash, flFrom))
1312         {
1313             lpFileOp->fAnyOperationsAborted = TRUE;
1314             return 0;
1315         }
1316
1317     for (i = 0; i < flFrom->dwNumFiles; i++)
1318     {
1319         bPathExists = TRUE;
1320         fileEntry = &flFrom->feFiles[i];
1321
1322         if (!IsAttribFile(fileEntry->attributes) &&
1323             (lpFileOp->fFlags & FOF_FILESONLY && fileEntry->bFromWildcard))
1324             continue;
1325
1326         if (bTrash)
1327         {
1328             BOOL bDelete;
1329             if (TRASH_TrashFile(fileEntry->szFullPath))
1330                 continue;
1331
1332             /* Note: Windows silently deletes the file in such a situation, we show a dialog */
1333             if (!(lpFileOp->fFlags & FOF_NOCONFIRMATION) || (lpFileOp->fFlags & FOF_WANTNUKEWARNING))
1334                 bDelete = SHELL_ConfirmDialogW(lpFileOp->hwnd, ASK_CANT_TRASH_ITEM, fileEntry->szFullPath, NULL);
1335             else
1336                 bDelete = TRUE;
1337
1338             if (!bDelete)
1339             {
1340                 lpFileOp->fAnyOperationsAborted = TRUE;
1341                 break;
1342             }
1343         }
1344         
1345         /* delete the file or directory */
1346         if (IsAttribFile(fileEntry->attributes))
1347             bPathExists = DeleteFileW(fileEntry->szFullPath);
1348         else
1349             bPathExists = SHELL_DeleteDirectoryW(lpFileOp->hwnd, fileEntry->szFullPath, FALSE);
1350
1351         if (!bPathExists)
1352             return ERROR_PATH_NOT_FOUND;
1353     }
1354
1355     return ERROR_SUCCESS;
1356 }
1357
1358 static void move_dir_to_dir(LPSHFILEOPSTRUCTW lpFileOp, const FILE_ENTRY *feFrom, LPCWSTR szDestPath)
1359 {
1360     WCHAR szFrom[MAX_PATH], szTo[MAX_PATH];
1361     SHFILEOPSTRUCTW fileOp;
1362
1363     static const WCHAR wildCardFiles[] = {'*','.','*',0};
1364
1365     if (IsDotDir(feFrom->szFilename))
1366         return;
1367
1368     SHNotifyCreateDirectoryW(szDestPath, NULL);
1369
1370     PathCombineW(szFrom, feFrom->szFullPath, wildCardFiles);
1371     szFrom[lstrlenW(szFrom) + 1] = '\0';
1372
1373     lstrcpyW(szTo, szDestPath);
1374     szTo[lstrlenW(szDestPath) + 1] = '\0';
1375
1376     memcpy(&fileOp, lpFileOp, sizeof(fileOp));
1377     fileOp.pFrom = szFrom;
1378     fileOp.pTo = szTo;
1379
1380     SHFileOperationW(&fileOp);
1381 }
1382
1383 /* moves a file or directory to another directory */
1384 static void move_to_dir(LPSHFILEOPSTRUCTW lpFileOp, const FILE_ENTRY *feFrom, const FILE_ENTRY *feTo)
1385 {
1386     WCHAR szDestPath[MAX_PATH];
1387
1388     PathCombineW(szDestPath, feTo->szFullPath, feFrom->szFilename);
1389
1390     if (IsAttribFile(feFrom->attributes))
1391         SHNotifyMoveFileW(feFrom->szFullPath, szDestPath);
1392     else if (!(lpFileOp->fFlags & FOF_FILESONLY && feFrom->bFromWildcard))
1393         move_dir_to_dir(lpFileOp, feFrom, szDestPath);
1394 }
1395
1396 /* the FO_MOVE operation */
1397 static HRESULT move_files(LPSHFILEOPSTRUCTW lpFileOp, const FILE_LIST *flFrom, const FILE_LIST *flTo)
1398 {
1399     DWORD i;
1400     const FILE_ENTRY *entryToMove;
1401     const FILE_ENTRY *fileDest;
1402
1403     if (!flFrom->dwNumFiles || !flTo->dwNumFiles)
1404         return ERROR_CANCELLED;
1405
1406     if (!(lpFileOp->fFlags & FOF_MULTIDESTFILES) &&
1407         flTo->dwNumFiles > 1 && flFrom->dwNumFiles > 1)
1408     {
1409         return ERROR_CANCELLED;
1410     }
1411
1412     if (!(lpFileOp->fFlags & FOF_MULTIDESTFILES) &&
1413         !flFrom->bAnyDirectories &&
1414         flFrom->dwNumFiles > flTo->dwNumFiles)
1415     {
1416         return ERROR_CANCELLED;
1417     }
1418
1419     if (!PathFileExistsW(flTo->feFiles[0].szDirectory))
1420         return ERROR_CANCELLED;
1421
1422     if ((lpFileOp->fFlags & FOF_MULTIDESTFILES) &&
1423         flFrom->dwNumFiles != flTo->dwNumFiles)
1424     {
1425         return ERROR_CANCELLED;
1426     }
1427
1428     fileDest = &flTo->feFiles[0];
1429     for (i = 0; i < flFrom->dwNumFiles; i++)
1430     {
1431         entryToMove = &flFrom->feFiles[i];
1432
1433         if (lpFileOp->fFlags & FOF_MULTIDESTFILES)
1434             fileDest = &flTo->feFiles[i];
1435
1436         if (!PathFileExistsW(fileDest->szDirectory))
1437             return ERROR_CANCELLED;
1438
1439         if (fileDest->bExists && IsAttribDir(fileDest->attributes))
1440             move_to_dir(lpFileOp, entryToMove, fileDest);
1441         else
1442             SHNotifyMoveFileW(entryToMove->szFullPath, fileDest->szFullPath);
1443     }
1444
1445     return ERROR_SUCCESS;
1446 }
1447
1448 /* the FO_RENAME files */
1449 static HRESULT rename_files(LPSHFILEOPSTRUCTW lpFileOp, const FILE_LIST *flFrom, const FILE_LIST *flTo)
1450 {
1451     const FILE_ENTRY *feFrom;
1452     const FILE_ENTRY *feTo;
1453
1454     if (flFrom->dwNumFiles != 1)
1455         return ERROR_GEN_FAILURE;
1456
1457     if (flTo->dwNumFiles != 1)
1458         return ERROR_CANCELLED;
1459
1460     feFrom = &flFrom->feFiles[0];
1461     feTo= &flTo->feFiles[0];
1462
1463     /* fail if destination doesn't exist */
1464     if (!feFrom->bExists)
1465         return ERROR_SHELL_INTERNAL_FILE_NOT_FOUND;
1466
1467     /* fail if destination already exists */
1468     if (feTo->bExists)
1469         return ERROR_ALREADY_EXISTS;
1470
1471     return SHNotifyMoveFileW(feFrom->szFullPath, feTo->szFullPath);
1472 }
1473
1474 /* alert the user if an unsupported flag is used */
1475 static void check_flags(FILEOP_FLAGS fFlags)
1476 {
1477     WORD wUnsupportedFlags = FOF_NO_CONNECTED_ELEMENTS |
1478         FOF_NOCOPYSECURITYATTRIBS | FOF_NORECURSEREPARSE |
1479         FOF_RENAMEONCOLLISION | FOF_WANTMAPPINGHANDLE;
1480
1481     if (fFlags & wUnsupportedFlags)
1482         FIXME("Unsupported flags: %04x\n", fFlags);
1483 }
1484
1485 /*************************************************************************
1486  * SHFileOperationW          [SHELL32.@]
1487  *
1488  * See SHFileOperationA
1489  */
1490 int WINAPI SHFileOperationW(LPSHFILEOPSTRUCTW lpFileOp)
1491 {
1492     FILE_OPERATION op;
1493     FILE_LIST flFrom, flTo;
1494     int ret = 0;
1495
1496     if (!lpFileOp)
1497         return ERROR_INVALID_PARAMETER;
1498
1499     check_flags(lpFileOp->fFlags);
1500
1501     ZeroMemory(&flFrom, sizeof(FILE_LIST));
1502     ZeroMemory(&flTo, sizeof(FILE_LIST));
1503
1504     if ((ret = parse_file_list(&flFrom, lpFileOp->pFrom)))
1505         return ret;
1506
1507     if (lpFileOp->wFunc != FO_DELETE)
1508         parse_file_list(&flTo, lpFileOp->pTo);
1509
1510     ZeroMemory(&op, sizeof(op));
1511     op.req = lpFileOp;
1512     op.bManyItems = (flFrom.dwNumFiles > 1);
1513
1514     switch (lpFileOp->wFunc)
1515     {
1516         case FO_COPY:
1517             ret = copy_files(&op, &flFrom, &flTo);
1518             break;
1519         case FO_DELETE:
1520             ret = delete_files(lpFileOp, &flFrom);
1521             break;
1522         case FO_MOVE:
1523             ret = move_files(lpFileOp, &flFrom, &flTo);
1524             break;
1525         case FO_RENAME:
1526             ret = rename_files(lpFileOp, &flFrom, &flTo);
1527             break;
1528         default:
1529             ret = ERROR_INVALID_PARAMETER;
1530             break;
1531     }
1532
1533     destroy_file_list(&flFrom);
1534
1535     if (lpFileOp->wFunc != FO_DELETE)
1536         destroy_file_list(&flTo);
1537
1538     if (ret == ERROR_CANCELLED)
1539         lpFileOp->fAnyOperationsAborted = TRUE;
1540     
1541     return ret;
1542 }
1543
1544 #define SHDSA_GetItemCount(hdsa) (*(int*)(hdsa))
1545
1546 /*************************************************************************
1547  * SHFreeNameMappings      [shell32.246]
1548  *
1549  * Free the mapping handle returned by SHFileoperation if FOF_WANTSMAPPINGHANDLE
1550  * was specified.
1551  *
1552  * PARAMS
1553  *  hNameMapping [I] handle to the name mappings used during renaming of files
1554  *
1555  * RETURNS
1556  *  Nothing
1557  */
1558 void WINAPI SHFreeNameMappings(HANDLE hNameMapping)
1559 {
1560         if (hNameMapping)
1561         {
1562           int i = SHDSA_GetItemCount((HDSA)hNameMapping) - 1;
1563
1564           for (; i>= 0; i--)
1565           {
1566             LPSHNAMEMAPPINGW lp = DSA_GetItemPtr((HDSA)hNameMapping, i);
1567
1568             SHFree(lp->pszOldPath);
1569             SHFree(lp->pszNewPath);
1570           }
1571           DSA_Destroy((HDSA)hNameMapping);
1572         }
1573 }
1574
1575 /*************************************************************************
1576  * SheGetDirA [SHELL32.@]
1577  *
1578  * drive = 0: returns the current directory path
1579  * drive > 0: returns the current directory path of the specified drive
1580  *            drive=1 -> A:  drive=2 -> B:  ...
1581  * returns 0 if successful
1582 */
1583 DWORD WINAPI SheGetDirA(DWORD drive, LPSTR buffer)
1584 {
1585     WCHAR org_path[MAX_PATH];
1586     DWORD ret;
1587     char drv_path[3];
1588
1589     /* change current directory to the specified drive */
1590     if (drive) {
1591         strcpy(drv_path, "A:");
1592         drv_path[0] += (char)drive-1;
1593
1594         GetCurrentDirectoryW(MAX_PATH, org_path);
1595
1596         SetCurrentDirectoryA(drv_path);
1597     }
1598
1599     /* query current directory path of the specified drive */
1600     ret = GetCurrentDirectoryA(MAX_PATH, buffer);
1601
1602     /* back to the original drive */
1603     if (drive)
1604         SetCurrentDirectoryW(org_path);
1605
1606     if (!ret)
1607         return GetLastError();
1608
1609     return 0;
1610 }
1611
1612 /*************************************************************************
1613  * SheGetDirW [SHELL32.@]
1614  *
1615  * drive = 0: returns the current directory path
1616  * drive > 0: returns the current directory path of the specified drive
1617  *            drive=1 -> A:  drive=2 -> B:  ...
1618  * returns 0 if successful
1619  */
1620 DWORD WINAPI SheGetDirW(DWORD drive, LPWSTR buffer)
1621 {
1622     WCHAR org_path[MAX_PATH];
1623     DWORD ret;
1624     char drv_path[3];
1625
1626     /* change current directory to the specified drive */
1627     if (drive) {
1628         strcpy(drv_path, "A:");
1629         drv_path[0] += (char)drive-1;
1630
1631         GetCurrentDirectoryW(MAX_PATH, org_path);
1632
1633         SetCurrentDirectoryA(drv_path);
1634     }
1635
1636     /* query current directory path of the specified drive */
1637     ret = GetCurrentDirectoryW(MAX_PATH, buffer);
1638
1639     /* back to the original drive */
1640     if (drive)
1641         SetCurrentDirectoryW(org_path);
1642
1643     if (!ret)
1644         return GetLastError();
1645
1646     return 0;
1647 }
1648
1649 /*************************************************************************
1650  * SheChangeDirA [SHELL32.@]
1651  *
1652  * changes the current directory to the specified path
1653  * and returns 0 if successful
1654  */
1655 DWORD WINAPI SheChangeDirA(LPSTR path)
1656 {
1657     if (SetCurrentDirectoryA(path))
1658         return 0;
1659     else
1660         return GetLastError();
1661 }
1662
1663 /*************************************************************************
1664  * SheChangeDirW [SHELL32.@]
1665  *
1666  * changes the current directory to the specified path
1667  * and returns 0 if successful
1668  */
1669 DWORD WINAPI SheChangeDirW(LPWSTR path)
1670 {
1671     if (SetCurrentDirectoryW(path))
1672         return 0;
1673     else
1674         return GetLastError();
1675 }
1676
1677 /*************************************************************************
1678  * IsNetDrive                   [SHELL32.66]
1679  */
1680 BOOL WINAPI IsNetDrive(DWORD drive)
1681 {
1682         char root[4];
1683         strcpy(root, "A:\\");
1684         root[0] += (char)drive;
1685         return (GetDriveTypeA(root) == DRIVE_REMOTE);
1686 }
1687
1688
1689 /*************************************************************************
1690  * RealDriveType                [SHELL32.524]
1691  */
1692 INT WINAPI RealDriveType(INT drive, BOOL bQueryNet)
1693 {
1694     char root[] = "A:\\";
1695     root[0] += (char)drive;
1696     return GetDriveTypeA(root);
1697 }
1698
1699 /***********************************************************************
1700  *              SHPathPrepareForWriteA (SHELL32.@)
1701  */
1702 HRESULT WINAPI SHPathPrepareForWriteA(HWND hwnd, IUnknown *modless, LPCSTR path, DWORD flags)
1703 {
1704     WCHAR wpath[MAX_PATH];
1705     MultiByteToWideChar( CP_ACP, 0, path, -1, wpath, MAX_PATH);
1706     return SHPathPrepareForWriteW(hwnd, modless, wpath, flags);
1707 }
1708
1709 /***********************************************************************
1710  *              SHPathPrepareForWriteW (SHELL32.@)
1711  */
1712 HRESULT WINAPI SHPathPrepareForWriteW(HWND hwnd, IUnknown *modless, LPCWSTR path, DWORD flags)
1713 {
1714     HRESULT res;
1715     DWORD err;
1716     LPCWSTR realpath;
1717     int len;
1718     WCHAR* last_slash;
1719     WCHAR* temppath=NULL;
1720
1721     TRACE("%p %p %s 0x%80x\n", hwnd, modless, debugstr_w(path), flags);
1722
1723     if (flags & ~(SHPPFW_DIRCREATE|SHPPFW_ASKDIRCREATE|SHPPFW_IGNOREFILENAME))
1724         FIXME("unimplemented flags 0x%08x\n", flags);
1725
1726     /* cut off filename if necessary */
1727     if (flags & SHPPFW_IGNOREFILENAME)
1728     {
1729         last_slash = StrRChrW(path, NULL, '\\');
1730         if (last_slash == NULL)
1731             len = 1;
1732         else
1733             len = last_slash - path + 1;
1734         temppath = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
1735         if (!temppath)
1736             return E_OUTOFMEMORY;
1737         StrCpyNW(temppath, path, len);
1738         realpath = temppath;
1739     }
1740     else
1741     {
1742         realpath = path;
1743     }
1744
1745     /* try to create the directory if asked to */
1746     if (flags & (SHPPFW_DIRCREATE|SHPPFW_ASKDIRCREATE))
1747     {
1748         if (flags & SHPPFW_ASKDIRCREATE)
1749             FIXME("treating SHPPFW_ASKDIRCREATE as SHPPFW_DIRCREATE\n");
1750
1751         SHCreateDirectoryExW(0, realpath, NULL);
1752     }
1753
1754     /* check if we can access the directory */
1755     res = GetFileAttributesW(realpath);
1756
1757     HeapFree(GetProcessHeap(), 0, temppath);
1758
1759     if (res == INVALID_FILE_ATTRIBUTES)
1760     {
1761         err = GetLastError();
1762         if (err == ERROR_FILE_NOT_FOUND)
1763             return HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND);
1764         return HRESULT_FROM_WIN32(err);
1765     }
1766     else if (res & FILE_ATTRIBUTE_DIRECTORY)
1767         return S_OK;
1768     else
1769         return HRESULT_FROM_WIN32(ERROR_DIRECTORY);
1770 }