Declaration, implemention and test for BuildTrusteeWithSid.
[wine] / dlls / shell32 / iconcache.c
1 /*
2  *      shell icon cache (SIC)
3  *
4  * Copyright 1998, 1999 Juergen Schmied
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  */
20
21 #include "config.h"
22 #include "wine/port.h"
23
24 #include <stdarg.h>
25 #include <string.h>
26 #include <sys/types.h>
27 #ifdef HAVE_UNISTD_H
28 # include <unistd.h>
29 #endif
30 #include "windef.h"
31 #include "winbase.h"
32 #include "wingdi.h"
33 #include "winuser.h"
34 #include "winreg.h"
35 #include "wine/winuser16.h"
36 #include "wine/winbase16.h"
37 #include "heap.h"
38 #include "wine/debug.h"
39
40 #include "shellapi.h"
41 #include "objbase.h"
42 #include "shlguid.h"
43 #include "pidl.h"
44 #include "shell32_main.h"
45 #include "undocshell.h"
46 #include "shlwapi.h"
47
48 WINE_DEFAULT_DEBUG_CHANNEL(shell);
49
50 /********************** THE ICON CACHE ********************************/
51
52 #define INVALID_INDEX -1
53
54 typedef struct
55 {
56         LPWSTR sSourceFile;     /* file (not path!) containing the icon */
57         DWORD dwSourceIndex;    /* index within the file, if it is a resoure ID it will be negated */
58         DWORD dwListIndex;      /* index within the iconlist */
59         DWORD dwFlags;          /* GIL_* flags */
60         DWORD dwAccessTime;
61 } SIC_ENTRY, * LPSIC_ENTRY;
62
63 static HDPA             sic_hdpa = 0;
64
65 static CRITICAL_SECTION SHELL32_SicCS;
66 static CRITICAL_SECTION_DEBUG critsect_debug =
67 {
68     0, 0, &SHELL32_SicCS,
69     { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
70       0, 0, { 0, (DWORD)(__FILE__ ": SHELL32_SicCS") }
71 };
72 static CRITICAL_SECTION SHELL32_SicCS = { &critsect_debug, -1, 0, 0, 0, 0 };
73
74 /*****************************************************************************
75  * SIC_CompareEntries
76  *
77  * NOTES
78  *  Callback for DPA_Search
79  */
80 static INT CALLBACK SIC_CompareEntries( LPVOID p1, LPVOID p2, LPARAM lparam)
81 {       TRACE("%p %p %8lx\n", p1, p2, lparam);
82
83         if (((LPSIC_ENTRY)p1)->dwSourceIndex != ((LPSIC_ENTRY)p2)->dwSourceIndex) /* first the faster one*/
84           return 1;
85
86         if (strcmpiW(((LPSIC_ENTRY)p1)->sSourceFile,((LPSIC_ENTRY)p2)->sSourceFile))
87           return 1;
88
89         return 0;
90 }
91 /*****************************************************************************
92  * SIC_IconAppend                       [internal]
93  *
94  * NOTES
95  *  appends an icon pair to the end of the cache
96  */
97 static INT SIC_IconAppend (LPCWSTR sSourceFile, INT dwSourceIndex, HICON hSmallIcon, HICON hBigIcon)
98 {       LPSIC_ENTRY lpsice;
99         INT ret, index, index1;
100         WCHAR path[MAX_PATH];
101         TRACE("%s %i %p %p\n", debugstr_w(sSourceFile), dwSourceIndex, hSmallIcon ,hBigIcon);
102
103         lpsice = (LPSIC_ENTRY) SHAlloc (sizeof (SIC_ENTRY));
104
105        GetFullPathNameW(sSourceFile, MAX_PATH, path, NULL);
106         lpsice->sSourceFile = HeapAlloc( GetProcessHeap(), 0, (strlenW(path)+1)*sizeof(WCHAR) );
107         strcpyW( lpsice->sSourceFile, path );
108
109         lpsice->dwSourceIndex = dwSourceIndex;
110
111         EnterCriticalSection(&SHELL32_SicCS);
112
113         index = DPA_InsertPtr(sic_hdpa, 0x7fff, lpsice);
114         if ( INVALID_INDEX == index )
115         {
116           HeapFree(GetProcessHeap(), 0, lpsice->sSourceFile);
117           SHFree(lpsice);
118           ret = INVALID_INDEX;
119         }
120         else
121         {
122           index = ImageList_AddIcon (ShellSmallIconList, hSmallIcon);
123           index1= ImageList_AddIcon (ShellBigIconList, hBigIcon);
124
125           if (index!=index1)
126           {
127             FIXME("iconlists out of sync 0x%x 0x%x\n", index, index1);
128           }
129           lpsice->dwListIndex = index;
130           ret = lpsice->dwListIndex;
131         }
132
133         LeaveCriticalSection(&SHELL32_SicCS);
134         return ret;
135 }
136 /****************************************************************************
137  * SIC_LoadIcon                         [internal]
138  *
139  * NOTES
140  *  gets small/big icon by number from a file
141  */
142 static INT SIC_LoadIcon (LPCWSTR sSourceFile, INT dwSourceIndex)
143 {       HICON   hiconLarge=0;
144         HICON   hiconSmall=0;
145
146         PrivateExtractIconsW( sSourceFile, dwSourceIndex, 32, 32, &hiconLarge, 0, 1, 0 );
147         PrivateExtractIconsW( sSourceFile, dwSourceIndex, 16, 16, &hiconSmall, 0, 1, 0 );
148
149         if ( !hiconLarge ||  !hiconSmall)
150         {
151           WARN("failure loading icon %i from %s (%p %p)\n", dwSourceIndex, debugstr_w(sSourceFile), hiconLarge, hiconSmall);
152           return -1;
153         }
154         return SIC_IconAppend (sSourceFile, dwSourceIndex, hiconSmall, hiconLarge);
155 }
156 /*****************************************************************************
157  * SIC_GetIconIndex                     [internal]
158  *
159  * Parameters
160  *      sSourceFile     [IN]    filename of file containing the icon
161  *      index           [IN]    index/resID (negated) in this file
162  *
163  * NOTES
164  *  look in the cache for a proper icon. if not available the icon is taken
165  *  from the file and cached
166  */
167 INT SIC_GetIconIndex (LPCWSTR sSourceFile, INT dwSourceIndex )
168 {
169         SIC_ENTRY sice;
170         INT ret, index = INVALID_INDEX;
171         WCHAR path[MAX_PATH];
172
173         TRACE("%s %i\n", debugstr_w(sSourceFile), dwSourceIndex);
174
175         GetFullPathNameW(sSourceFile, MAX_PATH, path, NULL);
176         sice.sSourceFile = path;
177         sice.dwSourceIndex = dwSourceIndex;
178
179         EnterCriticalSection(&SHELL32_SicCS);
180
181         if (NULL != DPA_GetPtr (sic_hdpa, 0))
182         {
183           /* search linear from position 0*/
184           index = DPA_Search (sic_hdpa, &sice, 0, SIC_CompareEntries, 0, 0);
185         }
186
187         if ( INVALID_INDEX == index )
188         {
189           ret = SIC_LoadIcon (sSourceFile, dwSourceIndex);
190         }
191         else
192         {
193           TRACE("-- found\n");
194           ret = ((LPSIC_ENTRY)DPA_GetPtr(sic_hdpa, index))->dwListIndex;
195         }
196
197         LeaveCriticalSection(&SHELL32_SicCS);
198         return ret;
199 }
200 /****************************************************************************
201  * SIC_GetIcon                          [internal]
202  *
203  * NOTES
204  *  retrieves the specified icon from the iconcache. if not found tries to load the icon
205  */
206 static HICON WINE_UNUSED SIC_GetIcon (LPCWSTR sSourceFile, INT dwSourceIndex, BOOL bSmallIcon )
207 {       INT index;
208
209         TRACE("%s %i\n", debugstr_w(sSourceFile), dwSourceIndex);
210
211         index = SIC_GetIconIndex(sSourceFile, dwSourceIndex);
212
213         if (INVALID_INDEX == index)
214         {
215           return (HICON)INVALID_INDEX;
216         }
217
218         if (bSmallIcon)
219           return ImageList_GetIcon(ShellSmallIconList, index, ILD_NORMAL);
220         return ImageList_GetIcon(ShellBigIconList, index, ILD_NORMAL);
221
222 }
223 /*****************************************************************************
224  * SIC_Initialize                       [internal]
225  *
226  * NOTES
227  *  hack to load the resources from the shell32.dll under a different dll name
228  *  will be removed when the resource-compiler is ready
229  */
230 BOOL SIC_Initialize(void)
231 {
232         HICON           hSm, hLg;
233         UINT            index;
234
235         TRACE("\n");
236
237         if (sic_hdpa)   /* already initialized?*/
238           return TRUE;
239
240         sic_hdpa = DPA_Create(16);
241
242         if (!sic_hdpa)
243         {
244           return(FALSE);
245         }
246
247        ShellSmallIconList = ImageList_Create(16,16,ILC_COLOR32|ILC_MASK,0,0x20);
248        ShellBigIconList = ImageList_Create(32,32,ILC_COLOR32|ILC_MASK,0,0x20);
249
250        ImageList_SetBkColor(ShellSmallIconList, CLR_NONE);
251        ImageList_SetBkColor(ShellBigIconList, CLR_NONE);
252
253         for (index=1; index<39; index++)
254         {
255           hSm = LoadImageA(shell32_hInstance, MAKEINTRESOURCEA(index), IMAGE_ICON, 16, 16,LR_SHARED);
256           hLg = LoadImageA(shell32_hInstance, MAKEINTRESOURCEA(index), IMAGE_ICON, 32, 32,LR_SHARED);
257
258           if(!hSm)
259           {
260             hSm = LoadImageA(shell32_hInstance, MAKEINTRESOURCEA(0), IMAGE_ICON, 16, 16,LR_SHARED);
261             hLg = LoadImageA(shell32_hInstance, MAKEINTRESOURCEA(0), IMAGE_ICON, 32, 32,LR_SHARED);
262           }
263          SIC_IconAppend (swShell32Name, index, hSm, hLg);
264         }
265
266         TRACE("hIconSmall=%p hIconBig=%p\n",ShellSmallIconList, ShellBigIconList);
267
268         return TRUE;
269 }
270 /*************************************************************************
271  * SIC_Destroy
272  *
273  * frees the cache
274  */
275 static INT CALLBACK sic_free( LPVOID ptr, LPVOID lparam )
276 {
277         HeapFree(GetProcessHeap(), 0, ((LPSIC_ENTRY)ptr)->sSourceFile);
278         SHFree(ptr);
279         return TRUE;
280 }
281
282 void SIC_Destroy(void)
283 {
284         TRACE("\n");
285
286         EnterCriticalSection(&SHELL32_SicCS);
287
288         if (sic_hdpa) DPA_DestroyCallback(sic_hdpa, sic_free, NULL );
289
290         sic_hdpa = NULL;
291         ImageList_Destroy(ShellSmallIconList);
292         ShellSmallIconList = 0;
293         ImageList_Destroy(ShellBigIconList);
294         ShellBigIconList = 0;
295
296         LeaveCriticalSection(&SHELL32_SicCS);
297         DeleteCriticalSection(&SHELL32_SicCS);
298 }
299
300 /*************************************************************************
301  * Shell_GetImageList                   [SHELL32.71]
302  *
303  * PARAMETERS
304  *  imglist[1|2] [OUT] pointer which receives imagelist handles
305  *
306  */
307 BOOL WINAPI Shell_GetImageList(HIMAGELIST * lpBigList, HIMAGELIST * lpSmallList)
308 {       TRACE("(%p,%p)\n",lpBigList,lpSmallList);
309         if (lpBigList)
310         { *lpBigList = ShellBigIconList;
311         }
312         if (lpSmallList)
313         { *lpSmallList = ShellSmallIconList;
314         }
315
316         return TRUE;
317 }
318 /*************************************************************************
319  * PidlToSicIndex                       [INTERNAL]
320  *
321  * PARAMETERS
322  *      sh      [IN]    IShellFolder
323  *      pidl    [IN]
324  *      bBigIcon [IN]
325  *      uFlags  [IN]    GIL_*
326  *      pIndex  [OUT]   index within the SIC
327  *
328  */
329 BOOL PidlToSicIndex (
330         IShellFolder * sh,
331         LPCITEMIDLIST pidl,
332         BOOL bBigIcon,
333         UINT uFlags,
334         int * pIndex)
335 {
336         IExtractIconW   *ei;
337         WCHAR           szIconFile[MAX_PATH];   /* file containing the icon */
338         INT             iSourceIndex;           /* index or resID(negated) in this file */
339         BOOL            ret = FALSE;
340         UINT            dwFlags = 0;
341
342         TRACE("sf=%p pidl=%p %s\n", sh, pidl, bBigIcon?"Big":"Small");
343
344         if (SUCCEEDED (IShellFolder_GetUIObjectOf(sh, 0, 1, &pidl, &IID_IExtractIconW, 0, (void **)&ei)))
345         {
346           if (SUCCEEDED(IExtractIconW_GetIconLocation(ei, uFlags, szIconFile, MAX_PATH, &iSourceIndex, &dwFlags)))
347           {
348             *pIndex = SIC_GetIconIndex(szIconFile, iSourceIndex);
349             ret = TRUE;
350           }
351           IExtractIconW_Release(ei);
352         }
353
354         if (INVALID_INDEX == *pIndex)   /* default icon when failed */
355           *pIndex = 1;
356
357         return ret;
358
359 }
360
361 /*************************************************************************
362  * SHMapPIDLToSystemImageListIndex      [SHELL32.77]
363  *
364  * PARAMETERS
365  *      sh      [IN]            pointer to an instance of IShellFolder
366  *      pidl    [IN]
367  *      pIndex  [OUT][OPTIONAL] SIC index for big icon
368  *
369  */
370 int WINAPI SHMapPIDLToSystemImageListIndex(
371         IShellFolder *sh,
372         LPCITEMIDLIST pidl,
373         int *pIndex)
374 {
375         int Index;
376
377         TRACE("(SF=%p,pidl=%p,%p)\n",sh,pidl,pIndex);
378         pdump(pidl);
379
380         if (pIndex)
381           PidlToSicIndex ( sh, pidl, 1, 0, pIndex);
382         PidlToSicIndex ( sh, pidl, 0, 0, &Index);
383         return Index;
384 }
385
386 /*************************************************************************
387  * Shell_GetCachedImageIndex            [SHELL32.72]
388  *
389  */
390 INT WINAPI Shell_GetCachedImageIndexA(LPCSTR szPath, INT nIndex, BOOL bSimulateDoc)
391 {
392         INT ret, len;
393         LPWSTR szTemp;
394
395         WARN("(%s,%08x,%08x) semi-stub.\n",debugstr_a(szPath), nIndex, bSimulateDoc);
396
397         len = MultiByteToWideChar( CP_ACP, 0, szPath, -1, NULL, 0 );
398         szTemp = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) );
399         MultiByteToWideChar( CP_ACP, 0, szPath, -1, szTemp, len );
400
401         ret = SIC_GetIconIndex( szTemp, nIndex );
402
403         HeapFree( GetProcessHeap(), 0, szTemp );
404
405         return ret;
406 }
407
408 INT WINAPI Shell_GetCachedImageIndexW(LPCWSTR szPath, INT nIndex, BOOL bSimulateDoc)
409 {
410         WARN("(%s,%08x,%08x) semi-stub.\n",debugstr_w(szPath), nIndex, bSimulateDoc);
411
412         return SIC_GetIconIndex(szPath, nIndex);
413 }
414
415 INT WINAPI Shell_GetCachedImageIndexAW(LPCVOID szPath, INT nIndex, BOOL bSimulateDoc)
416 {       if( SHELL_OsIsUnicode())
417           return Shell_GetCachedImageIndexW(szPath, nIndex, bSimulateDoc);
418         return Shell_GetCachedImageIndexA(szPath, nIndex, bSimulateDoc);
419 }
420
421 /*************************************************************************
422  * ExtractIconEx                        [SHELL32.@]
423  */
424 UINT WINAPI ExtractIconExAW(LPCVOID lpszFile, INT nIconIndex, HICON * phiconLarge, HICON * phiconSmall, UINT nIcons)
425 {       if (SHELL_OsIsUnicode())
426           return ExtractIconExW ( lpszFile, nIconIndex, phiconLarge, phiconSmall, nIcons);
427         return ExtractIconExA ( lpszFile, nIconIndex, phiconLarge, phiconSmall, nIcons);
428 }
429
430 /*************************************************************************
431  * ExtractIconExW                       [SHELL32.@]
432  * RETURNS
433  *  0 no icon found
434  *  -1 file is not valid
435  *  or number of icons extracted
436  */
437 UINT WINAPI ExtractIconExW(LPCWSTR lpszFile, INT nIconIndex, HICON * phiconLarge, HICON * phiconSmall, UINT nIcons)
438 {
439         TRACE("%s %i %p %p %i\n", debugstr_w(lpszFile), nIconIndex, phiconLarge, phiconSmall, nIcons);
440
441         return PrivateExtractIconExW(lpszFile, nIconIndex, phiconLarge, phiconSmall, nIcons);
442 }
443
444 /*************************************************************************
445  * ExtractIconExA                       [SHELL32.@]
446  */
447 UINT WINAPI ExtractIconExA(LPCSTR lpszFile, INT nIconIndex, HICON * phiconLarge, HICON * phiconSmall, UINT nIcons)
448 {
449     UINT ret;
450     INT len = MultiByteToWideChar(CP_ACP, 0, lpszFile, -1, NULL, 0);
451     LPWSTR lpwstrFile = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
452
453     TRACE("%s %i %p %p %i\n", lpszFile, nIconIndex, phiconLarge, phiconSmall, nIcons);
454
455     MultiByteToWideChar(CP_ACP, 0, lpszFile, -1, lpwstrFile, len);
456     ret = ExtractIconExW (lpwstrFile, nIconIndex, phiconLarge, phiconSmall, nIcons);
457     HeapFree(GetProcessHeap(), 0, lpwstrFile);
458     return ret;
459 }
460
461 /*************************************************************************
462  *                              ExtractAssociatedIconA (SHELL32.@)
463  *
464  * Return icon for given file (either from file itself or from associated
465  * executable) and patch parameters if needed.
466  */
467 HICON WINAPI ExtractAssociatedIconA(HINSTANCE hInst, LPSTR lpIconPath, LPWORD lpiIcon)
468 {       
469         HICON hIcon;
470         WORD wDummyIcon = 0;
471         
472         TRACE("\n");
473
474         if(lpiIcon == NULL)
475             lpiIcon = &wDummyIcon;
476
477         hIcon = ExtractIconA(hInst, lpIconPath, *lpiIcon);
478
479         if( hIcon < (HICON)2 )
480         { if( hIcon == (HICON)1 ) /* no icons found in given file */
481           { char  tempPath[0x80];
482             HINSTANCE uRet = FindExecutableA(lpIconPath,NULL,tempPath);
483
484             if( uRet > (HINSTANCE)32 && tempPath[0] )
485             { strcpy(lpIconPath,tempPath);
486               hIcon = ExtractIconA(hInst, lpIconPath, *lpiIcon);
487               if( hIcon > (HICON)2 )
488                 return hIcon;
489             }
490             else hIcon = 0;
491           }
492
493           if( hIcon == (HICON)1 )
494             *lpiIcon = 2;   /* MSDOS icon - we found .exe but no icons in it */
495           else
496             *lpiIcon = 6;   /* generic icon - found nothing */
497
498           if (GetModuleFileNameA(hInst, lpIconPath, 0x80))
499           {
500               /* terminate string (GetModuleFileName doesn't do if buffer is too small) */
501               lpIconPath[0x80 - 1] = '\0';
502               hIcon = LoadIconA( hInst, MAKEINTRESOURCEA(*lpiIcon));
503           }
504         }
505         return hIcon;
506 }
507
508 /*************************************************************************
509  *                              ExtractAssociatedIconExW (SHELL32.@)
510  *
511  * Return icon for given file (either from file itself or from associated
512  * executable) and patch parameters if needed.
513  */
514 HICON WINAPI ExtractAssociatedIconExW(HINSTANCE hInst, LPWSTR lpIconPath, LPWORD lpiIconIdx, LPWORD lpiIconId)
515 {
516   FIXME("%p %s %p %p): stub\n", hInst, debugstr_w(lpIconPath), lpiIconIdx, lpiIconId);
517   return 0;
518 }
519
520 /*************************************************************************
521  *                              ExtractAssociatedIconExA (SHELL32.@)
522  *
523  * Return icon for given file (either from file itself or from associated
524  * executable) and patch parameters if needed.
525  */
526 HICON WINAPI ExtractAssociatedIconExA(HINSTANCE hInst, LPSTR lpIconPath, LPWORD lpiIconIdx, LPWORD lpiIconId)
527 {
528   HICON ret;
529   INT len = MultiByteToWideChar( CP_ACP, 0, lpIconPath, -1, NULL, 0 );
530   LPWSTR lpwstrFile = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) );
531
532   TRACE("%p %s %p %p)\n", hInst, lpIconPath, lpiIconIdx, lpiIconId);
533
534   MultiByteToWideChar( CP_ACP, 0, lpIconPath, -1, lpwstrFile, len );
535   ret = ExtractAssociatedIconExW(hInst, lpwstrFile, lpiIconIdx, lpiIconId);
536   HeapFree(GetProcessHeap(), 0, lpwstrFile);
537   return ret;
538 }
539
540
541 /****************************************************************************
542  * SHDefExtractIconW            [SHELL32.@]
543  */
544 HRESULT WINAPI SHDefExtractIconW(LPCWSTR pszIconFile, int iIndex, UINT uFlags,
545                                  HICON* phiconLarge, HICON* phiconSmall, UINT nIconSize)
546 {
547         UINT ret;
548         HICON hIcons[2];
549         WARN("%s %d 0x%08x %p %p %d, semi-stub\n", debugstr_w(pszIconFile), iIndex, uFlags, phiconLarge, phiconSmall, nIconSize);
550
551         ret = PrivateExtractIconsW(pszIconFile, iIndex, nIconSize, nIconSize, hIcons, NULL, 2, LR_DEFAULTCOLOR);
552         /* FIXME: deal with uFlags parameter which contains GIL_ flags */
553         if (ret == 0xFFFFFFFF)
554           return E_FAIL;
555         if (ret > 0) {
556           *phiconLarge = hIcons[0];
557           *phiconSmall = hIcons[1];
558           return S_OK;
559         }
560         return S_FALSE;
561 }
562
563 /****************************************************************************
564  * SHDefExtractIconA            [SHELL32.@]
565  */
566 HRESULT WINAPI SHDefExtractIconA(LPCSTR pszIconFile, int iIndex, UINT uFlags,
567                                  HICON* phiconLarge, HICON* phiconSmall, UINT nIconSize)
568 {
569   HRESULT ret;
570   INT len = MultiByteToWideChar(CP_ACP, 0, pszIconFile, -1, NULL, 0);
571   LPWSTR lpwstrFile = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
572
573   TRACE("%s %d 0x%08x %p %p %d\n", pszIconFile, iIndex, uFlags, phiconLarge, phiconSmall, nIconSize);
574
575   MultiByteToWideChar(CP_ACP, 0, pszIconFile, -1, lpwstrFile, len);
576   ret = SHDefExtractIconW(lpwstrFile, iIndex, uFlags, phiconLarge, phiconSmall, nIconSize);
577   HeapFree(GetProcessHeap(), 0, lpwstrFile);
578   return ret;
579 }