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
9 * This library is free software; you can redistribute it and/or
10 * modify it under the terms of the GNU Lesser General Public
11 * License as published by the Free Software Foundation; either
12 * version 2.1 of the License, or (at your option) any later version.
14 * This library is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17 * Lesser General Public License for more details.
19 * You should have received a copy of the GNU Lesser General Public
20 * License along with this library; if not, write to the Free Software
21 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
24 * This program will read a Windows shortcut file using the IShellLink
25 * interface, then invoke wineshelllink with the appropriate arguments
26 * to create a KDE/Gnome menu entry for the shortcut.
28 * winemenubuilder [ -r ] <shortcut.lnk>
30 * If the -r parameter is passed, and the shortcut cannot be created,
31 * this program will add RunOnce entry to invoke itself at the next
32 * reboot. This covers the case when a ShortCut is created before the
33 * executable containing its icon.
38 #include "wine/port.h"
57 #include "wine/unicode.h"
58 #include "wine/debug.h"
61 WINE_DEFAULT_DEBUG_CHANNEL(menubuilder
);
63 #define in_desktop_dir(csidl) ((csidl)==CSIDL_DESKTOPDIRECTORY || \
64 (csidl)==CSIDL_COMMON_DESKTOPDIRECTORY)
65 #define in_startmenu(csidl) ((csidl)==CSIDL_STARTMENU || \
66 (csidl)==CSIDL_COMMON_STARTMENU)
68 /* link file formats */
89 GRPICONDIRENTRY idEntries
[1];
121 /* Icon extraction routines
123 * FIXME: should use PrivateExtractIcons and friends
124 * FIXME: should not use stdio
127 static BOOL
SaveIconResAsXPM(const BITMAPINFO
*pIcon
, const char *szXPMFileName
, LPCWSTR commentW
)
137 BOOL aColorUsed
[256] = {0};
142 if (!((pIcon
->bmiHeader
.biBitCount
== 4) || (pIcon
->bmiHeader
.biBitCount
== 8)))
145 if (!(fXPMFile
= fopen(szXPMFileName
, "w")))
148 i
= WideCharToMultiByte(CP_UNIXCP
, 0, commentW
, -1, NULL
, 0, NULL
, NULL
);
150 WideCharToMultiByte(CP_UNIXCP
, 0, commentW
, -1, comment
, i
, NULL
, NULL
);
152 nHeight
= pIcon
->bmiHeader
.biHeight
/ 2;
153 nXORWidthBytes
= 4 * ((pIcon
->bmiHeader
.biWidth
* pIcon
->bmiHeader
.biBitCount
/ 32)
154 + ((pIcon
->bmiHeader
.biWidth
* pIcon
->bmiHeader
.biBitCount
% 32) > 0));
155 nANDWidthBytes
= 4 * ((pIcon
->bmiHeader
.biWidth
/ 32)
156 + ((pIcon
->bmiHeader
.biWidth
% 32) > 0));
157 b8BitColors
= pIcon
->bmiHeader
.biBitCount
== 8;
158 nColors
= pIcon
->bmiHeader
.biClrUsed
? pIcon
->bmiHeader
.biClrUsed
159 : 1 << pIcon
->bmiHeader
.biBitCount
;
160 pXOR
= (const BYTE
*) pIcon
+ sizeof (BITMAPINFOHEADER
) + (nColors
* sizeof (RGBQUAD
));
161 pAND
= pXOR
+ nHeight
* nXORWidthBytes
;
163 #define MASK(x,y) (pAND[(x) / 8 + (nHeight - (y) - 1) * nANDWidthBytes] & (1 << (7 - (x) % 8)))
164 #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)
166 for (i
= 0; i
< nHeight
; i
++) {
167 for (j
= 0; j
< pIcon
->bmiHeader
.biWidth
; j
++) {
168 if (!aColorUsed
[COLOR(j
,i
)] && !MASK(j
,i
))
170 aColorUsed
[COLOR(j
,i
)] = TRUE
;
176 if (fprintf(fXPMFile
, "/* XPM */\n/* %s */\nstatic char *icon[] = {\n", comment
) <= 0)
178 if (fprintf(fXPMFile
, "\"%d %d %d %d\",\n",
179 (int) pIcon
->bmiHeader
.biWidth
, nHeight
, nColorsUsed
+ 1, 2) <=0)
182 for (i
= 0; i
< nColors
; i
++) {
184 if (fprintf(fXPMFile
, "\"%.2X c #%.2X%.2X%.2X\",\n", i
, pIcon
->bmiColors
[i
].rgbRed
,
185 pIcon
->bmiColors
[i
].rgbGreen
, pIcon
->bmiColors
[i
].rgbBlue
) <= 0)
188 if (fprintf(fXPMFile
, "\" c None\"") <= 0)
191 for (i
= 0; i
< nHeight
; i
++)
193 if (fprintf(fXPMFile
, ",\n\"") <= 0)
195 for (j
= 0; j
< pIcon
->bmiHeader
.biWidth
; j
++)
199 if (fprintf(fXPMFile
, " ") <= 0)
203 if (fprintf(fXPMFile
, "%.2X", COLOR(j
,i
)) <= 0)
206 if (fprintf(fXPMFile
, "\"") <= 0)
209 if (fprintf(fXPMFile
, "};\n") <= 0)
222 unlink( szXPMFileName
);
226 static BOOL CALLBACK
EnumResNameProc(HMODULE hModule
, LPCWSTR lpszType
, LPWSTR lpszName
, LONG_PTR lParam
)
228 ENUMRESSTRUCT
*sEnumRes
= (ENUMRESSTRUCT
*) lParam
;
230 if (!sEnumRes
->nIndex
--)
232 *sEnumRes
->pResInfo
= FindResourceW(hModule
, lpszName
, (LPCWSTR
)RT_GROUP_ICON
);
239 static BOOL
extract_icon32(LPCWSTR szFileName
, int nIndex
, const char *szXPMFileName
)
243 LPCWSTR lpName
= NULL
;
245 GRPICONDIR
*pIconDir
;
247 ENUMRESSTRUCT sEnumRes
;
253 hModule
= LoadLibraryExW(szFileName
, 0, LOAD_LIBRARY_AS_DATAFILE
);
256 WINE_ERR("LoadLibraryExW (%s) failed, error %ld\n",
257 wine_dbgstr_w(szFileName
), GetLastError());
263 hResInfo
= FindResourceW(hModule
, MAKEINTRESOURCEW(-nIndex
), (LPCWSTR
)RT_GROUP_ICON
);
264 WINE_TRACE("FindResourceW (%s) called, return %p, error %ld\n",
265 wine_dbgstr_w(szFileName
), hResInfo
, GetLastError());
270 sEnumRes
.pResInfo
= &hResInfo
;
271 sEnumRes
.nIndex
= nIndex
;
272 EnumResourceNamesW(hModule
, (LPCWSTR
)RT_GROUP_ICON
, EnumResNameProc
, (LONG_PTR
)&sEnumRes
);
277 if ((hResData
= LoadResource(hModule
, hResInfo
)))
279 if ((pIconDir
= LockResource(hResData
)))
281 for (i
= 0; i
< pIconDir
->idCount
; i
++)
283 if ((pIconDir
->idEntries
[i
].wBitCount
>= nMaxBits
) && (pIconDir
->idEntries
[i
].wBitCount
<= 8))
285 nMaxBits
= pIconDir
->idEntries
[i
].wBitCount
;
287 if ((pIconDir
->idEntries
[i
].bHeight
* pIconDir
->idEntries
[i
].bWidth
) >= nMax
)
289 lpName
= MAKEINTRESOURCEW(pIconDir
->idEntries
[i
].nID
);
290 nMax
= pIconDir
->idEntries
[i
].bHeight
* pIconDir
->idEntries
[i
].bWidth
;
296 FreeResource(hResData
);
301 WINE_ERR("ExtractFromEXEDLL failed, error %ld\n", GetLastError());
302 FreeLibrary(hModule
);
306 if ((hResInfo
= FindResourceW(hModule
, lpName
, (LPCWSTR
)RT_ICON
)))
308 if ((hResData
= LoadResource(hModule
, hResInfo
)))
310 if ((pIcon
= LockResource(hResData
)))
312 if(SaveIconResAsXPM(pIcon
, szXPMFileName
, szFileName
))
316 FreeResource(hResData
);
320 FreeLibrary(hModule
);
324 static BOOL
ExtractFromEXEDLL(LPCWSTR szFileName
, int nIndex
, const char *szXPMFileName
)
326 if (!extract_icon32(szFileName
, nIndex
, szXPMFileName
) /*&&
327 !extract_icon16(szFileName, szXPMFileName)*/)
332 static int ExtractFromICO(LPCWSTR szFileName
, const char *szXPMFileName
)
336 ICONDIRENTRY
*pIconDirEntry
;
343 filename
= wine_get_unix_file_name(szFileName
);
344 if (!(fICOFile
= fopen(filename
, "r")))
347 if (fread(&iconDir
, sizeof (ICONDIR
), 1, fICOFile
) != 1)
349 if ((iconDir
.idReserved
!= 0) || (iconDir
.idType
!= 1))
352 if ((pIconDirEntry
= malloc(iconDir
.idCount
* sizeof (ICONDIRENTRY
))) == NULL
)
354 if (fread(pIconDirEntry
, sizeof (ICONDIRENTRY
), iconDir
.idCount
, fICOFile
) != iconDir
.idCount
)
357 for (i
= 0; i
< iconDir
.idCount
; i
++)
358 if ((pIconDirEntry
[i
].bHeight
* pIconDirEntry
[i
].bWidth
) > nMax
)
361 nMax
= pIconDirEntry
[i
].bHeight
* pIconDirEntry
[i
].bWidth
;
363 if ((pIcon
= malloc(pIconDirEntry
[nIndex
].dwBytesInRes
)) == NULL
)
365 if (fseek(fICOFile
, pIconDirEntry
[nIndex
].dwImageOffset
, SEEK_SET
))
367 if (fread(pIcon
, pIconDirEntry
[nIndex
].dwBytesInRes
, 1, fICOFile
) != 1)
370 if(!SaveIconResAsXPM(pIcon
, szXPMFileName
, szFileName
))
376 HeapFree(GetProcessHeap(), 0, filename
);
386 HeapFree(GetProcessHeap(), 0, filename
);
390 static BOOL
create_default_icon( const char *filename
, const char* comment
)
395 if (!(fXPM
= fopen(filename
, "w"))) return FALSE
;
396 if (fprintf(fXPM
, "/* XPM */\n/* %s */\nstatic char * icon[] = {", comment
) <= 0)
398 for (i
= 0; i
< sizeof(wine_xpm
)/sizeof(wine_xpm
[0]); i
++) {
399 if (fprintf( fXPM
, "\n\"%s\",", wine_xpm
[i
]) <= 0)
402 if (fprintf( fXPM
, "};\n" ) <=0)
413 static unsigned short crc16(const char* string
)
415 unsigned short crc
= 0;
418 for (i
= 0; string
[i
] != 0; i
++)
421 for (j
= 0; j
< 8; c
>>= 1, j
++)
423 xor_poly
= (c
^ crc
) & 1;
432 /* extract an icon from an exe or icon file; helper for IPersistFile_fnSave */
433 static char *extract_icon( LPCWSTR path
, int index
)
437 char *iconsdir
, *ico_path
, *ico_name
, *xpm_path
;
442 /* Where should we save the icon? */
443 WINE_TRACE("path=[%s] index=%d\n", wine_dbgstr_w(path
), index
);
444 iconsdir
=NULL
; /* Default is no icon */
445 /* @@ Wine registry key: HKCU\Software\Wine\WineMenuBuilder */
446 if (!RegOpenKeyA( HKEY_CURRENT_USER
, "Software\\Wine\\WineMenuBuilder", &hkey
))
448 static const WCHAR IconsDirW
[] = {'I','c','o','n','s','D','i','r',0};
452 if (!RegQueryValueExW(hkey
, IconsDirW
, 0, NULL
, NULL
, &size
))
454 iconsdirW
= HeapAlloc(GetProcessHeap(), 0, size
);
455 RegQueryValueExW(hkey
, IconsDirW
, 0, NULL
, (LPBYTE
)iconsdirW
, &size
);
457 if (!(iconsdir
= wine_get_unix_file_name(iconsdirW
)))
459 int n
= WideCharToMultiByte(CP_UNIXCP
, 0, iconsdirW
, -1, NULL
, 0, NULL
, NULL
);
460 iconsdir
= HeapAlloc(GetProcessHeap(), 0, n
);
461 WideCharToMultiByte(CP_UNIXCP
, 0, iconsdirW
, -1, iconsdir
, n
, NULL
, NULL
);
463 HeapFree(GetProcessHeap(), 0, iconsdirW
);
470 WCHAR path
[MAX_PATH
];
472 if (GetTempPathW(MAX_PATH
, path
)) iconsdir
= wine_get_unix_file_name(path
);
476 return NULL
; /* No icon created */
480 HeapFree(GetProcessHeap(), 0, iconsdir
);
481 return NULL
; /* No icon created */
484 /* If icon path begins with a '*' then this is a deferred call */
491 /* Determine the icon base name */
492 n
= WideCharToMultiByte(CP_UNIXCP
, 0, path
, -1, NULL
, 0, NULL
, NULL
);
493 ico_path
= HeapAlloc(GetProcessHeap(), 0, n
);
494 WideCharToMultiByte(CP_UNIXCP
, 0, path
, -1, ico_path
, n
, NULL
, NULL
);
497 if (*s
=='/' || *s
=='\\') {
505 if (*ico_name
=='\\') *ico_name
++='\0';
506 s
=strrchr(ico_name
,'.');
509 /* Compute the source-path hash */
512 /* Try to treat the source file as an exe */
513 xpm_path
=HeapAlloc(GetProcessHeap(), 0, strlen(iconsdir
)+1+4+1+strlen(ico_name
)+1+12+1+3);
514 sprintf(xpm_path
,"%s/%04x_%s.%d.xpm",iconsdir
,crc
,ico_name
,index
);
515 if (ExtractFromEXEDLL( path
, index
, xpm_path
))
518 /* Must be something else, ignore the index in that case */
519 sprintf(xpm_path
,"%s/%04x_%s.xpm",iconsdir
,crc
,ico_name
);
520 if (ExtractFromICO( path
, xpm_path
))
523 if (create_default_icon( xpm_path
, ico_path
))
526 HeapFree( GetProcessHeap(), 0, xpm_path
);
530 HeapFree(GetProcessHeap(), 0, iconsdir
);
531 HeapFree(GetProcessHeap(), 0, ico_path
);
535 static BOOL
DeferToRunOnce(LPWSTR link
)
539 static const WCHAR szRunOnce
[] = {
540 'S','o','f','t','w','a','r','e','\\',
541 'M','i','c','r','o','s','o','f','t','\\',
542 'W','i','n','d','o','w','s','\\',
543 'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\',
544 'R','u','n','O','n','c','e',0
546 static const WCHAR szFormat
[] = { '%','s',' ','"','%','s','"',0 };
548 WCHAR szExecutable
[MAX_PATH
];
550 WINE_TRACE( "Deferring icon creation to reboot.\n");
552 len
= GetModuleFileNameW( 0, szExecutable
, MAX_PATH
);
553 if (!len
|| len
>= MAX_PATH
) return FALSE
;
555 len
= ( lstrlenW( link
) + lstrlenW( szExecutable
) + 4)*sizeof(WCHAR
);
556 buffer
= HeapAlloc( GetProcessHeap(), 0, len
);
560 wsprintfW( buffer
, szFormat
, szExecutable
, link
);
562 r
= RegCreateKeyExW(HKEY_LOCAL_MACHINE
, szRunOnce
, 0,
563 NULL
, REG_OPTION_NON_VOLATILE
, KEY_WRITE
, NULL
, &hkey
, NULL
);
564 if ( r
== ERROR_SUCCESS
)
566 r
= RegSetValueExW(hkey
, link
, 0, REG_SZ
,
567 (LPBYTE
) buffer
, (lstrlenW(buffer
) + 1)*sizeof(WCHAR
));
570 HeapFree(GetProcessHeap(), 0, buffer
);
575 /* This escapes \ in filenames */
576 static LPSTR
escape(LPCWSTR arg
)
583 while((esc
= strchrW(esc
, '\\')))
589 len
+= WideCharToMultiByte(CP_UNIXCP
, 0, arg
, -1, NULL
, 0, NULL
, NULL
);
590 narg
= HeapAlloc(GetProcessHeap(), 0, len
);
595 n
= WideCharToMultiByte(CP_UNIXCP
, 0, arg
, 1, x
, len
, NULL
, NULL
);
599 *x
++='\\'; /* escape \ */
606 static int fork_and_wait( const char *linker
, const char *link_name
, const char *path
,
607 int desktop
, const char *args
, const char *icon_name
,
608 const char *workdir
, const char *description
)
611 const char *argv
[20];
614 WINE_TRACE( "linker app='%s' link='%s' mode=%s "
615 "path='%s' args='%s' icon='%s' workdir='%s' descr='%s'\n",
616 linker
, link_name
, desktop
? "desktop" : "menu",
617 path
, args
, icon_name
, workdir
, description
);
619 argv
[pos
++] = linker
;
620 argv
[pos
++] = "--link";
621 argv
[pos
++] = link_name
;
622 argv
[pos
++] = "--path";
624 argv
[pos
++] = desktop
? "--desktop" : "--menu";
625 if (args
&& strlen(args
))
627 argv
[pos
++] = "--args";
632 argv
[pos
++] = "--icon";
633 argv
[pos
++] = icon_name
;
635 if (workdir
&& strlen(workdir
))
637 argv
[pos
++] = "--workdir";
638 argv
[pos
++] = workdir
;
640 if (description
&& strlen(description
))
642 argv
[pos
++] = "--descr";
643 argv
[pos
++] = description
;
647 retcode
=spawnvp( _P_WAIT
, linker
, argv
);
649 WINE_ERR("%s returned %d\n",linker
,retcode
);
653 static char *cleanup_link( LPCWSTR link
)
655 char *unix_file_name
;
658 unix_file_name
= wine_get_unix_file_name(link
);
661 WINE_ERR("target link %s not found\n", wine_dbgstr_w(link
));
665 link_name
= unix_file_name
;
666 p
= strrchr( link_name
, '/' );
670 p
= strrchr( link_name
, '.' );
674 p
= HeapAlloc(GetProcessHeap(), 0, strlen(link_name
) + 1);
675 strcpy(p
, link_name
);
676 HeapFree(GetProcessHeap(), 0, unix_file_name
);
681 /***********************************************************************
685 * returns TRUE if successful
686 * *loc will contain CS_DESKTOPDIRECTORY, CS_STARTMENU, CS_STARTUP
688 static BOOL
GetLinkLocation( LPCWSTR linkfile
, DWORD
*loc
)
690 WCHAR filename
[MAX_PATH
], buffer
[MAX_PATH
];
691 DWORD len
, i
, r
, filelen
;
692 const DWORD locations
[] = {
693 CSIDL_STARTUP
, CSIDL_DESKTOPDIRECTORY
, CSIDL_STARTMENU
,
694 CSIDL_COMMON_STARTUP
, CSIDL_COMMON_DESKTOPDIRECTORY
,
695 CSIDL_COMMON_STARTMENU
};
697 WINE_TRACE("%s\n", wine_dbgstr_w(linkfile
));
698 filelen
=GetFullPathNameW( linkfile
, MAX_PATH
, filename
, NULL
);
699 if (filelen
==0 || filelen
>MAX_PATH
)
702 WINE_TRACE("%s\n", wine_dbgstr_w(filename
));
704 for( i
=0; i
<sizeof(locations
)/sizeof(locations
[0]); i
++ )
706 if (!SHGetSpecialFolderPathW( 0, buffer
, locations
[i
], FALSE
))
709 len
= lstrlenW(buffer
);
713 if (len
> filelen
|| filename
[len
]!='\\')
715 /* do a lstrcmpinW */
717 r
= lstrcmpiW( filename
, buffer
);
718 filename
[len
] = '\\';
722 /* return the remainder of the string and link type */
730 /* gets the target path directly or through MSI */
731 static HRESULT
get_path( IShellLinkW
*sl
, LPWSTR szPath
, DWORD sz
)
733 IShellLinkDataList
*dl
= NULL
;
734 EXP_DARWIN_LINK
*dar
= NULL
;
738 hr
= IShellLinkW_GetPath( sl
, szPath
, MAX_PATH
, NULL
, SLGP_RAWPATH
);
739 if (hr
== S_OK
&& szPath
[0])
742 hr
= IShellLinkW_QueryInterface( sl
, &IID_IShellLinkDataList
, (LPVOID
*) &dl
);
746 hr
= IShellLinkDataList_CopyDataBlock( dl
, EXP_DARWIN_ID_SIG
, (LPVOID
*) &dar
);
749 CommandLineFromMsiDescriptor( dar
->szwDarwinID
, szPath
, &sz
);
753 IShellLinkDataList_Release( dl
);
757 static BOOL
InvokeShellLinker( IShellLinkW
*sl
, LPCWSTR link
, BOOL bAgain
)
759 char *link_name
= NULL
, *icon_name
= NULL
, *work_dir
= NULL
;
760 char *escaped_path
= NULL
, *escaped_args
= NULL
, *escaped_description
= NULL
;
761 WCHAR szDescription
[INFOTIPSIZE
], szPath
[MAX_PATH
], szWorkDir
[MAX_PATH
];
762 WCHAR szArgs
[INFOTIPSIZE
], szIconPath
[MAX_PATH
];
763 int iIconId
= 0, r
= -1;
768 WINE_ERR("Link name is null\n");
772 if( !GetLinkLocation( link
, &csidl
) )
774 WINE_WARN("Unknown link location '%s'. Ignoring.\n",wine_dbgstr_w(link
));
777 if (!in_desktop_dir(csidl
) && !in_startmenu(csidl
))
779 WINE_WARN("Not under desktop or start menu. Ignoring.\n");
784 IShellLinkW_GetWorkingDirectory( sl
, szWorkDir
, MAX_PATH
);
785 WINE_TRACE("workdir : %s\n", wine_dbgstr_w(szWorkDir
));
787 szDescription
[0] = 0;
788 IShellLinkW_GetDescription( sl
, szDescription
, INFOTIPSIZE
);
789 WINE_TRACE("description: %s\n", wine_dbgstr_w(szDescription
));
791 get_path( sl
, szPath
, MAX_PATH
);
792 WINE_TRACE("path : %s\n", wine_dbgstr_w(szPath
));
795 IShellLinkW_GetArguments( sl
, szArgs
, INFOTIPSIZE
);
796 WINE_TRACE("args : %s\n", wine_dbgstr_w(szArgs
));
799 IShellLinkW_GetIconLocation( sl
, szIconPath
, MAX_PATH
, &iIconId
);
800 WINE_TRACE("icon file : %s\n", wine_dbgstr_w(szIconPath
) );
804 LPITEMIDLIST pidl
= NULL
;
805 IShellLinkW_GetIDList( sl
, &pidl
);
806 if( pidl
&& SHGetPathFromIDListW( pidl
, szPath
) )
807 WINE_TRACE("pidl path : %s\n", wine_dbgstr_w(szPath
));
810 /* extract the icon */
812 icon_name
= extract_icon( szIconPath
, iIconId
);
814 icon_name
= extract_icon( szPath
, iIconId
);
816 /* fail - try once again at reboot time */
821 WINE_WARN("Unable to extract icon, deferring.\n");
824 WINE_ERR("failed to extract icon.\n");
830 static const WCHAR exeW
[] = {'.','e','x','e',0};
833 /* check for .exe extension */
834 if (!(p
= strrchrW( szPath
, '.' ))) return FALSE
;
835 if (strchrW( p
, '\\' ) || strchrW( p
, '/' )) return FALSE
;
836 if (lstrcmpiW( p
, exeW
)) return FALSE
;
838 /* convert app working dir */
840 work_dir
= wine_get_unix_file_name( szWorkDir
);
844 static const WCHAR startW
[] = {
845 '\\','c','o','m','m','a','n','d',
846 '\\','s','t','a','r','t','.','e','x','e',0};
848 /* if there's no path... try run the link itself */
849 lstrcpynW(szArgs
, link
, MAX_PATH
);
850 GetWindowsDirectoryW(szPath
, MAX_PATH
);
851 lstrcatW(szPath
, startW
);
854 link_name
= cleanup_link( link
);
857 WINE_ERR("Couldn't clean up link name %s\n", wine_dbgstr_w(link
));
861 /* escape the path and parameters */
862 escaped_path
= escape(szPath
);
863 escaped_args
= escape(szArgs
);
864 escaped_description
= escape(szDescription
);
866 r
= fork_and_wait("wineshelllink", link_name
, escaped_path
,
867 in_desktop_dir(csidl
), escaped_args
, icon_name
,
868 work_dir
? work_dir
: "", escaped_description
);
871 HeapFree( GetProcessHeap(), 0, icon_name
);
872 HeapFree( GetProcessHeap(), 0, work_dir
);
873 HeapFree( GetProcessHeap(), 0, link_name
);
874 HeapFree( GetProcessHeap(), 0, escaped_args
);
875 HeapFree( GetProcessHeap(), 0, escaped_path
);
876 HeapFree( GetProcessHeap(), 0, escaped_description
);
880 WINE_ERR("failed to fork and exec wineshelllink\n" );
888 static BOOL
Process_Link( LPCWSTR linkname
, BOOL bAgain
)
893 WCHAR fullname
[MAX_PATH
];
896 WINE_TRACE("%s, again %d\n", wine_dbgstr_w(linkname
), bAgain
);
900 WINE_ERR("link name missing\n");
904 len
=GetFullPathNameW( linkname
, MAX_PATH
, fullname
, NULL
);
905 if (len
==0 || len
>MAX_PATH
)
907 WINE_ERR("couldn't get full path of link file\n");
911 r
= CoInitialize( NULL
);
914 WINE_ERR("CoInitialize failed\n");
918 r
= CoCreateInstance( &CLSID_ShellLink
, NULL
, CLSCTX_INPROC_SERVER
,
919 &IID_IShellLinkW
, (LPVOID
*) &sl
);
922 WINE_ERR("No IID_IShellLink\n");
926 r
= IShellLinkW_QueryInterface( sl
, &IID_IPersistFile
, (LPVOID
*) &pf
);
929 WINE_ERR("No IID_IPersistFile\n");
933 r
= IPersistFile_Load( pf
, fullname
, STGM_READ
);
936 /* If something fails (eg. Couldn't extract icon)
937 * defer this menu entry to reboot via runonce
939 if( ! InvokeShellLinker( sl
, fullname
, bAgain
) && bAgain
)
940 DeferToRunOnce( fullname
);
942 WINE_TRACE("Success.\n");
945 IPersistFile_Release( pf
);
946 IShellLinkW_Release( sl
);
954 static CHAR
*next_token( LPSTR
*p
)
956 LPSTR token
= NULL
, t
= *p
;
969 /* unquote the token */
971 t
= strchr( token
, '"' );
980 t
= strchr( token
, ' ' );
990 /***********************************************************************
994 int PASCAL
WinMain (HINSTANCE hInstance
, HINSTANCE prev
, LPSTR cmdline
, int show
)
996 LPSTR token
= NULL
, p
;
998 HANDLE hsem
= CreateSemaphoreA( NULL
, 1, 1, "winemenubuilder_semaphore");
1001 /* running multiple instances of wineshelllink
1002 at the same time may be dangerous */
1003 if( WAIT_OBJECT_0
!= WaitForSingleObject( hsem
, INFINITE
) )
1009 for( p
= cmdline
; p
&& *p
; )
1011 token
= next_token( &p
);
1014 if( !lstrcmpA( token
, "-r" ) )
1016 else if( token
[0] == '-' )
1018 WINE_ERR( "unknown option %s\n",token
);
1022 WCHAR link
[MAX_PATH
];
1024 MultiByteToWideChar( CP_ACP
, 0, token
, -1, link
, sizeof(link
)/sizeof(WCHAR
) );
1025 if( !Process_Link( link
, bAgain
) )
1027 WINE_ERR( "failed to build menu item for %s\n",token
);
1033 ReleaseSemaphore( hsem
, 1, NULL
);
1034 CloseHandle( hsem
);