2 * Helper program to build unix menu entries
4 * Copyright 1997 Marcus Meissner
5 * Copyright 1998 Juergen Schmied
6 * Copyright 2003 Mike McCormack for CodeWeavers
7 * Copyright 2004 Dmitry Timoshkov
8 * Copyright 2005 Bill Medland
9 * Copyright 2008 Damjan Jovanovic
11 * This library is free software; you can redistribute it and/or
12 * modify it under the terms of the GNU Lesser General Public
13 * License as published by the Free Software Foundation; either
14 * version 2.1 of the License, or (at your option) any later version.
16 * This library is distributed in the hope that it will be useful,
17 * but WITHOUT ANY WARRANTY; without even the implied warranty of
18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
19 * Lesser General Public License for more details.
21 * You should have received a copy of the GNU Lesser General Public
22 * License along with this library; if not, write to the Free Software
23 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
26 * This program is used to replicate the Windows desktop and start menu
27 * into the native desktop's copies. Desktop entries are merged directly
28 * into the native desktop. The Windows Start Menu corresponds to a Wine
29 * entry within the native "start" menu and replicates the whole tree
30 * structure of the Windows Start Menu. Currently it does not differentiate
31 * between the user's desktop/start menu and the "All Users" copies.
33 * This program will read a Windows shortcut file using the IShellLink
34 * interface, then create a KDE/Gnome menu entry for the shortcut.
36 * winemenubuilder [ -w ] <shortcut.lnk>
38 * If the -w parameter is passed, and the shortcut cannot be created,
39 * this program will wait for the parent process to finish and then try
40 * again. This covers the case when a ShortCut is created before the
41 * executable containing its icon.
44 * Handle data lnk files. There is no icon in the file; the icon is in
45 * the handler for the file type (or pointed to by the lnk file). Also it
46 * might be better to use a native handler (e.g. a native acroread for pdf
48 * Differentiate between the user's entries and the "All Users" entries.
49 * If it is possible to add the desktop files to the native system's
50 * shared location for an "All Users" entry then do so. As a suggestion the
51 * shared menu Wine base could be writable to the wine group, or a wineadm
53 * Clean up fd.o menu icons and .directory files when the menu is deleted
55 * Generate icons for file open handlers to go into the "Open with..."
56 * list. What does Windows use, the default icon for the .EXE file? It's
57 * not in the registry.
58 * Associate applications under HKCR\Applications to open any MIME type
59 * (by associating with application/octet-stream, or how?).
60 * Clean up fd.o MIME types when they are deleted in Windows, their icons
61 * too. Very hard - once we associate them with fd.o, we can't tell whether
62 * they are ours or not, and the extension <-> MIME type mapping isn't
64 * Wine's HKCR is broken - it doesn't merge HKCU\Software\Classes, so apps
65 * that write associations there won't associate (#17019).
69 #include "wine/port.h"
94 #include "wine/unicode.h"
95 #include "wine/debug.h"
96 #include "wine/library.h"
97 #include "wine/list.h"
105 WINE_DEFAULT_DEBUG_CHANNEL(menubuilder
);
107 #define in_desktop_dir(csidl) ((csidl)==CSIDL_DESKTOPDIRECTORY || \
108 (csidl)==CSIDL_COMMON_DESKTOPDIRECTORY)
109 #define in_startmenu(csidl) ((csidl)==CSIDL_STARTMENU || \
110 (csidl)==CSIDL_COMMON_STARTMENU)
112 /* link file formats */
114 #include "pshpack1.h"
133 GRPICONDIRENTRY idEntries
[1];
171 static char *xdg_config_dir
;
172 static char *xdg_data_dir
;
173 static char *xdg_desktop_dir
;
175 /* Icon extraction routines
177 * FIXME: should use PrivateExtractIcons and friends
178 * FIXME: should not use stdio
181 #define MASK(x,y) (pAND[(x) / 8 + (nHeight - (y) - 1) * nANDWidthBytes] & (1 << (7 - (x) % 8)))
183 /* PNG-specific code */
186 static void *libpng_handle
;
187 #define MAKE_FUNCPTR(f) static typeof(f) * p##f
188 MAKE_FUNCPTR(png_create_info_struct
);
189 MAKE_FUNCPTR(png_create_write_struct
);
190 MAKE_FUNCPTR(png_destroy_write_struct
);
191 MAKE_FUNCPTR(png_get_error_ptr
);
192 MAKE_FUNCPTR(png_init_io
);
193 MAKE_FUNCPTR(png_set_bgr
);
194 MAKE_FUNCPTR(png_set_error_fn
);
195 MAKE_FUNCPTR(png_set_text
);
196 MAKE_FUNCPTR(png_set_IHDR
);
197 MAKE_FUNCPTR(png_write_end
);
198 MAKE_FUNCPTR(png_write_info
);
199 MAKE_FUNCPTR(png_write_row
);
202 static void *load_libpng(void)
204 if ((libpng_handle
= wine_dlopen(SONAME_LIBPNG
, RTLD_NOW
, NULL
, 0)) != NULL
)
206 #define LOAD_FUNCPTR(f) \
207 if((p##f = wine_dlsym(libpng_handle, #f, NULL, 0)) == NULL) { \
208 libpng_handle = NULL; \
211 LOAD_FUNCPTR(png_create_info_struct
);
212 LOAD_FUNCPTR(png_create_write_struct
);
213 LOAD_FUNCPTR(png_destroy_write_struct
);
214 LOAD_FUNCPTR(png_get_error_ptr
);
215 LOAD_FUNCPTR(png_init_io
);
216 LOAD_FUNCPTR(png_set_bgr
);
217 LOAD_FUNCPTR(png_set_error_fn
);
218 LOAD_FUNCPTR(png_set_IHDR
);
219 LOAD_FUNCPTR(png_set_text
);
220 LOAD_FUNCPTR(png_write_end
);
221 LOAD_FUNCPTR(png_write_info
);
222 LOAD_FUNCPTR(png_write_row
);
225 return libpng_handle
;
228 static void user_error_fn(png_structp png_ptr
, png_const_charp error_message
)
232 /* This uses setjmp/longjmp just like the default. We can't use the
233 * default because there's no way to access the jmp buffer in the png_struct
234 * that works in 1.2 and 1.4 and allows us to dynamically load libpng. */
235 WINE_ERR("PNG error: %s\n", wine_dbgstr_an(error_message
, -1));
236 pjmpbuf
= ppng_get_error_ptr(png_ptr
);
237 longjmp(*pjmpbuf
, 1);
240 static void user_warning_fn(png_structp png_ptr
, png_const_charp warning_message
)
242 WINE_WARN("PNG warning: %s\n", wine_dbgstr_an(warning_message
, -1));
245 static BOOL
SaveIconResAsPNG(const BITMAPINFO
*pIcon
, const char *png_filename
, LPCWSTR commentW
)
247 static const char comment_key
[] = "Created from";
252 int nXORWidthBytes
, nANDWidthBytes
, color_type
= 0, i
, j
;
253 BYTE
*row
, *copy
= NULL
;
254 const BYTE
*pXOR
, *pAND
= NULL
;
255 int nWidth
= pIcon
->bmiHeader
.biWidth
;
256 int nHeight
= pIcon
->bmiHeader
.biHeight
;
257 int nBpp
= pIcon
->bmiHeader
.biBitCount
;
263 color_type
|= PNG_COLOR_MASK_ALPHA
;
266 color_type
|= PNG_COLOR_MASK_COLOR
;
272 if (!libpng_handle
&& !load_libpng())
274 WINE_WARN("Unable to load libpng\n");
278 if (!(fp
= fopen(png_filename
, "w")))
280 WINE_ERR("unable to open '%s' for writing: %s\n", png_filename
, strerror(errno
));
284 nXORWidthBytes
= 4 * ((nWidth
* nBpp
+ 31) / 32);
285 nANDWidthBytes
= 4 * ((nWidth
+ 31 ) / 32);
286 pXOR
= (const BYTE
*) pIcon
+ sizeof(BITMAPINFOHEADER
) + pIcon
->bmiHeader
.biClrUsed
* sizeof(RGBQUAD
);
287 if (nHeight
> nWidth
)
290 pAND
= pXOR
+ nHeight
* nXORWidthBytes
;
293 /* Apply mask if present */
298 /* copy bytes before modifying them */
299 copy
= HeapAlloc( GetProcessHeap(), 0, nHeight
* nXORWidthBytes
);
300 memcpy( copy
, pXOR
, nHeight
* nXORWidthBytes
);
303 /* image and mask are upside down reversed */
304 row
= copy
+ (nHeight
- 1) * nXORWidthBytes
;
306 /* top left corner */
307 bgColor
.rgbRed
= row
[0];
308 bgColor
.rgbGreen
= row
[1];
309 bgColor
.rgbBlue
= row
[2];
310 bgColor
.rgbReserved
= 0;
312 for (i
= 0; i
< nHeight
; i
++, row
-= nXORWidthBytes
)
313 for (j
= 0; j
< nWidth
; j
++, row
+= nBpp
>> 3)
316 RGBQUAD
*pixel
= (RGBQUAD
*)row
;
317 pixel
->rgbBlue
= bgColor
.rgbBlue
;
318 pixel
->rgbGreen
= bgColor
.rgbGreen
;
319 pixel
->rgbRed
= bgColor
.rgbRed
;
321 pixel
->rgbReserved
= bgColor
.rgbReserved
;
327 if (!(png_ptr
= ppng_create_write_struct(PNG_LIBPNG_VER_STRING
, NULL
, NULL
, NULL
)) ||
328 !(info_ptr
= ppng_create_info_struct(png_ptr
)))
333 /* All future errors jump here */
336 ppng_set_error_fn(png_ptr
, &jmpbuf
, user_error_fn
, user_warning_fn
);
338 ppng_init_io(png_ptr
, fp
);
339 ppng_set_IHDR(png_ptr
, info_ptr
, nWidth
, nHeight
, 8,
342 PNG_COMPRESSION_TYPE_DEFAULT
,
343 PNG_FILTER_TYPE_DEFAULT
);
346 comment
.compression
= PNG_TEXT_COMPRESSION_NONE
;
347 comment
.key
= (png_charp
)comment_key
;
348 i
= WideCharToMultiByte(CP_UNIXCP
, 0, commentW
, -1, NULL
, 0, NULL
, NULL
);
349 comment
.text
= HeapAlloc(GetProcessHeap(), 0, i
);
350 WideCharToMultiByte(CP_UNIXCP
, 0, commentW
, -1, comment
.text
, i
, NULL
, NULL
);
351 comment
.text_length
= i
- 1;
352 ppng_set_text(png_ptr
, info_ptr
, &comment
, 1);
355 ppng_write_info(png_ptr
, info_ptr
);
356 ppng_set_bgr(png_ptr
);
357 for (i
= nHeight
- 1; i
>= 0 ; i
--)
358 ppng_write_row(png_ptr
, (png_bytep
)pXOR
+ nXORWidthBytes
* i
);
359 ppng_write_end(png_ptr
, info_ptr
);
361 ppng_destroy_write_struct(&png_ptr
, &info_ptr
);
362 if (png_ptr
) ppng_destroy_write_struct(&png_ptr
, NULL
);
364 HeapFree(GetProcessHeap(), 0, copy
);
365 HeapFree(GetProcessHeap(), 0, comment
.text
);
369 if (png_ptr
) ppng_destroy_write_struct(&png_ptr
, NULL
);
371 unlink(png_filename
);
372 HeapFree(GetProcessHeap(), 0, copy
);
373 HeapFree(GetProcessHeap(), 0, comment
.text
);
376 #endif /* SONAME_LIBPNG */
378 static BOOL
SaveIconResAsXPM(const BITMAPINFO
*pIcon
, const char *szXPMFileName
, LPCWSTR commentW
)
388 BOOL aColorUsed
[256] = {0};
393 if (!((pIcon
->bmiHeader
.biBitCount
== 4) || (pIcon
->bmiHeader
.biBitCount
== 8)))
395 WINE_FIXME("Unsupported color depth %d-bit\n", pIcon
->bmiHeader
.biBitCount
);
399 if (!(fXPMFile
= fopen(szXPMFileName
, "w")))
401 WINE_TRACE("unable to open '%s' for writing: %s\n", szXPMFileName
, strerror(errno
));
405 i
= WideCharToMultiByte(CP_UNIXCP
, 0, commentW
, -1, NULL
, 0, NULL
, NULL
);
406 comment
= HeapAlloc(GetProcessHeap(), 0, i
);
407 WideCharToMultiByte(CP_UNIXCP
, 0, commentW
, -1, comment
, i
, NULL
, NULL
);
409 nHeight
= pIcon
->bmiHeader
.biHeight
/ 2;
410 nXORWidthBytes
= 4 * ((pIcon
->bmiHeader
.biWidth
* pIcon
->bmiHeader
.biBitCount
/ 32)
411 + ((pIcon
->bmiHeader
.biWidth
* pIcon
->bmiHeader
.biBitCount
% 32) > 0));
412 nANDWidthBytes
= 4 * ((pIcon
->bmiHeader
.biWidth
/ 32)
413 + ((pIcon
->bmiHeader
.biWidth
% 32) > 0));
414 b8BitColors
= pIcon
->bmiHeader
.biBitCount
== 8;
415 nColors
= pIcon
->bmiHeader
.biClrUsed
? pIcon
->bmiHeader
.biClrUsed
416 : 1 << pIcon
->bmiHeader
.biBitCount
;
417 pXOR
= (const BYTE
*) pIcon
+ sizeof (BITMAPINFOHEADER
) + (nColors
* sizeof (RGBQUAD
));
418 pAND
= pXOR
+ nHeight
* nXORWidthBytes
;
420 #define COLOR(x,y) (b8BitColors ? pXOR[(x) + (nHeight - (y) - 1) * nXORWidthBytes] : (x) % 2 ? pXOR[(x) / 2 + (nHeight - (y) - 1) * nXORWidthBytes] & 0xF : (pXOR[(x) / 2 + (nHeight - (y) - 1) * nXORWidthBytes] & 0xF0) >> 4)
422 for (i
= 0; i
< nHeight
; i
++) {
423 for (j
= 0; j
< pIcon
->bmiHeader
.biWidth
; j
++) {
424 if (!aColorUsed
[COLOR(j
,i
)] && !MASK(j
,i
))
426 aColorUsed
[COLOR(j
,i
)] = TRUE
;
432 if (fprintf(fXPMFile
, "/* XPM */\n/* %s */\nstatic char *icon[] = {\n", comment
) <= 0)
434 if (fprintf(fXPMFile
, "\"%d %d %d %d\",\n",
435 (int) pIcon
->bmiHeader
.biWidth
, nHeight
, nColorsUsed
+ 1, 2) <=0)
438 for (i
= 0; i
< nColors
; i
++) {
440 if (fprintf(fXPMFile
, "\"%.2X c #%.2X%.2X%.2X\",\n", i
, pIcon
->bmiColors
[i
].rgbRed
,
441 pIcon
->bmiColors
[i
].rgbGreen
, pIcon
->bmiColors
[i
].rgbBlue
) <= 0)
444 if (fprintf(fXPMFile
, "\" c None\"") <= 0)
447 for (i
= 0; i
< nHeight
; i
++)
449 if (fprintf(fXPMFile
, ",\n\"") <= 0)
451 for (j
= 0; j
< pIcon
->bmiHeader
.biWidth
; j
++)
455 if (fprintf(fXPMFile
, " ") <= 0)
459 if (fprintf(fXPMFile
, "%.2X", COLOR(j
,i
)) <= 0)
462 if (fprintf(fXPMFile
, "\"") <= 0)
465 if (fprintf(fXPMFile
, "};\n") <= 0)
471 HeapFree(GetProcessHeap(), 0, comment
);
476 HeapFree(GetProcessHeap(), 0, comment
);
478 unlink( szXPMFileName
);
482 static BOOL CALLBACK
EnumResNameProc(HMODULE hModule
, LPCWSTR lpszType
, LPWSTR lpszName
, LONG_PTR lParam
)
484 ENUMRESSTRUCT
*sEnumRes
= (ENUMRESSTRUCT
*) lParam
;
486 if (!sEnumRes
->nIndex
--)
488 *sEnumRes
->pResInfo
= FindResourceW(hModule
, lpszName
, (LPCWSTR
)RT_GROUP_ICON
);
495 static BOOL
extract_icon32(LPCWSTR szFileName
, int nIndex
, char *szXPMFileName
)
499 LPCWSTR lpName
= NULL
;
501 GRPICONDIR
*pIconDir
;
503 ENUMRESSTRUCT sEnumRes
;
509 hModule
= LoadLibraryExW(szFileName
, 0, LOAD_LIBRARY_AS_DATAFILE
);
512 WINE_WARN("LoadLibraryExW (%s) failed, error %d\n",
513 wine_dbgstr_w(szFileName
), GetLastError());
519 hResInfo
= FindResourceW(hModule
, MAKEINTRESOURCEW(-nIndex
), (LPCWSTR
)RT_GROUP_ICON
);
520 WINE_TRACE("FindResourceW (%s) called, return %p, error %d\n",
521 wine_dbgstr_w(szFileName
), hResInfo
, GetLastError());
526 sEnumRes
.pResInfo
= &hResInfo
;
527 sEnumRes
.nIndex
= nIndex
;
528 if (!EnumResourceNamesW(hModule
, (LPCWSTR
)RT_GROUP_ICON
,
529 EnumResNameProc
, (LONG_PTR
)&sEnumRes
) &&
530 sEnumRes
.nIndex
!= -1)
532 WINE_TRACE("EnumResourceNamesW failed, error %d\n", GetLastError());
538 if ((hResData
= LoadResource(hModule
, hResInfo
)))
540 if ((pIconDir
= LockResource(hResData
)))
542 for (i
= 0; i
< pIconDir
->idCount
; i
++)
544 if (pIconDir
->idEntries
[i
].wBitCount
>= nMaxBits
)
546 if ((pIconDir
->idEntries
[i
].bHeight
* pIconDir
->idEntries
[i
].bWidth
) >= nMax
)
548 lpName
= MAKEINTRESOURCEW(pIconDir
->idEntries
[i
].nID
);
549 nMax
= pIconDir
->idEntries
[i
].bHeight
* pIconDir
->idEntries
[i
].bWidth
;
550 nMaxBits
= pIconDir
->idEntries
[i
].wBitCount
;
556 FreeResource(hResData
);
561 WINE_WARN("found no icon\n");
562 FreeLibrary(hModule
);
566 if ((hResInfo
= FindResourceW(hModule
, lpName
, (LPCWSTR
)RT_ICON
)))
568 if ((hResData
= LoadResource(hModule
, hResInfo
)))
570 if ((pIcon
= LockResource(hResData
)))
573 if (SaveIconResAsPNG(pIcon
, szXPMFileName
, szFileName
))
578 memcpy(szXPMFileName
+ strlen(szXPMFileName
) - 3, "xpm", 3);
579 if (SaveIconResAsXPM(pIcon
, szXPMFileName
, szFileName
))
584 FreeResource(hResData
);
588 FreeLibrary(hModule
);
592 static BOOL
ExtractFromEXEDLL(LPCWSTR szFileName
, int nIndex
, char *szXPMFileName
)
594 if (!extract_icon32(szFileName
, nIndex
, szXPMFileName
) /*&&
595 !extract_icon16(szFileName, szXPMFileName)*/)
600 static int ExtractFromICO(LPCWSTR szFileName
, char *szXPMFileName
)
602 FILE *fICOFile
= NULL
;
604 ICONDIRENTRY
*pIconDirEntry
= NULL
;
605 int nMax
= 0, nMaxBits
= 0;
609 char *filename
= NULL
;
611 filename
= wine_get_unix_file_name(szFileName
);
612 if (!(fICOFile
= fopen(filename
, "r")))
614 WINE_TRACE("unable to open '%s' for reading: %s\n", filename
, strerror(errno
));
618 if (fread(&iconDir
, sizeof (ICONDIR
), 1, fICOFile
) != 1 ||
619 (iconDir
.idReserved
!= 0) || (iconDir
.idType
!= 1))
621 WINE_WARN("Invalid ico file format\n");
625 if ((pIconDirEntry
= HeapAlloc(GetProcessHeap(), 0, iconDir
.idCount
* sizeof (ICONDIRENTRY
))) == NULL
)
627 if (fread(pIconDirEntry
, sizeof (ICONDIRENTRY
), iconDir
.idCount
, fICOFile
) != iconDir
.idCount
)
630 for (i
= 0; i
< iconDir
.idCount
; i
++)
632 WINE_TRACE("[%d]: %d x %d @ %d\n", i
, pIconDirEntry
[i
].bWidth
, pIconDirEntry
[i
].bHeight
, pIconDirEntry
[i
].wBitCount
);
633 if (pIconDirEntry
[i
].wBitCount
>= nMaxBits
&&
634 (pIconDirEntry
[i
].bHeight
* pIconDirEntry
[i
].bWidth
) >= nMax
)
637 nMax
= pIconDirEntry
[i
].bHeight
* pIconDirEntry
[i
].bWidth
;
638 nMaxBits
= pIconDirEntry
[i
].wBitCount
;
641 WINE_TRACE("Selected: %d\n", nIndex
);
643 if ((pIcon
= HeapAlloc(GetProcessHeap(), 0, pIconDirEntry
[nIndex
].dwBytesInRes
)) == NULL
)
645 if (fseek(fICOFile
, pIconDirEntry
[nIndex
].dwImageOffset
, SEEK_SET
))
647 if (fread(pIcon
, pIconDirEntry
[nIndex
].dwBytesInRes
, 1, fICOFile
) != 1)
651 /* Prefer PNG over XPM */
653 if (!SaveIconResAsPNG(pIcon
, szXPMFileName
, szFileName
))
656 memcpy(szXPMFileName
+ strlen(szXPMFileName
) - 3, "xpm", 3);
657 if (!SaveIconResAsXPM(pIcon
, szXPMFileName
, szFileName
))
661 HeapFree(GetProcessHeap(), 0, pIcon
);
662 HeapFree(GetProcessHeap(), 0, pIconDirEntry
);
664 HeapFree(GetProcessHeap(), 0, filename
);
668 HeapFree(GetProcessHeap(), 0, pIcon
);
669 HeapFree(GetProcessHeap(), 0, pIconDirEntry
);
670 if (fICOFile
) fclose(fICOFile
);
671 HeapFree(GetProcessHeap(), 0, filename
);
675 static BOOL
create_default_icon( const char *filename
, const char* comment
)
680 if (!(fXPM
= fopen(filename
, "w"))) return FALSE
;
681 if (fprintf(fXPM
, "/* XPM */\n/* %s */\nstatic char * icon[] = {", comment
) <= 0)
683 for (i
= 0; i
< sizeof(wine_xpm
)/sizeof(wine_xpm
[0]); i
++) {
684 if (fprintf( fXPM
, "\n\"%s\",", wine_xpm
[i
]) <= 0)
687 if (fprintf( fXPM
, "};\n" ) <=0)
698 static unsigned short crc16(const char* string
)
700 unsigned short crc
= 0;
703 for (i
= 0; string
[i
] != 0; i
++)
706 for (j
= 0; j
< 8; c
>>= 1, j
++)
708 xor_poly
= (c
^ crc
) & 1;
717 static char *strdupA( const char *str
)
721 if (!str
) return NULL
;
722 if ((ret
= HeapAlloc( GetProcessHeap(), 0, strlen(str
) + 1 ))) strcpy( ret
, str
);
726 static char* heap_printf(const char *format
, ...)
733 va_start(args
, format
);
736 buffer
= HeapAlloc(GetProcessHeap(), 0, size
);
739 n
= vsnprintf(buffer
, size
, format
, args
);
746 HeapFree(GetProcessHeap(), 0, buffer
);
749 if (!buffer
) return NULL
;
750 ret
= HeapReAlloc(GetProcessHeap(), 0, buffer
, strlen(buffer
) + 1 );
751 if (!ret
) ret
= buffer
;
755 static void write_xml_text(FILE *file
, const char *text
)
758 for (i
= 0; text
[i
]; i
++)
761 fputs("&", file
);
762 else if (text
[i
] == '<')
764 else if (text
[i
] == '>')
766 else if (text
[i
] == '\'')
767 fputs("'", file
);
768 else if (text
[i
] == '"')
769 fputs(""", file
);
771 fputc(text
[i
], file
);
775 static BOOL
create_directories(char *directory
)
780 for (i
= 0; directory
[i
]; i
++)
782 if (i
> 0 && directory
[i
] == '/')
785 mkdir(directory
, 0777);
789 if (mkdir(directory
, 0777) && errno
!= EEXIST
)
795 static char* wchars_to_utf8_chars(LPCWSTR string
)
798 INT size
= WideCharToMultiByte(CP_UTF8
, 0, string
, -1, NULL
, 0, NULL
, NULL
);
799 ret
= HeapAlloc(GetProcessHeap(), 0, size
);
801 WideCharToMultiByte(CP_UTF8
, 0, string
, -1, ret
, size
, NULL
, NULL
);
805 static char* wchars_to_unix_chars(LPCWSTR string
)
808 INT size
= WideCharToMultiByte(CP_UNIXCP
, 0, string
, -1, NULL
, 0, NULL
, NULL
);
809 ret
= HeapAlloc(GetProcessHeap(), 0, size
);
811 WideCharToMultiByte(CP_UNIXCP
, 0, string
, -1, ret
, size
, NULL
, NULL
);
815 static WCHAR
* utf8_chars_to_wchars(LPCSTR string
)
818 INT size
= MultiByteToWideChar(CP_UTF8
, 0, string
, -1, NULL
, 0);
819 ret
= HeapAlloc(GetProcessHeap(), 0, size
* sizeof(WCHAR
));
821 MultiByteToWideChar(CP_UTF8
, 0, string
, -1, ret
, size
);
825 /* extract an icon from an exe or icon file; helper for IPersistFile_fnSave */
826 static char *extract_icon( LPCWSTR path
, int index
, const char *destFilename
, BOOL bWait
)
829 char *iconsdir
= NULL
, *ico_path
= NULL
, *ico_name
, *xpm_path
= NULL
;
833 /* Where should we save the icon? */
834 WINE_TRACE("path=[%s] index=%d\n", wine_dbgstr_w(path
), index
);
835 iconsdir
= heap_printf("%s/icons", xdg_data_dir
);
838 if (mkdir(iconsdir
, 0777) && errno
!= EEXIST
)
840 WINE_WARN("couldn't make icons directory %s\n", wine_dbgstr_a(iconsdir
));
846 WINE_TRACE("no icon created\n");
850 /* Determine the icon base name */
851 n
= WideCharToMultiByte(CP_UNIXCP
, 0, path
, -1, NULL
, 0, NULL
, NULL
);
852 ico_path
= HeapAlloc(GetProcessHeap(), 0, n
);
853 WideCharToMultiByte(CP_UNIXCP
, 0, path
, -1, ico_path
, n
, NULL
, NULL
);
856 if (*s
=='/' || *s
=='\\') {
864 if (*ico_name
=='\\') *ico_name
++='\0';
865 s
=strrchr(ico_name
,'.');
868 /* Compute the source-path hash */
871 /* Try to treat the source file as an exe */
873 xpm_path
=heap_printf("%s/%s.png",iconsdir
,destFilename
);
875 xpm_path
=heap_printf("%s/%04x_%s.%d.png",iconsdir
,crc
,ico_name
,index
);
876 if (xpm_path
== NULL
)
878 WINE_ERR("could not extract icon %s, out of memory\n", wine_dbgstr_a(ico_name
));
882 if (ExtractFromEXEDLL( path
, index
, xpm_path
))
885 /* Must be something else, ignore the index in that case */
887 sprintf(xpm_path
,"%s/%s.png",iconsdir
,destFilename
);
889 sprintf(xpm_path
,"%s/%04x_%s.png",iconsdir
,crc
,ico_name
);
890 if (ExtractFromICO( path
, xpm_path
))
895 sprintf(xpm_path
,"%s/%s.xpm",iconsdir
,destFilename
);
897 sprintf(xpm_path
,"%s/%04x_%s.xpm",iconsdir
,crc
,ico_name
);
898 if (create_default_icon( xpm_path
, ico_path
))
902 HeapFree( GetProcessHeap(), 0, xpm_path
);
906 HeapFree(GetProcessHeap(), 0, iconsdir
);
907 HeapFree(GetProcessHeap(), 0, ico_path
);
911 static HKEY
open_menus_reg_key(void)
913 static const WCHAR Software_Wine_FileOpenAssociationsW
[] = {
914 'S','o','f','t','w','a','r','e','\\','W','i','n','e','\\','M','e','n','u','F','i','l','e','s',0};
917 ret
= RegCreateKeyW(HKEY_CURRENT_USER
, Software_Wine_FileOpenAssociationsW
, &assocKey
);
918 if (ret
== ERROR_SUCCESS
)
924 static DWORD
register_menus_entry(const char *unix_file
, const char *windows_file
)
927 WCHAR
*windows_fileW
;
931 size
= MultiByteToWideChar(CP_UNIXCP
, 0, unix_file
, -1, NULL
, 0);
932 unix_fileW
= HeapAlloc(GetProcessHeap(), 0, size
* sizeof(WCHAR
));
935 MultiByteToWideChar(CP_UNIXCP
, 0, unix_file
, -1, unix_fileW
, size
);
936 size
= MultiByteToWideChar(CP_UNIXCP
, 0, windows_file
, -1, NULL
, 0);
937 windows_fileW
= HeapAlloc(GetProcessHeap(), 0, size
* sizeof(WCHAR
));
941 MultiByteToWideChar(CP_UNIXCP
, 0, windows_file
, -1, windows_fileW
, size
);
942 hkey
= open_menus_reg_key();
945 ret
= RegSetValueExW(hkey
, unix_fileW
, 0, REG_SZ
, (const BYTE
*)windows_fileW
,
946 (strlenW(windows_fileW
) + 1) * sizeof(WCHAR
));
950 ret
= GetLastError();
951 HeapFree(GetProcessHeap(), 0, windows_fileW
);
954 ret
= ERROR_NOT_ENOUGH_MEMORY
;
955 HeapFree(GetProcessHeap(), 0, unix_fileW
);
958 ret
= ERROR_NOT_ENOUGH_MEMORY
;
962 static BOOL
write_desktop_entry(const char *unix_link
, const char *location
, const char *linkname
,
963 const char *path
, const char *args
, const char *descr
,
964 const char *workdir
, const char *icon
)
968 WINE_TRACE("(%s,%s,%s,%s,%s,%s,%s,%s)\n", wine_dbgstr_a(unix_link
), wine_dbgstr_a(location
),
969 wine_dbgstr_a(linkname
), wine_dbgstr_a(path
), wine_dbgstr_a(args
),
970 wine_dbgstr_a(descr
), wine_dbgstr_a(workdir
), wine_dbgstr_a(icon
));
972 file
= fopen(location
, "w");
976 fprintf(file
, "[Desktop Entry]\n");
977 fprintf(file
, "Name=%s\n", linkname
);
978 fprintf(file
, "Exec=env WINEPREFIX=\"%s\" wine %s %s\n",
979 wine_get_config_dir(), path
, args
);
980 fprintf(file
, "Type=Application\n");
981 fprintf(file
, "StartupNotify=true\n");
982 if (descr
&& lstrlenA(descr
))
983 fprintf(file
, "Comment=%s\n", descr
);
984 if (workdir
&& lstrlenA(workdir
))
985 fprintf(file
, "Path=%s\n", workdir
);
986 if (icon
&& lstrlenA(icon
))
987 fprintf(file
, "Icon=%s\n", icon
);
993 DWORD ret
= register_menus_entry(location
, unix_link
);
994 if (ret
!= ERROR_SUCCESS
)
1001 static BOOL
write_directory_entry(const char *directory
, const char *location
)
1005 WINE_TRACE("(%s,%s)\n", wine_dbgstr_a(directory
), wine_dbgstr_a(location
));
1007 file
= fopen(location
, "w");
1011 fprintf(file
, "[Desktop Entry]\n");
1012 fprintf(file
, "Type=Directory\n");
1013 if (strcmp(directory
, "wine") == 0)
1015 fprintf(file
, "Name=Wine\n");
1016 fprintf(file
, "Icon=wine\n");
1020 fprintf(file
, "Name=%s\n", directory
);
1021 fprintf(file
, "Icon=folder\n");
1028 static BOOL
write_menu_file(const char *unix_link
, const char *filename
)
1031 FILE *tempfile
= NULL
;
1034 char *menuPath
= NULL
;
1039 WINE_TRACE("(%s)\n", wine_dbgstr_a(filename
));
1043 tempfilename
= heap_printf("%s/wine-menu-XXXXXX", xdg_config_dir
);
1046 int tempfd
= mkstemps(tempfilename
, 0);
1049 tempfile
= fdopen(tempfd
, "w");
1055 else if (errno
== EEXIST
)
1057 HeapFree(GetProcessHeap(), 0, tempfilename
);
1060 HeapFree(GetProcessHeap(), 0, tempfilename
);
1065 fprintf(tempfile
, "<!DOCTYPE Menu PUBLIC \"-//freedesktop//DTD Menu 1.0//EN\"\n");
1066 fprintf(tempfile
, "\"http://www.freedesktop.org/standards/menu-spec/menu-1.0.dtd\">\n");
1067 fprintf(tempfile
, "<Menu>\n");
1068 fprintf(tempfile
, " <Name>Applications</Name>\n");
1070 name
= HeapAlloc(GetProcessHeap(), 0, lstrlenA(filename
) + 1);
1071 if (name
== NULL
) goto end
;
1073 for (i
= 0; filename
[i
]; i
++)
1075 name
[i
] = filename
[i
];
1076 if (filename
[i
] == '/')
1078 char *dir_file_name
;
1081 fprintf(tempfile
, " <Menu>\n");
1082 fprintf(tempfile
, " <Name>%s", count
? "" : "wine-");
1083 write_xml_text(tempfile
, name
);
1084 fprintf(tempfile
, "</Name>\n");
1085 fprintf(tempfile
, " <Directory>%s", count
? "" : "wine-");
1086 write_xml_text(tempfile
, name
);
1087 fprintf(tempfile
, ".directory</Directory>\n");
1088 dir_file_name
= heap_printf("%s/desktop-directories/%s%s.directory",
1089 xdg_data_dir
, count
? "" : "wine-", name
);
1092 if (stat(dir_file_name
, &st
) != 0 && errno
== ENOENT
)
1093 write_directory_entry(lastEntry
, dir_file_name
);
1094 HeapFree(GetProcessHeap(), 0, dir_file_name
);
1097 lastEntry
= &name
[i
+1];
1103 fprintf(tempfile
, " <Include>\n");
1104 fprintf(tempfile
, " <Filename>");
1105 write_xml_text(tempfile
, name
);
1106 fprintf(tempfile
, "</Filename>\n");
1107 fprintf(tempfile
, " </Include>\n");
1108 for (i
= 0; i
< count
; i
++)
1109 fprintf(tempfile
, " </Menu>\n");
1110 fprintf(tempfile
, "</Menu>\n");
1112 menuPath
= heap_printf("%s/%s", xdg_config_dir
, name
);
1113 if (menuPath
== NULL
) goto end
;
1114 strcpy(menuPath
+ strlen(menuPath
) - strlen(".desktop"), ".menu");
1121 ret
= (rename(tempfilename
, menuPath
) == 0);
1122 if (!ret
&& tempfilename
)
1123 remove(tempfilename
);
1124 HeapFree(GetProcessHeap(), 0, tempfilename
);
1126 register_menus_entry(menuPath
, unix_link
);
1127 HeapFree(GetProcessHeap(), 0, name
);
1128 HeapFree(GetProcessHeap(), 0, menuPath
);
1132 static BOOL
write_menu_entry(const char *unix_link
, const char *link
, const char *path
, const char *args
,
1133 const char *descr
, const char *workdir
, const char *icon
)
1135 const char *linkname
;
1136 char *desktopPath
= NULL
;
1138 char *filename
= NULL
;
1141 WINE_TRACE("(%s, %s, %s, %s, %s, %s, %s)\n", wine_dbgstr_a(unix_link
), wine_dbgstr_a(link
),
1142 wine_dbgstr_a(path
), wine_dbgstr_a(args
), wine_dbgstr_a(descr
),
1143 wine_dbgstr_a(workdir
), wine_dbgstr_a(icon
));
1145 linkname
= strrchr(link
, '/');
1146 if (linkname
== NULL
)
1151 desktopPath
= heap_printf("%s/applications/wine/%s.desktop", xdg_data_dir
, link
);
1154 WINE_WARN("out of memory creating menu entry\n");
1158 desktopDir
= strrchr(desktopPath
, '/');
1160 if (!create_directories(desktopPath
))
1162 WINE_WARN("couldn't make parent directories for %s\n", wine_dbgstr_a(desktopPath
));
1167 if (!write_desktop_entry(unix_link
, desktopPath
, linkname
, path
, args
, descr
, workdir
, icon
))
1169 WINE_WARN("couldn't make desktop entry %s\n", wine_dbgstr_a(desktopPath
));
1174 filename
= heap_printf("wine/%s.desktop", link
);
1175 if (!filename
|| !write_menu_file(unix_link
, filename
))
1177 WINE_WARN("couldn't make menu file %s\n", wine_dbgstr_a(filename
));
1182 HeapFree(GetProcessHeap(), 0, desktopPath
);
1183 HeapFree(GetProcessHeap(), 0, filename
);
1187 /* This escapes reserved characters in .desktop files' Exec keys. */
1188 static LPSTR
escape(LPCWSTR arg
)
1191 WCHAR
*escaped_string
;
1194 escaped_string
= HeapAlloc(GetProcessHeap(), 0, (4 * strlenW(arg
) + 1) * sizeof(WCHAR
));
1195 if (escaped_string
== NULL
) return NULL
;
1196 for (i
= j
= 0; arg
[i
]; i
++)
1201 escaped_string
[j
++] = '\\';
1202 escaped_string
[j
++] = '\\';
1203 escaped_string
[j
++] = '\\';
1204 escaped_string
[j
++] = '\\';
1224 escaped_string
[j
++] = '\\';
1225 escaped_string
[j
++] = '\\';
1228 escaped_string
[j
++] = arg
[i
];
1232 escaped_string
[j
] = 0;
1234 utf8_string
= wchars_to_utf8_chars(escaped_string
);
1235 if (utf8_string
== NULL
)
1237 WINE_ERR("out of memory\n");
1242 HeapFree(GetProcessHeap(), 0, escaped_string
);
1246 /* Return a heap-allocated copy of the unix format difference between the two
1247 * Windows-format paths.
1248 * locn is the owning location
1249 * link is within locn
1251 static char *relative_path( LPCWSTR link
, LPCWSTR locn
)
1253 char *unix_locn
, *unix_link
;
1254 char *relative
= NULL
;
1256 unix_locn
= wine_get_unix_file_name(locn
);
1257 unix_link
= wine_get_unix_file_name(link
);
1258 if (unix_locn
&& unix_link
)
1260 size_t len_unix_locn
, len_unix_link
;
1261 len_unix_locn
= strlen (unix_locn
);
1262 len_unix_link
= strlen (unix_link
);
1263 if (len_unix_locn
< len_unix_link
&& memcmp (unix_locn
, unix_link
, len_unix_locn
) == 0 && unix_link
[len_unix_locn
] == '/')
1266 char *p
= strrchr (unix_link
+ len_unix_locn
, '/');
1267 p
= strrchr (p
, '.');
1271 len_unix_link
= p
- unix_link
;
1273 len_rel
= len_unix_link
- len_unix_locn
;
1274 relative
= HeapAlloc(GetProcessHeap(), 0, len_rel
);
1277 memcpy (relative
, unix_link
+ len_unix_locn
+ 1, len_rel
);
1282 WINE_WARN("Could not separate the relative link path of %s in %s\n", wine_dbgstr_w(link
), wine_dbgstr_w(locn
));
1283 HeapFree(GetProcessHeap(), 0, unix_locn
);
1284 HeapFree(GetProcessHeap(), 0, unix_link
);
1288 /***********************************************************************
1292 * returns TRUE if successful
1293 * *loc will contain CS_DESKTOPDIRECTORY, CS_STARTMENU, CS_STARTUP etc.
1294 * *relative will contain the address of a heap-allocated copy of the portion
1295 * of the filename that is within the specified location, in unix form
1297 static BOOL
GetLinkLocation( LPCWSTR linkfile
, DWORD
*loc
, char **relative
)
1299 WCHAR filename
[MAX_PATH
], shortfilename
[MAX_PATH
], buffer
[MAX_PATH
];
1300 DWORD len
, i
, r
, filelen
;
1301 const DWORD locations
[] = {
1302 CSIDL_STARTUP
, CSIDL_DESKTOPDIRECTORY
, CSIDL_STARTMENU
,
1303 CSIDL_COMMON_STARTUP
, CSIDL_COMMON_DESKTOPDIRECTORY
,
1304 CSIDL_COMMON_STARTMENU
};
1306 WINE_TRACE("%s\n", wine_dbgstr_w(linkfile
));
1307 filelen
=GetFullPathNameW( linkfile
, MAX_PATH
, shortfilename
, NULL
);
1308 if (filelen
==0 || filelen
>MAX_PATH
)
1311 WINE_TRACE("%s\n", wine_dbgstr_w(shortfilename
));
1313 /* the CSLU Toolkit uses a short path name when creating .lnk files;
1314 * expand or our hardcoded list won't match.
1316 filelen
=GetLongPathNameW(shortfilename
, filename
, MAX_PATH
);
1317 if (filelen
==0 || filelen
>MAX_PATH
)
1320 WINE_TRACE("%s\n", wine_dbgstr_w(filename
));
1322 for( i
=0; i
<sizeof(locations
)/sizeof(locations
[0]); i
++ )
1324 if (!SHGetSpecialFolderPathW( 0, buffer
, locations
[i
], FALSE
))
1327 len
= lstrlenW(buffer
);
1328 if (len
>= MAX_PATH
)
1329 continue; /* We've just trashed memory! Hopefully we are OK */
1331 if (len
> filelen
|| filename
[len
]!='\\')
1333 /* do a lstrcmpinW */
1335 r
= lstrcmpiW( filename
, buffer
);
1336 filename
[len
] = '\\';
1340 /* return the remainder of the string and link type */
1341 *loc
= locations
[i
];
1342 *relative
= relative_path (filename
, buffer
);
1343 return (*relative
!= NULL
);
1349 /* gets the target path directly or through MSI */
1350 static HRESULT
get_cmdline( IShellLinkW
*sl
, LPWSTR szPath
, DWORD pathSize
,
1351 LPWSTR szArgs
, DWORD argsSize
)
1353 IShellLinkDataList
*dl
= NULL
;
1354 EXP_DARWIN_LINK
*dar
= NULL
;
1360 hr
= IShellLinkW_GetPath( sl
, szPath
, pathSize
, NULL
, SLGP_RAWPATH
);
1361 if (hr
== S_OK
&& szPath
[0])
1363 IShellLinkW_GetArguments( sl
, szArgs
, argsSize
);
1367 hr
= IShellLinkW_QueryInterface( sl
, &IID_IShellLinkDataList
, (LPVOID
*) &dl
);
1371 hr
= IShellLinkDataList_CopyDataBlock( dl
, EXP_DARWIN_ID_SIG
, (LPVOID
*) &dar
);
1378 hr
= CommandLineFromMsiDescriptor( dar
->szwDarwinID
, NULL
, &cmdSize
);
1379 if (hr
== ERROR_SUCCESS
)
1382 szCmdline
= HeapAlloc( GetProcessHeap(), 0, cmdSize
*sizeof(WCHAR
) );
1383 hr
= CommandLineFromMsiDescriptor( dar
->szwDarwinID
, szCmdline
, &cmdSize
);
1384 WINE_TRACE(" command : %s\n", wine_dbgstr_w(szCmdline
));
1385 if (hr
== ERROR_SUCCESS
)
1388 int bcount
, in_quotes
;
1390 /* Extract the application path */
1397 if ((*s
==0x0009 || *s
==0x0020) && !in_quotes
)
1399 /* skip the remaining spaces */
1402 } while (*s
==0x0009 || *s
==0x0020);
1405 else if (*s
==0x005c)
1411 else if (*s
==0x0022)
1414 if ((bcount
& 1)==0)
1416 /* Preceded by an even number of '\', this is
1417 * half that number of '\', plus a quote which
1421 in_quotes
=!in_quotes
;
1426 /* Preceded by an odd number of '\', this is
1427 * half that number of '\' followed by a '"'
1437 /* a regular character */
1441 if ((d
-szPath
) == pathSize
)
1443 /* Keep processing the path till we get to the
1444 * arguments, but 'stand still'
1449 /* Close the application path */
1452 lstrcpynW(szArgs
, s
, argsSize
);
1454 HeapFree( GetProcessHeap(), 0, szCmdline
);
1459 IShellLinkDataList_Release( dl
);
1463 static WCHAR
* assoc_query(ASSOCSTR assocStr
, LPCWSTR name
, LPCWSTR extra
)
1466 WCHAR
*value
= NULL
;
1468 hr
= AssocQueryStringW(0, assocStr
, name
, extra
, NULL
, &size
);
1471 value
= HeapAlloc(GetProcessHeap(), 0, size
* sizeof(WCHAR
));
1474 hr
= AssocQueryStringW(0, assocStr
, name
, extra
, value
, &size
);
1477 HeapFree(GetProcessHeap(), 0, value
);
1485 static char *slashes_to_minuses(const char *string
)
1488 char *ret
= HeapAlloc(GetProcessHeap(), 0, lstrlenA(string
) + 1);
1491 for (i
= 0; string
[i
]; i
++)
1493 if (string
[i
] == '/')
1504 static BOOL
next_line(FILE *file
, char **line
, int *size
)
1511 *line
= HeapAlloc(GetProcessHeap(), 0, *size
);
1513 while (*line
!= NULL
)
1515 if (fgets(&(*line
)[pos
], *size
- pos
, file
) == NULL
)
1517 HeapFree(GetProcessHeap(), 0, *line
);
1523 pos
= strlen(*line
);
1524 cr
= strchr(*line
, '\n');
1529 line2
= HeapReAlloc(GetProcessHeap(), 0, *line
, *size
);
1534 HeapFree(GetProcessHeap(), 0, *line
);
1547 static BOOL
add_mimes(const char *xdg_data_dir
, struct list
*mime_types
)
1549 char *globs_filename
= NULL
;
1551 globs_filename
= heap_printf("%s/mime/globs", xdg_data_dir
);
1554 FILE *globs_file
= fopen(globs_filename
, "r");
1555 if (globs_file
) /* doesn't have to exist */
1559 while (ret
&& (ret
= next_line(globs_file
, &line
, &size
)) && line
)
1562 struct xdg_mime_type
*mime_type_entry
= NULL
;
1563 if (line
[0] != '#' && (pos
= strchr(line
, ':')))
1565 mime_type_entry
= HeapAlloc(GetProcessHeap(), 0, sizeof(struct xdg_mime_type
));
1566 if (mime_type_entry
)
1569 mime_type_entry
->mimeType
= strdupA(line
);
1570 mime_type_entry
->glob
= strdupA(pos
+ 1);
1571 if (mime_type_entry
->mimeType
&& mime_type_entry
->glob
)
1572 list_add_tail(mime_types
, &mime_type_entry
->entry
);
1575 HeapFree(GetProcessHeap(), 0, mime_type_entry
->mimeType
);
1576 HeapFree(GetProcessHeap(), 0, mime_type_entry
->glob
);
1577 HeapFree(GetProcessHeap(), 0, mime_type_entry
);
1585 HeapFree(GetProcessHeap(), 0, line
);
1588 HeapFree(GetProcessHeap(), 0, globs_filename
);
1595 static void free_native_mime_types(struct list
*native_mime_types
)
1597 struct xdg_mime_type
*mime_type_entry
, *mime_type_entry2
;
1599 LIST_FOR_EACH_ENTRY_SAFE(mime_type_entry
, mime_type_entry2
, native_mime_types
, struct xdg_mime_type
, entry
)
1601 list_remove(&mime_type_entry
->entry
);
1602 HeapFree(GetProcessHeap(), 0, mime_type_entry
->glob
);
1603 HeapFree(GetProcessHeap(), 0, mime_type_entry
->mimeType
);
1604 HeapFree(GetProcessHeap(), 0, mime_type_entry
);
1606 HeapFree(GetProcessHeap(), 0, native_mime_types
);
1609 static BOOL
build_native_mime_types(const char *xdg_data_home
, struct list
**mime_types
)
1611 char *xdg_data_dirs
;
1616 xdg_data_dirs
= getenv("XDG_DATA_DIRS");
1617 if (xdg_data_dirs
== NULL
)
1618 xdg_data_dirs
= heap_printf("/usr/local/share/:/usr/share/");
1620 xdg_data_dirs
= strdupA(xdg_data_dirs
);
1624 *mime_types
= HeapAlloc(GetProcessHeap(), 0, sizeof(struct list
));
1630 list_init(*mime_types
);
1631 ret
= add_mimes(xdg_data_home
, *mime_types
);
1634 for (begin
= xdg_data_dirs
; (end
= strchr(begin
, ':')); begin
= end
+ 1)
1637 ret
= add_mimes(begin
, *mime_types
);
1643 ret
= add_mimes(begin
, *mime_types
);
1648 HeapFree(GetProcessHeap(), 0, xdg_data_dirs
);
1652 if (!ret
&& *mime_types
)
1654 free_native_mime_types(*mime_types
);
1660 static BOOL
match_glob(struct list
*native_mime_types
, const char *extension
,
1664 struct xdg_mime_type
*mime_type_entry
;
1665 int matchLength
= 0;
1669 LIST_FOR_EACH_ENTRY(mime_type_entry
, native_mime_types
, struct xdg_mime_type
, entry
)
1671 if (fnmatch(mime_type_entry
->glob
, extension
, 0) == 0)
1673 if (*match
== NULL
|| matchLength
< strlen(mime_type_entry
->glob
))
1675 *match
= mime_type_entry
->mimeType
;
1676 matchLength
= strlen(mime_type_entry
->glob
);
1683 *match
= strdupA(*match
);
1693 static BOOL
freedesktop_mime_type_for_extension(struct list
*native_mime_types
,
1694 const char *extensionA
,
1698 WCHAR
*lower_extensionW
;
1700 BOOL ret
= match_glob(native_mime_types
, extensionA
, mime_type
);
1701 if (ret
== FALSE
|| *mime_type
!= NULL
)
1703 len
= strlenW(extensionW
);
1704 lower_extensionW
= HeapAlloc(GetProcessHeap(), 0, (len
+ 1)*sizeof(WCHAR
));
1705 if (lower_extensionW
)
1707 char *lower_extensionA
;
1708 memcpy(lower_extensionW
, extensionW
, (len
+ 1)*sizeof(WCHAR
));
1709 strlwrW(lower_extensionW
);
1710 lower_extensionA
= wchars_to_utf8_chars(lower_extensionW
);
1711 if (lower_extensionA
)
1713 ret
= match_glob(native_mime_types
, lower_extensionA
, mime_type
);
1714 HeapFree(GetProcessHeap(), 0, lower_extensionA
);
1719 WINE_FIXME("out of memory\n");
1721 HeapFree(GetProcessHeap(), 0, lower_extensionW
);
1726 WINE_FIXME("out of memory\n");
1731 static WCHAR
* reg_get_valW(HKEY key
, LPCWSTR subkey
, LPCWSTR name
)
1734 if (RegGetValueW(key
, subkey
, name
, RRF_RT_REG_SZ
, NULL
, NULL
, &size
) == ERROR_SUCCESS
)
1736 WCHAR
*ret
= HeapAlloc(GetProcessHeap(), 0, size
);
1739 if (RegGetValueW(key
, subkey
, name
, RRF_RT_REG_SZ
, NULL
, ret
, &size
) == ERROR_SUCCESS
)
1742 HeapFree(GetProcessHeap(), 0, ret
);
1747 static CHAR
* reg_get_val_utf8(HKEY key
, LPCWSTR subkey
, LPCWSTR name
)
1749 WCHAR
*valW
= reg_get_valW(key
, subkey
, name
);
1752 char *val
= wchars_to_utf8_chars(valW
);
1753 HeapFree(GetProcessHeap(), 0, valW
);
1759 static HKEY
open_associations_reg_key(void)
1761 static const WCHAR Software_Wine_FileOpenAssociationsW
[] = {
1762 'S','o','f','t','w','a','r','e','\\','W','i','n','e','\\','F','i','l','e','O','p','e','n','A','s','s','o','c','i','a','t','i','o','n','s',0};
1764 if (RegCreateKeyW(HKEY_CURRENT_USER
, Software_Wine_FileOpenAssociationsW
, &assocKey
) == ERROR_SUCCESS
)
1769 static BOOL
has_association_changed(LPCWSTR extensionW
, LPCSTR mimeType
, LPCWSTR progId
, LPCSTR appName
, LPCWSTR docName
)
1771 static const WCHAR ProgIDW
[] = {'P','r','o','g','I','D',0};
1772 static const WCHAR DocNameW
[] = {'D','o','c','N','a','m','e',0};
1773 static const WCHAR MimeTypeW
[] = {'M','i','m','e','T','y','p','e',0};
1774 static const WCHAR AppNameW
[] = {'A','p','p','N','a','m','e',0};
1778 if ((assocKey
= open_associations_reg_key()))
1785 valueA
= reg_get_val_utf8(assocKey
, extensionW
, MimeTypeW
);
1786 if (!valueA
|| lstrcmpA(valueA
, mimeType
))
1788 HeapFree(GetProcessHeap(), 0, valueA
);
1790 value
= reg_get_valW(assocKey
, extensionW
, ProgIDW
);
1791 if (!value
|| strcmpW(value
, progId
))
1793 HeapFree(GetProcessHeap(), 0, value
);
1795 valueA
= reg_get_val_utf8(assocKey
, extensionW
, AppNameW
);
1796 if (!valueA
|| lstrcmpA(valueA
, appName
))
1798 HeapFree(GetProcessHeap(), 0, valueA
);
1800 value
= reg_get_valW(assocKey
, extensionW
, DocNameW
);
1801 if (docName
&& (!value
|| strcmpW(value
, docName
)))
1803 HeapFree(GetProcessHeap(), 0, value
);
1805 RegCloseKey(assocKey
);
1809 WINE_ERR("error opening associations registry key\n");
1815 static void update_association(LPCWSTR extension
, LPCSTR mimeType
, LPCWSTR progId
, LPCSTR appName
, LPCWSTR docName
, LPCSTR desktopFile
)
1817 static const WCHAR ProgIDW
[] = {'P','r','o','g','I','D',0};
1818 static const WCHAR DocNameW
[] = {'D','o','c','N','a','m','e',0};
1819 static const WCHAR MimeTypeW
[] = {'M','i','m','e','T','y','p','e',0};
1820 static const WCHAR AppNameW
[] = {'A','p','p','N','a','m','e',0};
1821 static const WCHAR DesktopFileW
[] = {'D','e','s','k','t','o','p','F','i','l','e',0};
1822 HKEY assocKey
= NULL
;
1824 WCHAR
*mimeTypeW
= NULL
;
1825 WCHAR
*appNameW
= NULL
;
1826 WCHAR
*desktopFileW
= NULL
;
1828 assocKey
= open_associations_reg_key();
1829 if (assocKey
== NULL
)
1831 WINE_ERR("could not open file associations key\n");
1835 if (RegCreateKeyW(assocKey
, extension
, &subkey
) != ERROR_SUCCESS
)
1837 WINE_ERR("could not create extension subkey\n");
1841 mimeTypeW
= utf8_chars_to_wchars(mimeType
);
1842 if (mimeTypeW
== NULL
)
1844 WINE_ERR("out of memory\n");
1848 appNameW
= utf8_chars_to_wchars(appName
);
1849 if (appNameW
== NULL
)
1851 WINE_ERR("out of memory\n");
1855 desktopFileW
= utf8_chars_to_wchars(desktopFile
);
1856 if (desktopFileW
== NULL
)
1858 WINE_ERR("out of memory\n");
1862 RegSetValueExW(subkey
, MimeTypeW
, 0, REG_SZ
, (const BYTE
*) mimeTypeW
, (lstrlenW(mimeTypeW
) + 1) * sizeof(WCHAR
));
1863 RegSetValueExW(subkey
, ProgIDW
, 0, REG_SZ
, (const BYTE
*) progId
, (lstrlenW(progId
) + 1) * sizeof(WCHAR
));
1864 RegSetValueExW(subkey
, AppNameW
, 0, REG_SZ
, (const BYTE
*) appNameW
, (lstrlenW(appNameW
) + 1) * sizeof(WCHAR
));
1866 RegSetValueExW(subkey
, DocNameW
, 0, REG_SZ
, (const BYTE
*) docName
, (lstrlenW(docName
) + 1) * sizeof(WCHAR
));
1867 RegSetValueExW(subkey
, DesktopFileW
, 0, REG_SZ
, (const BYTE
*) desktopFileW
, (lstrlenW(desktopFileW
) + 1) * sizeof(WCHAR
));
1870 RegCloseKey(assocKey
);
1871 RegCloseKey(subkey
);
1872 HeapFree(GetProcessHeap(), 0, mimeTypeW
);
1873 HeapFree(GetProcessHeap(), 0, appNameW
);
1874 HeapFree(GetProcessHeap(), 0, desktopFileW
);
1877 static BOOL
cleanup_associations(void)
1879 static const WCHAR openW
[] = {'o','p','e','n',0};
1880 static const WCHAR DesktopFileW
[] = {'D','e','s','k','t','o','p','F','i','l','e',0};
1882 BOOL hasChanged
= FALSE
;
1883 if ((assocKey
= open_associations_reg_key()))
1887 for (i
= 0; !done
; i
++)
1889 WCHAR
*extensionW
= NULL
;
1895 HeapFree(GetProcessHeap(), 0, extensionW
);
1896 extensionW
= HeapAlloc(GetProcessHeap(), 0, size
* sizeof(WCHAR
));
1897 if (extensionW
== NULL
)
1899 WINE_ERR("out of memory\n");
1900 ret
= ERROR_OUTOFMEMORY
;
1903 ret
= RegEnumKeyExW(assocKey
, i
, extensionW
, &size
, NULL
, NULL
, NULL
, NULL
);
1905 } while (ret
== ERROR_MORE_DATA
);
1907 if (ret
== ERROR_SUCCESS
)
1910 command
= assoc_query(ASSOCSTR_COMMAND
, extensionW
, openW
);
1911 if (command
== NULL
)
1913 char *desktopFile
= reg_get_val_utf8(assocKey
, extensionW
, DesktopFileW
);
1916 WINE_TRACE("removing file type association for %s\n", wine_dbgstr_w(extensionW
));
1917 remove(desktopFile
);
1919 RegDeleteKeyW(assocKey
, extensionW
);
1921 HeapFree(GetProcessHeap(), 0, desktopFile
);
1923 HeapFree(GetProcessHeap(), 0, command
);
1927 if (ret
!= ERROR_NO_MORE_ITEMS
)
1928 WINE_ERR("error %d while reading registry\n", ret
);
1931 HeapFree(GetProcessHeap(), 0, extensionW
);
1933 RegCloseKey(assocKey
);
1936 WINE_ERR("could not open file associations key\n");
1940 static BOOL
write_freedesktop_mime_type_entry(const char *packages_dir
, const char *dot_extension
,
1941 const char *mime_type
, const char *comment
)
1946 WINE_TRACE("writing MIME type %s, extension=%s, comment=%s\n", wine_dbgstr_a(mime_type
),
1947 wine_dbgstr_a(dot_extension
), wine_dbgstr_a(comment
));
1949 filename
= heap_printf("%s/x-wine-extension-%s.xml", packages_dir
, &dot_extension
[1]);
1952 FILE *packageFile
= fopen(filename
, "w");
1955 fprintf(packageFile
, "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n");
1956 fprintf(packageFile
, "<mime-info xmlns=\"http://www.freedesktop.org/standards/shared-mime-info\">\n");
1957 fprintf(packageFile
, " <mime-type type=\"");
1958 write_xml_text(packageFile
, mime_type
);
1959 fprintf(packageFile
, "\">\n");
1960 fprintf(packageFile
, " <glob pattern=\"*");
1961 write_xml_text(packageFile
, dot_extension
);
1962 fprintf(packageFile
, "\"/>\n");
1965 fprintf(packageFile
, " <comment>");
1966 write_xml_text(packageFile
, comment
);
1967 fprintf(packageFile
, "</comment>\n");
1969 fprintf(packageFile
, " </mime-type>\n");
1970 fprintf(packageFile
, "</mime-info>\n");
1972 fclose(packageFile
);
1975 WINE_ERR("error writing file %s\n", filename
);
1976 HeapFree(GetProcessHeap(), 0, filename
);
1979 WINE_ERR("out of memory\n");
1983 static BOOL
is_extension_blacklisted(LPCWSTR extension
)
1985 /* These are managed through external tools like wine.desktop, to evade malware created file type associations */
1986 static const WCHAR comW
[] = {'.','c','o','m',0};
1987 static const WCHAR exeW
[] = {'.','e','x','e',0};
1988 static const WCHAR msiW
[] = {'.','m','s','i',0};
1990 if (!strcmpiW(extension
, comW
) ||
1991 !strcmpiW(extension
, exeW
) ||
1992 !strcmpiW(extension
, msiW
))
1997 static const char* get_special_mime_type(LPCWSTR extension
)
1999 static const WCHAR lnkW
[] = {'.','l','n','k',0};
2000 if (!strcmpiW(extension
, lnkW
))
2001 return "application/x-ms-shortcut";
2005 static BOOL
write_freedesktop_association_entry(const char *desktopPath
, const char *dot_extension
,
2006 const char *friendlyAppName
, const char *mimeType
,
2012 WINE_TRACE("writing association for file type %s, friendlyAppName=%s, MIME type %s, progID=%s, to file %s\n",
2013 wine_dbgstr_a(dot_extension
), wine_dbgstr_a(friendlyAppName
), wine_dbgstr_a(mimeType
),
2014 wine_dbgstr_a(progId
), wine_dbgstr_a(desktopPath
));
2016 desktop
= fopen(desktopPath
, "w");
2019 fprintf(desktop
, "[Desktop Entry]\n");
2020 fprintf(desktop
, "Type=Application\n");
2021 fprintf(desktop
, "Name=%s\n", friendlyAppName
);
2022 fprintf(desktop
, "MimeType=%s\n", mimeType
);
2023 fprintf(desktop
, "Exec=wine start /ProgIDOpen %s %%f\n", progId
);
2024 fprintf(desktop
, "NoDisplay=true\n");
2025 fprintf(desktop
, "StartupNotify=true\n");
2030 WINE_ERR("error writing association file %s\n", wine_dbgstr_a(desktopPath
));
2034 static BOOL
generate_associations(const char *xdg_data_home
, const char *packages_dir
, const char *applications_dir
)
2036 static const WCHAR openW
[] = {'o','p','e','n',0};
2037 struct list
*nativeMimeTypes
= NULL
;
2040 BOOL hasChanged
= FALSE
;
2042 if (!build_native_mime_types(xdg_data_home
, &nativeMimeTypes
))
2044 WINE_ERR("could not build native MIME types\n");
2050 WCHAR
*extensionW
= NULL
;
2055 HeapFree(GetProcessHeap(), 0, extensionW
);
2056 extensionW
= HeapAlloc(GetProcessHeap(), 0, size
* sizeof(WCHAR
));
2057 if (extensionW
== NULL
)
2059 WINE_ERR("out of memory\n");
2060 ret
= ERROR_OUTOFMEMORY
;
2063 ret
= RegEnumKeyExW(HKEY_CLASSES_ROOT
, i
, extensionW
, &size
, NULL
, NULL
, NULL
, NULL
);
2065 } while (ret
== ERROR_MORE_DATA
);
2067 if (ret
== ERROR_SUCCESS
&& extensionW
[0] == '.' && !is_extension_blacklisted(extensionW
))
2069 char *extensionA
= NULL
;
2070 WCHAR
*commandW
= NULL
;
2071 WCHAR
*friendlyDocNameW
= NULL
;
2072 char *friendlyDocNameA
= NULL
;
2073 WCHAR
*iconW
= NULL
;
2075 WCHAR
*contentTypeW
= NULL
;
2076 char *mimeTypeA
= NULL
;
2077 WCHAR
*friendlyAppNameW
= NULL
;
2078 char *friendlyAppNameA
= NULL
;
2079 WCHAR
*progIdW
= NULL
;
2080 char *progIdA
= NULL
;
2082 extensionA
= wchars_to_utf8_chars(extensionW
);
2083 if (extensionA
== NULL
)
2085 WINE_ERR("out of memory\n");
2089 friendlyDocNameW
= assoc_query(ASSOCSTR_FRIENDLYDOCNAME
, extensionW
, NULL
);
2090 if (friendlyDocNameW
)
2092 friendlyDocNameA
= wchars_to_utf8_chars(friendlyDocNameW
);
2093 if (friendlyDocNameA
== NULL
)
2095 WINE_ERR("out of memory\n");
2100 iconW
= assoc_query(ASSOCSTR_DEFAULTICON
, extensionW
, NULL
);
2102 contentTypeW
= assoc_query(ASSOCSTR_CONTENTTYPE
, extensionW
, NULL
);
2104 strlwrW(contentTypeW
);
2106 if (!freedesktop_mime_type_for_extension(nativeMimeTypes
, extensionA
, extensionW
, &mimeTypeA
))
2109 if (mimeTypeA
== NULL
)
2111 if (contentTypeW
!= NULL
&& strchrW(contentTypeW
, '/'))
2112 mimeTypeA
= wchars_to_utf8_chars(contentTypeW
);
2113 else if ((get_special_mime_type(extensionW
)))
2114 mimeTypeA
= strdupA(get_special_mime_type(extensionW
));
2116 mimeTypeA
= heap_printf("application/x-wine-extension-%s", &extensionA
[1]);
2118 if (mimeTypeA
!= NULL
)
2120 /* Gnome seems to ignore the <icon> tag in MIME packages,
2121 * and the default name is more intuitive anyway.
2125 char *flattened_mime
= slashes_to_minuses(mimeTypeA
);
2129 WCHAR
*comma
= strrchrW(iconW
, ',');
2133 index
= atoiW(comma
+ 1);
2135 iconA
= extract_icon(iconW
, index
, flattened_mime
, FALSE
);
2136 HeapFree(GetProcessHeap(), 0, flattened_mime
);
2140 write_freedesktop_mime_type_entry(packages_dir
, extensionA
, mimeTypeA
, friendlyDocNameA
);
2145 WINE_FIXME("out of memory\n");
2150 commandW
= assoc_query(ASSOCSTR_COMMAND
, extensionW
, openW
);
2151 if (commandW
== NULL
)
2152 /* no command => no application is associated */
2155 friendlyAppNameW
= assoc_query(ASSOCSTR_FRIENDLYAPPNAME
, extensionW
, NULL
);
2156 if (friendlyAppNameW
)
2158 friendlyAppNameA
= wchars_to_utf8_chars(friendlyAppNameW
);
2159 if (friendlyAppNameA
== NULL
)
2161 WINE_ERR("out of memory\n");
2167 friendlyAppNameA
= heap_printf("A Wine application");
2168 if (friendlyAppNameA
== NULL
)
2170 WINE_ERR("out of memory\n");
2175 progIdW
= reg_get_valW(HKEY_CLASSES_ROOT
, extensionW
, NULL
);
2178 progIdA
= escape(progIdW
);
2179 if (progIdA
== NULL
)
2181 WINE_ERR("out of memory\n");
2186 goto end
; /* no progID => not a file type association */
2188 if (has_association_changed(extensionW
, mimeTypeA
, progIdW
, friendlyAppNameA
, friendlyDocNameW
))
2190 char *desktopPath
= heap_printf("%s/wine-extension-%s.desktop", applications_dir
, &extensionA
[1]);
2193 if (write_freedesktop_association_entry(desktopPath
, extensionA
, friendlyAppNameA
, mimeTypeA
, progIdA
))
2196 update_association(extensionW
, mimeTypeA
, progIdW
, friendlyAppNameA
, friendlyDocNameW
, desktopPath
);
2198 HeapFree(GetProcessHeap(), 0, desktopPath
);
2203 HeapFree(GetProcessHeap(), 0, extensionA
);
2204 HeapFree(GetProcessHeap(), 0, commandW
);
2205 HeapFree(GetProcessHeap(), 0, friendlyDocNameW
);
2206 HeapFree(GetProcessHeap(), 0, friendlyDocNameA
);
2207 HeapFree(GetProcessHeap(), 0, iconW
);
2208 HeapFree(GetProcessHeap(), 0, iconA
);
2209 HeapFree(GetProcessHeap(), 0, contentTypeW
);
2210 HeapFree(GetProcessHeap(), 0, mimeTypeA
);
2211 HeapFree(GetProcessHeap(), 0, friendlyAppNameW
);
2212 HeapFree(GetProcessHeap(), 0, friendlyAppNameA
);
2213 HeapFree(GetProcessHeap(), 0, progIdW
);
2214 HeapFree(GetProcessHeap(), 0, progIdA
);
2216 HeapFree(GetProcessHeap(), 0, extensionW
);
2217 if (ret
!= ERROR_SUCCESS
)
2221 free_native_mime_types(nativeMimeTypes
);
2225 static BOOL
InvokeShellLinker( IShellLinkW
*sl
, LPCWSTR link
, BOOL bWait
)
2227 static const WCHAR startW
[] = {'\\','c','o','m','m','a','n','d',
2228 '\\','s','t','a','r','t','.','e','x','e',0};
2229 char *link_name
= NULL
, *icon_name
= NULL
, *work_dir
= NULL
;
2230 char *escaped_path
= NULL
, *escaped_args
= NULL
, *description
= NULL
;
2231 WCHAR szTmp
[INFOTIPSIZE
];
2232 WCHAR szDescription
[INFOTIPSIZE
], szPath
[MAX_PATH
], szWorkDir
[MAX_PATH
];
2233 WCHAR szArgs
[INFOTIPSIZE
], szIconPath
[MAX_PATH
];
2234 int iIconId
= 0, r
= -1;
2237 char *unix_link
= NULL
;
2241 WINE_ERR("Link name is null\n");
2245 if( !GetLinkLocation( link
, &csidl
, &link_name
) )
2247 WINE_WARN("Unknown link location %s. Ignoring.\n",wine_dbgstr_w(link
));
2250 if (!in_desktop_dir(csidl
) && !in_startmenu(csidl
))
2252 WINE_WARN("Not under desktop or start menu. Ignoring.\n");
2255 WINE_TRACE("Link : %s\n", wine_dbgstr_a(link_name
));
2258 IShellLinkW_GetWorkingDirectory( sl
, szTmp
, MAX_PATH
);
2259 ExpandEnvironmentStringsW(szTmp
, szWorkDir
, MAX_PATH
);
2260 WINE_TRACE("workdir : %s\n", wine_dbgstr_w(szWorkDir
));
2263 IShellLinkW_GetDescription( sl
, szTmp
, INFOTIPSIZE
);
2264 ExpandEnvironmentStringsW(szTmp
, szDescription
, INFOTIPSIZE
);
2265 WINE_TRACE("description: %s\n", wine_dbgstr_w(szDescription
));
2267 get_cmdline( sl
, szTmp
, MAX_PATH
, szArgs
, INFOTIPSIZE
);
2268 ExpandEnvironmentStringsW(szTmp
, szPath
, MAX_PATH
);
2269 WINE_TRACE("path : %s\n", wine_dbgstr_w(szPath
));
2270 WINE_TRACE("args : %s\n", wine_dbgstr_w(szArgs
));
2273 IShellLinkW_GetIconLocation( sl
, szTmp
, MAX_PATH
, &iIconId
);
2274 ExpandEnvironmentStringsW(szTmp
, szIconPath
, MAX_PATH
);
2275 WINE_TRACE("icon file : %s\n", wine_dbgstr_w(szIconPath
) );
2279 LPITEMIDLIST pidl
= NULL
;
2280 IShellLinkW_GetIDList( sl
, &pidl
);
2281 if( pidl
&& SHGetPathFromIDListW( pidl
, szPath
) )
2282 WINE_TRACE("pidl path : %s\n", wine_dbgstr_w(szPath
));
2285 /* extract the icon */
2287 icon_name
= extract_icon( szIconPath
, iIconId
, NULL
, bWait
);
2289 icon_name
= extract_icon( szPath
, iIconId
, NULL
, bWait
);
2291 /* fail - try once again after parent process exit */
2296 WINE_WARN("Unable to extract icon, deferring.\n");
2299 WINE_ERR("failed to extract icon from %s\n",
2300 wine_dbgstr_w( szIconPath
[0] ? szIconPath
: szPath
));
2303 unix_link
= wine_get_unix_file_name(link
);
2304 if (unix_link
== NULL
)
2306 WINE_WARN("couldn't find unix path of %s\n", wine_dbgstr_w(link
));
2310 /* check the path */
2313 static const WCHAR exeW
[] = {'.','e','x','e',0};
2316 /* check for .exe extension */
2317 if (!(p
= strrchrW( szPath
, '.' )) ||
2318 strchrW( p
, '\\' ) || strchrW( p
, '/' ) ||
2319 lstrcmpiW( p
, exeW
))
2321 /* Not .exe - use 'start.exe' to launch this file */
2322 p
= szArgs
+ lstrlenW(szPath
) + 2;
2326 memmove( p
+1, szArgs
, min( (lstrlenW(szArgs
) + 1) * sizeof(szArgs
[0]),
2327 sizeof(szArgs
) - (p
+ 1 - szArgs
) * sizeof(szArgs
[0]) ) );
2333 lstrcpyW(szArgs
+ 1, szPath
);
2336 GetWindowsDirectoryW(szPath
, MAX_PATH
);
2337 lstrcatW(szPath
, startW
);
2340 /* convert app working dir */
2342 work_dir
= wine_get_unix_file_name( szWorkDir
);
2346 /* if there's no path... try run the link itself */
2347 lstrcpynW(szArgs
, link
, MAX_PATH
);
2348 GetWindowsDirectoryW(szPath
, MAX_PATH
);
2349 lstrcatW(szPath
, startW
);
2352 /* escape the path and parameters */
2353 escaped_path
= escape(szPath
);
2354 escaped_args
= escape(szArgs
);
2355 description
= wchars_to_utf8_chars(szDescription
);
2356 if (escaped_path
== NULL
|| escaped_args
== NULL
|| description
== NULL
)
2358 WINE_ERR("out of memory allocating/escaping parameters\n");
2362 /* building multiple menus concurrently has race conditions */
2363 hsem
= CreateSemaphoreA( NULL
, 1, 1, "winemenubuilder_semaphore");
2364 if( WAIT_OBJECT_0
!= MsgWaitForMultipleObjects( 1, &hsem
, FALSE
, INFINITE
, QS_ALLINPUT
) )
2366 WINE_ERR("failed wait for semaphore\n");
2370 if (in_desktop_dir(csidl
))
2373 const char *lastEntry
;
2374 lastEntry
= strrchr(link_name
, '/');
2375 if (lastEntry
== NULL
)
2376 lastEntry
= link_name
;
2379 location
= heap_printf("%s/%s.desktop", xdg_desktop_dir
, lastEntry
);
2382 r
= !write_desktop_entry(NULL
, location
, lastEntry
, escaped_path
, escaped_args
, description
, work_dir
, icon_name
);
2384 chmod(location
, 0755);
2385 HeapFree(GetProcessHeap(), 0, location
);
2390 WCHAR
*unix_linkW
= utf8_chars_to_wchars(unix_link
);
2393 char *escaped_lnk
= escape(unix_linkW
);
2396 char *menuarg
= heap_printf("/Unix %s", escaped_lnk
);
2399 r
= !write_menu_entry(unix_link
, link_name
, "start", menuarg
, description
, work_dir
, icon_name
);
2400 HeapFree(GetProcessHeap(), 0, menuarg
);
2402 HeapFree(GetProcessHeap(), 0, escaped_lnk
);
2404 HeapFree(GetProcessHeap(), 0, unix_linkW
);
2408 ReleaseSemaphore( hsem
, 1, NULL
);
2411 if (hsem
) CloseHandle( hsem
);
2412 HeapFree( GetProcessHeap(), 0, icon_name
);
2413 HeapFree( GetProcessHeap(), 0, work_dir
);
2414 HeapFree( GetProcessHeap(), 0, link_name
);
2415 HeapFree( GetProcessHeap(), 0, escaped_args
);
2416 HeapFree( GetProcessHeap(), 0, escaped_path
);
2417 HeapFree( GetProcessHeap(), 0, description
);
2418 HeapFree( GetProcessHeap(), 0, unix_link
);
2421 WINE_ERR("failed to build the menu\n" );
2426 static BOOL
InvokeShellLinkerForURL( IUniformResourceLocatorW
*url
, LPCWSTR link
, BOOL bWait
)
2428 char *link_name
= NULL
;
2431 char *escaped_urlPath
= NULL
;
2436 char *unix_link
= NULL
;
2440 WINE_ERR("Link name is null\n");
2444 if( !GetLinkLocation( link
, &csidl
, &link_name
) )
2446 WINE_WARN("Unknown link location %s. Ignoring.\n",wine_dbgstr_w(link
));
2449 if (!in_desktop_dir(csidl
) && !in_startmenu(csidl
))
2451 WINE_WARN("Not under desktop or start menu. Ignoring.\n");
2455 WINE_TRACE("Link : %s\n", wine_dbgstr_a(link_name
));
2457 hr
= url
->lpVtbl
->GetURL(url
, &urlPath
);
2463 WINE_TRACE("path : %s\n", wine_dbgstr_w(urlPath
));
2465 unix_link
= wine_get_unix_file_name(link
);
2466 if (unix_link
== NULL
)
2468 WINE_WARN("couldn't find unix path of %s\n", wine_dbgstr_w(link
));
2472 escaped_urlPath
= escape(urlPath
);
2473 if (escaped_urlPath
== NULL
)
2475 WINE_ERR("couldn't escape url, out of memory\n");
2479 hSem
= CreateSemaphoreA( NULL
, 1, 1, "winemenubuilder_semaphore");
2480 if( WAIT_OBJECT_0
!= MsgWaitForMultipleObjects( 1, &hSem
, FALSE
, INFINITE
, QS_ALLINPUT
) )
2482 WINE_ERR("failed wait for semaphore\n");
2485 if (in_desktop_dir(csidl
))
2488 const char *lastEntry
;
2489 lastEntry
= strrchr(link_name
, '/');
2490 if (lastEntry
== NULL
)
2491 lastEntry
= link_name
;
2494 location
= heap_printf("%s/%s.desktop", xdg_desktop_dir
, lastEntry
);
2497 r
= !write_desktop_entry(NULL
, location
, lastEntry
, "winebrowser", escaped_urlPath
, NULL
, NULL
, NULL
);
2499 chmod(location
, 0755);
2500 HeapFree(GetProcessHeap(), 0, location
);
2504 r
= !write_menu_entry(unix_link
, link_name
, "winebrowser", escaped_urlPath
, NULL
, NULL
, NULL
);
2506 ReleaseSemaphore(hSem
, 1, NULL
);
2511 HeapFree(GetProcessHeap(), 0, link_name
);
2512 CoTaskMemFree( urlPath
);
2513 HeapFree(GetProcessHeap(), 0, escaped_urlPath
);
2514 HeapFree(GetProcessHeap(), 0, unix_link
);
2518 static BOOL
WaitForParentProcess( void )
2520 PROCESSENTRY32 procentry
;
2521 HANDLE hsnapshot
= NULL
, hprocess
= NULL
;
2522 DWORD ourpid
= GetCurrentProcessId();
2523 BOOL ret
= FALSE
, rc
;
2525 WINE_TRACE("Waiting for parent process\n");
2526 if ((hsnapshot
= CreateToolhelp32Snapshot( TH32CS_SNAPPROCESS
, 0 )) ==
2527 INVALID_HANDLE_VALUE
)
2529 WINE_ERR("CreateToolhelp32Snapshot failed, error %d\n", GetLastError());
2533 procentry
.dwSize
= sizeof(PROCESSENTRY32
);
2534 rc
= Process32First( hsnapshot
, &procentry
);
2537 if (procentry
.th32ProcessID
== ourpid
) break;
2538 rc
= Process32Next( hsnapshot
, &procentry
);
2542 WINE_WARN("Unable to find current process id %d when listing processes\n", ourpid
);
2546 if ((hprocess
= OpenProcess( SYNCHRONIZE
, FALSE
, procentry
.th32ParentProcessID
)) ==
2549 WINE_WARN("OpenProcess failed pid=%d, error %d\n", procentry
.th32ParentProcessID
,
2554 if (MsgWaitForMultipleObjects( 1, &hprocess
, FALSE
, INFINITE
, QS_ALLINPUT
) == WAIT_OBJECT_0
)
2557 WINE_ERR("Unable to wait for parent process, error %d\n", GetLastError());
2560 if (hprocess
) CloseHandle( hprocess
);
2561 if (hsnapshot
) CloseHandle( hsnapshot
);
2565 static BOOL
Process_Link( LPCWSTR linkname
, BOOL bWait
)
2570 WCHAR fullname
[MAX_PATH
];
2573 WINE_TRACE("%s, wait %d\n", wine_dbgstr_w(linkname
), bWait
);
2577 WINE_ERR("link name missing\n");
2581 len
=GetFullPathNameW( linkname
, MAX_PATH
, fullname
, NULL
);
2582 if (len
==0 || len
>MAX_PATH
)
2584 WINE_ERR("couldn't get full path of link file\n");
2588 r
= CoInitialize( NULL
);
2591 WINE_ERR("CoInitialize failed\n");
2595 r
= CoCreateInstance( &CLSID_ShellLink
, NULL
, CLSCTX_INPROC_SERVER
,
2596 &IID_IShellLinkW
, (LPVOID
*) &sl
);
2599 WINE_ERR("No IID_IShellLink\n");
2603 r
= IShellLinkW_QueryInterface( sl
, &IID_IPersistFile
, (LPVOID
*) &pf
);
2606 WINE_ERR("No IID_IPersistFile\n");
2610 r
= IPersistFile_Load( pf
, fullname
, STGM_READ
);
2611 if( SUCCEEDED( r
) )
2613 /* If something fails (eg. Couldn't extract icon)
2614 * wait for parent process and try again
2616 if( ! InvokeShellLinker( sl
, fullname
, bWait
) && bWait
)
2618 WaitForParentProcess();
2619 InvokeShellLinker( sl
, fullname
, FALSE
);
2624 WINE_ERR("unable to load %s\n", wine_dbgstr_w(linkname
));
2627 IPersistFile_Release( pf
);
2628 IShellLinkW_Release( sl
);
2635 static BOOL
Process_URL( LPCWSTR urlname
, BOOL bWait
)
2637 IUniformResourceLocatorW
*url
;
2640 WCHAR fullname
[MAX_PATH
];
2643 WINE_TRACE("%s, wait %d\n", wine_dbgstr_w(urlname
), bWait
);
2647 WINE_ERR("URL name missing\n");
2651 len
=GetFullPathNameW( urlname
, MAX_PATH
, fullname
, NULL
);
2652 if (len
==0 || len
>MAX_PATH
)
2654 WINE_ERR("couldn't get full path of URL file\n");
2658 r
= CoInitialize( NULL
);
2661 WINE_ERR("CoInitialize failed\n");
2665 r
= CoCreateInstance( &CLSID_InternetShortcut
, NULL
, CLSCTX_INPROC_SERVER
,
2666 &IID_IUniformResourceLocatorW
, (LPVOID
*) &url
);
2669 WINE_ERR("No IID_IUniformResourceLocatorW\n");
2673 r
= url
->lpVtbl
->QueryInterface( url
, &IID_IPersistFile
, (LPVOID
*) &pf
);
2676 WINE_ERR("No IID_IPersistFile\n");
2679 r
= IPersistFile_Load( pf
, fullname
, STGM_READ
);
2680 if( SUCCEEDED( r
) )
2682 /* If something fails (eg. Couldn't extract icon)
2683 * wait for parent process and try again
2685 if( ! InvokeShellLinkerForURL( url
, fullname
, bWait
) && bWait
)
2687 WaitForParentProcess();
2688 InvokeShellLinkerForURL( url
, fullname
, FALSE
);
2692 IPersistFile_Release( pf
);
2693 url
->lpVtbl
->Release( url
);
2700 static void RefreshFileTypeAssociations(void)
2703 char *mime_dir
= NULL
;
2704 char *packages_dir
= NULL
;
2705 char *applications_dir
= NULL
;
2708 hSem
= CreateSemaphoreA( NULL
, 1, 1, "winemenubuilder_semaphore");
2709 if( WAIT_OBJECT_0
!= MsgWaitForMultipleObjects( 1, &hSem
, FALSE
, INFINITE
, QS_ALLINPUT
) )
2711 WINE_ERR("failed wait for semaphore\n");
2717 mime_dir
= heap_printf("%s/mime", xdg_data_dir
);
2718 if (mime_dir
== NULL
)
2720 WINE_ERR("out of memory\n");
2723 create_directories(mime_dir
);
2725 packages_dir
= heap_printf("%s/packages", mime_dir
);
2726 if (packages_dir
== NULL
)
2728 WINE_ERR("out of memory\n");
2731 create_directories(packages_dir
);
2733 applications_dir
= heap_printf("%s/applications", xdg_data_dir
);
2734 if (applications_dir
== NULL
)
2736 WINE_ERR("out of memory\n");
2739 create_directories(applications_dir
);
2741 hasChanged
= generate_associations(xdg_data_dir
, packages_dir
, applications_dir
);
2742 hasChanged
|= cleanup_associations();
2745 const char *argv
[3];
2747 argv
[0] = "update-mime-database";
2750 spawnvp( _P_NOWAIT
, argv
[0], argv
);
2752 argv
[0] = "update-desktop-database";
2753 argv
[1] = applications_dir
;
2754 spawnvp( _P_NOWAIT
, argv
[0], argv
);
2760 ReleaseSemaphore(hSem
, 1, NULL
);
2763 HeapFree(GetProcessHeap(), 0, mime_dir
);
2764 HeapFree(GetProcessHeap(), 0, packages_dir
);
2765 HeapFree(GetProcessHeap(), 0, applications_dir
);
2768 static void cleanup_menus(void)
2772 hkey
= open_menus_reg_key();
2776 LSTATUS lret
= ERROR_SUCCESS
;
2777 for (i
= 0; lret
== ERROR_SUCCESS
; )
2779 WCHAR
*value
= NULL
;
2781 DWORD valueSize
= 4096;
2782 DWORD dataSize
= 4096;
2785 lret
= ERROR_OUTOFMEMORY
;
2786 value
= HeapAlloc(GetProcessHeap(), 0, valueSize
* sizeof(WCHAR
));
2789 data
= HeapAlloc(GetProcessHeap(), 0, dataSize
* sizeof(WCHAR
));
2792 lret
= RegEnumValueW(hkey
, i
, value
, &valueSize
, NULL
, NULL
, (BYTE
*)data
, &dataSize
);
2793 if (lret
== ERROR_SUCCESS
|| lret
!= ERROR_MORE_DATA
)
2797 HeapFree(GetProcessHeap(), 0, value
);
2798 HeapFree(GetProcessHeap(), 0, data
);
2799 value
= data
= NULL
;
2801 if (lret
== ERROR_SUCCESS
)
2805 unix_file
= wchars_to_unix_chars(value
);
2806 windows_file
= wchars_to_unix_chars(data
);
2807 if (unix_file
!= NULL
&& windows_file
!= NULL
)
2809 struct stat filestats
;
2810 if (stat(windows_file
, &filestats
) < 0 && errno
== ENOENT
)
2812 WINE_TRACE("removing menu related file %s\n", unix_file
);
2814 RegDeleteValueW(hkey
, value
);
2821 WINE_ERR("out of memory enumerating menus\n");
2822 lret
= ERROR_OUTOFMEMORY
;
2824 HeapFree(GetProcessHeap(), 0, unix_file
);
2825 HeapFree(GetProcessHeap(), 0, windows_file
);
2827 else if (lret
!= ERROR_NO_MORE_ITEMS
)
2828 WINE_ERR("error %d reading registry\n", lret
);
2829 HeapFree(GetProcessHeap(), 0, value
);
2830 HeapFree(GetProcessHeap(), 0, data
);
2835 WINE_ERR("error opening registry key, menu cleanup failed\n");
2838 static WCHAR
*next_token( LPWSTR
*p
)
2840 LPWSTR token
= NULL
, t
= *p
;
2845 while( t
&& !token
)
2853 /* unquote the token */
2855 t
= strchrW( token
, '"' );
2864 t
= strchrW( token
, ' ' );
2874 static BOOL
init_xdg(void)
2876 WCHAR shellDesktopPath
[MAX_PATH
];
2877 HRESULT hr
= SHGetFolderPathW(NULL
, CSIDL_DESKTOP
, NULL
, SHGFP_TYPE_CURRENT
, shellDesktopPath
);
2879 xdg_desktop_dir
= wine_get_unix_file_name(shellDesktopPath
);
2880 if (xdg_desktop_dir
== NULL
)
2882 WINE_ERR("error looking up the desktop directory\n");
2886 if (getenv("XDG_CONFIG_HOME"))
2887 xdg_config_dir
= heap_printf("%s/menus/applications-merged", getenv("XDG_CONFIG_HOME"));
2889 xdg_config_dir
= heap_printf("%s/.config/menus/applications-merged", getenv("HOME"));
2892 create_directories(xdg_config_dir
);
2893 if (getenv("XDG_DATA_HOME"))
2894 xdg_data_dir
= strdupA(getenv("XDG_DATA_HOME"));
2896 xdg_data_dir
= heap_printf("%s/.local/share", getenv("HOME"));
2900 create_directories(xdg_data_dir
);
2901 buffer
= heap_printf("%s/desktop-directories", xdg_data_dir
);
2904 mkdir(buffer
, 0777);
2905 HeapFree(GetProcessHeap(), 0, buffer
);
2909 HeapFree(GetProcessHeap(), 0, xdg_config_dir
);
2911 WINE_ERR("out of memory\n");
2915 /***********************************************************************
2919 int PASCAL
wWinMain (HINSTANCE hInstance
, HINSTANCE prev
, LPWSTR cmdline
, int show
)
2921 static const WCHAR dash_aW
[] = {'-','a',0};
2922 static const WCHAR dash_rW
[] = {'-','r',0};
2923 static const WCHAR dash_uW
[] = {'-','u',0};
2924 static const WCHAR dash_wW
[] = {'-','w',0};
2926 LPWSTR token
= NULL
, p
;
2934 for( p
= cmdline
; p
&& *p
; )
2936 token
= next_token( &p
);
2939 if( !strcmpW( token
, dash_aW
) )
2941 RefreshFileTypeAssociations();
2944 if( !strcmpW( token
, dash_rW
) )
2949 if( !strcmpW( token
, dash_wW
) )
2951 else if ( !strcmpW( token
, dash_uW
) )
2953 else if( token
[0] == '-' )
2955 WINE_ERR( "unknown option %s\n", wine_dbgstr_w(token
) );
2962 bRet
= Process_URL( token
, bWait
);
2964 bRet
= Process_Link( token
, bWait
);
2967 WINE_ERR( "failed to build menu item for %s\n", wine_dbgstr_w(token
) );