?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
shell32/shelllink: Fix some string buffers leaks.
[wine/wine-gecko.git] / dlls / shell32 / shelllink.c
blobe304058488fc5df216e1f270f99c59da8baeb773
1 /*
2 *
3 * Copyright 1997 Marcus Meissner
4 * Copyright 1998 Juergen Schmied
5 * Copyright 2005 Mike McCormack
6 *
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20 *
21 * NOTES
22 * Nearly complete information about the binary formats
23 * of .lnk files available at http://www.wotsit.org
24 *
25 * You can use winedump to examine the contents of a link file:
26 * winedump lnk sc.lnk
27 *
28 * MSI advertised shortcuts are totally undocumented. They provide an
29 * icon for a program that is not yet installed, and invoke MSI to
30 * install the program when the shortcut is clicked on. They are
31 * created by passing a special string to SetPath, and the information
32 * in that string is parsed an stored.
33 */
34
35 #define COBJMACROS
36 #define NONAMELESSUNION
37
38 #include "wine/debug.h"
39 #include "winerror.h"
40 #include "windef.h"
41 #include "winbase.h"
42 #include "winnls.h"
43 #include "winreg.h"
44
45 #include "winuser.h"
46 #include "wingdi.h"
47 #include "shlobj.h"
48 #include "undocshell.h"
49
50 #include "pidl.h"
51 #include "shell32_main.h"
52 #include "shlguid.h"
53 #include "shlwapi.h"
54 #include "msi.h"
55 #include "appmgmt.h"
56
57 #include "initguid.h"
58
59 WINE_DEFAULT_DEBUG_CHANNEL(shell);
60
61 DEFINE_GUID( SHELL32_AdvtShortcutProduct,
62 0x9db1186f,0x40df,0x11d1,0xaa,0x8c,0x00,0xc0,0x4f,0xb6,0x78,0x63);
63 DEFINE_GUID( SHELL32_AdvtShortcutComponent,
64 0x9db1186e,0x40df,0x11d1,0xaa,0x8c,0x00,0xc0,0x4f,0xb6,0x78,0x63);
65
66 /* link file formats */
67
68 #include "pshpack1.h"
69
70 typedef struct _LINK_HEADER
71 {
72 DWORD dwSize; /* 0x00 size of the header - 0x4c */
73 GUID MagicGuid; /* 0x04 is CLSID_ShellLink */
74 DWORD dwFlags; /* 0x14 describes elements following */
75 DWORD dwFileAttr; /* 0x18 attributes of the target file */
76 FILETIME Time1; /* 0x1c */
77 FILETIME Time2; /* 0x24 */
78 FILETIME Time3; /* 0x2c */
79 DWORD dwFileLength; /* 0x34 File length */
80 DWORD nIcon; /* 0x38 icon number */
81 DWORD fStartup; /* 0x3c startup type */
82 DWORD wHotKey; /* 0x40 hotkey */
83 DWORD Unknown5; /* 0x44 */
84 DWORD Unknown6; /* 0x48 */
85 } LINK_HEADER, * PLINK_HEADER;
86
87 #define SHLINK_LOCAL 0
88 #define SHLINK_REMOTE 1
89
90 typedef struct _LOCATION_INFO
91 {
92 DWORD dwTotalSize;
93 DWORD dwHeaderSize;
94 DWORD dwFlags;
95 DWORD dwVolTableOfs;
96 DWORD dwLocalPathOfs;
97 DWORD dwNetworkVolTableOfs;
98 DWORD dwFinalPathOfs;
99 } LOCATION_INFO;
100
101 typedef struct _LOCAL_VOLUME_INFO
102 {
103 DWORD dwSize;
104 DWORD dwType;
105 DWORD dwVolSerial;
106 DWORD dwVolLabelOfs;
107 } LOCAL_VOLUME_INFO;
108
109 typedef struct volume_info_t
110 {
111 DWORD type;
112 DWORD serial;
113 WCHAR label[12]; /* assume 8.3 */
114 } volume_info;
115
116 #include "poppack.h"
117
118 static const IShellLinkAVtbl slvt;
119 static const IShellLinkWVtbl slvtw;
120 static const IPersistFileVtbl pfvt;
121 static const IPersistStreamVtbl psvt;
122 static const IShellLinkDataListVtbl dlvt;
123 static const IShellExtInitVtbl eivt;
124 static const IContextMenuVtbl cmvt;
125 static const IObjectWithSiteVtbl owsvt;
126
127 /* IShellLink Implementation */
128
129 typedef struct
130 {
131 const IShellLinkAVtbl *lpVtbl;
132 const IShellLinkWVtbl *lpvtblw;
133 const IPersistFileVtbl *lpvtblPersistFile;
134 const IPersistStreamVtbl *lpvtblPersistStream;
135 const IShellLinkDataListVtbl *lpvtblShellLinkDataList;
136 const IShellExtInitVtbl *lpvtblShellExtInit;
137 const IContextMenuVtbl *lpvtblContextMenu;
138 const IObjectWithSiteVtbl *lpvtblObjectWithSite;
139
140 LONG ref;
141
142 /* data structures according to the information in the link */
143 LPITEMIDLIST pPidl;
144 WORD wHotKey;
145 SYSTEMTIME time1;
146 SYSTEMTIME time2;
147 SYSTEMTIME time3;
148
149 DWORD iShowCmd;
150 LPWSTR sIcoPath;
151 INT iIcoNdx;
152 LPWSTR sPath;
153 LPWSTR sArgs;
154 LPWSTR sWorkDir;
155 LPWSTR sDescription;
156 LPWSTR sPathRel;
157 LPWSTR sProduct;
158 LPWSTR sComponent;
159 volume_info volume;
160
161 BOOL bDirty;
162 INT iIdOpen; /* id of the "Open" entry in the context menu */
163 IUnknown *site;
164 } IShellLinkImpl;
165
166 static inline IShellLinkImpl *impl_from_IShellLinkW( IShellLinkW *iface )
167 {
168 return (IShellLinkImpl *)((char*)iface - FIELD_OFFSET(IShellLinkImpl, lpvtblw));
169 }
170
171 static inline IShellLinkImpl *impl_from_IPersistFile( IPersistFile *iface )
172 {
173 return (IShellLinkImpl *)((char*)iface - FIELD_OFFSET(IShellLinkImpl, lpvtblPersistFile));
174 }
175
176 static inline IShellLinkImpl *impl_from_IPersistStream( IPersistStream *iface )
177 {
178 return (IShellLinkImpl *)((char*)iface - FIELD_OFFSET(IShellLinkImpl, lpvtblPersistStream));
179 }
180
181 static inline IShellLinkImpl *impl_from_IShellLinkDataList( IShellLinkDataList *iface )
182 {
183 return (IShellLinkImpl *)((char*)iface - FIELD_OFFSET(IShellLinkImpl, lpvtblShellLinkDataList));
184 }
185
186 static inline IShellLinkImpl *impl_from_IShellExtInit( IShellExtInit *iface )
187 {
188 return (IShellLinkImpl *)((char*)iface - FIELD_OFFSET(IShellLinkImpl, lpvtblShellExtInit));
189 }
190
191 static inline IShellLinkImpl *impl_from_IContextMenu( IContextMenu *iface )
192 {
193 return (IShellLinkImpl *)((char*)iface - FIELD_OFFSET(IShellLinkImpl, lpvtblContextMenu));
194 }
195
196 static inline IShellLinkImpl *impl_from_IObjectWithSite( IObjectWithSite *iface )
197 {
198 return (IShellLinkImpl *)((char*)iface - FIELD_OFFSET(IShellLinkImpl, lpvtblObjectWithSite));
199 }
200
201 static HRESULT ShellLink_UpdatePath(LPCWSTR sPathRel, LPCWSTR path, LPCWSTR sWorkDir, LPWSTR* psPath);
202
203 /* strdup on the process heap */
204 static inline LPWSTR HEAP_strdupAtoW( HANDLE heap, DWORD flags, LPCSTR str)
205 {
206 INT len = MultiByteToWideChar( CP_ACP, 0, str, -1, NULL, 0 );
207 LPWSTR p = HeapAlloc( heap, flags, len*sizeof (WCHAR) );
208 if( !p )
209 return p;
210 MultiByteToWideChar( CP_ACP, 0, str, -1, p, len );
211 return p;
212 }
213
214 static inline LPWSTR strdupW( LPCWSTR src )
215 {
216 LPWSTR dest;
217 if (!src) return NULL;
218 dest = HeapAlloc( GetProcessHeap(), 0, (lstrlenW(src)+1)*sizeof(WCHAR) );
219 if (dest)
220 lstrcpyW(dest, src);
221 return dest;
222 }
223
224 /**************************************************************************
225 * ShellLink::QueryInterface implementation
226 */
227 static HRESULT ShellLink_QueryInterface( IShellLinkImpl *This, REFIID riid, LPVOID *ppvObj)
228 {
229 TRACE("(%p)->(\n\tIID:\t%s)\n",This,debugstr_guid(riid));
230
231 *ppvObj = NULL;
232
233 if(IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IShellLinkA))
234 {
235 *ppvObj = This;
236 }
237 else if(IsEqualIID(riid, &IID_IShellLinkW))
238 {
239 *ppvObj = &(This->lpvtblw);
240 }
241 else if(IsEqualIID(riid, &IID_IPersistFile))
242 {
243 *ppvObj = &(This->lpvtblPersistFile);
244 }
245 else if(IsEqualIID(riid, &IID_IPersistStream))
246 {
247 *ppvObj = &(This->lpvtblPersistStream);
248 }
249 else if(IsEqualIID(riid, &IID_IShellLinkDataList))
250 {
251 *ppvObj = &(This->lpvtblShellLinkDataList);
252 }
253 else if(IsEqualIID(riid, &IID_IShellExtInit))
254 {
255 *ppvObj = &(This->lpvtblShellExtInit);
256 }
257 else if(IsEqualIID(riid, &IID_IContextMenu))
258 {
259 *ppvObj = &(This->lpvtblContextMenu);
260 }
261 else if(IsEqualIID(riid, &IID_IObjectWithSite))
262 {
263 *ppvObj = &(This->lpvtblObjectWithSite);
264 }
265
266 if(*ppvObj)
267 {
268 IUnknown_AddRef((IUnknown*)(*ppvObj));
269 TRACE("-- Interface: (%p)->(%p)\n",ppvObj,*ppvObj);
270 return S_OK;
271 }
272 ERR("-- Interface: E_NOINTERFACE\n");
273 return E_NOINTERFACE;
274 }
275
276 /**************************************************************************
277 * ShellLink::AddRef implementation
278 */
279 static ULONG ShellLink_AddRef( IShellLinkImpl *This )
280 {
281 ULONG refCount = InterlockedIncrement(&This->ref);
282
283 TRACE("(%p)->(count=%u)\n", This, refCount - 1);
284
285 return refCount;
286 }
287
288 /**************************************************************************
289 * ShellLink::Release implementation
290 */
291 static ULONG ShellLink_Release( IShellLinkImpl *This )
292 {
293 ULONG refCount = InterlockedDecrement(&This->ref);
294
295 TRACE("(%p)->(count=%u)\n", This, refCount + 1);
296
297 if (refCount)
298 return refCount;
299
300 TRACE("-- destroying IShellLink(%p)\n",This);
301
302 HeapFree(GetProcessHeap(), 0, This->sIcoPath);
303 HeapFree(GetProcessHeap(), 0, This->sArgs);
304 HeapFree(GetProcessHeap(), 0, This->sWorkDir);
305 HeapFree(GetProcessHeap(), 0, This->sDescription);
306 HeapFree(GetProcessHeap(), 0, This->sPath);
307 HeapFree(GetProcessHeap(), 0, This->sPathRel);
308 HeapFree(GetProcessHeap(), 0, This->sProduct);
309 HeapFree(GetProcessHeap(), 0, This->sComponent);
310
311 if (This->site)
312 IUnknown_Release( This->site );
313
314 if (This->pPidl)
315 ILFree(This->pPidl);
316
317 LocalFree(This);
318
319 return 0;
320 }
321
322 static HRESULT ShellLink_GetClassID( IShellLinkImpl *This, CLSID *pclsid )
323 {
324 TRACE("%p %p\n", This, pclsid);
325
326 *pclsid = CLSID_ShellLink;
327 return S_OK;
328 }
329
330 /**************************************************************************
331 * IPersistFile_QueryInterface
332 */
333 static HRESULT WINAPI IPersistFile_fnQueryInterface(
334 IPersistFile* iface,
335 REFIID riid,
336 LPVOID *ppvObj)
337 {
338 IShellLinkImpl *This = impl_from_IPersistFile(iface);
339 return ShellLink_QueryInterface( This, riid, ppvObj );
340 }
341
342 /******************************************************************************
343 * IPersistFile_AddRef
344 */
345 static ULONG WINAPI IPersistFile_fnAddRef(IPersistFile* iface)
346 {
347 IShellLinkImpl *This = impl_from_IPersistFile(iface);
348 return ShellLink_AddRef( This );
349 }
350
351 /******************************************************************************
352 * IPersistFile_Release
353 */
354 static ULONG WINAPI IPersistFile_fnRelease(IPersistFile* iface)
355 {
356 IShellLinkImpl *This = impl_from_IPersistFile(iface);
357 return IShellLinkA_Release((IShellLinkA*)This);
358 }
359
360 static HRESULT WINAPI IPersistFile_fnGetClassID(IPersistFile* iface, CLSID *pClassID)
361 {
362 IShellLinkImpl *This = impl_from_IPersistFile(iface);
363 return ShellLink_GetClassID( This, pClassID );
364 }
365
366 static HRESULT WINAPI IPersistFile_fnIsDirty(IPersistFile* iface)
367 {
368 IShellLinkImpl *This = impl_from_IPersistFile(iface);
369
370 TRACE("(%p)\n",This);
371
372 if (This->bDirty)
373 return S_OK;
374
375 return S_FALSE;
376 }
377
378 static HRESULT WINAPI IPersistFile_fnLoad(IPersistFile* iface, LPCOLESTR pszFileName, DWORD dwMode)
379 {
380 IShellLinkImpl *This = impl_from_IPersistFile(iface);
381 IPersistStream *StreamThis = (IPersistStream *)&This->lpvtblPersistStream;
382 HRESULT r;
383 IStream *stm;
384
385 TRACE("(%p, %s, %x)\n",This, debugstr_w(pszFileName), dwMode);
386
387 if( dwMode == 0 )
388 dwMode = STGM_READ | STGM_SHARE_DENY_WRITE;
389 r = SHCreateStreamOnFileW(pszFileName, dwMode, &stm);
390 if( SUCCEEDED( r ) )
391 {
392 r = IPersistStream_Load(StreamThis, stm);
393 ShellLink_UpdatePath(This->sPathRel, pszFileName, This->sWorkDir, &This->sPath);
394 IStream_Release( stm );
395 This->bDirty = FALSE;
396 }
397 TRACE("-- returning hr %08x\n", r);
398 return r;
399 }
400
401 BOOL run_winemenubuilder( const WCHAR *args )
402 {
403 static const WCHAR menubuilder[] = {'\\','w','i','n','e','m','e','n','u','b','u','i','l','d','e','r','.','e','x','e',0};
404 LONG len;
405 LPWSTR buffer;
406 STARTUPINFOW si;
407 PROCESS_INFORMATION pi;
408 BOOL ret;
409 WCHAR app[MAX_PATH];
410
411 GetSystemDirectoryW( app, MAX_PATH - sizeof(menubuilder)/sizeof(WCHAR) );
412 strcatW( app, menubuilder );
413
414 len = (strlenW( app ) + strlenW( args ) + 1) * sizeof(WCHAR);
415 buffer = HeapAlloc( GetProcessHeap(), 0, len );
416 if( !buffer )
417 return FALSE;
418
419 strcpyW( buffer, app );
420 strcatW( buffer, args );
421
422 TRACE("starting %s\n",debugstr_w(buffer));
423
424 memset(&si, 0, sizeof(si));
425 si.cb = sizeof(si);
426
427 ret = CreateProcessW( app, buffer, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi );
428
429 HeapFree( GetProcessHeap(), 0, buffer );
430
431 if (ret)
432 {
433 CloseHandle( pi.hProcess );
434 CloseHandle( pi.hThread );
435 }
436
437 return ret;
438 }
439
440 static BOOL StartLinkProcessor( LPCOLESTR szLink )
441 {
442 static const WCHAR szFormat[] = {' ','-','w',' ','"','%','s','"',0 };
443 LONG len;
444 LPWSTR buffer;
445 BOOL ret;
446
447 len = sizeof(szFormat) + lstrlenW( szLink ) * sizeof(WCHAR);
448 buffer = HeapAlloc( GetProcessHeap(), 0, len );
449 if( !buffer )
450 return FALSE;
451
452 wsprintfW( buffer, szFormat, szLink );
453 ret = run_winemenubuilder( buffer );
454 HeapFree( GetProcessHeap(), 0, buffer );
455 return ret;
456 }
457
458 static HRESULT WINAPI IPersistFile_fnSave(IPersistFile* iface, LPCOLESTR pszFileName, BOOL fRemember)
459 {
460 IShellLinkImpl *This = impl_from_IPersistFile(iface);
461 IPersistStream *StreamThis = (IPersistStream *)&This->lpvtblPersistStream;
462 HRESULT r;
463 IStream *stm;
464
465 TRACE("(%p)->(%s)\n",This,debugstr_w(pszFileName));
466
467 if (!pszFileName)
468 return E_FAIL;
469
470 r = SHCreateStreamOnFileW( pszFileName, STGM_READWRITE | STGM_CREATE | STGM_SHARE_EXCLUSIVE, &stm );
471 if( SUCCEEDED( r ) )
472 {
473 r = IPersistStream_Save(StreamThis, stm, FALSE);
474 IStream_Release( stm );
475
476 if( SUCCEEDED( r ) )
477 {
478 StartLinkProcessor( pszFileName );
479
480 This->bDirty = FALSE;
481 }
482 else
483 {
484 DeleteFileW( pszFileName );
485 WARN("Failed to create shortcut %s\n", debugstr_w(pszFileName) );
486 }
487 }
488
489 return r;
490 }
491
492 static HRESULT WINAPI IPersistFile_fnSaveCompleted(IPersistFile* iface, LPCOLESTR pszFileName)
493 {
494 IShellLinkImpl *This = impl_from_IPersistFile(iface);
495 FIXME("(%p)->(%s)\n",This,debugstr_w(pszFileName));
496 return NOERROR;
497 }
498
499 static HRESULT WINAPI IPersistFile_fnGetCurFile(IPersistFile* iface, LPOLESTR *ppszFileName)
500 {
501 IShellLinkImpl *This = impl_from_IPersistFile(iface);
502 FIXME("(%p)\n",This);
503 return NOERROR;
504 }
505
506 static const IPersistFileVtbl pfvt =
507 {
508 IPersistFile_fnQueryInterface,
509 IPersistFile_fnAddRef,
510 IPersistFile_fnRelease,
511 IPersistFile_fnGetClassID,
512 IPersistFile_fnIsDirty,
513 IPersistFile_fnLoad,
514 IPersistFile_fnSave,
515 IPersistFile_fnSaveCompleted,
516 IPersistFile_fnGetCurFile
517 };
518
519 /************************************************************************
520 * IPersistStream_QueryInterface
521 */
522 static HRESULT WINAPI IPersistStream_fnQueryInterface(
523 IPersistStream* iface,
524 REFIID riid,
525 VOID** ppvObj)
526 {
527 IShellLinkImpl *This = impl_from_IPersistStream(iface);
528 return ShellLink_QueryInterface( This, riid, ppvObj );
529 }
530
531 /************************************************************************
532 * IPersistStream_Release
533 */
534 static ULONG WINAPI IPersistStream_fnRelease(
535 IPersistStream* iface)
536 {
537 IShellLinkImpl *This = impl_from_IPersistStream(iface);
538 return IShellLinkA_Release((IShellLinkA*)This);
539 }
540
541 /************************************************************************
542 * IPersistStream_AddRef
543 */
544 static ULONG WINAPI IPersistStream_fnAddRef(
545 IPersistStream* iface)
546 {
547 IShellLinkImpl *This = impl_from_IPersistStream(iface);
548 return ShellLink_AddRef( This );
549 }
550
551 /************************************************************************
552 * IPersistStream_GetClassID
553 *
554 */
555 static HRESULT WINAPI IPersistStream_fnGetClassID(
556 IPersistStream* iface,
557 CLSID* pClassID)
558 {
559 IShellLinkImpl *This = impl_from_IPersistStream(iface);
560 return ShellLink_GetClassID( This, pClassID );
561 }
562
563 /************************************************************************
564 * IPersistStream_IsDirty (IPersistStream)
565 */
566 static HRESULT WINAPI IPersistStream_fnIsDirty(
567 IPersistStream* iface)
568 {
569 IShellLinkImpl *This = impl_from_IPersistStream(iface);
570
571 TRACE("(%p)\n", This);
572
573 return S_OK;
574 }
575
576
577 static HRESULT Stream_LoadString( IStream* stm, BOOL unicode, LPWSTR *pstr )
578 {
579 DWORD count;
580 USHORT len;
581 LPVOID temp;
582 LPWSTR str;
583 HRESULT r;
584
585 TRACE("%p\n", stm);
586
587 count = 0;
588 r = IStream_Read(stm, &len, sizeof(len), &count);
589 if ( FAILED (r) || ( count != sizeof(len) ) )
590 return E_FAIL;
591
592 if( unicode )
593 len *= sizeof (WCHAR);
594
595 TRACE("reading %d\n", len);
596 temp = HeapAlloc(GetProcessHeap(), 0, len+sizeof(WCHAR));
597 if( !temp )
598 return E_OUTOFMEMORY;
599 count = 0;
600 r = IStream_Read(stm, temp, len, &count);
601 if( FAILED (r) || ( count != len ) )
602 {
603 HeapFree( GetProcessHeap(), 0, temp );
604 return E_FAIL;
605 }
606
607 TRACE("read %s\n", debugstr_an(temp,len));
608
609 /* convert to unicode if necessary */
610 if( !unicode )
611 {
612 count = MultiByteToWideChar( CP_ACP, 0, temp, len, NULL, 0 );
613 str = HeapAlloc( GetProcessHeap(), 0, (count+1)*sizeof (WCHAR) );
614 if( !str )
615 {
616 HeapFree( GetProcessHeap(), 0, temp );
617 return E_OUTOFMEMORY;
618 }
619 MultiByteToWideChar( CP_ACP, 0, temp, len, str, count );
620 HeapFree( GetProcessHeap(), 0, temp );
621 }
622 else
623 {
624 count /= 2;
625 str = temp;
626 }
627 str[count] = 0;
628
629 *pstr = str;
630
631 return S_OK;
632 }
633
634 static HRESULT Stream_ReadChunk( IStream* stm, LPVOID *data )
635 {
636 DWORD size;
637 ULONG count;
638 HRESULT r;
639 struct sized_chunk {
640 DWORD size;
641 unsigned char data[1];
642 } *chunk;
643
644 TRACE("%p\n",stm);
645
646 r = IStream_Read( stm, &size, sizeof(size), &count );
647 if( FAILED( r ) || count != sizeof(size) )
648 return E_FAIL;
649
650 chunk = HeapAlloc( GetProcessHeap(), 0, size );
651 if( !chunk )
652 return E_OUTOFMEMORY;
653
654 chunk->size = size;
655 r = IStream_Read( stm, chunk->data, size - sizeof(size), &count );
656 if( FAILED( r ) || count != (size - sizeof(size)) )
657 {
658 HeapFree( GetProcessHeap(), 0, chunk );
659 return E_FAIL;
660 }
661
662 TRACE("Read %d bytes\n",chunk->size);
663
664 *data = chunk;
665
666 return S_OK;
667 }
668
669 static BOOL Stream_LoadVolume( LOCAL_VOLUME_INFO *vol, volume_info *volume )
670 {
671 const int label_sz = sizeof volume->label/sizeof volume->label[0];
672 LPSTR label;
673 int len;
674
675 volume->serial = vol->dwVolSerial;
676 volume->type = vol->dwType;
677
678 if( !vol->dwVolLabelOfs )
679 return FALSE;
680 if( vol->dwSize <= vol->dwVolLabelOfs )
681 return FALSE;
682 len = vol->dwSize - vol->dwVolLabelOfs;
683
684 label = (LPSTR) vol;
685 label += vol->dwVolLabelOfs;
686 MultiByteToWideChar( CP_ACP, 0, label, len, volume->label, label_sz-1);
687
688 return TRUE;
689 }
690
691 static LPWSTR Stream_LoadPath( LPCSTR p, DWORD maxlen )
692 {
693 int len = 0, wlen;
694 LPWSTR path;
695
696 while( p[len] && (len < maxlen) )
697 len++;
698
699 wlen = MultiByteToWideChar(CP_ACP, 0, p, len, NULL, 0);
700 path = HeapAlloc(GetProcessHeap(), 0, (wlen+1)*sizeof(WCHAR));
701 MultiByteToWideChar(CP_ACP, 0, p, len, path, wlen);
702 path[wlen] = 0;
703
704 return path;
705 }
706
707 static HRESULT Stream_LoadLocation( IStream *stm,
708 volume_info *volume, LPWSTR *path )
709 {
710 char *p = NULL;
711 LOCATION_INFO *loc;
712 HRESULT r;
713 DWORD n;
714
715 r = Stream_ReadChunk( stm, (LPVOID*) &p );
716 if( FAILED(r) )
717 return r;
718
719 loc = (LOCATION_INFO*) p;
720 if (loc->dwTotalSize < sizeof(LOCATION_INFO))
721 {
722 HeapFree( GetProcessHeap(), 0, p );
723 return E_FAIL;
724 }
725
726 /* if there's valid local volume information, load it */
727 if( loc->dwVolTableOfs &&
728 ((loc->dwVolTableOfs + sizeof(LOCAL_VOLUME_INFO)) <= loc->dwTotalSize) )
729 {
730 LOCAL_VOLUME_INFO *volume_info;
731
732 volume_info = (LOCAL_VOLUME_INFO*) &p[loc->dwVolTableOfs];
733 Stream_LoadVolume( volume_info, volume );
734 }
735
736 /* if there's a local path, load it */
737 n = loc->dwLocalPathOfs;
738 if( n && (n < loc->dwTotalSize) )
739 *path = Stream_LoadPath( &p[n], loc->dwTotalSize - n );
740
741 TRACE("type %d serial %08x name %s path %s\n", volume->type,
742 volume->serial, debugstr_w(volume->label), debugstr_w(*path));
743
744 HeapFree( GetProcessHeap(), 0, p );
745 return S_OK;
746 }
747
748 /*
749 * The format of the advertised shortcut info seems to be:
750 *
751 * Offset Description
752 * ------ -----------
753 *
754 * 0 Length of the block (4 bytes, usually 0x314)
755 * 4 tag (dword)
756 * 8 string data in ASCII
757 * 8+0x104 string data in UNICODE
758 *
759 * In the original Win32 implementation the buffers are not initialized
760 * to zero, so data trailing the string is random garbage.
761 */
762 static HRESULT Stream_LoadAdvertiseInfo( IStream* stm, LPWSTR *str )
763 {
764 DWORD size;
765 ULONG count;
766 HRESULT r;
767 EXP_DARWIN_LINK buffer;
768
769 TRACE("%p\n",stm);
770
771 r = IStream_Read( stm, &buffer.dbh.cbSize, sizeof (DWORD), &count );
772 if( FAILED( r ) )
773 return r;
774
775 /* make sure that we read the size of the structure even on error */
776 size = sizeof buffer - sizeof (DWORD);
777 if( buffer.dbh.cbSize != sizeof buffer )
778 {
779 ERR("Ooops. This structure is not as expected...\n");
780 return E_FAIL;
781 }
782
783 r = IStream_Read( stm, &buffer.dbh.dwSignature, size, &count );
784 if( FAILED( r ) )
785 return r;
786
787 if( count != size )
788 return E_FAIL;
789
790 TRACE("magic %08x string = %s\n", buffer.dbh.dwSignature, debugstr_w(buffer.szwDarwinID));
791
792 if( (buffer.dbh.dwSignature&0xffff0000) != 0xa0000000 )
793 {
794 ERR("Unknown magic number %08x in advertised shortcut\n", buffer.dbh.dwSignature);
795 return E_FAIL;
796 }
797
798 *str = HeapAlloc( GetProcessHeap(), 0,
799 (lstrlenW(buffer.szwDarwinID)+1) * sizeof(WCHAR) );
800 lstrcpyW( *str, buffer.szwDarwinID );
801
802 return S_OK;
803 }
804
805 /************************************************************************
806 * IPersistStream_Load (IPersistStream)
807 */
808 static HRESULT WINAPI IPersistStream_fnLoad(
809 IPersistStream* iface,
810 IStream* stm)
811 {
812 LINK_HEADER hdr;
813 ULONG dwBytesRead;
814 BOOL unicode;
815 HRESULT r;
816 DWORD zero;
817
818 IShellLinkImpl *This = impl_from_IPersistStream(iface);
819
820 TRACE("%p %p\n", This, stm);
821
822 if( !stm )
823 return STG_E_INVALIDPOINTER;
824
825 dwBytesRead = 0;
826 r = IStream_Read(stm, &hdr, sizeof(hdr), &dwBytesRead);
827 if( FAILED( r ) )
828 return r;
829
830 if( dwBytesRead != sizeof(hdr))
831 return E_FAIL;
832 if( hdr.dwSize != sizeof(hdr))
833 return E_FAIL;
834 if( !IsEqualIID(&hdr.MagicGuid, &CLSID_ShellLink) )
835 return E_FAIL;
836
837 /* free all the old stuff */
838 ILFree(This->pPidl);
839 This->pPidl = NULL;
840 memset( &This->volume, 0, sizeof This->volume );
841 HeapFree(GetProcessHeap(), 0, This->sPath);
842 This->sPath = NULL;
843 HeapFree(GetProcessHeap(), 0, This->sDescription);
844 This->sDescription = NULL;
845 HeapFree(GetProcessHeap(), 0, This->sPathRel);
846 This->sPathRel = NULL;
847 HeapFree(GetProcessHeap(), 0, This->sWorkDir);
848 This->sWorkDir = NULL;
849 HeapFree(GetProcessHeap(), 0, This->sArgs);
850 This->sArgs = NULL;
851 HeapFree(GetProcessHeap(), 0, This->sIcoPath);
852 This->sIcoPath = NULL;
853 HeapFree(GetProcessHeap(), 0, This->sProduct);
854 This->sProduct = NULL;
855 HeapFree(GetProcessHeap(), 0, This->sComponent);
856 This->sComponent = NULL;
857
858 This->wHotKey = (WORD)hdr.wHotKey;
859 This->iIcoNdx = hdr.nIcon;
860 FileTimeToSystemTime (&hdr.Time1, &This->time1);
861 FileTimeToSystemTime (&hdr.Time2, &This->time2);
862 FileTimeToSystemTime (&hdr.Time3, &This->time3);
863 if (TRACE_ON(shell))
864 {
865 WCHAR sTemp[MAX_PATH];
866 GetDateFormatW(LOCALE_USER_DEFAULT,DATE_SHORTDATE, &This->time1,
867 NULL, sTemp, sizeof(sTemp)/sizeof(*sTemp));
868 TRACE("-- time1: %s\n", debugstr_w(sTemp) );
869 GetDateFormatW(LOCALE_USER_DEFAULT,DATE_SHORTDATE, &This->time2,
870 NULL, sTemp, sizeof(sTemp)/sizeof(*sTemp));
871 TRACE("-- time2: %s\n", debugstr_w(sTemp) );
872 GetDateFormatW(LOCALE_USER_DEFAULT,DATE_SHORTDATE, &This->time3,
873 NULL, sTemp, sizeof(sTemp)/sizeof(*sTemp));
874 TRACE("-- time3: %s\n", debugstr_w(sTemp) );
875 }
876
877 /* load all the new stuff */
878 if( hdr.dwFlags & SLDF_HAS_ID_LIST )
879 {
880 r = ILLoadFromStream( stm, &This->pPidl );
881 if( FAILED( r ) )
882 return r;
883 }
884 pdump(This->pPidl);
885
886 /* load the location information */
887 if( hdr.dwFlags & SLDF_HAS_LINK_INFO )
888 r = Stream_LoadLocation( stm, &This->volume, &This->sPath );
889 if( FAILED( r ) )
890 goto end;
891
892 unicode = hdr.dwFlags & SLDF_UNICODE;
893 if( hdr.dwFlags & SLDF_HAS_NAME )
894 {
895 r = Stream_LoadString( stm, unicode, &This->sDescription );
896 TRACE("Description -> %s\n",debugstr_w(This->sDescription));
897 }
898 if( FAILED( r ) )
899 goto end;
900
901 if( hdr.dwFlags & SLDF_HAS_RELPATH )
902 {
903 r = Stream_LoadString( stm, unicode, &This->sPathRel );
904 TRACE("Relative Path-> %s\n",debugstr_w(This->sPathRel));
905 }
906 if( FAILED( r ) )
907 goto end;
908
909 if( hdr.dwFlags & SLDF_HAS_WORKINGDIR )
910 {
911 r = Stream_LoadString( stm, unicode, &This->sWorkDir );
912 TRACE("Working Dir -> %s\n",debugstr_w(This->sWorkDir));
913 }
914 if( FAILED( r ) )
915 goto end;
916
917 if( hdr.dwFlags & SLDF_HAS_ARGS )
918 {
919 r = Stream_LoadString( stm, unicode, &This->sArgs );
920 TRACE("Working Dir -> %s\n",debugstr_w(This->sArgs));
921 }
922 if( FAILED( r ) )
923 goto end;
924
925 if( hdr.dwFlags & SLDF_HAS_ICONLOCATION )
926 {
927 r = Stream_LoadString( stm, unicode, &This->sIcoPath );
928 TRACE("Icon file -> %s\n",debugstr_w(This->sIcoPath));
929 }
930 if( FAILED( r ) )
931 goto end;
932
933 if( hdr.dwFlags & SLDF_HAS_LOGO3ID )
934 {
935 r = Stream_LoadAdvertiseInfo( stm, &This->sProduct );
936 TRACE("Product -> %s\n",debugstr_w(This->sProduct));
937 }
938 if( FAILED( r ) )
939 goto end;
940
941 if( hdr.dwFlags & SLDF_HAS_DARWINID )
942 {
943 r = Stream_LoadAdvertiseInfo( stm, &This->sComponent );
944 TRACE("Component -> %s\n",debugstr_w(This->sComponent));
945 }
946 if( FAILED( r ) )
947 goto end;
948
949 r = IStream_Read(stm, &zero, sizeof zero, &dwBytesRead);
950 if( FAILED( r ) || zero || dwBytesRead != sizeof zero )
951 {
952 /* Some lnk files have extra data blocks starting with a
953 * DATABLOCK_HEADER. For instance EXP_SPECIAL_FOLDER and an unknown
954 * one with a 0xa0000003 signature. However these don't seem to matter
955 * too much.
956 */
957 WARN("Last word was not zero\n");
958 }
959
960 TRACE("OK\n");
961
962 pdump (This->pPidl);
963
964 return S_OK;
965 end:
966 return r;
967 }
968
969 /************************************************************************
970 * Stream_WriteString
971 *
972 * Helper function for IPersistStream_Save. Writes a unicode string
973 * with terminating nul byte to a stream, preceded by the its length.
974 */
975 static HRESULT Stream_WriteString( IStream* stm, LPCWSTR str )
976 {
977 USHORT len = lstrlenW( str ) + 1;
978 DWORD count;
979 HRESULT r;
980
981 r = IStream_Write( stm, &len, sizeof(len), &count );
982 if( FAILED( r ) )
983 return r;
984
985 len *= sizeof(WCHAR);
986
987 r = IStream_Write( stm, str, len, &count );
988 if( FAILED( r ) )
989 return r;
990
991 return S_OK;
992 }
993
994 /************************************************************************
995 * Stream_WriteLocationInfo
996 *
997 * Writes the location info to a stream
998 *
999 * FIXME: One day we might want to write the network volume information
1000 * and the final path.
1001 * Figure out how Windows deals with unicode paths here.
1002 */
1003 static HRESULT Stream_WriteLocationInfo( IStream* stm, LPCWSTR path,
1004 volume_info *volume )
1005 {
1006 DWORD total_size, path_size, volume_info_size, label_size, final_path_size;
1007 LOCAL_VOLUME_INFO *vol;
1008 LOCATION_INFO *loc;
1009 LPSTR szLabel, szPath, szFinalPath;
1010 ULONG count = 0;
1011 HRESULT hr;
1012
1013 TRACE("%p %s %p\n", stm, debugstr_w(path), volume);
1014
1015 /* figure out the size of everything */
1016 label_size = WideCharToMultiByte( CP_ACP, 0, volume->label, -1,
1017 NULL, 0, NULL, NULL );
1018 path_size = WideCharToMultiByte( CP_ACP, 0, path, -1,
1019 NULL, 0, NULL, NULL );
1020 volume_info_size = sizeof *vol + label_size;
1021 final_path_size = 1;
1022 total_size = sizeof *loc + volume_info_size + path_size + final_path_size;
1023
1024 /* create pointers to everything */
1025 loc = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, total_size);
1026 vol = (LOCAL_VOLUME_INFO*) &loc[1];
1027 szLabel = (LPSTR) &vol[1];
1028 szPath = &szLabel[label_size];
1029 szFinalPath = &szPath[path_size];
1030
1031 /* fill in the location information header */
1032 loc->dwTotalSize = total_size;
1033 loc->dwHeaderSize = sizeof (*loc);
1034 loc->dwFlags = 1;
1035 loc->dwVolTableOfs = sizeof (*loc);
1036 loc->dwLocalPathOfs = sizeof (*loc) + volume_info_size;
1037 loc->dwNetworkVolTableOfs = 0;
1038 loc->dwFinalPathOfs = sizeof (*loc) + volume_info_size + path_size;
1039
1040 /* fill in the volume information */
1041 vol->dwSize = volume_info_size;
1042 vol->dwType = volume->type;
1043 vol->dwVolSerial = volume->serial;
1044 vol->dwVolLabelOfs = sizeof (*vol);
1045
1046 /* copy in the strings */
1047 WideCharToMultiByte( CP_ACP, 0, volume->label, -1,
1048 szLabel, label_size, NULL, NULL );
1049 WideCharToMultiByte( CP_ACP, 0, path, -1,
1050 szPath, path_size, NULL, NULL );
1051 szFinalPath[0] = 0;
1052
1053 hr = IStream_Write( stm, loc, total_size, &count );
1054 HeapFree(GetProcessHeap(), 0, loc);
1055
1056 return hr;
1057 }
1058
1059 static EXP_DARWIN_LINK* shelllink_build_darwinid( LPCWSTR string, DWORD magic )
1060 {
1061 EXP_DARWIN_LINK *buffer;
1062
1063 buffer = LocalAlloc( LMEM_ZEROINIT, sizeof *buffer );
1064 buffer->dbh.cbSize = sizeof *buffer;
1065 buffer->dbh.dwSignature = magic;
1066 lstrcpynW( buffer->szwDarwinID, string, MAX_PATH );
1067 WideCharToMultiByte(CP_ACP, 0, string, -1, buffer->szDarwinID, MAX_PATH, NULL, NULL );
1068
1069 return buffer;
1070 }
1071
1072 static HRESULT Stream_WriteAdvertiseInfo( IStream* stm, LPCWSTR string, DWORD magic )
1073 {
1074 EXP_DARWIN_LINK *buffer;
1075 ULONG count;
1076
1077 TRACE("%p\n",stm);
1078
1079 buffer = shelllink_build_darwinid( string, magic );
1080
1081 return IStream_Write( stm, buffer, buffer->dbh.cbSize, &count );
1082 }
1083
1084 /************************************************************************
1085 * IPersistStream_Save (IPersistStream)
1086 *
1087 * FIXME: makes assumptions about byte order
1088 */
1089 static HRESULT WINAPI IPersistStream_fnSave(
1090 IPersistStream* iface,
1091 IStream* stm,
1092 BOOL fClearDirty)
1093 {
1094 LINK_HEADER header;
1095 ULONG count;
1096 DWORD zero;
1097 HRESULT r;
1098
1099 IShellLinkImpl *This = impl_from_IPersistStream(iface);
1100
1101 TRACE("%p %p %x\n", This, stm, fClearDirty);
1102
1103 memset(&header, 0, sizeof(header));
1104 header.dwSize = sizeof(header);
1105 header.fStartup = This->iShowCmd;
1106 header.MagicGuid = CLSID_ShellLink;
1107
1108 header.wHotKey = This->wHotKey;
1109 header.nIcon = This->iIcoNdx;
1110 header.dwFlags = SLDF_UNICODE; /* strings are in unicode */
1111 if( This->pPidl )
1112 header.dwFlags |= SLDF_HAS_ID_LIST;
1113 if( This->sPath )
1114 header.dwFlags |= SLDF_HAS_LINK_INFO;
1115 if( This->sDescription )
1116 header.dwFlags |= SLDF_HAS_NAME;
1117 if( This->sWorkDir )
1118 header.dwFlags |= SLDF_HAS_WORKINGDIR;
1119 if( This->sArgs )
1120 header.dwFlags |= SLDF_HAS_ARGS;
1121 if( This->sIcoPath )
1122 header.dwFlags |= SLDF_HAS_ICONLOCATION;
1123 if( This->sProduct )
1124 header.dwFlags |= SLDF_HAS_LOGO3ID;
1125 if( This->sComponent )
1126 header.dwFlags |= SLDF_HAS_DARWINID;
1127
1128 SystemTimeToFileTime ( &This->time1, &header.Time1 );
1129 SystemTimeToFileTime ( &This->time2, &header.Time2 );
1130 SystemTimeToFileTime ( &This->time3, &header.Time3 );
1131
1132 /* write the Shortcut header */
1133 r = IStream_Write( stm, &header, sizeof(header), &count );
1134 if( FAILED( r ) )
1135 {
1136 ERR("Write failed at %d\n",__LINE__);
1137 return r;
1138 }
1139
1140 TRACE("Writing pidl\n");
1141
1142 /* write the PIDL to the shortcut */
1143 if( This->pPidl )
1144 {
1145 r = ILSaveToStream( stm, This->pPidl );
1146 if( FAILED( r ) )
1147 {
1148 ERR("Failed to write PIDL at %d\n",__LINE__);
1149 return r;
1150 }
1151 }
1152
1153 if( This->sPath )
1154 Stream_WriteLocationInfo( stm, This->sPath, &This->volume );
1155
1156 if( This->sDescription )
1157 r = Stream_WriteString( stm, This->sDescription );
1158
1159 if( This->sPathRel )
1160 r = Stream_WriteString( stm, This->sPathRel );
1161
1162 if( This->sWorkDir )
1163 r = Stream_WriteString( stm, This->sWorkDir );
1164
1165 if( This->sArgs )
1166 r = Stream_WriteString( stm, This->sArgs );
1167
1168 if( This->sIcoPath )
1169 r = Stream_WriteString( stm, This->sIcoPath );
1170
1171 if( This->sProduct )
1172 r = Stream_WriteAdvertiseInfo( stm, This->sProduct, EXP_SZ_ICON_SIG );
1173
1174 if( This->sComponent )
1175 r = Stream_WriteAdvertiseInfo( stm, This->sComponent, EXP_DARWIN_ID_SIG );
1176
1177 /* the last field is a single zero dword */
1178 zero = 0;
1179 r = IStream_Write( stm, &zero, sizeof zero, &count );
1180
1181 return S_OK;
1182 }
1183
1184 /************************************************************************
1185 * IPersistStream_GetSizeMax (IPersistStream)
1186 */
1187 static HRESULT WINAPI IPersistStream_fnGetSizeMax(
1188 IPersistStream* iface,
1189 ULARGE_INTEGER* pcbSize)
1190 {
1191 IShellLinkImpl *This = impl_from_IPersistStream(iface);
1192
1193 TRACE("(%p)\n", This);
1194
1195 return E_NOTIMPL;
1196 }
1197
1198 static const IPersistStreamVtbl psvt =
1199 {
1200 IPersistStream_fnQueryInterface,
1201 IPersistStream_fnAddRef,
1202 IPersistStream_fnRelease,
1203 IPersistStream_fnGetClassID,
1204 IPersistStream_fnIsDirty,
1205 IPersistStream_fnLoad,
1206 IPersistStream_fnSave,
1207 IPersistStream_fnGetSizeMax
1208 };
1209
1210 /**************************************************************************
1211 * IShellLink_Constructor
1212 */
1213 HRESULT WINAPI IShellLink_Constructor( IUnknown *pUnkOuter,
1214 REFIID riid, LPVOID *ppv )
1215 {
1216 IShellLinkImpl * sl;
1217 HRESULT r;
1218
1219 TRACE("unkOut=%p riid=%s\n",pUnkOuter, debugstr_guid(riid));
1220
1221 *ppv = NULL;
1222
1223 if (pUnkOuter)
1224 return CLASS_E_NOAGGREGATION;
1225 sl = LocalAlloc(LMEM_ZEROINIT,sizeof(IShellLinkImpl));
1226 if (!sl)
1227 return E_OUTOFMEMORY;
1228
1229 sl->ref = 1;
1230 sl->lpVtbl = &slvt;
1231 sl->lpvtblw = &slvtw;
1232 sl->lpvtblPersistFile = &pfvt;
1233 sl->lpvtblPersistStream = &psvt;
1234 sl->lpvtblShellLinkDataList = &dlvt;
1235 sl->lpvtblShellExtInit = &eivt;
1236 sl->lpvtblContextMenu = &cmvt;
1237 sl->lpvtblObjectWithSite = &owsvt;
1238 sl->iShowCmd = SW_SHOWNORMAL;
1239 sl->bDirty = FALSE;
1240 sl->iIdOpen = -1;
1241 sl->site = NULL;
1242
1243 TRACE("(%p)->()\n",sl);
1244
1245 r = ShellLink_QueryInterface( sl, riid, ppv );
1246 ShellLink_Release( sl );
1247 return r;
1248 }
1249
1250
1251 static BOOL SHELL_ExistsFileW(LPCWSTR path)
1252 {
1253 if (INVALID_FILE_ATTRIBUTES == GetFileAttributesW(path))
1254 return FALSE;
1255 return TRUE;
1256 }
1257
1258 /**************************************************************************
1259 * ShellLink_UpdatePath
1260 * update absolute path in sPath using relative path in sPathRel
1261 */
1262 static HRESULT ShellLink_UpdatePath(LPCWSTR sPathRel, LPCWSTR path, LPCWSTR sWorkDir, LPWSTR* psPath)
1263 {
1264 if (!path || !psPath)
1265 return E_INVALIDARG;
1266
1267 if (!*psPath && sPathRel) {
1268 WCHAR buffer[2*MAX_PATH], abs_path[2*MAX_PATH];
1269 LPWSTR final = NULL;
1270
1271 /* first try if [directory of link file] + [relative path] finds an existing file */
1272
1273 GetFullPathNameW( path, MAX_PATH*2, buffer, &final );
1274 if( !final )
1275 final = buffer;
1276 lstrcpyW(final, sPathRel);
1277
1278 *abs_path = '\0';
1279
1280 if (SHELL_ExistsFileW(buffer)) {
1281 if (!GetFullPathNameW(buffer, MAX_PATH, abs_path, &final))
1282 lstrcpyW(abs_path, buffer);
1283 } else {
1284 /* try if [working directory] + [relative path] finds an existing file */
1285 if (sWorkDir) {
1286 lstrcpyW(buffer, sWorkDir);
1287 lstrcpyW(PathAddBackslashW(buffer), sPathRel);
1288
1289 if (SHELL_ExistsFileW(buffer))
1290 if (!GetFullPathNameW(buffer, MAX_PATH, abs_path, &final))
1291 lstrcpyW(abs_path, buffer);
1292 }
1293 }
1294
1295 /* FIXME: This is even not enough - not all shell links can be resolved using this algorithm. */
1296 if (!*abs_path)
1297 lstrcpyW(abs_path, sPathRel);
1298
1299 *psPath = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(abs_path)+1)*sizeof(WCHAR));
1300 if (!*psPath)
1301 return E_OUTOFMEMORY;
1302
1303 lstrcpyW(*psPath, abs_path);
1304 }
1305
1306 return S_OK;
1307 }
1308
1309 /**************************************************************************
1310 * IShellLink_ConstructFromFile
1311 */
1312 HRESULT WINAPI IShellLink_ConstructFromFile( IUnknown* pUnkOuter, REFIID riid,
1313 LPCITEMIDLIST pidl, LPVOID* ppv)
1314 {
1315 IShellLinkW* psl;
1316
1317 HRESULT hr = IShellLink_Constructor(NULL, riid, (LPVOID*)&psl);
1318
1319 if (SUCCEEDED(hr)) {
1320 IPersistFile* ppf;
1321
1322 *ppv = NULL;
1323
1324 hr = IShellLinkW_QueryInterface(psl, &IID_IPersistFile, (LPVOID*)&ppf);
1325
1326 if (SUCCEEDED(hr)) {
1327 WCHAR path[MAX_PATH];
1328
1329 if (SHGetPathFromIDListW(pidl, path))
1330 hr = IPersistFile_Load(ppf, path, 0);
1331 else
1332 hr = E_FAIL;
1333
1334 if (SUCCEEDED(hr))
1335 *ppv = psl;
1336
1337 IPersistFile_Release(ppf);
1338 }
1339
1340 if (!*ppv)
1341 IShellLinkW_Release(psl);
1342 }
1343
1344 return hr;
1345 }
1346
1347 /**************************************************************************
1348 * IShellLinkA_QueryInterface
1349 */
1350 static HRESULT WINAPI IShellLinkA_fnQueryInterface( IShellLinkA * iface, REFIID riid, LPVOID *ppvObj)
1351 {
1352 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1353 return ShellLink_QueryInterface( This, riid, ppvObj );
1354 }
1355
1356 /******************************************************************************
1357 * IShellLinkA_AddRef
1358 */
1359 static ULONG WINAPI IShellLinkA_fnAddRef(IShellLinkA * iface)
1360 {
1361 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1362 return ShellLink_AddRef( This );
1363 }
1364
1365 /******************************************************************************
1366 * IShellLinkA_Release
1367 */
1368 static ULONG WINAPI IShellLinkA_fnRelease(IShellLinkA * iface)
1369 {
1370 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1371 return ShellLink_Release( This );
1372 }
1373
1374 static HRESULT WINAPI IShellLinkA_fnGetPath(IShellLinkA * iface, LPSTR pszFile,
1375 INT cchMaxPath, WIN32_FIND_DATAA *pfd, DWORD fFlags)
1376 {
1377 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1378
1379 TRACE("(%p)->(pfile=%p len=%u find_data=%p flags=%u)(%s)\n",
1380 This, pszFile, cchMaxPath, pfd, fFlags, debugstr_w(This->sPath));
1381
1382 if (This->sComponent || This->sProduct)
1383 return S_FALSE;
1384
1385 if (cchMaxPath)
1386 pszFile[0] = 0;
1387 if (This->sPath)
1388 WideCharToMultiByte( CP_ACP, 0, This->sPath, -1,
1389 pszFile, cchMaxPath, NULL, NULL);
1390
1391 if (pfd) FIXME("(%p): WIN32_FIND_DATA is not yet filled.\n", This);
1392
1393 return S_OK;
1394 }
1395
1396 static HRESULT WINAPI IShellLinkA_fnGetIDList(IShellLinkA * iface, LPITEMIDLIST * ppidl)
1397 {
1398 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1399
1400 TRACE("(%p)->(ppidl=%p)\n",This, ppidl);
1401
1402 return IShellLinkW_GetIDList((IShellLinkW*)&(This->lpvtblw), ppidl);
1403 }
1404
1405 static HRESULT WINAPI IShellLinkA_fnSetIDList(IShellLinkA * iface, LPCITEMIDLIST pidl)
1406 {
1407 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1408
1409 TRACE("(%p)->(pidl=%p)\n",This, pidl);
1410
1411 if (This->pPidl)
1412 ILFree(This->pPidl);
1413 This->pPidl = ILClone (pidl);
1414 This->bDirty = TRUE;
1415
1416 return S_OK;
1417 }
1418
1419 static HRESULT WINAPI IShellLinkA_fnGetDescription(IShellLinkA * iface, LPSTR pszName,INT cchMaxName)
1420 {
1421 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1422
1423 TRACE("(%p)->(%p len=%u)\n",This, pszName, cchMaxName);
1424
1425 if( cchMaxName )
1426 pszName[0] = 0;
1427 if( This->sDescription )
1428 WideCharToMultiByte( CP_ACP, 0, This->sDescription, -1,
1429 pszName, cchMaxName, NULL, NULL);
1430
1431 return S_OK;
1432 }
1433
1434 static HRESULT WINAPI IShellLinkA_fnSetDescription(IShellLinkA * iface, LPCSTR pszName)
1435 {
1436 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1437
1438 TRACE("(%p)->(pName=%s)\n", This, pszName);
1439
1440 HeapFree(GetProcessHeap(), 0, This->sDescription);
1441 This->sDescription = HEAP_strdupAtoW( GetProcessHeap(), 0, pszName);
1442 if ( !This->sDescription )
1443 return E_OUTOFMEMORY;
1444
1445 This->bDirty = TRUE;
1446
1447 return S_OK;
1448 }
1449
1450 static HRESULT WINAPI IShellLinkA_fnGetWorkingDirectory(IShellLinkA * iface, LPSTR pszDir,INT cchMaxPath)
1451 {
1452 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1453
1454 TRACE("(%p)->(%p len=%u)\n", This, pszDir, cchMaxPath);
1455
1456 if( cchMaxPath )
1457 pszDir[0] = 0;
1458 if( This->sWorkDir )
1459 WideCharToMultiByte( CP_ACP, 0, This->sWorkDir, -1,
1460 pszDir, cchMaxPath, NULL, NULL);
1461
1462 return S_OK;
1463 }
1464
1465 static HRESULT WINAPI IShellLinkA_fnSetWorkingDirectory(IShellLinkA * iface, LPCSTR pszDir)
1466 {
1467 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1468
1469 TRACE("(%p)->(dir=%s)\n",This, pszDir);
1470
1471 HeapFree(GetProcessHeap(), 0, This->sWorkDir);
1472 This->sWorkDir = HEAP_strdupAtoW( GetProcessHeap(), 0, pszDir);
1473 if ( !This->sWorkDir )
1474 return E_OUTOFMEMORY;
1475
1476 This->bDirty = TRUE;
1477
1478 return S_OK;
1479 }
1480
1481 static HRESULT WINAPI IShellLinkA_fnGetArguments(IShellLinkA * iface, LPSTR pszArgs,INT cchMaxPath)
1482 {
1483 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1484
1485 TRACE("(%p)->(%p len=%u)\n", This, pszArgs, cchMaxPath);
1486
1487 if( cchMaxPath )
1488 pszArgs[0] = 0;
1489 if( This->sArgs )
1490 WideCharToMultiByte( CP_ACP, 0, This->sArgs, -1,
1491 pszArgs, cchMaxPath, NULL, NULL);
1492
1493 return S_OK;
1494 }
1495
1496 static HRESULT WINAPI IShellLinkA_fnSetArguments(IShellLinkA * iface, LPCSTR pszArgs)
1497 {
1498 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1499
1500 TRACE("(%p)->(args=%s)\n",This, pszArgs);
1501
1502 HeapFree(GetProcessHeap(), 0, This->sArgs);
1503 if (pszArgs)
1504 {
1505 This->sArgs = HEAP_strdupAtoW( GetProcessHeap(), 0, pszArgs);
1506 if( !This->sArgs )
1507 return E_OUTOFMEMORY;
1508 }
1509 else This->sArgs = NULL;
1510
1511 This->bDirty = TRUE;
1512
1513 return S_OK;
1514 }
1515
1516 static HRESULT WINAPI IShellLinkA_fnGetHotkey(IShellLinkA * iface, WORD *pwHotkey)
1517 {
1518 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1519
1520 TRACE("(%p)->(%p)(0x%08x)\n",This, pwHotkey, This->wHotKey);
1521
1522 *pwHotkey = This->wHotKey;
1523
1524 return S_OK;
1525 }
1526
1527 static HRESULT WINAPI IShellLinkA_fnSetHotkey(IShellLinkA * iface, WORD wHotkey)
1528 {
1529 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1530
1531 TRACE("(%p)->(hotkey=%x)\n",This, wHotkey);
1532
1533 This->wHotKey = wHotkey;
1534 This->bDirty = TRUE;
1535
1536 return S_OK;
1537 }
1538
1539 static HRESULT WINAPI IShellLinkA_fnGetShowCmd(IShellLinkA * iface, INT *piShowCmd)
1540 {
1541 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1542
1543 TRACE("(%p)->(%p)\n",This, piShowCmd);
1544 *piShowCmd = This->iShowCmd;
1545 return S_OK;
1546 }
1547
1548 static HRESULT WINAPI IShellLinkA_fnSetShowCmd(IShellLinkA * iface, INT iShowCmd)
1549 {
1550 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1551
1552 TRACE("(%p) %d\n",This, iShowCmd);
1553
1554 This->iShowCmd = iShowCmd;
1555 This->bDirty = TRUE;
1556
1557 return NOERROR;
1558 }
1559
1560 static HRESULT SHELL_PidlGeticonLocationA(IShellFolder* psf, LPCITEMIDLIST pidl,
1561 LPSTR pszIconPath, int cchIconPath, int* piIcon)
1562 {
1563 LPCITEMIDLIST pidlLast;
1564
1565 HRESULT hr = SHBindToParent(pidl, &IID_IShellFolder, (LPVOID*)&psf, &pidlLast);
1566
1567 if (SUCCEEDED(hr)) {
1568 IExtractIconA* pei;
1569
1570 hr = IShellFolder_GetUIObjectOf(psf, 0, 1, &pidlLast, &IID_IExtractIconA, NULL, (LPVOID*)&pei);
1571
1572 if (SUCCEEDED(hr)) {
1573 hr = IExtractIconA_GetIconLocation(pei, 0, pszIconPath, MAX_PATH, piIcon, NULL);
1574
1575 IExtractIconA_Release(pei);
1576 }
1577
1578 IShellFolder_Release(psf);
1579 }
1580
1581 return hr;
1582 }
1583
1584 static HRESULT WINAPI IShellLinkA_fnGetIconLocation(IShellLinkA * iface, LPSTR pszIconPath,INT cchIconPath,INT *piIcon)
1585 {
1586 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1587
1588 TRACE("(%p)->(%p len=%u iicon=%p)\n", This, pszIconPath, cchIconPath, piIcon);
1589
1590 pszIconPath[0] = 0;
1591 *piIcon = This->iIcoNdx;
1592
1593 if (This->sIcoPath)
1594 {
1595 WideCharToMultiByte(CP_ACP, 0, This->sIcoPath, -1, pszIconPath, cchIconPath, NULL, NULL);
1596 return S_OK;
1597 }
1598
1599 if (This->pPidl || This->sPath)
1600 {
1601 IShellFolder* pdsk;
1602
1603 HRESULT hr = SHGetDesktopFolder(&pdsk);
1604
1605 if (SUCCEEDED(hr))
1606 {
1607 /* first look for an icon using the PIDL (if present) */
1608 if (This->pPidl)
1609 hr = SHELL_PidlGeticonLocationA(pdsk, This->pPidl, pszIconPath, cchIconPath, piIcon);
1610 else
1611 hr = E_FAIL;
1612
1613 /* if we couldn't find an icon yet, look for it using the file system path */
1614 if (FAILED(hr) && This->sPath)
1615 {
1616 LPITEMIDLIST pidl;
1617
1618 hr = IShellFolder_ParseDisplayName(pdsk, 0, NULL, This->sPath, NULL, &pidl, NULL);
1619
1620 if (SUCCEEDED(hr)) {
1621 hr = SHELL_PidlGeticonLocationA(pdsk, pidl, pszIconPath, cchIconPath, piIcon);
1622
1623 SHFree(pidl);
1624 }
1625 }
1626
1627 IShellFolder_Release(pdsk);
1628 }
1629
1630 return hr;
1631 }
1632 return S_OK;
1633 }
1634
1635 static HRESULT WINAPI IShellLinkA_fnSetIconLocation(IShellLinkA * iface, LPCSTR pszIconPath,INT iIcon)
1636 {
1637 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1638
1639 TRACE("(%p)->(path=%s iicon=%u)\n",This, pszIconPath, iIcon);
1640
1641 HeapFree(GetProcessHeap(), 0, This->sIcoPath);
1642 This->sIcoPath = HEAP_strdupAtoW(GetProcessHeap(), 0, pszIconPath);
1643 if ( !This->sIcoPath )
1644 return E_OUTOFMEMORY;
1645
1646 This->iIcoNdx = iIcon;
1647 This->bDirty = TRUE;
1648
1649 return S_OK;
1650 }
1651
1652 static HRESULT WINAPI IShellLinkA_fnSetRelativePath(IShellLinkA * iface, LPCSTR pszPathRel, DWORD dwReserved)
1653 {
1654 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1655
1656 TRACE("(%p)->(path=%s %x)\n",This, pszPathRel, dwReserved);
1657
1658 HeapFree(GetProcessHeap(), 0, This->sPathRel);
1659 This->sPathRel = HEAP_strdupAtoW(GetProcessHeap(), 0, pszPathRel);
1660 This->bDirty = TRUE;
1661
1662 return ShellLink_UpdatePath(This->sPathRel, This->sPath, This->sWorkDir, &This->sPath);
1663 }
1664
1665 static HRESULT WINAPI IShellLinkA_fnResolve(IShellLinkA * iface, HWND hwnd, DWORD fFlags)
1666 {
1667 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1668
1669 TRACE("(%p)->(hwnd=%p flags=%x)\n",This, hwnd, fFlags);
1670
1671 return IShellLinkW_Resolve( (IShellLinkW*)&(This->lpvtblw), hwnd, fFlags );
1672 }
1673
1674 static HRESULT WINAPI IShellLinkA_fnSetPath(IShellLinkA * iface, LPCSTR pszFile)
1675 {
1676 HRESULT r;
1677 LPWSTR str;
1678 IShellLinkImpl *This = (IShellLinkImpl *)iface;
1679
1680 TRACE("(%p)->(path=%s)\n",This, pszFile);
1681
1682 if (!pszFile) return E_INVALIDARG;
1683
1684 str = HEAP_strdupAtoW(GetProcessHeap(), 0, pszFile);
1685 if( !str )
1686 return E_OUTOFMEMORY;
1687
1688 r = IShellLinkW_SetPath((IShellLinkW*)&(This->lpvtblw), str);
1689 HeapFree( GetProcessHeap(), 0, str );
1690
1691 return r;
1692 }
1693
1694 /**************************************************************************
1695 * IShellLink Implementation
1696 */
1697
1698 static const IShellLinkAVtbl slvt =
1699 {
1700 IShellLinkA_fnQueryInterface,
1701 IShellLinkA_fnAddRef,
1702 IShellLinkA_fnRelease,
1703 IShellLinkA_fnGetPath,
1704 IShellLinkA_fnGetIDList,
1705 IShellLinkA_fnSetIDList,
1706 IShellLinkA_fnGetDescription,
1707 IShellLinkA_fnSetDescription,
1708 IShellLinkA_fnGetWorkingDirectory,
1709 IShellLinkA_fnSetWorkingDirectory,
1710 IShellLinkA_fnGetArguments,
1711 IShellLinkA_fnSetArguments,
1712 IShellLinkA_fnGetHotkey,
1713 IShellLinkA_fnSetHotkey,
1714 IShellLinkA_fnGetShowCmd,
1715 IShellLinkA_fnSetShowCmd,
1716 IShellLinkA_fnGetIconLocation,
1717 IShellLinkA_fnSetIconLocation,
1718 IShellLinkA_fnSetRelativePath,
1719 IShellLinkA_fnResolve,
1720 IShellLinkA_fnSetPath
1721 };
1722
1723
1724 /**************************************************************************
1725 * IShellLinkW_fnQueryInterface
1726 */
1727 static HRESULT WINAPI IShellLinkW_fnQueryInterface(
1728 IShellLinkW * iface, REFIID riid, LPVOID *ppvObj)
1729 {
1730 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1731 return ShellLink_QueryInterface( This, riid, ppvObj );
1732 }
1733
1734 /******************************************************************************
1735 * IShellLinkW_fnAddRef
1736 */
1737 static ULONG WINAPI IShellLinkW_fnAddRef(IShellLinkW * iface)
1738 {
1739 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1740 return ShellLink_AddRef( This );
1741 }
1742
1743 /******************************************************************************
1744 * IShellLinkW_fnRelease
1745 */
1746 static ULONG WINAPI IShellLinkW_fnRelease(IShellLinkW * iface)
1747 {
1748 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1749 return ShellLink_Release( This );
1750 }
1751
1752 static HRESULT WINAPI IShellLinkW_fnGetPath(IShellLinkW * iface, LPWSTR pszFile,INT cchMaxPath, WIN32_FIND_DATAW *pfd, DWORD fFlags)
1753 {
1754 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1755
1756 TRACE("(%p)->(pfile=%p len=%u find_data=%p flags=%u)(%s)\n",
1757 This, pszFile, cchMaxPath, pfd, fFlags, debugstr_w(This->sPath));
1758
1759 if (This->sComponent || This->sProduct)
1760 return S_FALSE;
1761
1762 if (cchMaxPath)
1763 pszFile[0] = 0;
1764 if (This->sPath)
1765 lstrcpynW( pszFile, This->sPath, cchMaxPath );
1766
1767 if (pfd) FIXME("(%p): WIN32_FIND_DATA is not yet filled.\n", This);
1768
1769 return S_OK;
1770 }
1771
1772 static HRESULT WINAPI IShellLinkW_fnGetIDList(IShellLinkW * iface, LPITEMIDLIST * ppidl)
1773 {
1774 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1775
1776 TRACE("(%p)->(ppidl=%p)\n",This, ppidl);
1777
1778 if (!This->pPidl)
1779 {
1780 *ppidl = NULL;
1781 return S_FALSE;
1782 }
1783 *ppidl = ILClone(This->pPidl);
1784 return S_OK;
1785 }
1786
1787 static HRESULT WINAPI IShellLinkW_fnSetIDList(IShellLinkW * iface, LPCITEMIDLIST pidl)
1788 {
1789 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1790
1791 TRACE("(%p)->(pidl=%p)\n",This, pidl);
1792
1793 if( This->pPidl )
1794 ILFree( This->pPidl );
1795 This->pPidl = ILClone( pidl );
1796 if( !This->pPidl )
1797 return E_FAIL;
1798
1799 This->bDirty = TRUE;
1800
1801 return S_OK;
1802 }
1803
1804 static HRESULT WINAPI IShellLinkW_fnGetDescription(IShellLinkW * iface, LPWSTR pszName,INT cchMaxName)
1805 {
1806 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1807
1808 TRACE("(%p)->(%p len=%u)\n",This, pszName, cchMaxName);
1809
1810 pszName[0] = 0;
1811 if( This->sDescription )
1812 lstrcpynW( pszName, This->sDescription, cchMaxName );
1813
1814 return S_OK;
1815 }
1816
1817 static HRESULT WINAPI IShellLinkW_fnSetDescription(IShellLinkW * iface, LPCWSTR pszName)
1818 {
1819 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1820
1821 TRACE("(%p)->(desc=%s)\n",This, debugstr_w(pszName));
1822
1823 HeapFree(GetProcessHeap(), 0, This->sDescription);
1824 This->sDescription = HeapAlloc( GetProcessHeap(), 0,
1825 (lstrlenW( pszName )+1)*sizeof(WCHAR) );
1826 if ( !This->sDescription )
1827 return E_OUTOFMEMORY;
1828
1829 lstrcpyW( This->sDescription, pszName );
1830 This->bDirty = TRUE;
1831
1832 return S_OK;
1833 }
1834
1835 static HRESULT WINAPI IShellLinkW_fnGetWorkingDirectory(IShellLinkW * iface, LPWSTR pszDir,INT cchMaxPath)
1836 {
1837 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1838
1839 TRACE("(%p)->(%p len %u)\n", This, pszDir, cchMaxPath);
1840
1841 if( cchMaxPath )
1842 pszDir[0] = 0;
1843 if( This->sWorkDir )
1844 lstrcpynW( pszDir, This->sWorkDir, cchMaxPath );
1845
1846 return S_OK;
1847 }
1848
1849 static HRESULT WINAPI IShellLinkW_fnSetWorkingDirectory(IShellLinkW * iface, LPCWSTR pszDir)
1850 {
1851 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1852
1853 TRACE("(%p)->(dir=%s)\n",This, debugstr_w(pszDir));
1854
1855 HeapFree(GetProcessHeap(), 0, This->sWorkDir);
1856 This->sWorkDir = HeapAlloc( GetProcessHeap(), 0,
1857 (lstrlenW( pszDir )+1)*sizeof (WCHAR) );
1858 if ( !This->sWorkDir )
1859 return E_OUTOFMEMORY;
1860 lstrcpyW( This->sWorkDir, pszDir );
1861 This->bDirty = TRUE;
1862
1863 return S_OK;
1864 }
1865
1866 static HRESULT WINAPI IShellLinkW_fnGetArguments(IShellLinkW * iface, LPWSTR pszArgs,INT cchMaxPath)
1867 {
1868 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1869
1870 TRACE("(%p)->(%p len=%u)\n", This, pszArgs, cchMaxPath);
1871
1872 if( cchMaxPath )
1873 pszArgs[0] = 0;
1874 if( This->sArgs )
1875 lstrcpynW( pszArgs, This->sArgs, cchMaxPath );
1876
1877 return NOERROR;
1878 }
1879
1880 static HRESULT WINAPI IShellLinkW_fnSetArguments(IShellLinkW * iface, LPCWSTR pszArgs)
1881 {
1882 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1883
1884 TRACE("(%p)->(args=%s)\n",This, debugstr_w(pszArgs));
1885
1886 HeapFree(GetProcessHeap(), 0, This->sArgs);
1887 if (pszArgs)
1888 {
1889 This->sArgs = HeapAlloc( GetProcessHeap(), 0,
1890 (lstrlenW( pszArgs )+1)*sizeof (WCHAR) );
1891 if ( !This->sArgs )
1892 return E_OUTOFMEMORY;
1893 lstrcpyW( This->sArgs, pszArgs );
1894 }
1895 else This->sArgs = NULL;
1896
1897 This->bDirty = TRUE;
1898
1899 return S_OK;
1900 }
1901
1902 static HRESULT WINAPI IShellLinkW_fnGetHotkey(IShellLinkW * iface, WORD *pwHotkey)
1903 {
1904 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1905
1906 TRACE("(%p)->(%p)\n",This, pwHotkey);
1907
1908 *pwHotkey=This->wHotKey;
1909
1910 return S_OK;
1911 }
1912
1913 static HRESULT WINAPI IShellLinkW_fnSetHotkey(IShellLinkW * iface, WORD wHotkey)
1914 {
1915 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1916
1917 TRACE("(%p)->(hotkey=%x)\n",This, wHotkey);
1918
1919 This->wHotKey = wHotkey;
1920 This->bDirty = TRUE;
1921
1922 return S_OK;
1923 }
1924
1925 static HRESULT WINAPI IShellLinkW_fnGetShowCmd(IShellLinkW * iface, INT *piShowCmd)
1926 {
1927 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1928
1929 TRACE("(%p)->(%p)\n",This, piShowCmd);
1930
1931 *piShowCmd = This->iShowCmd;
1932
1933 return S_OK;
1934 }
1935
1936 static HRESULT WINAPI IShellLinkW_fnSetShowCmd(IShellLinkW * iface, INT iShowCmd)
1937 {
1938 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1939
1940 This->iShowCmd = iShowCmd;
1941 This->bDirty = TRUE;
1942
1943 return S_OK;
1944 }
1945
1946 static HRESULT SHELL_PidlGeticonLocationW(IShellFolder* psf, LPCITEMIDLIST pidl,
1947 LPWSTR pszIconPath, int cchIconPath, int* piIcon)
1948 {
1949 LPCITEMIDLIST pidlLast;
1950
1951 HRESULT hr = SHBindToParent(pidl, &IID_IShellFolder, (LPVOID*)&psf, &pidlLast);
1952
1953 if (SUCCEEDED(hr)) {
1954 IExtractIconW* pei;
1955
1956 hr = IShellFolder_GetUIObjectOf(psf, 0, 1, &pidlLast, &IID_IExtractIconW, NULL, (LPVOID*)&pei);
1957
1958 if (SUCCEEDED(hr)) {
1959 hr = IExtractIconW_GetIconLocation(pei, 0, pszIconPath, MAX_PATH, piIcon, NULL);
1960
1961 IExtractIconW_Release(pei);
1962 }
1963
1964 IShellFolder_Release(psf);
1965 }
1966
1967 return hr;
1968 }
1969
1970 static HRESULT WINAPI IShellLinkW_fnGetIconLocation(IShellLinkW * iface, LPWSTR pszIconPath,INT cchIconPath,INT *piIcon)
1971 {
1972 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
1973
1974 TRACE("(%p)->(%p len=%u iicon=%p)\n", This, pszIconPath, cchIconPath, piIcon);
1975
1976 pszIconPath[0] = 0;
1977 *piIcon = This->iIcoNdx;
1978
1979 if (This->sIcoPath)
1980 {
1981 lstrcpynW(pszIconPath, This->sIcoPath, cchIconPath);
1982 return S_OK;
1983 }
1984
1985 if (This->pPidl || This->sPath)
1986 {
1987 IShellFolder* pdsk;
1988
1989 HRESULT hr = SHGetDesktopFolder(&pdsk);
1990
1991 if (SUCCEEDED(hr))
1992 {
1993 /* first look for an icon using the PIDL (if present) */
1994 if (This->pPidl)
1995 hr = SHELL_PidlGeticonLocationW(pdsk, This->pPidl, pszIconPath, cchIconPath, piIcon);
1996 else
1997 hr = E_FAIL;
1998
1999 /* if we couldn't find an icon yet, look for it using the file system path */
2000 if (FAILED(hr) && This->sPath)
2001 {
2002 LPITEMIDLIST pidl;
2003
2004 hr = IShellFolder_ParseDisplayName(pdsk, 0, NULL, This->sPath, NULL, &pidl, NULL);
2005
2006 if (SUCCEEDED(hr))
2007 {
2008 hr = SHELL_PidlGeticonLocationW(pdsk, pidl, pszIconPath, cchIconPath, piIcon);
2009
2010 SHFree(pidl);
2011 }
2012 }
2013
2014 IShellFolder_Release(pdsk);
2015 }
2016 return hr;
2017 }
2018 return S_OK;
2019 }
2020
2021 static HRESULT WINAPI IShellLinkW_fnSetIconLocation(IShellLinkW * iface, LPCWSTR pszIconPath,INT iIcon)
2022 {
2023 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
2024
2025 TRACE("(%p)->(path=%s iicon=%u)\n",This, debugstr_w(pszIconPath), iIcon);
2026
2027 HeapFree(GetProcessHeap(), 0, This->sIcoPath);
2028 This->sIcoPath = HeapAlloc( GetProcessHeap(), 0,
2029 (lstrlenW( pszIconPath )+1)*sizeof (WCHAR) );
2030 if ( !This->sIcoPath )
2031 return E_OUTOFMEMORY;
2032 lstrcpyW( This->sIcoPath, pszIconPath );
2033
2034 This->iIcoNdx = iIcon;
2035 This->bDirty = TRUE;
2036
2037 return S_OK;
2038 }
2039
2040 static HRESULT WINAPI IShellLinkW_fnSetRelativePath(IShellLinkW * iface, LPCWSTR pszPathRel, DWORD dwReserved)
2041 {
2042 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
2043
2044 TRACE("(%p)->(path=%s %x)\n",This, debugstr_w(pszPathRel), dwReserved);
2045
2046 HeapFree(GetProcessHeap(), 0, This->sPathRel);
2047 This->sPathRel = HeapAlloc( GetProcessHeap(), 0,
2048 (lstrlenW( pszPathRel )+1) * sizeof (WCHAR) );
2049 if ( !This->sPathRel )
2050 return E_OUTOFMEMORY;
2051 lstrcpyW( This->sPathRel, pszPathRel );
2052 This->bDirty = TRUE;
2053
2054 return ShellLink_UpdatePath(This->sPathRel, This->sPath, This->sWorkDir, &This->sPath);
2055 }
2056
2057 static HRESULT WINAPI IShellLinkW_fnResolve(IShellLinkW * iface, HWND hwnd, DWORD fFlags)
2058 {
2059 HRESULT hr = S_OK;
2060 BOOL bSuccess;
2061
2062 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
2063
2064 TRACE("(%p)->(hwnd=%p flags=%x)\n",This, hwnd, fFlags);
2065
2066 /*FIXME: use IResolveShellLink interface */
2067
2068 if (!This->sPath && This->pPidl) {
2069 WCHAR buffer[MAX_PATH];
2070
2071 bSuccess = SHGetPathFromIDListW(This->pPidl, buffer);
2072
2073 if (bSuccess && *buffer) {
2074 This->sPath = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(buffer)+1)*sizeof(WCHAR));
2075 if (!This->sPath)
2076 return E_OUTOFMEMORY;
2077
2078 lstrcpyW(This->sPath, buffer);
2079
2080 This->bDirty = TRUE;
2081 } else
2082 hr = S_OK; /* don't report an error occurred while just caching information */
2083 }
2084
2085 if (!This->sIcoPath && This->sPath) {
2086 This->sIcoPath = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(This->sPath)+1)*sizeof(WCHAR));
2087 if (!This->sIcoPath)
2088 return E_OUTOFMEMORY;
2089
2090 lstrcpyW(This->sIcoPath, This->sPath);
2091 This->iIcoNdx = 0;
2092
2093 This->bDirty = TRUE;
2094 }
2095
2096 return hr;
2097 }
2098
2099 static LPWSTR ShellLink_GetAdvertisedArg(LPCWSTR str)
2100 {
2101 LPWSTR ret;
2102 LPCWSTR p;
2103 DWORD len;
2104
2105 if( !str )
2106 return NULL;
2107
2108 p = strchrW( str, ':' );
2109 if( !p )
2110 return NULL;
2111 len = p - str;
2112 ret = HeapAlloc( GetProcessHeap(), 0, sizeof(WCHAR)*(len+1));
2113 if( !ret )
2114 return ret;
2115 memcpy( ret, str, sizeof(WCHAR)*len );
2116 ret[len] = 0;
2117 return ret;
2118 }
2119
2120 static HRESULT ShellLink_SetAdvertiseInfo(IShellLinkImpl *This, LPCWSTR str)
2121 {
2122 LPCWSTR szComponent = NULL, szProduct = NULL, p;
2123 WCHAR szGuid[39];
2124 HRESULT r;
2125 GUID guid;
2126 int len;
2127
2128 while( str[0] )
2129 {
2130 /* each segment must start with two colons */
2131 if( str[0] != ':' || str[1] != ':' )
2132 return E_FAIL;
2133
2134 /* the last segment is just two colons */
2135 if( !str[2] )
2136 break;
2137 str += 2;
2138
2139 /* there must be a colon straight after a guid */
2140 p = strchrW( str, ':' );
2141 if( !p )
2142 return E_FAIL;
2143 len = p - str;
2144 if( len != 38 )
2145 return E_FAIL;
2146
2147 /* get the guid, and check it's validly formatted */
2148 memcpy( szGuid, str, sizeof(WCHAR)*len );
2149 szGuid[len] = 0;
2150 r = CLSIDFromString( szGuid, &guid );
2151 if( r != S_OK )
2152 return r;
2153 str = p + 1;
2154
2155 /* match it up to a guid that we care about */
2156 if( IsEqualGUID( &guid, &SHELL32_AdvtShortcutComponent ) && !szComponent )
2157 szComponent = str;
2158 else if( IsEqualGUID( &guid, &SHELL32_AdvtShortcutProduct ) && !szProduct )
2159 szProduct = str;
2160 else
2161 return E_FAIL;
2162
2163 /* skip to the next field */
2164 str = strchrW( str, ':' );
2165 if( !str )
2166 return E_FAIL;
2167 }
2168
2169 /* we have to have a component for an advertised shortcut */
2170 if( !szComponent )
2171 return E_FAIL;
2172
2173 This->sComponent = ShellLink_GetAdvertisedArg( szComponent );
2174 This->sProduct = ShellLink_GetAdvertisedArg( szProduct );
2175
2176 TRACE("Component = %s\n", debugstr_w(This->sComponent));
2177 TRACE("Product = %s\n", debugstr_w(This->sProduct));
2178
2179 return S_OK;
2180 }
2181
2182 static BOOL ShellLink_GetVolumeInfo(LPCWSTR path, volume_info *volume)
2183 {
2184 const int label_sz = sizeof volume->label/sizeof volume->label[0];
2185 WCHAR drive[4] = { path[0], ':', '\\', 0 };
2186 BOOL r;
2187
2188 volume->type = GetDriveTypeW(drive);
2189 r = GetVolumeInformationW(drive, volume->label, label_sz,
2190 &volume->serial, NULL, NULL, NULL, 0);
2191 TRACE("r = %d type %d serial %08x name %s\n", r,
2192 volume->type, volume->serial, debugstr_w(volume->label));
2193 return r;
2194 }
2195
2196 static HRESULT WINAPI IShellLinkW_fnSetPath(IShellLinkW * iface, LPCWSTR pszFile)
2197 {
2198 IShellLinkImpl *This = impl_from_IShellLinkW(iface);
2199 WCHAR buffer[MAX_PATH];
2200 LPWSTR fname, unquoted = NULL;
2201 HRESULT hr = S_OK;
2202 UINT len;
2203
2204 TRACE("(%p)->(path=%s)\n",This, debugstr_w(pszFile));
2205
2206 if (!pszFile) return E_INVALIDARG;
2207
2208 /* quotes at the ends of the string are stripped */
2209 len = lstrlenW(pszFile);
2210 if (pszFile[0] == '"' && pszFile[len-1] == '"')
2211 {
2212 unquoted = strdupW(pszFile);
2213 PathUnquoteSpacesW(unquoted);
2214 pszFile = unquoted;
2215 }
2216
2217 /* any other quote marks are invalid */
2218 if (strchrW(pszFile, '"'))
2219 {
2220 HeapFree(GetProcessHeap(), 0, unquoted);
2221 return S_FALSE;
2222 }
2223
2224 HeapFree(GetProcessHeap(), 0, This->sPath);
2225 This->sPath = NULL;
2226
2227 HeapFree(GetProcessHeap(), 0, This->sComponent);
2228 This->sComponent = NULL;
2229
2230 if (This->pPidl)
2231 ILFree(This->pPidl);
2232 This->pPidl = NULL;
2233
2234 if (S_OK != ShellLink_SetAdvertiseInfo( This, pszFile ))
2235 {
2236 if (*pszFile == '\0')
2237 *buffer = '\0';
2238 else if (!GetFullPathNameW(pszFile, MAX_PATH, buffer, &fname))
2239 return E_FAIL;
2240 else if(!PathFileExistsW(buffer) &&
2241 !SearchPathW(NULL, pszFile, NULL, MAX_PATH, buffer, NULL))
2242 hr = S_FALSE;
2243
2244 This->pPidl = SHSimpleIDListFromPathW(pszFile);
2245 ShellLink_GetVolumeInfo(buffer, &This->volume);
2246
2247 This->sPath = HeapAlloc( GetProcessHeap(), 0,
2248 (lstrlenW( buffer )+1) * sizeof (WCHAR) );
2249 if (!This->sPath)
2250 {
2251 HeapFree(GetProcessHeap(), 0, unquoted);
2252 return E_OUTOFMEMORY;
2253 }
2254
2255 lstrcpyW(This->sPath, buffer);
2256 }
2257 This->bDirty = TRUE;
2258 HeapFree(GetProcessHeap(), 0, unquoted);
2259
2260 return hr;
2261 }
2262
2263 /**************************************************************************
2264 * IShellLinkW Implementation
2265 */
2266
2267 static const IShellLinkWVtbl slvtw =
2268 {
2269 IShellLinkW_fnQueryInterface,
2270 IShellLinkW_fnAddRef,
2271 IShellLinkW_fnRelease,
2272 IShellLinkW_fnGetPath,
2273 IShellLinkW_fnGetIDList,
2274 IShellLinkW_fnSetIDList,
2275 IShellLinkW_fnGetDescription,
2276 IShellLinkW_fnSetDescription,
2277 IShellLinkW_fnGetWorkingDirectory,
2278 IShellLinkW_fnSetWorkingDirectory,
2279 IShellLinkW_fnGetArguments,
2280 IShellLinkW_fnSetArguments,
2281 IShellLinkW_fnGetHotkey,
2282 IShellLinkW_fnSetHotkey,
2283 IShellLinkW_fnGetShowCmd,
2284 IShellLinkW_fnSetShowCmd,
2285 IShellLinkW_fnGetIconLocation,
2286 IShellLinkW_fnSetIconLocation,
2287 IShellLinkW_fnSetRelativePath,
2288 IShellLinkW_fnResolve,
2289 IShellLinkW_fnSetPath
2290 };
2291
2292 static HRESULT WINAPI
2293 ShellLink_DataList_QueryInterface( IShellLinkDataList* iface, REFIID riid, void** ppvObject)
2294 {
2295 IShellLinkImpl *This = impl_from_IShellLinkDataList(iface);
2296 return IShellLinkA_QueryInterface((IShellLinkA*)This, riid, ppvObject);
2297 }
2298
2299 static ULONG WINAPI
2300 ShellLink_DataList_AddRef( IShellLinkDataList* iface )
2301 {
2302 IShellLinkImpl *This = impl_from_IShellLinkDataList(iface);
2303 return IShellLinkA_AddRef((IShellLinkA*)This);
2304 }
2305
2306 static ULONG WINAPI
2307 ShellLink_DataList_Release( IShellLinkDataList* iface )
2308 {
2309 IShellLinkImpl *This = impl_from_IShellLinkDataList(iface);
2310 return ShellLink_Release( This );
2311 }
2312
2313 static HRESULT WINAPI
2314 ShellLink_AddDataBlock( IShellLinkDataList* iface, void* pDataBlock )
2315 {
2316 FIXME("\n");
2317 return E_NOTIMPL;
2318 }
2319
2320 static HRESULT WINAPI
2321 ShellLink_CopyDataBlock( IShellLinkDataList* iface, DWORD dwSig, void** ppDataBlock )
2322 {
2323 IShellLinkImpl *This = impl_from_IShellLinkDataList(iface);
2324 LPVOID block = NULL;
2325 HRESULT r = E_FAIL;
2326
2327 TRACE("%p %08x %p\n", iface, dwSig, ppDataBlock );
2328
2329 switch (dwSig)
2330 {
2331 case EXP_DARWIN_ID_SIG:
2332 if (!This->sComponent)
2333 break;
2334 block = shelllink_build_darwinid( This->sComponent, dwSig );
2335 r = S_OK;
2336 break;
2337 case EXP_SZ_LINK_SIG:
2338 case NT_CONSOLE_PROPS_SIG:
2339 case NT_FE_CONSOLE_PROPS_SIG:
2340 case EXP_SPECIAL_FOLDER_SIG:
2341 case EXP_SZ_ICON_SIG:
2342 FIXME("valid but unhandled datablock %08x\n", dwSig);
2343 break;
2344 default:
2345 ERR("unknown datablock %08x\n", dwSig);
2346 }
2347 *ppDataBlock = block;
2348 return r;
2349 }
2350
2351 static HRESULT WINAPI
2352 ShellLink_RemoveDataBlock( IShellLinkDataList* iface, DWORD dwSig )
2353 {
2354 FIXME("\n");
2355 return E_NOTIMPL;
2356 }
2357
2358 static HRESULT WINAPI
2359 ShellLink_GetFlags( IShellLinkDataList* iface, DWORD* pdwFlags )
2360 {
2361 IShellLinkImpl *This = impl_from_IShellLinkDataList(iface);
2362 DWORD flags = 0;
2363
2364 FIXME("%p %p\n", This, pdwFlags );
2365
2366 /* FIXME: add more */
2367 if (This->sArgs)
2368 flags |= SLDF_HAS_ARGS;
2369 if (This->sComponent)
2370 flags |= SLDF_HAS_DARWINID;
2371 if (This->sIcoPath)
2372 flags |= SLDF_HAS_ICONLOCATION;
2373 if (This->sProduct)
2374 flags |= SLDF_HAS_LOGO3ID;
2375 if (This->pPidl)
2376 flags |= SLDF_HAS_ID_LIST;
2377
2378 *pdwFlags = flags;
2379
2380 return S_OK;
2381 }
2382
2383 static HRESULT WINAPI
2384 ShellLink_SetFlags( IShellLinkDataList* iface, DWORD dwFlags )
2385 {
2386 FIXME("\n");
2387 return E_NOTIMPL;
2388 }
2389
2390 static const IShellLinkDataListVtbl dlvt =
2391 {
2392 ShellLink_DataList_QueryInterface,
2393 ShellLink_DataList_AddRef,
2394 ShellLink_DataList_Release,
2395 ShellLink_AddDataBlock,
2396 ShellLink_CopyDataBlock,
2397 ShellLink_RemoveDataBlock,
2398 ShellLink_GetFlags,
2399 ShellLink_SetFlags
2400 };
2401
2402 static HRESULT WINAPI
2403 ShellLink_ExtInit_QueryInterface( IShellExtInit* iface, REFIID riid, void** ppvObject )
2404 {
2405 IShellLinkImpl *This = impl_from_IShellExtInit(iface);
2406 return IShellLinkA_QueryInterface((IShellLinkA*)This, riid, ppvObject);
2407 }
2408
2409 static ULONG WINAPI
2410 ShellLink_ExtInit_AddRef( IShellExtInit* iface )
2411 {
2412 IShellLinkImpl *This = impl_from_IShellExtInit(iface);
2413 return IShellLinkA_AddRef((IShellLinkA*)This);
2414 }
2415
2416 static ULONG WINAPI
2417 ShellLink_ExtInit_Release( IShellExtInit* iface )
2418 {
2419 IShellLinkImpl *This = impl_from_IShellExtInit(iface);
2420 return ShellLink_Release( This );
2421 }
2422
2423 /**************************************************************************
2424 * ShellLink implementation of IShellExtInit::Initialize()
2425 *
2426 * Loads the shelllink from the dataobject the shell is pointing to.
2427 */
2428 static HRESULT WINAPI
2429 ShellLink_ExtInit_Initialize( IShellExtInit* iface, LPCITEMIDLIST pidlFolder,
2430 IDataObject *pdtobj, HKEY hkeyProgID )
2431 {
2432 IShellLinkImpl *This = impl_from_IShellExtInit(iface);
2433 FORMATETC format;
2434 STGMEDIUM stgm;
2435 UINT count;
2436 HRESULT r = E_FAIL;
2437
2438 TRACE("%p %p %p %p\n", This, pidlFolder, pdtobj, hkeyProgID );
2439
2440 if( !pdtobj )
2441 return r;
2442
2443 format.cfFormat = CF_HDROP;
2444 format.ptd = NULL;
2445 format.dwAspect = DVASPECT_CONTENT;
2446 format.lindex = -1;
2447 format.tymed = TYMED_HGLOBAL;
2448
2449 if( FAILED( IDataObject_GetData( pdtobj, &format, &stgm ) ) )
2450 return r;
2451
2452 count = DragQueryFileW( stgm.u.hGlobal, -1, NULL, 0 );
2453 if( count == 1 )
2454 {
2455 LPWSTR path;
2456
2457 count = DragQueryFileW( stgm.u.hGlobal, 0, NULL, 0 );
2458 count++;
2459 path = HeapAlloc( GetProcessHeap(), 0, count*sizeof(WCHAR) );
2460 if( path )
2461 {
2462 IPersistFile *pf = (IPersistFile*) &This->lpvtblPersistFile;
2463
2464 count = DragQueryFileW( stgm.u.hGlobal, 0, path, count );
2465 r = IPersistFile_Load( pf, path, 0 );
2466 HeapFree( GetProcessHeap(), 0, path );
2467 }
2468 }
2469 ReleaseStgMedium( &stgm );
2470
2471 return r;
2472 }
2473
2474 static const IShellExtInitVtbl eivt =
2475 {
2476 ShellLink_ExtInit_QueryInterface,
2477 ShellLink_ExtInit_AddRef,
2478 ShellLink_ExtInit_Release,
2479 ShellLink_ExtInit_Initialize
2480 };
2481
2482 static HRESULT WINAPI
2483 ShellLink_ContextMenu_QueryInterface( IContextMenu* iface, REFIID riid, void** ppvObject )
2484 {
2485 IShellLinkImpl *This = impl_from_IContextMenu(iface);
2486 return IShellLinkA_QueryInterface((IShellLinkA*)This, riid, ppvObject);
2487 }
2488
2489 static ULONG WINAPI
2490 ShellLink_ContextMenu_AddRef( IContextMenu* iface )
2491 {
2492 IShellLinkImpl *This = impl_from_IContextMenu(iface);
2493 return IShellLinkA_AddRef((IShellLinkA*)This);
2494 }
2495
2496 static ULONG WINAPI
2497 ShellLink_ContextMenu_Release( IContextMenu* iface )
2498 {
2499 IShellLinkImpl *This = impl_from_IContextMenu(iface);
2500 return ShellLink_Release( This );
2501 }
2502
2503 static HRESULT WINAPI
2504 ShellLink_QueryContextMenu( IContextMenu* iface, HMENU hmenu, UINT indexMenu,
2505 UINT idCmdFirst, UINT idCmdLast, UINT uFlags )
2506 {
2507 IShellLinkImpl *This = impl_from_IContextMenu(iface);
2508 static WCHAR szOpen[] = { 'O','p','e','n',0 };
2509 MENUITEMINFOW mii;
2510 int id = 1;
2511
2512 TRACE("%p %p %u %u %u %u\n", This,
2513 hmenu, indexMenu, idCmdFirst, idCmdLast, uFlags );
2514
2515 if ( !hmenu )
2516 return E_INVALIDARG;
2517
2518 memset( &mii, 0, sizeof mii );
2519 mii.cbSize = sizeof mii;
2520 mii.fMask = MIIM_TYPE | MIIM_ID | MIIM_STATE;
2521 mii.dwTypeData = szOpen;
2522 mii.cch = strlenW( mii.dwTypeData );
2523 mii.wID = idCmdFirst + id++;
2524 mii.fState = MFS_DEFAULT | MFS_ENABLED;
2525 mii.fType = MFT_STRING;
2526 if (!InsertMenuItemW( hmenu, indexMenu, TRUE, &mii ))
2527 return E_FAIL;
2528 This->iIdOpen = 0;
2529
2530 return MAKE_HRESULT( SEVERITY_SUCCESS, 0, id );
2531 }
2532
2533 static LPWSTR
2534 shelllink_get_msi_component_path( LPWSTR component )
2535 {
2536 LPWSTR path;
2537 DWORD r, sz = 0;
2538
2539 r = CommandLineFromMsiDescriptor( component, NULL, &sz );
2540 if (r != ERROR_SUCCESS)
2541 return NULL;
2542
2543 sz++;
2544 path = HeapAlloc( GetProcessHeap(), 0, sz*sizeof(WCHAR) );
2545 r = CommandLineFromMsiDescriptor( component, path, &sz );
2546 if (r != ERROR_SUCCESS)
2547 {
2548 HeapFree( GetProcessHeap(), 0, path );
2549 path = NULL;
2550 }
2551
2552 TRACE("returning %s\n", debugstr_w( path ) );
2553
2554 return path;
2555 }
2556
2557 static HRESULT WINAPI
2558 ShellLink_InvokeCommand( IContextMenu* iface, LPCMINVOKECOMMANDINFO lpici )
2559 {
2560 IShellLinkImpl *This = impl_from_IContextMenu(iface);
2561 static const WCHAR szOpen[] = { 'O','p','e','n',0 };
2562 SHELLEXECUTEINFOW sei;
2563 HWND hwnd = NULL; /* FIXME: get using interface set from IObjectWithSite */
2564 LPWSTR args = NULL;
2565 LPWSTR path = NULL;
2566 HRESULT r;
2567
2568 TRACE("%p %p\n", This, lpici );
2569
2570 if ( lpici->cbSize < sizeof (CMINVOKECOMMANDINFO) )
2571 return E_INVALIDARG;
2572
2573 if ( lpici->lpVerb != MAKEINTRESOURCEA(This->iIdOpen) )
2574 {
2575 ERR("Unknown id %p != %d\n", lpici->lpVerb, This->iIdOpen );
2576 return E_INVALIDARG;
2577 }
2578
2579 r = IShellLinkW_Resolve( (IShellLinkW*)&(This->lpvtblw), hwnd, 0 );
2580 if ( FAILED( r ) )
2581 return r;
2582
2583 if ( This->sComponent )
2584 {
2585 path = shelllink_get_msi_component_path( This->sComponent );
2586 if (!path)
2587 return E_FAIL;
2588 }
2589 else
2590 path = strdupW( This->sPath );
2591
2592 if ( lpici->cbSize == sizeof (CMINVOKECOMMANDINFOEX) &&
2593 ( lpici->fMask & CMIC_MASK_UNICODE ) )
2594 {
2595 LPCMINVOKECOMMANDINFOEX iciex = (LPCMINVOKECOMMANDINFOEX) lpici;
2596 DWORD len = 2;
2597
2598 if ( This->sArgs )
2599 len += lstrlenW( This->sArgs );
2600 if ( iciex->lpParametersW )
2601 len += lstrlenW( iciex->lpParametersW );
2602
2603 args = HeapAlloc( GetProcessHeap(), 0, len*sizeof(WCHAR) );
2604 args[0] = 0;
2605 if ( This->sArgs )
2606 lstrcatW( args, This->sArgs );
2607 if ( iciex->lpParametersW )
2608 {
2609 static const WCHAR space[] = { ' ', 0 };
2610 lstrcatW( args, space );
2611 lstrcatW( args, iciex->lpParametersW );
2612 }
2613 }
2614
2615 memset( &sei, 0, sizeof sei );
2616 sei.cbSize = sizeof sei;
2617 sei.fMask = SEE_MASK_UNICODE | (lpici->fMask & (SEE_MASK_NOASYNC|SEE_MASK_ASYNCOK|SEE_MASK_FLAG_NO_UI));
2618 sei.lpFile = path;
2619 sei.nShow = This->iShowCmd;
2620 sei.lpIDList = This->pPidl;
2621 sei.lpDirectory = This->sWorkDir;
2622 sei.lpParameters = args;
2623 sei.lpVerb = szOpen;
2624
2625 if( ShellExecuteExW( &sei ) )
2626 r = S_OK;
2627 else
2628 r = E_FAIL;
2629
2630 HeapFree( GetProcessHeap(), 0, args );
2631 HeapFree( GetProcessHeap(), 0, path );
2632
2633 return r;
2634 }
2635
2636 static HRESULT WINAPI
2637 ShellLink_GetCommandString( IContextMenu* iface, UINT_PTR idCmd, UINT uType,
2638 UINT* pwReserved, LPSTR pszName, UINT cchMax )
2639 {
2640 IShellLinkImpl *This = impl_from_IContextMenu(iface);
2641
2642 FIXME("%p %lu %u %p %p %u\n", This,
2643 idCmd, uType, pwReserved, pszName, cchMax );
2644
2645 return E_NOTIMPL;
2646 }
2647
2648 static const IContextMenuVtbl cmvt =
2649 {
2650 ShellLink_ContextMenu_QueryInterface,
2651 ShellLink_ContextMenu_AddRef,
2652 ShellLink_ContextMenu_Release,
2653 ShellLink_QueryContextMenu,
2654 ShellLink_InvokeCommand,
2655 ShellLink_GetCommandString
2656 };
2657
2658 static HRESULT WINAPI
2659 ShellLink_ObjectWithSite_QueryInterface( IObjectWithSite* iface, REFIID riid, void** ppvObject )
2660 {
2661 IShellLinkImpl *This = impl_from_IObjectWithSite(iface);
2662 return ShellLink_QueryInterface( This, riid, ppvObject );
2663 }
2664
2665 static ULONG WINAPI
2666 ShellLink_ObjectWithSite_AddRef( IObjectWithSite* iface )
2667 {
2668 IShellLinkImpl *This = impl_from_IObjectWithSite(iface);
2669 return ShellLink_AddRef( This );
2670 }
2671
2672 static ULONG WINAPI
2673 ShellLink_ObjectWithSite_Release( IObjectWithSite* iface )
2674 {
2675 IShellLinkImpl *This = impl_from_IObjectWithSite(iface);
2676 return ShellLink_Release( This );
2677 }
2678
2679 static HRESULT WINAPI
2680 ShellLink_GetSite( IObjectWithSite *iface, REFIID iid, void ** ppvSite )
2681 {
2682 IShellLinkImpl *This = impl_from_IObjectWithSite(iface);
2683
2684 TRACE("%p %s %p\n", This, debugstr_guid( iid ), ppvSite );
2685
2686 if ( !This->site )
2687 return E_FAIL;
2688 return IUnknown_QueryInterface( This->site, iid, ppvSite );
2689 }
2690
2691 static HRESULT WINAPI
2692 ShellLink_SetSite( IObjectWithSite *iface, IUnknown *punk )
2693 {
2694 IShellLinkImpl *This = impl_from_IObjectWithSite(iface);
2695
2696 TRACE("%p %p\n", iface, punk);
2697
2698 if ( punk )
2699 IUnknown_AddRef( punk );
2700 This->site = punk;
2701
2702 return S_OK;
2703 }
2704
2705 static const IObjectWithSiteVtbl owsvt =
2706 {
2707 ShellLink_ObjectWithSite_QueryInterface,
2708 ShellLink_ObjectWithSite_AddRef,
2709 ShellLink_ObjectWithSite_Release,
2710 ShellLink_SetSite,
2711 ShellLink_GetSite,
2712 };
Wine tree for Wine Gecko development