Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-06-22 22:11:39 +00:00
Code Issues Packages Projects Releases Wiki Activity
reactos/reactos/dll/win32/shell32/shelllink.cpp
Rafal Harabien 35da3cbf00 [SHELL32] 
- Fix exception in shortcut properties dialog

svn path=/trunk/; revision=54957
2012-01-14 13:46:33 +00:00

2210 lines
58 KiB
C++
Raw Blame History

/*
*
* Copyright 1997 Marcus Meissner
* Copyright 1998 Juergen Schmied
* Copyright 2005 Mike McCormack
* Copyright 2009 Andrew Hill
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*
* NOTES
* Nearly complete information about the binary formats
* of .lnk files available at http://www.wotsit.org
*
* You can use winedump to examine the contents of a link file:
* winedump lnk sc.lnk
*
* MSI advertised shortcuts are totally undocumented. They provide an
* icon for a program that is not yet installed, and invoke MSI to
* install the program when the shortcut is clicked on. They are
* created by passing a special string to SetPath, and the information
* in that string is parsed an stored.
*/
#include <precomp.h>
WINE_DEFAULT_DEBUG_CHANNEL(shell);
#define SHLINK_LOCAL 0
#define SHLINK_REMOTE 1
#define MAX_PROPERTY_SHEET_PAGE 32
/* link file formats */
#include "pshpack1.h"
struct LINK_HEADER
{
DWORD dwSize; /* 0x00 size of the header - 0x4c */
GUID MagicGuid; /* 0x04 is CLSID_ShellLink */
DWORD dwFlags; /* 0x14 describes elements following */
DWORD dwFileAttr; /* 0x18 attributes of the target file */
FILETIME Time1; /* 0x1c */
FILETIME Time2; /* 0x24 */
FILETIME Time3; /* 0x2c */
DWORD dwFileLength; /* 0x34 File length */
DWORD nIcon; /* 0x38 icon number */
DWORD fStartup; /* 0x3c startup type */
DWORD wHotKey; /* 0x40 hotkey */
DWORD Unknown5; /* 0x44 */
DWORD Unknown6; /* 0x48 */
};
struct LOCATION_INFO
{
DWORD dwTotalSize;
DWORD dwHeaderSize;
DWORD dwFlags;
DWORD dwVolTableOfs;
DWORD dwLocalPathOfs;
DWORD dwNetworkVolTableOfs;
DWORD dwFinalPathOfs;
};
struct LOCAL_VOLUME_INFO
{
DWORD dwSize;
DWORD dwType;
DWORD dwVolSerial;
DWORD dwVolLabelOfs;
};
struct volume_info
{
DWORD type;
DWORD serial;
WCHAR label[12]; /* assume 8.3 */
};
#include "poppack.h"
/* IShellLink Implementation */
static HRESULT ShellLink_UpdatePath(LPCWSTR sPathRel, LPCWSTR path, LPCWSTR sWorkDir, LPWSTR* psPath);
/* strdup on the process heap */
static LPWSTR __inline HEAP_strdupAtoW( HANDLE heap, DWORD flags, LPCSTR str)
{
INT len;
LPWSTR p;
assert(str);
len = MultiByteToWideChar( CP_ACP, 0, str, -1, NULL, 0 );
p = (LPWSTR)HeapAlloc( heap, flags, len*sizeof (WCHAR) );
if( !p )
return p;
MultiByteToWideChar( CP_ACP, 0, str, -1, p, len );
return p;
}
static LPWSTR __inline strdupW( LPCWSTR src )
{
LPWSTR dest;
if (!src) return NULL;
dest = (LPWSTR)HeapAlloc( GetProcessHeap(), 0, (wcslen(src)+1)*sizeof(WCHAR) );
if (dest)
wcscpy(dest, src);
return dest;
}
CShellLink::CShellLink()
{
pPidl = NULL;
wHotKey = 0;
memset(&time1, 0, sizeof(time1));
memset(&time2, 0, sizeof(time2));
memset(&time3, 0, sizeof(time3));
iShowCmd = SW_SHOWNORMAL;
sIcoPath = NULL;
iIcoNdx = 0;
sPath = NULL;
sArgs = NULL;
sWorkDir = NULL;
sDescription = NULL;
sPathRel = NULL;
sProduct = NULL;
sComponent = NULL;
memset(&volume, 0, sizeof(volume));
sLinkPath = NULL;
bRunAs = FALSE;
bDirty = FALSE;
iIdOpen = -1;
}
CShellLink::~CShellLink()
{
TRACE("-- destroying IShellLink(%p)\n", this);
HeapFree(GetProcessHeap(), 0, sIcoPath);
HeapFree(GetProcessHeap(), 0, sArgs);
HeapFree(GetProcessHeap(), 0, sWorkDir);
HeapFree(GetProcessHeap(), 0, sDescription);
HeapFree(GetProcessHeap(), 0, sPath);
HeapFree(GetProcessHeap(), 0, sLinkPath);
if (pPidl)
ILFree(pPidl);
}
HRESULT WINAPI CShellLink::GetClassID(CLSID *pclsid )
{
TRACE("%p %p\n", this, pclsid);
if (pclsid == NULL)
return E_POINTER;
*pclsid = CLSID_ShellLink;
return S_OK;
}
HRESULT WINAPI CShellLink::IsDirty()
{
TRACE("(%p)\n",this);
if (bDirty)
return S_OK;
return S_FALSE;
}
HRESULT WINAPI CShellLink::Load(LPCOLESTR pszFileName, DWORD dwMode)
{
HRESULT r;
CComPtr<IStream> stm;
TRACE("(%p, %s, %x)\n",this, debugstr_w(pszFileName), dwMode);
if (dwMode == 0)
dwMode = STGM_READ | STGM_SHARE_DENY_WRITE;
r = SHCreateStreamOnFileW(pszFileName, dwMode, &stm);
if (SUCCEEDED(r))
{
HeapFree(GetProcessHeap(), 0, sLinkPath);
sLinkPath = strdupW(pszFileName);
r = Load(stm);
ShellLink_UpdatePath(sPathRel, pszFileName, sWorkDir, &sPath);
bDirty = FALSE;
}
TRACE("-- returning hr %08x\n", r);
return r;
}
static BOOL StartLinkProcessor( LPCOLESTR szLink )
{
static const WCHAR szFormat[] = {
'w','i','n','e','m','e','n','u','b','u','i','l','d','e','r','.','e','x','e',
' ','-','w',' ','"','%','s','"',0 };
LONG len;
LPWSTR buffer;
STARTUPINFOW si;
PROCESS_INFORMATION pi;
BOOL ret;
len = sizeof(szFormat) + wcslen( szLink ) * sizeof(WCHAR);
buffer = (LPWSTR)HeapAlloc( GetProcessHeap(), 0, len );
if( !buffer )
return FALSE;
swprintf( buffer, szFormat, szLink );
TRACE("starting %s\n",debugstr_w(buffer));
memset(&si, 0, sizeof(si));
si.cb = sizeof(si);
ret = CreateProcessW( NULL, buffer, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi );
HeapFree( GetProcessHeap(), 0, buffer );
if (ret)
{
CloseHandle( pi.hProcess );
CloseHandle( pi.hThread );
}
return ret;
}
HRESULT WINAPI CShellLink::Save(LPCOLESTR pszFileName, BOOL fRemember)
{
HRESULT r;
CComPtr<IStream> stm;
TRACE("(%p)->(%s)\n", this, debugstr_w(pszFileName));
if (!pszFileName)
return E_FAIL;
r = SHCreateStreamOnFileW( pszFileName, STGM_READWRITE | STGM_CREATE | STGM_SHARE_EXCLUSIVE, &stm );
if( SUCCEEDED( r ) )
{
r = Save(stm, FALSE);
if( SUCCEEDED( r ) )
{
if ( sLinkPath )
{
HeapFree(GetProcessHeap(), 0, sLinkPath);
}
sLinkPath = (LPWSTR)HeapAlloc(GetProcessHeap(), 0, (wcslen(pszFileName)+1) * sizeof(WCHAR));
if ( sLinkPath )
{
wcscpy(sLinkPath, pszFileName);
}
StartLinkProcessor( pszFileName );
bDirty = FALSE;
}
else
{
DeleteFileW( pszFileName );
WARN("Failed to create shortcut %s\n", debugstr_w(pszFileName) );
}
}
return r;
}
HRESULT WINAPI CShellLink::SaveCompleted(LPCOLESTR pszFileName)
{
FIXME("(%p)->(%s)\n", this, debugstr_w(pszFileName));
return NOERROR;
}
HRESULT WINAPI CShellLink::GetCurFile(LPOLESTR *ppszFileName)
{
*ppszFileName = NULL;
if ( !sLinkPath)
{
/* IPersistFile::GetCurFile called before IPersistFile::Save */
return S_FALSE;
}
*ppszFileName = (LPOLESTR)CoTaskMemAlloc((wcslen(sLinkPath)+1) * sizeof(WCHAR));
if (!*ppszFileName)
{
/* out of memory */
return E_OUTOFMEMORY;
}
/* copy last saved filename */
wcscpy(*ppszFileName, sLinkPath);
return NOERROR;
}
/************************************************************************
* IPersistStream_IsDirty (IPersistStream)
*/
static HRESULT Stream_LoadString( IStream* stm, BOOL unicode, LPWSTR *pstr )
{
DWORD count;
USHORT len;
LPSTR temp;
LPWSTR str;
HRESULT r;
TRACE("%p\n", stm);
count = 0;
r = stm->Read(&len, sizeof(len), &count);
if ( FAILED (r) || ( count != sizeof(len) ) )
return E_FAIL;
if( unicode )
len *= sizeof (WCHAR);
TRACE("reading %d\n", len);
temp = (LPSTR)HeapAlloc(GetProcessHeap(), 0, len+sizeof(WCHAR));
if( !temp )
return E_OUTOFMEMORY;
count = 0;
r = stm->Read(temp, len, &count);
if( FAILED (r) || ( count != len ) )
{
HeapFree( GetProcessHeap(), 0, temp );
return E_FAIL;
}
TRACE("read %s\n", debugstr_an(temp,len));
/* convert to unicode if necessary */
if( !unicode )
{
count = MultiByteToWideChar( CP_ACP, 0, temp, len, NULL, 0 );
str = (LPWSTR)HeapAlloc( GetProcessHeap(), 0, (count+1)*sizeof (WCHAR) );
if( !str )
{
HeapFree( GetProcessHeap(), 0, temp );
return E_OUTOFMEMORY;
}
MultiByteToWideChar( CP_ACP, 0, temp, len, str, count );
HeapFree( GetProcessHeap(), 0, temp );
}
else
{
count /= 2;
str = (LPWSTR)temp;
}
str[count] = 0;
*pstr = str;
return S_OK;
}
static HRESULT Stream_ReadChunk( IStream* stm, LPVOID *data )
{
DWORD size;
ULONG count;
HRESULT r;
struct sized_chunk {
DWORD size;
unsigned char data[1];
} *chunk;
TRACE("%p\n",stm);
r = stm->Read(&size, sizeof(size), &count );
if( FAILED( r ) || count != sizeof(size) )
return E_FAIL;
chunk = (sized_chunk *)HeapAlloc( GetProcessHeap(), 0, size );
if( !chunk )
return E_OUTOFMEMORY;
chunk->size = size;
r = stm->Read(chunk->data, size - sizeof(size), &count );
if( FAILED( r ) || count != (size - sizeof(size)) )
{
HeapFree( GetProcessHeap(), 0, chunk );
return E_FAIL;
}
TRACE("Read %d bytes\n",chunk->size);
*data = chunk;
return S_OK;
}
static BOOL Stream_LoadVolume( LOCAL_VOLUME_INFO *vol, CShellLink::volume_info *volume )
{
const int label_sz = sizeof volume->label/sizeof volume->label[0];
LPSTR label;
int len;
volume->serial = vol->dwVolSerial;
volume->type = vol->dwType;
if( !vol->dwVolLabelOfs )
return FALSE;
if( vol->dwSize <= vol->dwVolLabelOfs )
return FALSE;
len = vol->dwSize - vol->dwVolLabelOfs;
label = (LPSTR) vol;
label += vol->dwVolLabelOfs;
MultiByteToWideChar( CP_ACP, 0, label, len, volume->label, label_sz-1);
return TRUE;
}
static LPWSTR Stream_LoadPath( LPCSTR p, DWORD maxlen )
{
unsigned int len = 0, wlen;
LPWSTR path;
while( p[len] && (len < maxlen) )
len++;
wlen = MultiByteToWideChar(CP_ACP, 0, p, len, NULL, 0);
path = (LPWSTR)HeapAlloc(GetProcessHeap(), 0, (wlen+1)*sizeof(WCHAR));
MultiByteToWideChar(CP_ACP, 0, p, len, path, wlen);
path[wlen] = 0;
return path;
}
static HRESULT Stream_LoadLocation( IStream *stm,
CShellLink::volume_info *volume, LPWSTR *path )
{
char *p = NULL;
LOCATION_INFO *loc;
HRESULT r;
DWORD n;
r = Stream_ReadChunk( stm, (LPVOID*) &p );
if( FAILED(r) )
return r;
loc = (LOCATION_INFO*) p;
if (loc->dwTotalSize < sizeof(LOCATION_INFO))
{
HeapFree( GetProcessHeap(), 0, p );
return E_FAIL;
}
/* if there's valid local volume information, load it */
if( loc->dwVolTableOfs &&
((loc->dwVolTableOfs + sizeof(LOCAL_VOLUME_INFO)) <= loc->dwTotalSize) )
{
LOCAL_VOLUME_INFO *volume_info;
volume_info = (LOCAL_VOLUME_INFO*) &p[loc->dwVolTableOfs];
Stream_LoadVolume( volume_info, volume );
}
/* if there's a local path, load it */
n = loc->dwLocalPathOfs;
if( n && (n < loc->dwTotalSize) )
*path = Stream_LoadPath( &p[n], loc->dwTotalSize - n );
TRACE("type %d serial %08x name %s path %s\n", volume->type,
volume->serial, debugstr_w(volume->label), debugstr_w(*path));
HeapFree( GetProcessHeap(), 0, p );
return S_OK;
}
/*
* The format of the advertised shortcut info seems to be:
*
* Offset Description
* ------ -----------
*
* 0 Length of the block (4 bytes, usually 0x314)
* 4 tag (dword)
* 8 string data in ASCII
* 8+0x104 string data in UNICODE
*
* In the original Win32 implementation the buffers are not initialized
* to zero, so data trailing the string is random garbage.
*/
static HRESULT Stream_LoadAdvertiseInfo( IStream* stm, LPWSTR *str )
{
DWORD size;
ULONG count;
HRESULT r;
EXP_DARWIN_LINK buffer;
TRACE("%p\n",stm);
r = stm->Read(&buffer.dbh.cbSize, sizeof (DWORD), &count );
if( FAILED( r ) )
return r;
/* make sure that we read the size of the structure even on error */
size = sizeof buffer - sizeof (DWORD);
if( buffer.dbh.cbSize != sizeof buffer )
{
ERR("Ooops. This structure is not as expected...\n");
return E_FAIL;
}
r = stm->Read(&buffer.dbh.dwSignature, size, &count );
if( FAILED( r ) )
return r;
if( count != size )
return E_FAIL;
TRACE("magic %08x string = %s\n", buffer.dbh.dwSignature, debugstr_w(buffer.szwDarwinID));
if( (buffer.dbh.dwSignature&0xffff0000) != 0xa0000000 )
{
ERR("Unknown magic number %08x in advertised shortcut\n", buffer.dbh.dwSignature);
return E_FAIL;
}
*str = (LPWSTR)HeapAlloc( GetProcessHeap(), 0,
(wcslen(buffer.szwDarwinID)+1) * sizeof(WCHAR) );
wcscpy( *str, buffer.szwDarwinID );
return S_OK;
}
/************************************************************************
* IPersistStream_Load (IPersistStream)
*/
HRESULT WINAPI CShellLink::Load(IStream *stm)
{
LINK_HEADER hdr;
ULONG dwBytesRead;
BOOL unicode;
HRESULT r;
DWORD zero;
TRACE("%p %p\n", this, stm);
if (!stm)
return STG_E_INVALIDPOINTER;
dwBytesRead = 0;
r = stm->Read(&hdr, sizeof(hdr), &dwBytesRead);
if (FAILED(r))
return r;
if (dwBytesRead != sizeof(hdr))
return E_FAIL;
if (hdr.dwSize != sizeof(hdr))
return E_FAIL;
if (!IsEqualIID(hdr.MagicGuid, CLSID_ShellLink))
return E_FAIL;
/* free all the old stuff */
ILFree(pPidl);
pPidl = NULL;
memset( &volume, 0, sizeof volume );
HeapFree(GetProcessHeap(), 0, sPath);
sPath = NULL;
HeapFree(GetProcessHeap(), 0, sDescription);
sDescription = NULL;
HeapFree(GetProcessHeap(), 0, sPathRel);
sPathRel = NULL;
HeapFree(GetProcessHeap(), 0, sWorkDir);
sWorkDir = NULL;
HeapFree(GetProcessHeap(), 0, sArgs);
sArgs = NULL;
HeapFree(GetProcessHeap(), 0, sIcoPath);
sIcoPath = NULL;
HeapFree(GetProcessHeap(), 0, sProduct);
sProduct = NULL;
HeapFree(GetProcessHeap(), 0, sComponent);
sComponent = NULL;
iShowCmd = hdr.fStartup;
wHotKey = (WORD)hdr.wHotKey;
iIcoNdx = hdr.nIcon;
FileTimeToSystemTime (&hdr.Time1, &time1);
FileTimeToSystemTime (&hdr.Time2, &time2);
FileTimeToSystemTime (&hdr.Time3, &time3);
if (TRACE_ON(shell))
{
WCHAR sTemp[MAX_PATH];
GetDateFormatW(LOCALE_USER_DEFAULT,DATE_SHORTDATE, &time1,
NULL, sTemp, sizeof(sTemp)/sizeof(*sTemp));
TRACE("-- time1: %s\n", debugstr_w(sTemp) );
GetDateFormatW(LOCALE_USER_DEFAULT,DATE_SHORTDATE, &time2,
NULL, sTemp, sizeof(sTemp)/sizeof(*sTemp));
TRACE("-- time2: %s\n", debugstr_w(sTemp) );
GetDateFormatW(LOCALE_USER_DEFAULT,DATE_SHORTDATE, &time3,
NULL, sTemp, sizeof(sTemp)/sizeof(*sTemp));
TRACE("-- time3: %s\n", debugstr_w(sTemp) );
}
/* load all the new stuff */
if( hdr.dwFlags & SLDF_HAS_ID_LIST )
{
r = ILLoadFromStream( stm, &pPidl );
if( FAILED( r ) )
return r;
}
pdump(pPidl);
/* load the location information */
if( hdr.dwFlags & SLDF_HAS_LINK_INFO )
r = Stream_LoadLocation( stm, &volume, &sPath );
if( FAILED( r ) )
goto end;
unicode = hdr.dwFlags & SLDF_UNICODE;
if( hdr.dwFlags & SLDF_HAS_NAME )
{
r = Stream_LoadString( stm, unicode, &sDescription );
TRACE("Description -> %s\n",debugstr_w(sDescription));
}
if( FAILED( r ) )
goto end;
if( hdr.dwFlags & SLDF_HAS_RELPATH )
{
r = Stream_LoadString( stm, unicode, &sPathRel );
TRACE("Relative Path-> %s\n",debugstr_w(sPathRel));
}
if( FAILED( r ) )
goto end;
if( hdr.dwFlags & SLDF_HAS_WORKINGDIR )
{
r = Stream_LoadString( stm, unicode, &sWorkDir );
TRACE("Working Dir -> %s\n",debugstr_w(sWorkDir));
}
if( FAILED( r ) )
goto end;
if( hdr.dwFlags & SLDF_HAS_ARGS )
{
r = Stream_LoadString( stm, unicode, &sArgs );
TRACE("Working Dir -> %s\n",debugstr_w(sArgs));
}
if( FAILED( r ) )
goto end;
if( hdr.dwFlags & SLDF_HAS_ICONLOCATION )
{
r = Stream_LoadString( stm, unicode, &sIcoPath );
TRACE("Icon file -> %s\n",debugstr_w(sIcoPath));
}
if( FAILED( r ) )
goto end;
#if (NTDDI_VERSION < NTDDI_LONGHORN)
if( hdr.dwFlags & SLDF_HAS_LOGO3ID )
{
r = Stream_LoadAdvertiseInfo( stm, &sProduct );
TRACE("Product -> %s\n",debugstr_w(sProduct));
}
if( FAILED( r ) )
goto end;
#endif
if( hdr.dwFlags & SLDF_HAS_DARWINID )
{
r = Stream_LoadAdvertiseInfo( stm, &sComponent );
TRACE("Component -> %s\n",debugstr_w(sComponent));
}
if( hdr.dwFlags & SLDF_RUNAS_USER )
{
bRunAs = TRUE;
}
else
{
bRunAs = FALSE;
}
if( FAILED( r ) )
goto end;
r = stm->Read(&zero, sizeof zero, &dwBytesRead);
if( FAILED( r ) || zero || dwBytesRead != sizeof zero )
ERR("Last word was not zero\n");
TRACE("OK\n");
pdump (pPidl);
return S_OK;
end:
return r;
}
/************************************************************************
* Stream_WriteString
*
* Helper function for IPersistStream_Save. Writes a unicode string
* with terminating nul byte to a stream, preceded by the its length.
*/
static HRESULT Stream_WriteString( IStream* stm, LPCWSTR str )
{
USHORT len = wcslen( str ) + 1;
DWORD count;
HRESULT r;
r = stm->Write(&len, sizeof(len), &count );
if( FAILED( r ) )
return r;
len *= sizeof(WCHAR);
r = stm->Write(str, len, &count );
if( FAILED( r ) )
return r;
return S_OK;
}
/************************************************************************
* Stream_WriteLocationInfo
*
* Writes the location info to a stream
*
* FIXME: One day we might want to write the network volume information
* and the final path.
* Figure out how Windows deals with unicode paths here.
*/
static HRESULT Stream_WriteLocationInfo( IStream* stm, LPCWSTR path,
CShellLink::volume_info *volume )
{
DWORD total_size, path_size, volume_info_size, label_size, final_path_size;
LOCAL_VOLUME_INFO *vol;
LOCATION_INFO *loc;
LPSTR szLabel, szPath, szFinalPath;
ULONG count = 0;
HRESULT hr;
TRACE("%p %s %p\n", stm, debugstr_w(path), volume);
/* figure out the size of everything */
label_size = WideCharToMultiByte( CP_ACP, 0, volume->label, -1,
NULL, 0, NULL, NULL );
path_size = WideCharToMultiByte( CP_ACP, 0, path, -1,
NULL, 0, NULL, NULL );
volume_info_size = sizeof *vol + label_size;
final_path_size = 1;
total_size = sizeof *loc + volume_info_size + path_size + final_path_size;
/* create pointers to everything */
loc = (LOCATION_INFO *)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, total_size);
vol = (LOCAL_VOLUME_INFO*) &loc[1];
szLabel = (LPSTR) &vol[1];
szPath = &szLabel[label_size];
szFinalPath = &szPath[path_size];
/* fill in the location information header */
loc->dwTotalSize = total_size;
loc->dwHeaderSize = sizeof (*loc);
loc->dwFlags = 1;
loc->dwVolTableOfs = sizeof (*loc);
loc->dwLocalPathOfs = sizeof (*loc) + volume_info_size;
loc->dwNetworkVolTableOfs = 0;
loc->dwFinalPathOfs = sizeof (*loc) + volume_info_size + path_size;
/* fill in the volume information */
vol->dwSize = volume_info_size;
vol->dwType = volume->type;
vol->dwVolSerial = volume->serial;
vol->dwVolLabelOfs = sizeof (*vol);
/* copy in the strings */
WideCharToMultiByte( CP_ACP, 0, volume->label, -1,
szLabel, label_size, NULL, NULL );
WideCharToMultiByte( CP_ACP, 0, path, -1,
szPath, path_size, NULL, NULL );
szFinalPath[0] = 0;
hr = stm->Write(loc, total_size, &count );
HeapFree(GetProcessHeap(), 0, loc);
return hr;
}
static EXP_DARWIN_LINK* shelllink_build_darwinid( LPCWSTR string, DWORD magic )
{
EXP_DARWIN_LINK *buffer;
buffer = (EXP_DARWIN_LINK *)LocalAlloc( LMEM_ZEROINIT, sizeof *buffer );
buffer->dbh.cbSize = sizeof *buffer;
buffer->dbh.dwSignature = magic;
lstrcpynW( buffer->szwDarwinID, string, MAX_PATH );
WideCharToMultiByte(CP_ACP, 0, string, -1, buffer->szDarwinID, MAX_PATH, NULL, NULL );
return buffer;
}
static HRESULT Stream_WriteAdvertiseInfo( IStream* stm, LPCWSTR string, DWORD magic )
{
EXP_DARWIN_LINK *buffer;
ULONG count;
TRACE("%p\n",stm);
buffer = shelllink_build_darwinid( string, magic );
return stm->Write(buffer, buffer->dbh.cbSize, &count );
}
/************************************************************************
* IPersistStream_Save (IPersistStream)
*
* FIXME: makes assumptions about byte order
*/
HRESULT WINAPI CShellLink::Save(IStream *stm, BOOL fClearDirty)
{
LINK_HEADER header;
ULONG count;
DWORD zero;
HRESULT r;
TRACE("%p %p %x\n", this, stm, fClearDirty);
memset(&header, 0, sizeof(header));
header.dwSize = sizeof(header);
header.fStartup = iShowCmd;
header.MagicGuid = CLSID_ShellLink;
header.wHotKey = wHotKey;
header.nIcon = iIcoNdx;
header.dwFlags = SLDF_UNICODE; /* strings are in unicode */
if( pPidl )
header.dwFlags |= SLDF_HAS_ID_LIST;
if( sPath )
header.dwFlags |= SLDF_HAS_LINK_INFO;
if( sDescription )
header.dwFlags |= SLDF_HAS_NAME;
if( sWorkDir )
header.dwFlags |= SLDF_HAS_WORKINGDIR;
if( sArgs )
header.dwFlags |= SLDF_HAS_ARGS;
if( sIcoPath )
header.dwFlags |= SLDF_HAS_ICONLOCATION;
#if (NTDDI_VERSION < NTDDI_LONGHORN)
if( sProduct )
header.dwFlags |= SLDF_HAS_LOGO3ID;
#endif
if( sComponent )
header.dwFlags |= SLDF_HAS_DARWINID;
if( bRunAs )
header.dwFlags |= SLDF_RUNAS_USER;
SystemTimeToFileTime ( &time1, &header.Time1 );
SystemTimeToFileTime ( &time2, &header.Time2 );
SystemTimeToFileTime ( &time3, &header.Time3 );
/* write the Shortcut header */
r = stm->Write(&header, sizeof(header), &count );
if( FAILED( r ) )
{
ERR("Write failed at %d\n",__LINE__);
return r;
}
TRACE("Writing pidl\n");
/* write the PIDL to the shortcut */
if( pPidl )
{
r = ILSaveToStream( stm, pPidl );
if( FAILED( r ) )
{
ERR("Failed to write PIDL at %d\n",__LINE__);
return r;
}
}
if( sPath )
Stream_WriteLocationInfo( stm, sPath, &volume );
if( sDescription )
r = Stream_WriteString( stm, sDescription );
if( sPathRel )
r = Stream_WriteString( stm, sPathRel );
if( sWorkDir )
r = Stream_WriteString( stm, sWorkDir );
if( sArgs )
r = Stream_WriteString( stm, sArgs );
if( sIcoPath )
r = Stream_WriteString( stm, sIcoPath );
if( sProduct )
r = Stream_WriteAdvertiseInfo( stm, sProduct, EXP_SZ_ICON_SIG );
if( sComponent )
r = Stream_WriteAdvertiseInfo( stm, sComponent, EXP_DARWIN_ID_SIG );
/* the last field is a single zero dword */
zero = 0;
r = stm->Write(&zero, sizeof zero, &count );
return S_OK;
}
/************************************************************************
* IPersistStream_GetSizeMax (IPersistStream)
*/
HRESULT WINAPI CShellLink::GetSizeMax(ULARGE_INTEGER *pcbSize)
{
TRACE("(%p)\n", this);
return E_NOTIMPL;
}
static BOOL SHELL_ExistsFileW(LPCWSTR path)
{
if (INVALID_FILE_ATTRIBUTES == GetFileAttributesW(path))
return FALSE;
return TRUE;
}
/**************************************************************************
* ShellLink_UpdatePath
* update absolute path in sPath using relative path in sPathRel
*/
static HRESULT ShellLink_UpdatePath(LPCWSTR sPathRel, LPCWSTR path, LPCWSTR sWorkDir, LPWSTR* psPath)
{
if (!path || !psPath)
return E_INVALIDARG;
if (!*psPath && sPathRel) {
WCHAR buffer[2*MAX_PATH], abs_path[2*MAX_PATH];
LPWSTR final = NULL;
/* first try if [directory of link file] + [relative path] finds an existing file */
GetFullPathNameW( path, MAX_PATH*2, buffer, &final );
if( !final )
final = buffer;
wcscpy(final, sPathRel);
*abs_path = '\0';
if (SHELL_ExistsFileW(buffer)) {
if (!GetFullPathNameW(buffer, MAX_PATH, abs_path, &final))
wcscpy(abs_path, buffer);
} else {
/* try if [working directory] + [relative path] finds an existing file */
if (sWorkDir) {
wcscpy(buffer, sWorkDir);
wcscpy(PathAddBackslashW(buffer), sPathRel);
if (SHELL_ExistsFileW(buffer))
if (!GetFullPathNameW(buffer, MAX_PATH, abs_path, &final))
wcscpy(abs_path, buffer);
}
}
/* FIXME: This is even not enough - not all shell links can be resolved using this algorithm. */
if (!*abs_path)
wcscpy(abs_path, sPathRel);
*psPath = (LPWSTR)HeapAlloc(GetProcessHeap(), 0, (wcslen(abs_path)+1)*sizeof(WCHAR));
if (!*psPath)
return E_OUTOFMEMORY;
wcscpy(*psPath, abs_path);
}
return S_OK;
}
HRESULT WINAPI CShellLink::GetPath(LPSTR pszFile, INT cchMaxPath, WIN32_FIND_DATAA *pfd, DWORD fFlags)
{
TRACE("(%p)->(pfile=%p len=%u find_data=%p flags=%u)(%s)\n",
this, pszFile, cchMaxPath, pfd, fFlags, debugstr_w(sPath));
if (sComponent || sProduct)
return S_FALSE;
if (cchMaxPath)
pszFile[0] = 0;
if (sPath)
WideCharToMultiByte( CP_ACP, 0, sPath, -1,
pszFile, cchMaxPath, NULL, NULL);
if (pfd) FIXME("(%p): WIN32_FIND_DATA is not yet filled.\n", this);
return S_OK;
}
HRESULT WINAPI CShellLink::GetIDList(LPITEMIDLIST * ppidl)
{
TRACE("(%p)->(ppidl=%p)\n",this, ppidl);
if (!pPidl)
{
*ppidl = NULL;
return S_FALSE;
}
*ppidl = ILClone(pPidl);
return S_OK;
}
HRESULT WINAPI CShellLink::SetIDList(LPCITEMIDLIST pidl)
{
TRACE("(%p)->(pidl=%p)\n",this, pidl);
if( pPidl )
ILFree( pPidl );
pPidl = ILClone( pidl );
if( !pPidl )
return E_FAIL;
bDirty = TRUE;
return S_OK;
}
HRESULT WINAPI CShellLink::GetDescription(LPSTR pszName,INT cchMaxName)
{
TRACE("(%p)->(%p len=%u)\n",this, pszName, cchMaxName);
if( cchMaxName )
pszName[0] = 0;
if( sDescription )
WideCharToMultiByte( CP_ACP, 0, sDescription, -1,
pszName, cchMaxName, NULL, NULL);
return S_OK;
}
HRESULT WINAPI CShellLink::SetDescription(LPCSTR pszName)
{
TRACE("(%p)->(pName=%s)\n", this, pszName);
HeapFree(GetProcessHeap(), 0, sDescription);
sDescription = NULL;
if ( pszName ) {
sDescription = HEAP_strdupAtoW( GetProcessHeap(), 0, pszName);
if ( !sDescription )
return E_OUTOFMEMORY;
}
bDirty = TRUE;
return S_OK;
}
HRESULT WINAPI CShellLink::GetWorkingDirectory(LPSTR pszDir,INT cchMaxPath)
{
TRACE("(%p)->(%p len=%u)\n", this, pszDir, cchMaxPath);
if( cchMaxPath )
pszDir[0] = 0;
if( sWorkDir )
WideCharToMultiByte( CP_ACP, 0, sWorkDir, -1,
pszDir, cchMaxPath, NULL, NULL);
return S_OK;
}
HRESULT WINAPI CShellLink::SetWorkingDirectory(LPCSTR pszDir)
{
TRACE("(%p)->(dir=%s)\n",this, pszDir);
HeapFree(GetProcessHeap(), 0, sWorkDir);
sWorkDir = NULL;
if ( pszDir ) {
sWorkDir = HEAP_strdupAtoW( GetProcessHeap(), 0, pszDir);
if ( !sWorkDir )
return E_OUTOFMEMORY;
}
bDirty = TRUE;
return S_OK;
}
HRESULT WINAPI CShellLink::GetArguments(LPSTR pszArgs,INT cchMaxPath)
{
TRACE("(%p)->(%p len=%u)\n", this, pszArgs, cchMaxPath);
if( cchMaxPath )
pszArgs[0] = 0;
if( sArgs )
WideCharToMultiByte( CP_ACP, 0, sArgs, -1,
pszArgs, cchMaxPath, NULL, NULL);
return S_OK;
}
HRESULT WINAPI CShellLink::SetArguments(LPCSTR pszArgs)
{
TRACE("(%p)->(args=%s)\n",this, pszArgs);
HeapFree(GetProcessHeap(), 0, sArgs);
sArgs = NULL;
if ( pszArgs ) {
sArgs = HEAP_strdupAtoW( GetProcessHeap(), 0, pszArgs);
if( !sArgs )
return E_OUTOFMEMORY;
}
bDirty = TRUE;
return S_OK;
}
HRESULT WINAPI CShellLink::GetHotkey(WORD *pwHotkey)
{
TRACE("(%p)->(%p)(0x%08x)\n",this, pwHotkey, wHotKey);
*pwHotkey = wHotKey;
return S_OK;
}
HRESULT WINAPI CShellLink::SetHotkey(WORD wHotkey)
{
TRACE("(%p)->(hotkey=%x)\n",this, wHotkey);
wHotKey = wHotkey;
bDirty = TRUE;
return S_OK;
}
HRESULT WINAPI CShellLink::GetShowCmd(INT *piShowCmd)
{
TRACE("(%p)->(%p)\n",this, piShowCmd);
*piShowCmd = iShowCmd;
return S_OK;
}
HRESULT WINAPI CShellLink::SetShowCmd(INT iShowCmd)
{
TRACE("(%p) %d\n",this, iShowCmd);
this->iShowCmd = iShowCmd;
bDirty = TRUE;
return NOERROR;
}
static HRESULT SHELL_PidlGeticonLocationA(IShellFolder* psf, LPCITEMIDLIST pidl,
LPSTR pszIconPath, int cchIconPath, int* piIcon)
{
LPCITEMIDLIST pidlLast;
HRESULT hr = SHBindToParent(pidl, IID_IShellFolder, (LPVOID*)&psf, &pidlLast);
if (SUCCEEDED(hr)) {
CComPtr<IExtractIconA> pei;
hr = psf->GetUIObjectOf(0, 1, &pidlLast, IID_IExtractIconA, NULL, (LPVOID*)&pei);
if (SUCCEEDED(hr)) {
hr = pei->GetIconLocation(0, pszIconPath, MAX_PATH, piIcon, NULL);
}
psf->Release();
}
return hr;
}
HRESULT WINAPI CShellLink::GetIconLocation(LPSTR pszIconPath,INT cchIconPath,INT *piIcon)
{
TRACE("(%p)->(%p len=%u iicon=%p)\n", this, pszIconPath, cchIconPath, piIcon);
pszIconPath[0] = 0;
*piIcon = iIcoNdx;
if (sIcoPath)
{
WideCharToMultiByte(CP_ACP, 0, sIcoPath, -1, pszIconPath, cchIconPath, NULL, NULL);
return S_OK;
}
if (pPidl || sPath)
{
CComPtr<IShellFolder> pdsk;
HRESULT hr = SHGetDesktopFolder(&pdsk);
if (SUCCEEDED(hr))
{
/* first look for an icon using the PIDL (if present) */
if (pPidl)
hr = SHELL_PidlGeticonLocationA(pdsk, pPidl, pszIconPath, cchIconPath, piIcon);
else
hr = E_FAIL;
/* if we couldn't find an icon yet, look for it using the file system path */
if (FAILED(hr) && sPath)
{
LPITEMIDLIST pidl;
hr = pdsk->ParseDisplayName(0, NULL, sPath, NULL, &pidl, NULL);
if (SUCCEEDED(hr))
{
hr = SHELL_PidlGeticonLocationA(pdsk, pidl, pszIconPath, cchIconPath, piIcon);
SHFree(pidl);
}
}
}
return hr;
}
return S_OK;
}
HRESULT WINAPI CShellLink::SetIconLocation(LPCSTR pszIconPath,INT iIcon)
{
TRACE("(%p)->(path=%s iicon=%u)\n",this, pszIconPath, iIcon);
HeapFree(GetProcessHeap(), 0, sIcoPath);
sIcoPath = NULL;
if ( pszIconPath )
{
sIcoPath = HEAP_strdupAtoW(GetProcessHeap(), 0, pszIconPath);
if ( !sIcoPath )
return E_OUTOFMEMORY;
}
iIcoNdx = iIcon;
bDirty = TRUE;
return S_OK;
}
HRESULT WINAPI CShellLink::SetRelativePath(LPCSTR pszPathRel, DWORD dwReserved)
{
TRACE("(%p)->(path=%s %x)\n",this, pszPathRel, dwReserved);
HeapFree(GetProcessHeap(), 0, sPathRel);
sPathRel = NULL;
if ( pszPathRel )
{
sPathRel = HEAP_strdupAtoW(GetProcessHeap(), 0, pszPathRel);
bDirty = TRUE;
}
return ShellLink_UpdatePath(sPathRel, sPath, sWorkDir, &sPath);
}
HRESULT WINAPI CShellLink::Resolve(HWND hwnd, DWORD fFlags)
{
HRESULT hr = S_OK;
BOOL bSuccess;
TRACE("(%p)->(hwnd=%p flags=%x)\n",this, hwnd, fFlags);
/*FIXME: use IResolveShellLink interface */
if (!sPath && pPidl)
{
WCHAR buffer[MAX_PATH];
bSuccess = SHGetPathFromIDListW(pPidl, buffer);
if (bSuccess && *buffer)
{
sPath = (LPWSTR)HeapAlloc(GetProcessHeap(), 0, (wcslen(buffer)+1)*sizeof(WCHAR));
if (!sPath)
return E_OUTOFMEMORY;
wcscpy(sPath, buffer);
bDirty = TRUE;
}
else
hr = S_OK; /* don't report an error occurred while just caching information */
}
if (!sIcoPath && sPath)
{
sIcoPath = (LPWSTR)HeapAlloc(GetProcessHeap(), 0, (wcslen(sPath)+1)*sizeof(WCHAR));
if (!sIcoPath)
return E_OUTOFMEMORY;
wcscpy(sIcoPath, sPath);
iIcoNdx = 0;
bDirty = TRUE;
}
return hr;
}
HRESULT WINAPI CShellLink::SetPath(LPCSTR pszFile)
{
HRESULT r;
LPWSTR str;
TRACE("(%p)->(path=%s)\n",this, pszFile);
if (pszFile == NULL)
return E_INVALIDARG;
str = HEAP_strdupAtoW(GetProcessHeap(), 0, pszFile);
if (!str)
return E_OUTOFMEMORY;
r = SetPath(str);
HeapFree( GetProcessHeap(), 0, str );
return r;
}
HRESULT WINAPI CShellLink::GetPath(LPWSTR pszFile,INT cchMaxPath, WIN32_FIND_DATAW *pfd, DWORD fFlags)
{
TRACE("(%p)->(pfile=%p len=%u find_data=%p flags=%u)(%s)\n",
this, pszFile, cchMaxPath, pfd, fFlags, debugstr_w(sPath));
if (sComponent || sProduct)
return S_FALSE;
if (cchMaxPath)
pszFile[0] = 0;
if (sPath)
lstrcpynW( pszFile, sPath, cchMaxPath );
if (pfd) FIXME("(%p): WIN32_FIND_DATA is not yet filled.\n", this);
return S_OK;
}
HRESULT WINAPI CShellLink::GetDescription(LPWSTR pszName,INT cchMaxName)
{
TRACE("(%p)->(%p len=%u)\n",this, pszName, cchMaxName);
pszName[0] = 0;
if (sDescription)
lstrcpynW( pszName, sDescription, cchMaxName );
return S_OK;
}
HRESULT WINAPI CShellLink::SetDescription(LPCWSTR pszName)
{
TRACE("(%p)->(desc=%s)\n",this, debugstr_w(pszName));
HeapFree(GetProcessHeap(), 0, sDescription);
sDescription = (LPWSTR)HeapAlloc( GetProcessHeap(), 0,
(wcslen( pszName )+1)*sizeof(WCHAR) );
if ( !sDescription )
return E_OUTOFMEMORY;
wcscpy( sDescription, pszName );
bDirty = TRUE;
return S_OK;
}
HRESULT WINAPI CShellLink::GetWorkingDirectory(LPWSTR pszDir,INT cchMaxPath)
{
TRACE("(%p)->(%p len %u)\n", this, pszDir, cchMaxPath);
if( cchMaxPath )
pszDir[0] = 0;
if( sWorkDir )
lstrcpynW( pszDir, sWorkDir, cchMaxPath );
return S_OK;
}
HRESULT WINAPI CShellLink::SetWorkingDirectory(LPCWSTR pszDir)
{
TRACE("(%p)->(dir=%s)\n",this, debugstr_w(pszDir));
HeapFree(GetProcessHeap(), 0, sWorkDir);
sWorkDir = (LPWSTR)HeapAlloc( GetProcessHeap(), 0,
(wcslen( pszDir )+1)*sizeof (WCHAR) );
if ( !sWorkDir )
return E_OUTOFMEMORY;
wcscpy( sWorkDir, pszDir );
bDirty = TRUE;
return S_OK;
}
HRESULT WINAPI CShellLink::GetArguments(LPWSTR pszArgs,INT cchMaxPath)
{
TRACE("(%p)->(%p len=%u)\n", this, pszArgs, cchMaxPath);
if( cchMaxPath )
pszArgs[0] = 0;
if( sArgs )
lstrcpynW( pszArgs, sArgs, cchMaxPath );
return NOERROR;
}
HRESULT WINAPI CShellLink::SetArguments(LPCWSTR pszArgs)
{
TRACE("(%p)->(args=%s)\n",this, debugstr_w(pszArgs));
HeapFree(GetProcessHeap(), 0, sArgs);
sArgs = (LPWSTR)HeapAlloc( GetProcessHeap(), 0,
(wcslen( pszArgs )+1)*sizeof (WCHAR) );
if ( !sArgs )
return E_OUTOFMEMORY;
wcscpy( sArgs, pszArgs );
bDirty = TRUE;
return S_OK;
}
static HRESULT SHELL_PidlGeticonLocationW(IShellFolder* psf, LPCITEMIDLIST pidl,
LPWSTR pszIconPath, int cchIconPath, int* piIcon)
{
LPCITEMIDLIST pidlLast;
UINT wFlags;
HRESULT hr = SHBindToParent(pidl, IID_IShellFolder, (LPVOID*)&psf, &pidlLast);
if (SUCCEEDED(hr)) {
CComPtr<IExtractIconW> pei;
hr = psf->GetUIObjectOf(0, 1, &pidlLast, IID_IExtractIconW, NULL, (LPVOID*)&pei);
if (SUCCEEDED(hr)) {
hr = pei->GetIconLocation(0, pszIconPath, MAX_PATH, piIcon, &wFlags);
}
psf->Release();
}
return hr;
}
HRESULT WINAPI CShellLink::GetIconLocation(LPWSTR pszIconPath,INT cchIconPath,INT *piIcon)
{
TRACE("(%p)->(%p len=%u iicon=%p)\n", this, pszIconPath, cchIconPath, piIcon);
pszIconPath[0] = 0;
*piIcon = iIcoNdx;
if (sIcoPath)
{
lstrcpynW(pszIconPath, sIcoPath, cchIconPath);
return S_OK;
}
if (pPidl || sPath)
{
CComPtr<IShellFolder> pdsk;
HRESULT hr = SHGetDesktopFolder(&pdsk);
if (SUCCEEDED(hr))
{
/* first look for an icon using the PIDL (if present) */
if (pPidl)
hr = SHELL_PidlGeticonLocationW(pdsk, pPidl, pszIconPath, cchIconPath, piIcon);
else
hr = E_FAIL;
/* if we couldn't find an icon yet, look for it using the file system path */
if (FAILED(hr) && sPath)
{
LPITEMIDLIST pidl;
hr = pdsk->ParseDisplayName(0, NULL, sPath, NULL, &pidl, NULL);
if (SUCCEEDED(hr))
{
hr = SHELL_PidlGeticonLocationW(pdsk, pidl, pszIconPath, cchIconPath, piIcon);
SHFree(pidl);
}
}
}
return hr;
}
return S_OK;
}
HRESULT WINAPI CShellLink::SetIconLocation(LPCWSTR pszIconPath,INT iIcon)
{
TRACE("(%p)->(path=%s iicon=%u)\n",this, debugstr_w(pszIconPath), iIcon);
HeapFree(GetProcessHeap(), 0, sIcoPath);
sIcoPath = (LPWSTR)HeapAlloc( GetProcessHeap(), 0,
(wcslen( pszIconPath )+1)*sizeof (WCHAR) );
if ( !sIcoPath )
return E_OUTOFMEMORY;
wcscpy( sIcoPath, pszIconPath );
iIcoNdx = iIcon;
bDirty = TRUE;
return S_OK;
}
HRESULT WINAPI CShellLink::SetRelativePath(LPCWSTR pszPathRel, DWORD dwReserved)
{
TRACE("(%p)->(path=%s %x)\n",this, debugstr_w(pszPathRel), dwReserved);
HeapFree(GetProcessHeap(), 0, sPathRel);
sPathRel = (LPWSTR)HeapAlloc( GetProcessHeap(), 0,
(wcslen( pszPathRel )+1) * sizeof (WCHAR) );
if ( !sPathRel )
return E_OUTOFMEMORY;
wcscpy( sPathRel, pszPathRel );
bDirty = TRUE;
return ShellLink_UpdatePath(sPathRel, sPath, sWorkDir, &sPath);
}
LPWSTR CShellLink::ShellLink_GetAdvertisedArg(LPCWSTR str)
{
LPWSTR ret;
LPCWSTR p;
DWORD len;
if( !str )
return NULL;
p = wcschr( str, ':' );
if( !p )
return NULL;
len = p - str;
ret = (LPWSTR)HeapAlloc( GetProcessHeap(), 0, sizeof(WCHAR)*(len+1));
if( !ret )
return ret;
memcpy( ret, str, sizeof(WCHAR)*len );
ret[len] = 0;
return ret;
}
HRESULT CShellLink::ShellLink_SetAdvertiseInfo(LPCWSTR str)
{
LPCWSTR szComponent = NULL, szProduct = NULL, p;
WCHAR szGuid[39];
HRESULT r;
GUID guid;
int len;
while( str[0] )
{
/* each segment must start with two colons */
if( str[0] != ':' || str[1] != ':' )
return E_FAIL;
/* the last segment is just two colons */
if( !str[2] )
break;
str += 2;
/* there must be a colon straight after a guid */
p = wcschr( str, ':' );
if( !p )
return E_FAIL;
len = p - str;
if( len != 38 )
return E_FAIL;
/* get the guid, and check it's validly formatted */
memcpy( szGuid, str, sizeof(WCHAR)*len );
szGuid[len] = 0;
r = CLSIDFromString( szGuid, &guid );
if( r != S_OK )
return r;
str = p + 1;
/* match it up to a guid that we care about */
if( IsEqualGUID( guid, SHELL32_AdvtShortcutComponent ) && !szComponent )
szComponent = str;
else if( IsEqualGUID(guid, SHELL32_AdvtShortcutProduct ) && !szProduct )
szProduct = str;
else
return E_FAIL;
/* skip to the next field */
str = wcschr( str, ':' );
if( !str )
return E_FAIL;
}
/* we have to have a component for an advertised shortcut */
if( !szComponent )
return E_FAIL;
sComponent = ShellLink_GetAdvertisedArg( szComponent );
sProduct = ShellLink_GetAdvertisedArg( szProduct );
TRACE("Component = %s\n", debugstr_w(sComponent));
TRACE("Product = %s\n", debugstr_w(sProduct));
return S_OK;
}
static BOOL ShellLink_GetVolumeInfo(LPCWSTR path, CShellLink::volume_info *volume)
{
const int label_sz = sizeof volume->label/sizeof volume->label[0];
WCHAR drive[4] = { path[0], ':', '\\', 0 };
BOOL r;
volume->type = GetDriveTypeW(drive);
r = GetVolumeInformationW(drive, volume->label, label_sz, &volume->serial, NULL, NULL, NULL, 0);
TRACE("r = %d type %d serial %08x name %s\n", r,
volume->type, volume->serial, debugstr_w(volume->label));
return r;
}
HRESULT WINAPI CShellLink::SetPath(LPCWSTR pszFile)
{
WCHAR buffer[MAX_PATH];
LPWSTR fname, unquoted = NULL;
HRESULT hr = S_OK;
UINT len;
TRACE("(%p)->(path=%s)\n",this, debugstr_w(pszFile));
if (!pszFile) return E_INVALIDARG;
/* quotes at the ends of the string are stripped */
len = wcslen(pszFile);
if (pszFile[0] == '"' && pszFile[len-1] == '"')
{
unquoted = strdupW(pszFile);
PathUnquoteSpacesW(unquoted);
pszFile = unquoted;
}
/* any other quote marks are invalid */
if (wcschr(pszFile, '"'))
{
HeapFree(GetProcessHeap(), 0, unquoted);
return S_FALSE;
}
HeapFree(GetProcessHeap(), 0, sPath);
sPath = NULL;
HeapFree(GetProcessHeap(), 0, sComponent);
sComponent = NULL;
if (pPidl)
ILFree(pPidl);
pPidl = NULL;
if (S_OK != ShellLink_SetAdvertiseInfo(pszFile ))
{
if (*pszFile == '\0')
*buffer = '\0';
else if (!GetFullPathNameW(pszFile, MAX_PATH, buffer, &fname))
return E_FAIL;
else if(!PathFileExistsW(buffer) &&
!SearchPathW(NULL, pszFile, NULL, MAX_PATH, buffer, NULL))
hr = S_FALSE;
pPidl = SHSimpleIDListFromPathW(pszFile);
ShellLink_GetVolumeInfo(buffer, &volume);
sPath = (LPWSTR)HeapAlloc( GetProcessHeap(), 0,
(wcslen( buffer )+1) * sizeof (WCHAR) );
if (!sPath)
return E_OUTOFMEMORY;
wcscpy(sPath, buffer);
}
bDirty = TRUE;
HeapFree(GetProcessHeap(), 0, unquoted);
return hr;
}
HRESULT WINAPI CShellLink::AddDataBlock(void* pDataBlock )
{
FIXME("\n");
return E_NOTIMPL;
}
HRESULT WINAPI CShellLink::CopyDataBlock(DWORD dwSig, void** ppDataBlock )
{
LPVOID block = NULL;
HRESULT r = E_FAIL;
TRACE("%p %08x %p\n", this, dwSig, ppDataBlock );
switch (dwSig)
{
case EXP_DARWIN_ID_SIG:
if (!sComponent)
break;
block = shelllink_build_darwinid( sComponent, dwSig );
r = S_OK;
break;
case EXP_SZ_LINK_SIG:
case NT_CONSOLE_PROPS_SIG:
case NT_FE_CONSOLE_PROPS_SIG:
case EXP_SPECIAL_FOLDER_SIG:
case EXP_SZ_ICON_SIG:
FIXME("valid but unhandled datablock %08x\n", dwSig);
break;
default:
ERR("unknown datablock %08x\n", dwSig);
}
*ppDataBlock = block;
return r;
}
HRESULT WINAPI CShellLink::RemoveDataBlock(DWORD dwSig )
{
FIXME("\n");
return E_NOTIMPL;
}
HRESULT WINAPI CShellLink::GetFlags(DWORD* pdwFlags )
{
DWORD flags = 0;
FIXME("%p %p\n", this, pdwFlags );
/* FIXME: add more */
if (sArgs)
flags |= SLDF_HAS_ARGS;
if (sComponent)
flags |= SLDF_HAS_DARWINID;
if (sIcoPath)
flags |= SLDF_HAS_ICONLOCATION;
#if (NTDDI_VERSION < NTDDI_LONGHORN)
if (sProduct)
flags |= SLDF_HAS_LOGO3ID;
#endif
if (pPidl)
flags |= SLDF_HAS_ID_LIST;
*pdwFlags = flags;
return S_OK;
}
HRESULT WINAPI CShellLink::SetFlags(DWORD dwFlags )
{
FIXME("\n");
return E_NOTIMPL;
}
/**************************************************************************
* CShellLink implementation of IShellExtInit::Initialize()
*
* Loads the shelllink from the dataobject the shell is pointing to.
*/
HRESULT WINAPI CShellLink::Initialize(LPCITEMIDLIST pidlFolder, IDataObject *pdtobj, HKEY hkeyProgID )
{
FORMATETC format;
STGMEDIUM stgm;
UINT count;
HRESULT r = E_FAIL;
TRACE("%p %p %p %p\n", this, pidlFolder, pdtobj, hkeyProgID );
if( !pdtobj )
return r;
format.cfFormat = CF_HDROP;
format.ptd = NULL;
format.dwAspect = DVASPECT_CONTENT;
format.lindex = -1;
format.tymed = TYMED_HGLOBAL;
if( FAILED(pdtobj->GetData(&format, &stgm ) ) )
return r;
count = DragQueryFileW((HDROP)stgm.hGlobal, -1, NULL, 0 );
if( count == 1 )
{
LPWSTR path;
count = DragQueryFileW((HDROP)stgm.hGlobal, 0, NULL, 0 );
count++;
path = (LPWSTR)HeapAlloc( GetProcessHeap(), 0, count*sizeof(WCHAR) );
if( path )
{
count = DragQueryFileW((HDROP)stgm.hGlobal, 0, path, count );
r = Load(path, 0 );
HeapFree(GetProcessHeap(), 0, path );
}
}
ReleaseStgMedium(&stgm );
return r;
}
HRESULT WINAPI CShellLink::QueryContextMenu(HMENU hMenu, UINT indexMenu, UINT idCmdFirst, UINT idCmdLast, UINT uFlags)
{
WCHAR wszOpen[20];
MENUITEMINFOW mii;
int id = 1;
TRACE("%p %p %u %u %u %u\n", this,
hMenu, indexMenu, idCmdFirst, idCmdLast, uFlags );
if (!hMenu)
return E_INVALIDARG;
if (!LoadStringW(shell32_hInstance, IDS_OPEN_VERB, wszOpen, sizeof(wszOpen)/sizeof(WCHAR)))
wszOpen[0] = L'\0';
memset( &mii, 0, sizeof(mii) );
mii.cbSize = sizeof (mii);
mii.fMask = MIIM_TYPE | MIIM_ID | MIIM_STATE;
mii.dwTypeData = wszOpen;
mii.cch = wcslen(mii.dwTypeData);
mii.wID = idCmdFirst + id++;
mii.fState = MFS_DEFAULT | MFS_ENABLED;
mii.fType = MFT_STRING;
if (!InsertMenuItemW(hMenu, indexMenu, TRUE, &mii))
return E_FAIL;
iIdOpen = 1;
return MAKE_HRESULT(SEVERITY_SUCCESS, 0, id);
}
static LPWSTR
shelllink_get_msi_component_path( LPWSTR component )
{
LPWSTR path;
DWORD r, sz = 0;
r = CommandLineFromMsiDescriptor( component, NULL, &sz );
if (r != ERROR_SUCCESS)
return NULL;
sz++;
path = (LPWSTR)HeapAlloc( GetProcessHeap(), 0, sz*sizeof(WCHAR) );
r = CommandLineFromMsiDescriptor( component, path, &sz );
if (r != ERROR_SUCCESS)
{
HeapFree( GetProcessHeap(), 0, path );
path = NULL;
}
TRACE("returning %s\n", debugstr_w( path ) );
return path;
}
HRESULT WINAPI CShellLink::InvokeCommand(LPCMINVOKECOMMANDINFO lpici)
{
static const WCHAR szOpen[] = { 'o','p','e','n',0 };
static const WCHAR szCplOpen[] = { 'c','p','l','o','p','e','n',0 };
SHELLEXECUTEINFOW sei;
HWND hwnd = NULL; /* FIXME: get using interface set from IObjectWithSite */
LPWSTR args = NULL;
LPWSTR path = NULL;
HRESULT r;
TRACE("%p %p\n", this, lpici );
if ( lpici->cbSize < sizeof (CMINVOKECOMMANDINFO) )
return E_INVALIDARG;
r = Resolve(hwnd, 0 );
if ( FAILED( r ) )
{
TRACE("failed to resolve component with error 0x%08x", r);
return r;
}
if ( sComponent )
{
path = shelllink_get_msi_component_path( sComponent );
if (!path)
return E_FAIL;
}
else
path = strdupW( sPath );
if ( lpici->cbSize == sizeof (CMINVOKECOMMANDINFOEX) &&
( lpici->fMask & CMIC_MASK_UNICODE ) )
{
LPCMINVOKECOMMANDINFOEX iciex = (LPCMINVOKECOMMANDINFOEX) lpici;
DWORD len = 2;
if ( sArgs )
len += wcslen( sArgs );
if ( iciex->lpParametersW )
len += wcslen( iciex->lpParametersW );
args = (LPWSTR)HeapAlloc( GetProcessHeap(), 0, len*sizeof(WCHAR) );
args[0] = 0;
if ( sArgs )
wcscat( args, sArgs );
if ( iciex->lpParametersW )
{
static const WCHAR space[] = { ' ', 0 };
wcscat( args, space );
wcscat( args, iciex->lpParametersW );
}
}
else if (sArgs != NULL)
{
args = strdupW(sArgs);
}
memset( &sei, 0, sizeof sei );
sei.cbSize = sizeof sei;
sei.fMask = SEE_MASK_UNICODE | (lpici->fMask & (SEE_MASK_NOASYNC|SEE_MASK_ASYNCOK|SEE_MASK_FLAG_NO_UI));
sei.lpFile = path;
sei.nShow = iShowCmd;
sei.lpDirectory = sWorkDir;
sei.lpParameters = args;
sei.lpVerb = szOpen;
// HACK for ShellExecuteExW
if (!wcsstr(sPath, L".cpl"))
sei.lpVerb = szOpen;
else
sei.lpVerb = szCplOpen;
if( ShellExecuteExW( &sei ) )
r = S_OK;
else
r = E_FAIL;
HeapFree( GetProcessHeap(), 0, args );
HeapFree( GetProcessHeap(), 0, path );
return r;
}
HRESULT WINAPI CShellLink::GetCommandString(UINT_PTR idCmd, UINT uType, UINT* pwReserved, LPSTR pszName, UINT cchMax)
{
FIXME("%p %lu %u %p %p %u\n", this, idCmd, uType, pwReserved, pszName, cchMax );
return E_NOTIMPL;
}
INT_PTR CALLBACK ExtendedShortcutProc(HWND hwndDlg, UINT uMsg,
WPARAM wParam, LPARAM lParam)
{
HWND hDlgCtrl;
switch(uMsg)
{
case WM_INITDIALOG:
if (lParam)
{
hDlgCtrl = GetDlgItem(hwndDlg, 14000);
SendMessage(hDlgCtrl, BM_SETCHECK, BST_CHECKED, 0);
}
return TRUE;
case WM_COMMAND:
hDlgCtrl = GetDlgItem(hwndDlg, 14000);
if (LOWORD(wParam) == IDOK)
{
if ( SendMessage(hDlgCtrl, BM_GETCHECK, 0, 0) == BST_CHECKED )
EndDialog(hwndDlg, 1);
else
EndDialog(hwndDlg, 0);
}
else if (LOWORD(wParam) == IDCANCEL)
{
EndDialog(hwndDlg, -1);
}
else if (LOWORD(wParam) == 14000)
{
if ( SendMessage(hDlgCtrl, BM_GETCHECK, 0, 0) == BST_CHECKED)
SendMessage(hDlgCtrl, BM_SETCHECK, BST_UNCHECKED, 0);
else
SendMessage(hDlgCtrl, BM_SETCHECK, BST_CHECKED, 0);
}
}
return FALSE;
}
/**************************************************************************
* SH_ShellLinkDlgProc
*
* dialog proc of the shortcut property dialog
*/
INT_PTR CALLBACK CShellLink::SH_ShellLinkDlgProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
CShellLink *pThis = (CShellLink *)GetWindowLongPtr(hwndDlg, DWLP_USER);
switch(uMsg)
{
case WM_INITDIALOG:
{
LPPROPSHEETPAGEW ppsp = (LPPROPSHEETPAGEW)lParam;
if (ppsp == NULL)
break;
TRACE("ShellLink_DlgProc (WM_INITDIALOG hwnd %p lParam %p ppsplParam %x)\n",hwndDlg, lParam, ppsp->lParam);
pThis = (CShellLink *)ppsp->lParam;
SetWindowLongPtr(hwndDlg, DWLP_USER, (LONG_PTR)pThis);
TRACE("sArgs: %S sComponent: %S sDescription: %S sIcoPath: %S sPath: %S sPathRel: %S sProduct: %S sWorkDir: %S\n", pThis->sArgs, pThis->sComponent, pThis->sDescription,
pThis->sIcoPath, pThis->sPath, pThis->sPathRel, pThis->sProduct, pThis->sWorkDir);
/* target location */
if (pThis->sWorkDir)
SetDlgItemTextW(hwndDlg, 14007, PathFindFileName(pThis->sWorkDir));
/* target path */
if (pThis->sPath)
SetDlgItemTextW(hwndDlg, 14009, pThis->sPath);
/* working dir */
if (pThis->sWorkDir)
SetDlgItemTextW(hwndDlg, 14011, pThis->sWorkDir);
/* description */
if (pThis->sDescription)
SetDlgItemTextW(hwndDlg, 14019, pThis->sDescription);
return TRUE;
}
case WM_NOTIFY:
{
LPPSHNOTIFY lppsn = (LPPSHNOTIFY)lParam;
if (lppsn->hdr.code == PSN_APPLY)
{
WCHAR wszBuf[MAX_PATH];
/* set working directory */
GetDlgItemTextW(hwndDlg, 14011, wszBuf, MAX_PATH);
pThis->SetWorkingDirectory(wszBuf);
/* set link destination */
GetDlgItemTextW(hwndDlg, 14009, wszBuf, MAX_PATH);
if (!PathFileExistsW(wszBuf))
{
//FIXME load localized error msg
MessageBoxW(hwndDlg, L"file not existing", wszBuf, MB_OK);
SetWindowLongPtr(hwndDlg, DWL_MSGRESULT, PSNRET_INVALID_NOCHANGEPAGE);
return TRUE;
}
WCHAR *pwszExt = PathFindExtensionW(wszBuf);
if (!wcsicmp(pwszExt, L".lnk"))
{
// FIXME load localized error msg
MessageBoxW(hwndDlg, L"You cannot create a link to a shortcut", L"Error", MB_ICONERROR);
SetWindowLongPtr(hwndDlg, DWL_MSGRESULT, PSNRET_INVALID_NOCHANGEPAGE);
return TRUE;
}
pThis->SetPath(wszBuf);
TRACE("This %p sLinkPath %S\n", pThis, pThis->sLinkPath);
pThis->Save(pThis->sLinkPath, TRUE);
SetWindowLongPtr(hwndDlg, DWL_MSGRESULT, PSNRET_NOERROR);
return TRUE;
}
break;
}
case WM_COMMAND:
switch(LOWORD(wParam))
{
case 14020:
///
/// FIXME
/// open target directory
///
return TRUE;
case 14021:
{
WCHAR wszPath[MAX_PATH] = L"";
if (pThis->sIcoPath)
wcscpy(wszPath, pThis->sIcoPath);
INT IconIndex = pThis->iIcoNdx;
if (PickIconDlg(hwndDlg, wszPath, MAX_PATH, &IconIndex))
{
pThis->SetIconLocation(wszPath, IconIndex);
///
/// FIXME redraw icon
}
return TRUE;
}
case 14022:
{
INT_PTR result = DialogBoxParamW(shell32_hInstance, MAKEINTRESOURCEW(SHELL_EXTENDED_SHORTCUT_DLG), hwndDlg, ExtendedShortcutProc, (LPARAM)pThis->bRunAs);
if (result == 1 || result == 0)
{
if (pThis->bRunAs != result)
{
PropSheet_Changed(GetParent(hwndDlg), hwndDlg);
}
pThis->bRunAs = result;
}
return TRUE;
}
}
switch(HIWORD(wParam))
{
case EN_CHANGE:
PropSheet_Changed(GetParent(hwndDlg), hwndDlg);
break;
}
break;
default:
break;
}
return FALSE;
}
/**************************************************************************
* ShellLink_IShellPropSheetExt interface
*/
HRESULT WINAPI CShellLink::AddPages(LPFNADDPROPSHEETPAGE pfnAddPage, LPARAM lParam)
{
HPROPSHEETPAGE hPage;
BOOL bRet;
hPage = SH_CreatePropertySheetPage("SHELL_GENERAL_SHORTCUT_DLG", SH_ShellLinkDlgProc, (LPARAM)this, NULL);
if (hPage == NULL)
{
ERR("failed to create property sheet page\n");
return E_FAIL;
}
bRet = pfnAddPage(hPage, lParam);
if (bRet)
return S_OK;
else
return E_FAIL;
}
HRESULT WINAPI CShellLink::ReplacePage(UINT uPageID, LPFNADDPROPSHEETPAGE pfnReplacePage, LPARAM lParam)
{
TRACE("(%p) (uPageID %u, pfnReplacePage %p lParam %p\n", this, uPageID, pfnReplacePage, lParam);
return E_NOTIMPL;
}
HRESULT WINAPI CShellLink::SetSite(IUnknown *punk)
{
TRACE("%p %p\n", this, punk);
site = punk;
return S_OK;
}
HRESULT WINAPI CShellLink::GetSite(REFIID iid, void ** ppvSite)
{
TRACE("%p %s %p\n", this, debugstr_guid(&iid), ppvSite );
if (site == NULL)
return E_FAIL;
return site->QueryInterface(iid, ppvSite );
}
/**************************************************************************
* IShellLink_ConstructFromFile
*/
HRESULT WINAPI IShellLink_ConstructFromFile(IUnknown *pUnkOuter, REFIID riid, LPCITEMIDLIST pidl, LPVOID *ppv)
{
CComPtr<IUnknown> psl;
HRESULT hr = CShellLink::_CreatorClass::CreateInstance(NULL, riid, (void**)&psl);
if (SUCCEEDED(hr))
{
CComPtr<IPersistFile> ppf;
*ppv = NULL;
hr = psl->QueryInterface(IID_IPersistFile, (LPVOID*)&ppf);
if (SUCCEEDED(hr))
{
WCHAR path[MAX_PATH];
if (SHGetPathFromIDListW(pidl, path))
hr = ppf->Load(path, 0);
else
hr = E_FAIL;
if (SUCCEEDED(hr))
*ppv = psl.Detach();
}
}
return hr;
}
Reference in a new issue View git blame Copy permalink