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