rpcrt4: Retrieve the maximum token length from the security provider rather than...
[wine] / dlls / shell32 / folders.c
1 /*
2  *      Copyright 1997  Marcus Meissner
3  *      Copyright 1998  Juergen Schmied
4  *
5  * This library is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Lesser General Public
7  * License as published by the Free Software Foundation; either
8  * version 2.1 of the License, or (at your option) any later version.
9  *
10  * This library is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * Lesser General Public License for more details.
14  *
15  * You should have received a copy of the GNU Lesser General Public
16  * License along with this library; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
18  */
19
20 #include "config.h"
21 #include "wine/port.h"
22
23 #include <stdlib.h>
24 #include <stdarg.h>
25 #include <stdio.h>
26 #include <string.h>
27
28 #define COBJMACROS
29
30 #include "windef.h"
31 #include "winbase.h"
32 #include "winerror.h"
33 #include "objbase.h"
34 #include "undocshell.h"
35 #include "shlguid.h"
36 #include "winreg.h"
37
38 #include "wine/debug.h"
39
40 #include "pidl.h"
41 #include "shell32_main.h"
42 #include "shfldr.h"
43 #include "shresdef.h"
44
45 WINE_DEFAULT_DEBUG_CHANNEL(shell);
46
47 /***********************************************************************
48 *   IExtractIconW implementation
49 */
50 typedef struct
51 {
52         const IExtractIconWVtbl *lpVtbl;
53         LONG               ref;
54         const IPersistFileVtbl  *lpvtblPersistFile;
55         const IExtractIconAVtbl *lpvtblExtractIconA;
56         LPITEMIDLIST       pidl;
57 } IExtractIconWImpl;
58
59 static const IExtractIconAVtbl eiavt;
60 static const IExtractIconWVtbl eivt;
61 static const IPersistFileVtbl pfvt;
62
63 static inline IExtractIconW *impl_from_IPersistFile( IPersistFile *iface )
64 {
65     return (IExtractIconW *)((char*)iface - FIELD_OFFSET(IExtractIconWImpl, lpvtblPersistFile));
66 }
67
68 static inline IExtractIconW *impl_from_IExtractIconA( IExtractIconA *iface )
69 {
70     return (IExtractIconW *)((char*)iface - FIELD_OFFSET(IExtractIconWImpl, lpvtblExtractIconA));
71 }
72
73
74 /**************************************************************************
75 *  IExtractIconW_Constructor
76 */
77 IExtractIconW* IExtractIconW_Constructor(LPCITEMIDLIST pidl)
78 {
79         IExtractIconWImpl* ei;
80         
81         TRACE("%p\n", pidl);
82
83         ei = HeapAlloc(GetProcessHeap(),0,sizeof(IExtractIconWImpl));
84         ei->ref=1;
85         ei->lpVtbl = &eivt;
86         ei->lpvtblPersistFile = &pfvt;
87         ei->lpvtblExtractIconA = &eiavt;
88         ei->pidl=ILClone(pidl);
89
90         pdump(pidl);
91
92         TRACE("(%p)\n", ei);
93         return (IExtractIconW *)ei;
94 }
95 /**************************************************************************
96  *  IExtractIconW_QueryInterface
97  */
98 static HRESULT WINAPI IExtractIconW_fnQueryInterface(IExtractIconW *iface, REFIID riid, LPVOID *ppvObj)
99 {
100         IExtractIconWImpl *This = (IExtractIconWImpl *)iface;
101
102         TRACE("(%p)->(\n\tIID:\t%s,%p)\n", This, debugstr_guid(riid), ppvObj);
103
104         *ppvObj = NULL;
105
106         if (IsEqualIID(riid, &IID_IUnknown))                            /*IUnknown*/
107         {
108           *ppvObj = This;
109         }
110         else if (IsEqualIID(riid, &IID_IPersistFile))   /*IExtractIcon*/
111         {
112           *ppvObj = (IPersistFile*)&(This->lpvtblPersistFile);
113         }
114         else if (IsEqualIID(riid, &IID_IExtractIconA))  /*IExtractIcon*/
115         {
116           *ppvObj = (IExtractIconA*)&(This->lpvtblExtractIconA);
117         }
118         else if (IsEqualIID(riid, &IID_IExtractIconW))  /*IExtractIcon*/
119         {
120           *ppvObj = (IExtractIconW*)This;
121         }
122
123         if(*ppvObj)
124         {
125           IExtractIconW_AddRef((IExtractIconW*) *ppvObj);
126           TRACE("-- Interface: (%p)->(%p)\n",ppvObj,*ppvObj);
127           return S_OK;
128         }
129         TRACE("-- Interface: E_NOINTERFACE\n");
130         return E_NOINTERFACE;
131 }
132
133 /**************************************************************************
134 *  IExtractIconW_AddRef
135 */
136 static ULONG WINAPI IExtractIconW_fnAddRef(IExtractIconW * iface)
137 {
138         IExtractIconWImpl *This = (IExtractIconWImpl *)iface;
139         ULONG refCount = InterlockedIncrement(&This->ref);
140
141         TRACE("(%p)->(count=%u)\n", This, refCount - 1);
142
143         return refCount;
144 }
145 /**************************************************************************
146 *  IExtractIconW_Release
147 */
148 static ULONG WINAPI IExtractIconW_fnRelease(IExtractIconW * iface)
149 {
150         IExtractIconWImpl *This = (IExtractIconWImpl *)iface;
151         ULONG refCount = InterlockedDecrement(&This->ref);
152
153         TRACE("(%p)->(count=%u)\n", This, refCount + 1);
154
155         if (!refCount)
156         {
157           TRACE(" destroying IExtractIcon(%p)\n",This);
158           SHFree(This->pidl);
159           HeapFree(GetProcessHeap(),0,This);
160           return 0;
161         }
162         return refCount;
163 }
164
165 static HRESULT getIconLocationForFolder(IExtractIconW *iface, UINT uFlags,
166  LPWSTR szIconFile, UINT cchMax, int *piIndex, UINT *pwFlags)
167 {
168     IExtractIconWImpl *This = (IExtractIconWImpl *)iface;
169     int icon_idx;
170     WCHAR wszPath[MAX_PATH];
171     WCHAR wszCLSIDValue[CHARS_IN_GUID];
172     static const WCHAR shellClassInfo[] = { '.','S','h','e','l','l','C','l','a','s','s','I','n','f','o',0 };
173     static const WCHAR iconFile[] = { 'I','c','o','n','F','i','l','e',0 };
174     static const WCHAR clsid[] = { 'C','L','S','I','D',0 };
175     static const WCHAR clsid2[] = { 'C','L','S','I','D','2',0 };
176     static const WCHAR iconIndex[] = { 'I','c','o','n','I','n','d','e','x',0 };
177
178     if (SHELL32_GetCustomFolderAttribute(This->pidl, shellClassInfo, iconFile,
179         wszPath, MAX_PATH))
180     {
181         WCHAR wszIconIndex[10];
182         SHELL32_GetCustomFolderAttribute(This->pidl, shellClassInfo, iconIndex,
183             wszIconIndex, 10);
184         *piIndex = atoiW(wszIconIndex);
185     }
186     else if (SHELL32_GetCustomFolderAttribute(This->pidl, shellClassInfo, clsid,
187         wszCLSIDValue, CHARS_IN_GUID) &&
188         HCR_GetDefaultIconW(wszCLSIDValue, szIconFile, cchMax, &icon_idx))
189     {
190        *piIndex = icon_idx;
191     }
192     else if (SHELL32_GetCustomFolderAttribute(This->pidl, shellClassInfo, clsid2,
193         wszCLSIDValue, CHARS_IN_GUID) &&
194         HCR_GetDefaultIconW(wszCLSIDValue, szIconFile, cchMax, &icon_idx))
195     {
196        *piIndex = icon_idx;
197     }
198     else
199     {
200         static const WCHAR folder[] = { 'F','o','l','d','e','r',0 };
201
202         if (!HCR_GetDefaultIconW(folder, szIconFile, cchMax, &icon_idx))
203         {
204             lstrcpynW(szIconFile, swShell32Name, cchMax);
205             icon_idx = -IDI_SHELL_FOLDER;
206         }
207
208         if (uFlags & GIL_OPENICON)
209             *piIndex = icon_idx<0? icon_idx-1: icon_idx+1;
210         else
211             *piIndex = icon_idx;
212     }
213
214     return S_OK;
215 }
216
217 WCHAR swShell32Name[MAX_PATH];
218
219 /**************************************************************************
220 *  IExtractIconW_GetIconLocation
221 *
222 * mapping filetype to icon
223 */
224 static HRESULT WINAPI IExtractIconW_fnGetIconLocation(
225         IExtractIconW * iface,
226         UINT uFlags,            /* GIL_ flags */
227         LPWSTR szIconFile,
228         UINT cchMax,
229         int * piIndex,
230         UINT * pwFlags)         /* returned GIL_ flags */
231 {
232         IExtractIconWImpl *This = (IExtractIconWImpl *)iface;
233
234         char    sTemp[MAX_PATH];
235         int             icon_idx;
236         GUID const * riid;
237         LPITEMIDLIST    pSimplePidl = ILFindLastID(This->pidl);
238
239         TRACE("(%p) (flags=%u %p %u %p %p)\n", This, uFlags, szIconFile, cchMax, piIndex, pwFlags);
240
241         if (pwFlags)
242           *pwFlags = 0;
243
244         if (_ILIsDesktop(pSimplePidl))
245         {
246           lstrcpynW(szIconFile, swShell32Name, cchMax);
247           *piIndex = -IDI_SHELL_DESKTOP;
248         }
249
250         /* my computer and other shell extensions */
251         else if ((riid = _ILGetGUIDPointer(pSimplePidl)))
252         {
253           static const WCHAR fmt[] = { 'C','L','S','I','D','\\',
254        '{','%','0','8','l','x','-','%','0','4','x','-','%','0','4','x','-',
255        '%','0','2','x','%','0','2','x','-','%','0','2','x', '%','0','2','x',
256        '%','0','2','x','%','0','2','x','%','0','2','x','%','0','2','x','}',0 };
257           WCHAR xriid[50];
258
259           sprintfW(xriid, fmt,
260                   riid->Data1, riid->Data2, riid->Data3,
261                   riid->Data4[0], riid->Data4[1], riid->Data4[2], riid->Data4[3],
262                   riid->Data4[4], riid->Data4[5], riid->Data4[6], riid->Data4[7]);
263
264           if (HCR_GetDefaultIconW(xriid, szIconFile, cchMax, &icon_idx))
265           {
266             *piIndex = icon_idx;
267           }
268           else
269           {
270             lstrcpynW(szIconFile, swShell32Name, cchMax);
271             if(IsEqualGUID(riid, &CLSID_MyComputer))
272                 *piIndex = -IDI_SHELL_MY_COMPUTER;
273             else if(IsEqualGUID(riid, &CLSID_MyDocuments))
274                 *piIndex = -IDI_SHELL_MY_DOCUMENTS;
275             else if(IsEqualGUID(riid, &CLSID_NetworkPlaces))
276                 *piIndex = -IDI_SHELL_MY_NETWORK_PLACES;
277             else if(IsEqualGUID(riid, &CLSID_UnixFolder) ||
278                     IsEqualGUID(riid, &CLSID_UnixDosFolder))
279                 *piIndex = -IDI_SHELL_DRIVE;
280             else
281                 *piIndex = -IDI_SHELL_FOLDER;
282           }
283         }
284
285         else if (_ILIsDrive (pSimplePidl))
286         {
287           static const WCHAR drive[] = { 'D','r','i','v','e',0 };
288
289           int icon_idx = -1;
290
291           if (_ILGetDrive(pSimplePidl, sTemp, MAX_PATH))
292           {
293                 switch(GetDriveTypeA(sTemp))
294                 {
295                   case DRIVE_REMOVABLE:   icon_idx = IDI_SHELL_FLOPPY;        break;
296                   case DRIVE_CDROM:       icon_idx = IDI_SHELL_CDROM;         break;
297                   case DRIVE_REMOTE:      icon_idx = IDI_SHELL_NETDRIVE;      break;
298                   case DRIVE_RAMDISK:     icon_idx = IDI_SHELL_RAMDISK;       break;
299                 }
300           }
301
302           if (icon_idx != -1)
303           {
304                 lstrcpynW(szIconFile, swShell32Name, cchMax);
305                 *piIndex = -icon_idx;
306           }
307           else
308           {
309                 if (HCR_GetDefaultIconW(drive, szIconFile, cchMax, &icon_idx))
310                 {
311                   *piIndex = icon_idx;
312                 }
313                 else
314                 {
315                   lstrcpynW(szIconFile, swShell32Name, cchMax);
316                   *piIndex = -IDI_SHELL_DRIVE;
317                 }
318           }
319         }
320         else if (_ILIsFolder (pSimplePidl))
321         {
322             getIconLocationForFolder(iface, uFlags, szIconFile, cchMax, piIndex,
323                                      pwFlags);
324         }
325         else
326         {
327           BOOL found = FALSE;
328
329           if (_ILIsCPanelStruct(pSimplePidl))
330           {
331             if (SUCCEEDED(CPanel_GetIconLocationW(pSimplePidl, szIconFile, cchMax, piIndex)))
332                 found = TRUE;
333           }
334           else if (_ILGetExtension(pSimplePidl, sTemp, MAX_PATH))
335           {
336             if (HCR_MapTypeToValueA(sTemp, sTemp, MAX_PATH, TRUE)
337                 && HCR_GetDefaultIconA(sTemp, sTemp, MAX_PATH, &icon_idx))
338             {
339               if (!lstrcmpA("%1", sTemp))               /* icon is in the file */
340               {
341                 SHGetPathFromIDListW(This->pidl, szIconFile);
342                 *piIndex = 0;
343               }
344               else
345               {
346                 MultiByteToWideChar(CP_ACP, 0, sTemp, -1, szIconFile, cchMax);
347                 *piIndex = icon_idx;
348               }
349
350               found = TRUE;
351             }
352             else if (!lstrcmpiA(sTemp, "lnkfile"))
353             {
354               /* extract icon from shell shortcut */
355               IShellFolder* dsf;
356               IShellLinkW* psl;
357
358               if (SUCCEEDED(SHGetDesktopFolder(&dsf)))
359               {
360                 HRESULT hr = IShellFolder_GetUIObjectOf(dsf, NULL, 1, (LPCITEMIDLIST*)&This->pidl, &IID_IShellLinkW, NULL, (LPVOID*)&psl);
361
362                 if (SUCCEEDED(hr))
363                 {
364                   hr = IShellLinkW_GetIconLocation(psl, szIconFile, MAX_PATH, piIndex);
365
366                   if (SUCCEEDED(hr) && *szIconFile)
367                     found = TRUE;
368
369                   IShellLinkW_Release(psl);
370                 }
371
372                 IShellFolder_Release(dsf);
373               }
374             }
375           }
376
377           if (!found)                                   /* default icon */
378           {
379             lstrcpynW(szIconFile, swShell32Name, cchMax);
380             *piIndex = 0;
381           }
382         }
383
384         TRACE("-- %s %x\n", debugstr_w(szIconFile), *piIndex);
385         return NOERROR;
386 }
387
388 /**************************************************************************
389 *  IExtractIconW_Extract
390 */
391 static HRESULT WINAPI IExtractIconW_fnExtract(IExtractIconW * iface, LPCWSTR pszFile, UINT nIconIndex, HICON *phiconLarge, HICON *phiconSmall, UINT nIconSize)
392 {
393         IExtractIconWImpl *This = (IExtractIconWImpl *)iface;
394         int index;
395
396         FIXME("(%p) (file=%p index=%d %p %p size=%08x) semi-stub\n", This, debugstr_w(pszFile), (signed)nIconIndex,
397               phiconLarge, phiconSmall, nIconSize);
398
399         index = SIC_GetIconIndex(pszFile, nIconIndex, 0);
400
401         if (phiconLarge)
402           *phiconLarge = ImageList_GetIcon(ShellBigIconList, index, ILD_TRANSPARENT);
403
404         if (phiconSmall)
405           *phiconSmall = ImageList_GetIcon(ShellSmallIconList, index, ILD_TRANSPARENT);
406
407         return S_OK;
408 }
409
410 static const IExtractIconWVtbl eivt =
411 {
412         IExtractIconW_fnQueryInterface,
413         IExtractIconW_fnAddRef,
414         IExtractIconW_fnRelease,
415         IExtractIconW_fnGetIconLocation,
416         IExtractIconW_fnExtract
417 };
418
419 /**************************************************************************
420 *  IExtractIconA_Constructor
421 */
422 IExtractIconA* IExtractIconA_Constructor(LPCITEMIDLIST pidl)
423 {
424         IExtractIconWImpl *This = (IExtractIconWImpl *)IExtractIconW_Constructor(pidl);
425         IExtractIconA *eia = (IExtractIconA *)&This->lpvtblExtractIconA;
426         
427         TRACE("(%p)->(%p)\n", This, eia);
428         return eia;
429 }
430 /**************************************************************************
431  *  IExtractIconA_QueryInterface
432  */
433 static HRESULT WINAPI IExtractIconA_fnQueryInterface(IExtractIconA * iface, REFIID riid, LPVOID *ppvObj)
434 {
435         IExtractIconW *This = impl_from_IExtractIconA(iface);
436
437         return IExtractIconW_QueryInterface(This, riid, ppvObj);
438 }
439
440 /**************************************************************************
441 *  IExtractIconA_AddRef
442 */
443 static ULONG WINAPI IExtractIconA_fnAddRef(IExtractIconA * iface)
444 {
445         IExtractIconW *This = impl_from_IExtractIconA(iface);
446
447         return IExtractIconW_AddRef(This);
448 }
449 /**************************************************************************
450 *  IExtractIconA_Release
451 */
452 static ULONG WINAPI IExtractIconA_fnRelease(IExtractIconA * iface)
453 {
454         IExtractIconW *This = impl_from_IExtractIconA(iface);
455
456         return IExtractIconW_AddRef(This);
457 }
458 /**************************************************************************
459 *  IExtractIconA_GetIconLocation
460 *
461 * mapping filetype to icon
462 */
463 static HRESULT WINAPI IExtractIconA_fnGetIconLocation(
464         IExtractIconA * iface,
465         UINT uFlags,
466         LPSTR szIconFile,
467         UINT cchMax,
468         int * piIndex,
469         UINT * pwFlags)
470 {
471         HRESULT ret;
472         LPWSTR lpwstrFile = HeapAlloc(GetProcessHeap(), 0, cchMax * sizeof(WCHAR));
473         IExtractIconW *This = impl_from_IExtractIconA(iface);
474         
475         TRACE("(%p) (flags=%u %p %u %p %p)\n", This, uFlags, szIconFile, cchMax, piIndex, pwFlags);
476
477         ret = IExtractIconW_GetIconLocation(This, uFlags, lpwstrFile, cchMax, piIndex, pwFlags);
478         WideCharToMultiByte(CP_ACP, 0, lpwstrFile, -1, szIconFile, cchMax, NULL, NULL);
479         HeapFree(GetProcessHeap(), 0, lpwstrFile);
480
481         TRACE("-- %s %x\n", szIconFile, *piIndex);
482         return ret;
483 }
484 /**************************************************************************
485 *  IExtractIconA_Extract
486 */
487 static HRESULT WINAPI IExtractIconA_fnExtract(IExtractIconA * iface, LPCSTR pszFile, UINT nIconIndex, HICON *phiconLarge, HICON *phiconSmall, UINT nIconSize)
488 {
489         HRESULT ret;
490         INT len = MultiByteToWideChar(CP_ACP, 0, pszFile, -1, NULL, 0);
491         LPWSTR lpwstrFile = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
492         IExtractIconW *This = impl_from_IExtractIconA(iface);
493
494         TRACE("(%p) (file=%p index=%u %p %p size=%u)\n", This, pszFile, nIconIndex, phiconLarge, phiconSmall, nIconSize);
495
496         MultiByteToWideChar(CP_ACP, 0, pszFile, -1, lpwstrFile, len);
497         ret = IExtractIconW_Extract(This, lpwstrFile, nIconIndex, phiconLarge, phiconSmall, nIconSize);
498         HeapFree(GetProcessHeap(), 0, lpwstrFile);
499         return ret;
500 }
501
502 static const IExtractIconAVtbl eiavt =
503 {
504         IExtractIconA_fnQueryInterface,
505         IExtractIconA_fnAddRef,
506         IExtractIconA_fnRelease,
507         IExtractIconA_fnGetIconLocation,
508         IExtractIconA_fnExtract
509 };
510
511 /************************************************************************
512  * IEIPersistFile_QueryInterface (IUnknown)
513  */
514 static HRESULT WINAPI IEIPersistFile_fnQueryInterface(
515         IPersistFile    *iface,
516         REFIID          iid,
517         LPVOID          *ppvObj)
518 {
519         IExtractIconW *This = impl_from_IPersistFile(iface);
520
521         return IExtractIconW_QueryInterface(This, iid, ppvObj);
522 }
523
524 /************************************************************************
525  * IEIPersistFile_AddRef (IUnknown)
526  */
527 static ULONG WINAPI IEIPersistFile_fnAddRef(
528         IPersistFile    *iface)
529 {
530         IExtractIconW *This = impl_from_IPersistFile(iface);
531
532         return IExtractIconW_AddRef(This);
533 }
534
535 /************************************************************************
536  * IEIPersistFile_Release (IUnknown)
537  */
538 static ULONG WINAPI IEIPersistFile_fnRelease(
539         IPersistFile    *iface)
540 {
541         IExtractIconW *This = impl_from_IPersistFile(iface);
542
543         return IExtractIconW_Release(This);
544 }
545
546 /************************************************************************
547  * IEIPersistFile_GetClassID (IPersist)
548  */
549 static HRESULT WINAPI IEIPersistFile_fnGetClassID(
550         IPersistFile    *iface,
551         LPCLSID         lpClassId)
552 {
553         CLSID StdFolderID = { 0x00000000, 0x0000, 0x0000, {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00} };
554
555         if (lpClassId==NULL)
556           return E_POINTER;
557
558         memcpy(lpClassId, &StdFolderID, sizeof(StdFolderID));
559
560         return S_OK;
561 }
562
563 /************************************************************************
564  * IEIPersistFile_Load (IPersistFile)
565  */
566 static HRESULT WINAPI IEIPersistFile_fnLoad(IPersistFile* iface, LPCOLESTR pszFileName, DWORD dwMode)
567 {
568         IExtractIconW *This = impl_from_IPersistFile(iface);
569         FIXME("%p\n", This);
570         return E_NOTIMPL;
571
572 }
573
574 static const IPersistFileVtbl pfvt =
575 {
576         IEIPersistFile_fnQueryInterface,
577         IEIPersistFile_fnAddRef,
578         IEIPersistFile_fnRelease,
579         IEIPersistFile_fnGetClassID,
580         (void *) 0xdeadbeef /* IEIPersistFile_fnIsDirty */,
581         IEIPersistFile_fnLoad,
582         (void *) 0xdeadbeef /* IEIPersistFile_fnSave */,
583         (void *) 0xdeadbeef /* IEIPersistFile_fnSaveCompleted */,
584         (void *) 0xdeadbeef /* IEIPersistFile_fnGetCurFile */
585 };