344 lines
9.0 KiB
C++
344 lines
9.0 KiB
C++
/*
|
|
* intshcut.cpp - IUnknown implementation for InternetShortcut class.
|
|
*/
|
|
|
|
|
|
/* Headers
|
|
**********/
|
|
|
|
#include "project.hpp"
|
|
#pragma hdrstop
|
|
|
|
#include "assoc.h"
|
|
#include "clsfact.h"
|
|
|
|
|
|
/* Global Constants
|
|
*******************/
|
|
|
|
#pragma data_seg(DATA_SEG_READ_ONLY)
|
|
|
|
PUBLIC_DATA const int g_nDefaultShowCmd = SW_NORMAL;
|
|
|
|
#pragma data_seg()
|
|
|
|
|
|
/****************************** Public Functions *****************************/
|
|
|
|
|
|
PUBLIC_CODE HRESULT IsProtocolRegistered(PCSTR pcszProtocol)
|
|
{
|
|
HRESULT hr;
|
|
PSTR pszKey;
|
|
|
|
ASSERT(IS_VALID_STRING_PTR(pcszProtocol, CSTR));
|
|
|
|
hr = GetProtocolKey(pcszProtocol, EMPTY_STRING, &pszKey);
|
|
|
|
if (hr == S_OK)
|
|
{
|
|
hr = (GetRegKeyValue(g_hkeyURLProtocols, pszKey, g_cszURLProtocol, NULL,
|
|
NULL, NULL) == ERROR_SUCCESS)
|
|
? S_OK
|
|
: URL_E_UNREGISTERED_PROTOCOL;
|
|
|
|
delete pszKey;
|
|
pszKey = NULL;
|
|
}
|
|
|
|
if (hr != S_OK) {
|
|
TRACE_OUT(("IsProtocolRegistered(): Protocol \"%s\" is not registered.",
|
|
pcszProtocol));
|
|
}
|
|
|
|
return(hr);
|
|
}
|
|
|
|
|
|
PUBLIC_CODE HRESULT ValidateURL(PCSTR pcszURL)
|
|
{
|
|
HRESULT hr;
|
|
PSTR pszProtocol;
|
|
|
|
ASSERT(IS_VALID_STRING_PTR(pcszURL, CSTR));
|
|
|
|
hr = CopyURLProtocol(pcszURL, &pszProtocol);
|
|
|
|
if (hr == S_OK)
|
|
{
|
|
hr = IsProtocolRegistered(pszProtocol);
|
|
|
|
delete pszProtocol;
|
|
pszProtocol = NULL;
|
|
}
|
|
|
|
return(hr);
|
|
}
|
|
|
|
|
|
PUBLIC_CODE HRESULT ValidateWorkingDirectory(PCSTR pcszWorkingDirectory)
|
|
{
|
|
ASSERT(IS_VALID_STRING_PTR(pcszWorkingDirectory, CSTR));
|
|
|
|
return(IsPathDirectory(pcszWorkingDirectory) ? S_OK : E_PATH_NOT_FOUND);
|
|
}
|
|
|
|
|
|
#ifdef DEBUG
|
|
|
|
PUBLIC_CODE BOOL IsValidPCInternetShortcut(PCInternetShortcut pcintshcut)
|
|
{
|
|
return(IS_VALID_READ_PTR(pcintshcut, CInternetShortcut) &&
|
|
FLAGS_ARE_VALID(pcintshcut->m_dwFlags, ALL_INTSHCUT_FLAGS) &&
|
|
(! pcintshcut->m_pszFile ||
|
|
IS_VALID_STRING_PTR(pcintshcut->m_pszFile, STR)) &&
|
|
(! pcintshcut->m_pszURL ||
|
|
IS_VALID_STRING_PTR(pcintshcut->m_pszURL, STR)) &&
|
|
((! pcintshcut->m_pszIconFile &&
|
|
! pcintshcut->m_niIcon) ||
|
|
EVAL(IsValidIconIndex(S_OK, pcintshcut->m_pszIconFile, MAX_PATH_LEN, pcintshcut->m_niIcon))) &&
|
|
(! pcintshcut->m_pszWorkingDirectory ||
|
|
EVAL(IsFullPath(pcintshcut->m_pszWorkingDirectory))) &&
|
|
EVAL(IsValidShowCmd(pcintshcut->m_nShowCmd)) &&
|
|
EVAL(! pcintshcut->m_pszFolder ||
|
|
IsValidPath(pcintshcut->m_pszFolder)) &&
|
|
EVAL(! pcintshcut->m_wHotkey ||
|
|
IsValidHotkey(pcintshcut->m_wHotkey)) &&
|
|
IS_VALID_STRUCT_PTR((PCRefCount)pcintshcut, CRefCount) &&
|
|
IS_VALID_INTERFACE_PTR((PCIDataObject)pcintshcut, IDataObject) &&
|
|
IS_VALID_INTERFACE_PTR((PCIExtractIcon)pcintshcut, IExtractIcon) &&
|
|
IS_VALID_INTERFACE_PTR((PCINewShortcutHook)pcintshcut, INewShortcutHook) &&
|
|
IS_VALID_INTERFACE_PTR((PCIPersistFile)pcintshcut, IPersistFile) &&
|
|
IS_VALID_INTERFACE_PTR((PCIPersistStream)pcintshcut, IPersistStream) &&
|
|
IS_VALID_INTERFACE_PTR((PCIShellExecuteHook)pcintshcut, IShellExecuteHook) &&
|
|
IS_VALID_INTERFACE_PTR((PCIShellExtInit)pcintshcut, IShellExtInit) &&
|
|
IS_VALID_INTERFACE_PTR((PCIShellLink)pcintshcut, IShellLink) &&
|
|
IS_VALID_INTERFACE_PTR((PCIShellPropSheetExt)pcintshcut, IShellPropSheetExt) &&
|
|
IS_VALID_INTERFACE_PTR((PCIUniformResourceLocator)pcintshcut, IUniformResourceLocator));
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
/********************************** Methods **********************************/
|
|
|
|
|
|
#pragma warning(disable:4705) /* "statement has no effect" warning - cl bug, see KB Q98989 */
|
|
|
|
InternetShortcut::InternetShortcut()
|
|
{
|
|
DebugEntry(InternetShortcut::InternetShortcut);
|
|
|
|
// Don't validate this until after construction.
|
|
|
|
m_dwFlags = 0;
|
|
m_pszFile = NULL;
|
|
m_pszURL = NULL;
|
|
m_pszIconFile = NULL;
|
|
m_niIcon = 0;
|
|
m_pszWorkingDirectory = NULL;
|
|
m_nShowCmd = g_nDefaultShowCmd;
|
|
m_pszFolder = NULL;
|
|
m_wHotkey = 0;
|
|
|
|
ASSERT(IS_VALID_STRUCT_PTR(this, CInternetShortcut));
|
|
|
|
DebugExitVOID(InternetShortcut::InternetShortcut);
|
|
|
|
return;
|
|
}
|
|
|
|
#pragma warning(default:4705) /* "statement has no effect" warning - cl bug, see KB Q98989 */
|
|
|
|
|
|
InternetShortcut::~InternetShortcut(void)
|
|
{
|
|
DebugEntry(InternetShortcut::~InternetShortcut);
|
|
|
|
ASSERT(IS_VALID_STRUCT_PTR(this, CInternetShortcut));
|
|
|
|
if (m_pszFile)
|
|
{
|
|
delete m_pszFile;
|
|
m_pszFile = NULL;
|
|
}
|
|
|
|
if (m_pszURL)
|
|
{
|
|
delete m_pszURL;
|
|
m_pszURL = NULL;
|
|
}
|
|
|
|
if (m_pszIconFile)
|
|
{
|
|
delete m_pszIconFile;
|
|
m_pszIconFile = NULL;
|
|
m_niIcon = 0;
|
|
}
|
|
|
|
if (m_pszWorkingDirectory)
|
|
{
|
|
delete m_pszWorkingDirectory;
|
|
m_pszWorkingDirectory = NULL;
|
|
}
|
|
|
|
if (m_pszFolder)
|
|
{
|
|
delete m_pszFolder;
|
|
m_pszFolder = NULL;
|
|
}
|
|
|
|
ASSERT(IS_VALID_STRUCT_PTR(this, CInternetShortcut));
|
|
|
|
DebugExitVOID(InternetShortcut::~InternetShortcut);
|
|
|
|
return;
|
|
}
|
|
|
|
|
|
ULONG STDMETHODCALLTYPE InternetShortcut::AddRef(void)
|
|
{
|
|
ULONG ulcRef;
|
|
|
|
DebugEntry(InternetShortcut::AddRef);
|
|
|
|
ASSERT(IS_VALID_STRUCT_PTR(this, CInternetShortcut));
|
|
|
|
ulcRef = RefCount::AddRef();
|
|
|
|
ASSERT(IS_VALID_STRUCT_PTR(this, CInternetShortcut));
|
|
|
|
DebugExitULONG(InternetShortcut::AddRef, ulcRef);
|
|
|
|
return(ulcRef);
|
|
}
|
|
|
|
|
|
ULONG STDMETHODCALLTYPE InternetShortcut::Release(void)
|
|
{
|
|
ULONG ulcRef;
|
|
|
|
DebugEntry(InternetShortcut::Release);
|
|
|
|
ASSERT(IS_VALID_STRUCT_PTR(this, CInternetShortcut));
|
|
|
|
ulcRef = RefCount::Release();
|
|
|
|
DebugExitULONG(InternetShortcut::Release, ulcRef);
|
|
|
|
return(ulcRef);
|
|
}
|
|
|
|
|
|
HRESULT STDMETHODCALLTYPE InternetShortcut::QueryInterface(REFIID riid,
|
|
PVOID *ppvObject)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
DebugEntry(InternetShortcut::QueryInterface);
|
|
|
|
ASSERT(IS_VALID_STRUCT_PTR(this, CInternetShortcut));
|
|
ASSERT(IsValidREFIID(riid));
|
|
ASSERT(IS_VALID_WRITE_PTR(ppvObject, PVOID));
|
|
|
|
if (riid == IID_IDataObject)
|
|
{
|
|
*ppvObject = (PIDataObject)this;
|
|
TRACE_OUT(("InternetShortcut::QueryInterface(): Returning IDataObject."));
|
|
}
|
|
else if (riid == IID_IExtractIcon)
|
|
{
|
|
*ppvObject = (PIExtractIcon)this;
|
|
TRACE_OUT(("InternetShortcut::QueryInterface(): Returning IExtractIcon."));
|
|
}
|
|
else if (riid == IID_INewShortcutHook)
|
|
{
|
|
*ppvObject = (PINewShortcutHook)this;
|
|
TRACE_OUT(("InternetShortcut::QueryInterface(): Returning INewShortcutHook."));
|
|
}
|
|
else if (riid == IID_IPersist)
|
|
{
|
|
*ppvObject = (PIPersist)(PIPersistStream)this;
|
|
TRACE_OUT(("InternetShortcut::QueryInterface(): Returning IPersist."));
|
|
}
|
|
else if (riid == IID_IPersistFile)
|
|
{
|
|
*ppvObject = (PIPersistFile)this;
|
|
TRACE_OUT(("InternetShortcut::QueryInterface(): Returning IPersistFile."));
|
|
}
|
|
else if (riid == IID_IPersistStream)
|
|
{
|
|
*ppvObject = (PIPersistStream)this;
|
|
TRACE_OUT(("InternetShortcut::QueryInterface(): Returning IPersistStream."));
|
|
}
|
|
else if (riid == IID_IShellExecuteHook)
|
|
{
|
|
*ppvObject = (PIShellExecuteHook)this;
|
|
TRACE_OUT(("InternetShortcut::QueryInterface(): Returning IShellExecuteHook."));
|
|
}
|
|
else if (riid == IID_IShellExtInit)
|
|
{
|
|
*ppvObject = (PIShellExtInit)this;
|
|
TRACE_OUT(("InternetShortcut::QueryInterface(): Returning IShellExtInit."));
|
|
}
|
|
else if (riid == IID_IShellLink)
|
|
{
|
|
*ppvObject = (PIShellLink)this;
|
|
TRACE_OUT(("InternetShortcut::QueryInterface(): Returning IShellLink."));
|
|
}
|
|
else if (riid == IID_IShellPropSheetExt)
|
|
{
|
|
*ppvObject = (PIShellPropSheetExt)this;
|
|
TRACE_OUT(("InternetShortcut::QueryInterface(): Returning IShellPropSheetExt."));
|
|
}
|
|
else if (riid == IID_IUniformResourceLocator)
|
|
{
|
|
*ppvObject = (PIUniformResourceLocator)this;
|
|
TRACE_OUT(("InternetShortcut::QueryInterface(): Returning IUniformResourceLocator."));
|
|
}
|
|
else if (riid == IID_IUnknown)
|
|
{
|
|
*ppvObject = (PIUnknown)(PIUniformResourceLocator)this;
|
|
TRACE_OUT(("InternetShortcut::QueryInterface(): Returning IUnknown."));
|
|
}
|
|
else
|
|
{
|
|
TRACE_OUT(("InternetShortcut::QueryInterface(): Called on unknown interface."));
|
|
*ppvObject = NULL;
|
|
hr = E_NOINTERFACE;
|
|
}
|
|
|
|
if (hr == S_OK)
|
|
AddRef();
|
|
|
|
ASSERT(IS_VALID_STRUCT_PTR(this, CInternetShortcut));
|
|
|
|
DebugExitHRESULT(InternetShortcut::QueryInterface, hr);
|
|
|
|
return(hr);
|
|
}
|
|
|
|
|
|
extern "C"
|
|
STDAPI CreateInstance_Intshcut(IUnknown *punkOuter, REFIID riid, void **ppvOut)
|
|
{
|
|
HRESULT hres;
|
|
|
|
*ppvOut = NULL;
|
|
|
|
if (punkOuter)
|
|
return CLASS_E_NOAGGREGATION;
|
|
|
|
InternetShortcut *pintshcut = new(InternetShortcut);
|
|
if (pintshcut)
|
|
{
|
|
hres = pintshcut->QueryInterface(riid, ppvOut);
|
|
pintshcut->Release();
|
|
}
|
|
else
|
|
hres = E_OUTOFMEMORY;
|
|
|
|
return hres;
|
|
}
|
|
|