able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/windows/winstate/cobra/utils/linkpif/linkpif.c

1736 lines
52 KiB
C
Raw Normal View History

Add source files
2020-09-26 03:20:57 -05:00
/*++
Copyright (c) 1996 Microsoft Corporation
Module Name:
linkpif.c
Abstract:
Functions to query and modify LNK and PIF files.
Author:
Calin Negreanu (calinn) 07-Sep-1998
Revision History:
--*/
//
// Includes
//
#include "pch.h"
#include <pif.h> // private\windows\inc
//
// Debug constants
//
#define DBG_VERSION "LnkPif"
//
// Strings
//
// None
//
// Constants
//
// None
//
// Macros
//
// None
//
// Types
//
// None
//
// Globals
//
// None
//
// Macro expansion list
//
// None
//
// Private function prototypes
//
// None
//
// Macro expansion definition
//
// None
//
// Code
//
BOOL
InitCOMLinkA (
OUT IShellLinkA **ShellLink,
OUT IPersistFile **PersistFile
)
{
HRESULT hres;
BOOL result;
//
// Initialize COM
//
CoInitialize (NULL);
*ShellLink = NULL;
*PersistFile = NULL;
result = FALSE;
__try {
//
// Get a pointer to the IShellLink interface.
//
hres = CoCreateInstance (&CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, &IID_IShellLinkA, ShellLink);
if (!SUCCEEDED (hres)) {
__leave;
}
//
// Get a pointer to the IPersistFile interface.
//
hres = (*ShellLink)->lpVtbl->QueryInterface ((*ShellLink), &IID_IPersistFile, PersistFile);
if (!SUCCEEDED (hres)) {
__leave;
}
result = TRUE;
}
__finally {
if (!result) {
if (*PersistFile) {
(*PersistFile)->lpVtbl->Release (*PersistFile);
*PersistFile = NULL;
}
if (*ShellLink) {
(*ShellLink)->lpVtbl->Release (*ShellLink);
*ShellLink = NULL;
}
}
}
if (!result) {
//
// Free COM
//
CoUninitialize ();
}
return result;
}
BOOL
InitCOMLinkW (
OUT IShellLinkW **ShellLink,
OUT IPersistFile **PersistFile
)
{
HRESULT hres;
BOOL result;
//
// Initialize COM
//
CoInitialize (NULL);
*ShellLink = NULL;
*PersistFile = NULL;
result = FALSE;
__try {
//
// Get a pointer to the IShellLink interface.
//
hres = CoCreateInstance (&CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, &IID_IShellLinkW, ShellLink);
if (!SUCCEEDED (hres)) {
__leave;
}
//
// Get a pointer to the IPersistFile interface.
//
hres = (*ShellLink)->lpVtbl->QueryInterface ((*ShellLink), &IID_IPersistFile, PersistFile);
if (!SUCCEEDED (hres)) {
__leave;
}
result = TRUE;
}
__finally {
if (!result) {
if (*PersistFile) {
(*PersistFile)->lpVtbl->Release (*PersistFile);
*PersistFile = NULL;
}
if (*ShellLink) {
(*ShellLink)->lpVtbl->Release (*ShellLink);
*ShellLink = NULL;
}
}
}
if (!result) {
//
// Free COM
//
CoUninitialize ();
}
return result;
}
BOOL
FreeCOMLinkA (
IN OUT IShellLinkA **ShellLink,
IN OUT IPersistFile **PersistFile
)
{
if (*PersistFile) {
(*PersistFile)->lpVtbl->Release (*PersistFile);
*PersistFile = NULL;
}
if (*ShellLink) {
(*ShellLink)->lpVtbl->Release (*ShellLink);
*ShellLink = NULL;
}
//
// Free COM
//
CoUninitialize ();
return TRUE;
}
BOOL
FreeCOMLinkW (
IN OUT IShellLinkW **ShellLink,
IN OUT IPersistFile **PersistFile
)
{
if (*PersistFile) {
(*PersistFile)->lpVtbl->Release (*PersistFile);
*PersistFile = NULL;
}
if (*ShellLink) {
(*ShellLink)->lpVtbl->Release (*ShellLink);
*ShellLink = NULL;
}
//
// Free COM
//
CoUninitialize ();
return TRUE;
}
PVOID
pFindEnhPifSignature (
IN PVOID FileImage,
IN PCSTR Signature
)
/*++
Routine Description:
pFindEnhPifSignature finds a certain PIF structure inside a PIF file (if it exists)
based on a signature.
Arguments:
FileImage - image of the PIF file mapped into memory
Signature - structure signature
Return Value:
address of the PIF structure, or NULL if non existent
--*/
{
PBYTE tempPtr;
PBYTE lastPtr;
PVOID result = NULL;
BOOL finished = FALSE;
PPIFEXTHDR pifExtHdr;
lastPtr = (PBYTE) FileImage;
tempPtr = (PBYTE) FileImage;
tempPtr += sizeof (STDPIF);
pifExtHdr = (PPIFEXTHDR) tempPtr;
__try {
do {
if (tempPtr < lastPtr) {
result = NULL;
break;
} else {
lastPtr = tempPtr;
}
finished = pifExtHdr->extnxthdrfloff == LASTHDRPTR;
if (StringMatchA (pifExtHdr->extsig, Signature)) {
result = tempPtr + sizeof (PIFEXTHDR);
break;
}
else {
tempPtr = (PBYTE)FileImage + pifExtHdr->extnxthdrfloff;
pifExtHdr = (PPIFEXTHDR) tempPtr;
}
} while (!finished);
}
__except (1) {
// something went wrong trying to access PIF file. Let's exit with NULL
return NULL;
}
return result;
}
BOOL
ExtractPifInfoA(
IN PCSTR FileName,
OUT PCSTR *Target,
OUT PCSTR *Params,
OUT PCSTR *WorkDir,
OUT PCSTR *IconPath,
OUT PINT IconNumber,
OUT BOOL *MsDosMode,
OUT PLNK_EXTRA_DATAA ExtraData OPTIONAL
)
{
PVOID fileImage = NULL;
HANDLE mapHandle = NULL;
HANDLE fileHandle = INVALID_HANDLE_VALUE;
CHAR tempStr [MEMDB_MAX];
CHAR target1 [MEMDB_MAX];
PSTR strPtr;
PSTR dontCare;
PSTDPIF stdPif;
PWENHPIF40 wenhPif40;
PW386PIF30 w386ext30;
BOOL result = TRUE;
if (Target) {
*Target = NULL;
}
if (Params) {
*Params = NULL;
}
if (WorkDir) {
*WorkDir = NULL;
}
if (IconPath) {
*IconPath = NULL;
}
*IconNumber = 0;
*MsDosMode = FALSE;
if (ExtraData) {
ZeroMemory (ExtraData, sizeof(LNK_EXTRA_DATA));
}
__try {
fileImage = MapFileIntoMemoryA (FileName, &fileHandle, &mapHandle);
if (fileImage == NULL) {
__leave;
}
__try {
stdPif = (PSTDPIF) fileImage;
//
// getting working directory
//
_mbsncpy (tempStr, stdPif->defpath, PIFDEFPATHSIZE);
// we might have a path terminated with a wack, we don't want that
strPtr = _mbsdec (tempStr, GetEndOfStringA (tempStr));
if (strPtr) {
if (_mbsnextc (strPtr) == '\\') {
*strPtr = 0;
}
}
// now get the long path.
if (WorkDir) {
*WorkDir = DuplicatePathStringA (tempStr, 0);
}
//
// getting PIFs target
//
_mbsncpy (target1, stdPif->startfile, PIFSTARTLOCSIZE);
// in most cases, the target is without a path. We try to build the path, either
// by using WorkDir or by calling SearchPath to look for this file.
if (*target1) {//non empty target
strPtr = _mbsrchr (target1, '\\');
if (!strPtr) {
if (WorkDir && (*WorkDir)[0]) {
StringCopyA (tempStr, *WorkDir);
StringCatA (tempStr, "\\");
StringCatA (tempStr, target1);
}
else {
if (!SearchPathA (NULL, target1, NULL, MEMDB_MAX, tempStr, &dontCare)) {
DEBUGMSG ((DBG_WARNING, "Could not find path for PIF target:%s", FileName));
StringCopyA (tempStr, target1);
}
}
}
else {
StringCopyA (tempStr, target1);
}
// now get the long path
if (Target) {
*Target = DuplicatePathStringA (tempStr, 0);
}
}
//
// getting PIFs arguments
//
_mbsncpy (tempStr, stdPif->params, PIFPARAMSSIZE);
if (Params) {
*Params = DuplicatePathStringA (tempStr, 0);
}
//
// let's try to read the WENHPIF40 structure
//
wenhPif40 = pFindEnhPifSignature (fileImage, WENHHDRSIG40);
if (wenhPif40) {
if (IconPath) {
*IconPath = DuplicatePathStringA (wenhPif40->achIconFileProp, 0);
}
*IconNumber = wenhPif40->wIconIndexProp;
if (ExtraData) {
ExtraData->xSize = 80;
ExtraData->ySize = wenhPif40->vidProp.cScreenLines;
if (ExtraData->ySize < 25) {
ExtraData->ySize = 25;
}
ExtraData->QuickEdit = !(wenhPif40->mseProp.flMse & MSE_WINDOWENABLE);
ExtraData->CurrentCodePage = wenhPif40->fntProp.wCurrentCP;
// now let's do some crazy things trying to get the font used
{
LOGFONTA logFont;
HDC dc;
HFONT font;
HGDIOBJ oldObject;
TEXTMETRIC tm;
ZeroMemory (&logFont, sizeof (LOGFONTA));
logFont.lfHeight = wenhPif40->fntProp.cyFontActual;
logFont.lfWidth = wenhPif40->fntProp.cxFontActual;
logFont.lfEscapement = 0;
logFont.lfOrientation = 0;
logFont.lfWeight = FW_DONTCARE;
logFont.lfItalic = FALSE;
logFont.lfUnderline = FALSE;
logFont.lfStrikeOut = FALSE;
logFont.lfCharSet = DEFAULT_CHARSET;
logFont.lfOutPrecision = OUT_DEFAULT_PRECIS;
logFont.lfClipPrecision = CLIP_DEFAULT_PRECIS;
logFont.lfQuality = DEFAULT_QUALITY;
logFont.lfPitchAndFamily = DEFAULT_PITCH;
if (wenhPif40->fntProp.flFnt & FNT_TT) {
_mbsncpy (logFont.lfFaceName, wenhPif40->fntProp.achTTFaceName, LF_FACESIZE);
_mbsncpy (ExtraData->FontName, wenhPif40->fntProp.achTTFaceName, LF_FACESIZE);
} else {
_mbsncpy (logFont.lfFaceName, wenhPif40->fntProp.achRasterFaceName, LF_FACESIZE);
_mbsncpy (ExtraData->FontName, wenhPif40->fntProp.achRasterFaceName, LF_FACESIZE);
}
dc = CreateDCA ("DISPLAY", NULL, NULL, NULL);
if (dc) {
font = CreateFontIndirectA (&logFont);
if (font) {
oldObject = SelectObject (dc, font);
if (GetTextMetrics (dc, &tm)) {
ExtraData->xFontSize = tm.tmAveCharWidth;
ExtraData->yFontSize = tm.tmHeight;
ExtraData->FontWeight = tm.tmWeight;
ExtraData->FontFamily = tm.tmPitchAndFamily;
}
SelectObject (dc, oldObject);
DeleteObject (font);
}
DeleteDC (dc);
}
}
}
}
w386ext30 = pFindEnhPifSignature (fileImage, W386HDRSIG30);
if (w386ext30) {
if (((w386ext30->PfW386Flags & fRealMode ) == fRealMode ) ||
((w386ext30->PfW386Flags & fRealModeSilent) == fRealModeSilent)
) {
*MsDosMode = TRUE;
}
if (ExtraData) {
ExtraData->FullScreen = (w386ext30->PfW386Flags & fFullScreen) != 0;
}
}
}
__except (1) {
// something went wrong when we tried to read or write PIF file,
result = FALSE;
}
}
__finally {
UnmapFile (fileImage, mapHandle, fileHandle);
}
return result;
}
BOOL
ExtractPifInfoW(
IN PCWSTR FileName,
OUT PCWSTR *Target,
OUT PCWSTR *Params,
OUT PCWSTR *WorkDir,
OUT PCWSTR *IconPath,
OUT PINT IconNumber,
OUT BOOL *MsDosMode,
OUT PLNK_EXTRA_DATAW ExtraData OPTIONAL
)
{
PCSTR aTarget = NULL;
PCSTR aParams = NULL;
PCSTR aWorkDir = NULL;
PCSTR aIconPath = NULL;
PCSTR aFileName;
PCWSTR tempStrW;
BOOL result;
LNK_EXTRA_DATAA extraDataA;
aFileName = ConvertWtoA (FileName);
result = ExtractPifInfoA (
aFileName,
&aTarget,
&aParams,
&aWorkDir,
&aIconPath,
IconNumber,
MsDosMode,
ExtraData?&extraDataA:NULL
);
FreeConvertedStr (aFileName);
if (Target) {
*Target = NULL;
if (aTarget) {
tempStrW = ConvertAtoW (aTarget);
*Target = DuplicatePathStringW (tempStrW, 0);
FreeConvertedStr (tempStrW);
}
}
if (aTarget) {
FreePathStringA (aTarget);
}
if (Params) {
*Params = NULL;
if (aParams) {
tempStrW = ConvertAtoW (aParams);
*Params = DuplicatePathStringW (tempStrW, 0);
FreeConvertedStr (tempStrW);
}
}
if (aParams) {
FreePathStringA (aParams);
}
if (WorkDir) {
*WorkDir = NULL;
if (aWorkDir) {
tempStrW = ConvertAtoW (aWorkDir);
*WorkDir = DuplicatePathStringW (tempStrW, 0);
FreeConvertedStr (tempStrW);
}
}
if (aWorkDir) {
FreePathStringA (aWorkDir);
}
if (IconPath) {
*IconPath = NULL;
if (aIconPath) {
tempStrW = ConvertAtoW (aIconPath);
*IconPath = DuplicatePathStringW (tempStrW, 0);
FreeConvertedStr (tempStrW);
}
}
if (aIconPath) {
FreePathStringA (aIconPath);
}
if (ExtraData) {
ExtraData->FullScreen = extraDataA.FullScreen;
ExtraData->xSize = extraDataA.xSize;
ExtraData->ySize = extraDataA.ySize;
ExtraData->QuickEdit = extraDataA.QuickEdit;
tempStrW = ConvertAtoW (extraDataA.FontName);
StringCopyW (ExtraData->FontName, tempStrW);
FreeConvertedStr (tempStrW);
ExtraData->xFontSize = extraDataA.xFontSize;
ExtraData->yFontSize = extraDataA.yFontSize;
ExtraData->FontWeight = extraDataA.FontWeight;
ExtraData->FontFamily = extraDataA.FontFamily;
ExtraData->CurrentCodePage = extraDataA.CurrentCodePage;
}
return result;
}
BOOL
ExtractShellLinkInfoA (
IN PCSTR FileName,
OUT PCSTR *Target,
OUT PCSTR *Params,
OUT PCSTR *WorkDir,
OUT PCSTR *IconPath,
OUT PINT IconNumber,
OUT PWORD HotKey,
IN IShellLinkA *ShellLink,
IN IPersistFile *PersistFile
)
{
CHAR tempStr [MEMDB_MAX];
PCSTR sanitizedStr = NULL;
PCWSTR fileNameW;
PSTR strPtr;
HRESULT hres;
WIN32_FIND_DATAA fd;
IShellLinkDataList *shellLinkDataList;
LPEXP_SZ_LINK expSzLink;
if (Target) {
*Target = NULL;
}
if (Params) {
*Params = NULL;
}
if (WorkDir) {
*WorkDir = NULL;
}
if (IconPath) {
*IconPath = NULL;
}
fileNameW = ConvertAtoW (FileName);
hres = PersistFile->lpVtbl->Load(PersistFile, fileNameW, STGM_READ);
FreeConvertedStr (fileNameW);
if (!SUCCEEDED(hres)) {
DEBUGMSGA((DBG_WARNING, "Cannot load link %s", FileName));
return FALSE;
}
//
// Get the link target
//
hres = ShellLink->lpVtbl->GetPath (
ShellLink,
tempStr,
sizeof (tempStr),
&fd,
SLGP_RAWPATH
);
if (!SUCCEEDED(hres)) {
DEBUGMSGA((DBG_WARNING, "Cannot read target for link %s", FileName));
return FALSE;
}
if (Target) {
sanitizedStr = SanitizePathA (tempStr);
*Target = DuplicatePathStringA (sanitizedStr?sanitizedStr:"", 0);
if (sanitizedStr) {
FreePathStringA (sanitizedStr);
sanitizedStr = NULL;
}
}
//
// Get the link working directory
//
hres = ShellLink->lpVtbl->GetWorkingDirectory (
ShellLink,
tempStr,
sizeof (tempStr)
);
if (!SUCCEEDED(hres)) {
DEBUGMSGA((DBG_WARNING, "Cannot read target for link %s", FileName));
return FALSE;
}
if (WorkDir) {
sanitizedStr = SanitizePathA (tempStr);
if (sanitizedStr) {
strPtr = (PSTR)GetEndOfStringA (sanitizedStr);
if (strPtr) {
strPtr = _mbsdec (sanitizedStr, strPtr);
if (strPtr) {
if (_mbsnextc (strPtr) == '\\') {
*strPtr = 0;
}
}
}
}
*WorkDir = DuplicatePathStringA (sanitizedStr?sanitizedStr:"", 0);
if (sanitizedStr) {
FreePathStringA (sanitizedStr);
sanitizedStr = NULL;
}
}
//
// Get the arguments.
//
hres = ShellLink->lpVtbl->GetArguments (
ShellLink,
tempStr,
MEMDB_MAX
);
if (!SUCCEEDED(hres)) {
DEBUGMSGA((DBG_WARNING, "Cannot read arguments for link %s", FileName));
return FALSE;
}
if (Params) {
*Params = DuplicatePathStringA (tempStr, 0);
}
//
// Get icon path
//
hres = ShellLink->lpVtbl->GetIconLocation (
ShellLink,
tempStr,
sizeof (tempStr),
IconNumber
);
if (!SUCCEEDED(hres)) {
DEBUGMSGA((DBG_WARNING, "Cannot read icon path for link %s", FileName));
return FALSE;
}
if (IconPath) {
sanitizedStr = SanitizePathA (tempStr);
*IconPath = DuplicatePathStringA (sanitizedStr?sanitizedStr:"", 0);
if (sanitizedStr) {
FreePathStringA (sanitizedStr);
sanitizedStr = NULL;
}
// One more thing: let's see if the actual icon path is a EXPAND_SZ
hres = ShellLink->lpVtbl->QueryInterface (ShellLink, &IID_IShellLinkDataList, &shellLinkDataList);
if (SUCCEEDED(hres)) {
hres = shellLinkDataList->lpVtbl->CopyDataBlock (shellLinkDataList, EXP_SZ_ICON_SIG, (LPVOID*)&expSzLink);
if (SUCCEEDED(hres)) {
if (expSzLink->szTarget [0]) {
FreePathStringA (*IconPath);
sanitizedStr = SanitizePathA (expSzLink->szTarget);
*IconPath = DuplicatePathStringA (sanitizedStr?sanitizedStr:"", 0);
if (sanitizedStr) {
FreePathStringA (sanitizedStr);
sanitizedStr = NULL;
}
}
LocalFree (expSzLink);
}
shellLinkDataList->lpVtbl->Release (shellLinkDataList);
}
}
//
// Get hot key
//
hres = ShellLink->lpVtbl->GetHotkey (
ShellLink,
HotKey
);
if (!SUCCEEDED(hres)) {
DEBUGMSGA((DBG_WARNING, "Cannot read hot key for link %s", FileName));
return FALSE;
}
return TRUE;
}
BOOL
ExtractShellLinkInfoW (
IN PCWSTR FileName,
OUT PCWSTR *Target,
OUT PCWSTR *Params,
OUT PCWSTR *WorkDir,
OUT PCWSTR *IconPath,
OUT PINT IconNumber,
OUT PWORD HotKey,
IN IShellLinkW *ShellLink,
IN IPersistFile *PersistFile
)
{
WCHAR tempStr [MEMDB_MAX];
PCWSTR sanitizedStr = NULL;
PWSTR strPtr;
HRESULT hres;
WIN32_FIND_DATAW fd;
IShellLinkDataList *shellLinkDataList;
LPEXP_SZ_LINK expSzLink;
if (Target) {
*Target = NULL;
}
if (Params) {
*Params = NULL;
}
if (WorkDir) {
*WorkDir = NULL;
}
if (IconPath) {
*IconPath = NULL;
}
hres = PersistFile->lpVtbl->Load(PersistFile, FileName, STGM_READ);
if (!SUCCEEDED(hres)) {
DEBUGMSGW((DBG_WARNING, "Cannot load link %s", FileName));
return FALSE;
}
//
// Get the link target
//
hres = ShellLink->lpVtbl->GetPath (
ShellLink,
tempStr,
sizeof (tempStr),
&fd,
SLGP_RAWPATH
);
if (!SUCCEEDED(hres)) {
DEBUGMSGA((DBG_WARNING, "Cannot read target for link %s", FileName));
return FALSE;
}
if (Target) {
sanitizedStr = SanitizePathW (tempStr);
*Target = DuplicatePathStringW (sanitizedStr?sanitizedStr:L"", 0);
if (sanitizedStr) {
FreePathStringW (sanitizedStr);
sanitizedStr = NULL;
}
}
//
// Get the link working directory
//
hres = ShellLink->lpVtbl->GetWorkingDirectory (
ShellLink,
tempStr,
sizeof (tempStr)
);
if (!SUCCEEDED(hres)) {
DEBUGMSGW((DBG_WARNING, "Cannot read target for link %s", FileName));
return FALSE;
}
if (WorkDir) {
sanitizedStr = SanitizePathW (tempStr);
if (sanitizedStr) {
strPtr = GetEndOfStringW (sanitizedStr) - 1;
if (strPtr >= sanitizedStr) {
if (*strPtr == L'\\') {
*strPtr = 0;
}
}
}
*WorkDir = DuplicatePathStringW (sanitizedStr?sanitizedStr:L"", 0);
if (sanitizedStr) {
FreePathStringW (sanitizedStr);
sanitizedStr = NULL;
}
}
//
// Get the arguments.
//
hres = ShellLink->lpVtbl->GetArguments (
ShellLink,
tempStr,
MEMDB_MAX
);
if (!SUCCEEDED(hres)) {
DEBUGMSGW((DBG_WARNING, "Cannot read arguments for link %s", FileName));
return FALSE;
}
if (Params) {
*Params = DuplicatePathStringW (tempStr, 0);
}
//
// Get icon path
//
hres = ShellLink->lpVtbl->GetIconLocation (
ShellLink,
tempStr,
sizeof (tempStr),
IconNumber
);
if (!SUCCEEDED(hres)) {
DEBUGMSGW((DBG_WARNING, "Cannot read icon path for link %s", FileName));
return FALSE;
}
if (IconPath) {
sanitizedStr = SanitizePathW (tempStr);
*IconPath = DuplicatePathStringW (sanitizedStr?sanitizedStr:L"", 0);
if (sanitizedStr) {
FreePathStringW (sanitizedStr);
sanitizedStr = NULL;
}
// One more thing: let's see if the actual icon path is a EXPAND_SZ
hres = ShellLink->lpVtbl->QueryInterface (ShellLink, &IID_IShellLinkDataList, &shellLinkDataList);
if (SUCCEEDED(hres)) {
hres = shellLinkDataList->lpVtbl->CopyDataBlock (shellLinkDataList, EXP_SZ_ICON_SIG, (LPVOID*)&expSzLink);
if (SUCCEEDED(hres)) {
if (expSzLink->swzTarget [0]) {
FreePathStringW (*IconPath);
sanitizedStr = SanitizePathW (expSzLink->swzTarget);
*IconPath = DuplicatePathStringW (sanitizedStr?sanitizedStr:L"", 0);
if (sanitizedStr) {
FreePathStringW (sanitizedStr);
sanitizedStr = NULL;
}
}
LocalFree (expSzLink);
}
shellLinkDataList->lpVtbl->Release (shellLinkDataList);
}
}
//
// Get hot key
//
hres = ShellLink->lpVtbl->GetHotkey (
ShellLink,
HotKey
);
if (!SUCCEEDED(hres)) {
DEBUGMSGW((DBG_WARNING, "Cannot read hot key for link %s", FileName));
return FALSE;
}
return TRUE;
}
BOOL
ExtractShortcutInfoA (
IN PCSTR FileName,
OUT PCSTR *Target,
OUT PCSTR *Params,
OUT PCSTR *WorkDir,
OUT PCSTR *IconPath,
OUT PINT IconNumber,
OUT PWORD HotKey,
OUT BOOL *DosApp,
OUT BOOL *MsDosMode,
OUT PLNK_EXTRA_DATAA ExtraData, OPTIONAL
IN IShellLinkA *ShellLink,
IN IPersistFile *PersistFile
)
{
PCSTR shortcutExt = NULL;
*MsDosMode = FALSE;
*DosApp = FALSE;
*HotKey = 0;
shortcutExt = GetFileExtensionFromPathA (FileName);
if (shortcutExt != NULL) {
if (StringIMatchA (shortcutExt, "LNK")) {
return ExtractShellLinkInfoA (
FileName,
Target,
Params,
WorkDir,
IconPath,
IconNumber,
HotKey,
ShellLink,
PersistFile
);
} else if (StringIMatchA (shortcutExt, "PIF")) {
*DosApp = TRUE;
return ExtractPifInfoA (
FileName,
Target,
Params,
WorkDir,
IconPath,
IconNumber,
MsDosMode,
ExtraData
);
} else {
return FALSE;
}
} else {
return FALSE;
}
}
BOOL
ExtractShortcutInfoW (
IN PCWSTR FileName,
OUT PCWSTR *Target,
OUT PCWSTR *Params,
OUT PCWSTR *WorkDir,
OUT PCWSTR *IconPath,
OUT PINT IconNumber,
OUT PWORD HotKey,
OUT BOOL *DosApp,
OUT BOOL *MsDosMode,
OUT PLNK_EXTRA_DATAW ExtraData, OPTIONAL
IN IShellLinkW *ShellLink,
IN IPersistFile *PersistFile
)
{
PCWSTR shortcutExt = NULL;
*MsDosMode = FALSE;
*DosApp = FALSE;
*HotKey = 0;
shortcutExt = GetFileExtensionFromPathW (FileName);
if (shortcutExt != NULL) {
if (StringIMatchW (shortcutExt, L"LNK")) {
return ExtractShellLinkInfoW (
FileName,
Target,
Params,
WorkDir,
IconPath,
IconNumber,
HotKey,
ShellLink,
PersistFile
);
} else if (StringIMatchW (shortcutExt, L"PIF")) {
*DosApp = TRUE;
return ExtractPifInfoW (
FileName,
Target,
Params,
WorkDir,
IconPath,
IconNumber,
MsDosMode,
ExtraData
);
} else {
return FALSE;
}
} else {
return FALSE;
}
}
BOOL
ModifyShellLinkFileA (
IN PCSTR FileName,
IN PCSTR Target, OPTIONAL
IN PCSTR Params, OPTIONAL
IN PCSTR WorkDir, OPTIONAL
IN PCSTR IconPath, OPTIONAL
IN INT IconNumber,
IN WORD HotKey,
IN PLNK_EXTRA_DATAA ExtraData, OPTIONAL
IN IShellLinkA *ShellLink,
IN IPersistFile *PersistFile
)
{
PCWSTR fileNameW = NULL;
PCWSTR faceNameW;
HRESULT comResult;
__try {
if (!DoesFileExistA (FileName)) {
__leave;
}
if (((Target == NULL) || (Target [0] == 0)) &&
((Params == NULL) || (Params [0] == 0)) &&
((WorkDir == NULL) || (WorkDir [0] == 0)) &&
((IconPath == NULL) || (IconPath [0] == 0)) &&
(HotKey == 0) &&
(ExtraData == NULL)
) {
__leave;
}
fileNameW = ConvertAtoW (FileName);
comResult = PersistFile->lpVtbl->Load(PersistFile, fileNameW, STGM_READ);
if (comResult != S_OK) {
LOGA ((LOG_ERROR, "LINKEDIT: Load failed for %s", FileName));
__leave;
}
if (Target != NULL) {
comResult = ShellLink->lpVtbl->SetPath (ShellLink, Target);
if (comResult != S_OK) {
DEBUGMSGA ((DBG_WARNING, "LINKEDIT: SetPath failed for %s", FileName));
}
}
if (Params != NULL) {
comResult = ShellLink->lpVtbl->SetArguments (ShellLink, Params);
if (comResult != S_OK) {
DEBUGMSGA ((DBG_WARNING, "LINKEDIT: SetArguments failed for %s", FileName));
}
}
if (WorkDir != NULL) {
comResult = ShellLink->lpVtbl->SetWorkingDirectory (ShellLink, WorkDir);
if (comResult != S_OK) {
DEBUGMSGA ((DBG_WARNING, "LINKEDIT: SetWorkingDirectory failed for %s", FileName));
}
}
if (IconPath != NULL) {
comResult = ShellLink->lpVtbl->SetIconLocation (ShellLink, IconPath, IconNumber);
if (comResult != S_OK) {
DEBUGMSGA ((DBG_WARNING, "LINKEDIT: SetIconLocation failed for %s", FileName));
}
}
// NTRAID#NTBUG9-153303-2000/08/01-jimschm Add HotKey processing here
//
// add NT_CONSOLE_PROPS
//
if (ExtraData) {
HRESULT hres;
NT_CONSOLE_PROPS props;
NT_CONSOLE_PROPS *oldProps;
IShellLinkDataList *psldl;
//
// Get a pointer to the IShellLinkDataList interface.
//
hres = ShellLink->lpVtbl->QueryInterface (ShellLink, &IID_IShellLinkDataList, &psldl);
if (!SUCCEEDED (hres)) {
DEBUGMSGA ((DBG_WARNING, "Cannot get IShellLinkDataList interface"));
__leave;
}
ZeroMemory (&props, sizeof (NT_CONSOLE_PROPS));
props.cbSize = sizeof (NT_CONSOLE_PROPS);
props.dwSignature = NT_CONSOLE_PROPS_SIG;
//
// let's try to get the extra data
//
comResult = psldl->lpVtbl->CopyDataBlock (psldl, NT_CONSOLE_PROPS_SIG, &oldProps);
if ((comResult != S_OK) || (oldProps->cbSize != props.cbSize)) {
// no extra data exists. We need to fill some good data for this console
props.wFillAttribute = 0x0007;
props.wPopupFillAttribute = 0x00f5;
props.dwWindowOrigin.X = 0;
props.dwWindowOrigin.Y = 0;
props.nFont = 0;
props.nInputBufferSize = 0;
props.uCursorSize = 0x0019;
props.bInsertMode = FALSE;
props.bAutoPosition = TRUE;
props.uHistoryBufferSize = 0x0032;
props.uNumberOfHistoryBuffers = 0x0004;
props.bHistoryNoDup = FALSE;
props.ColorTable [0] = 0x00000000;
props.ColorTable [1] = 0x00800000;
props.ColorTable [2] = 0x00008000;
props.ColorTable [3] = 0x00808000;
props.ColorTable [4] = 0x00000080;
props.ColorTable [5] = 0x00800080;
props.ColorTable [6] = 0x00008080;
props.ColorTable [7] = 0x00c0c0c0;
props.ColorTable [8] = 0x00808080;
props.ColorTable [9] = 0x00ff0000;
props.ColorTable [10] = 0x0000ff00;
props.ColorTable [11] = 0x00ffff00;
props.ColorTable [12] = 0x000000ff;
props.ColorTable [13] = 0x00ff00ff;
props.ColorTable [14] = 0x0000ffff;
props.ColorTable [15] = 0x00ffffff;
} else {
props.wFillAttribute = oldProps->wFillAttribute;
props.wPopupFillAttribute = oldProps->wPopupFillAttribute;
props.dwWindowOrigin.X = oldProps->dwWindowOrigin.X;
props.dwWindowOrigin.Y = oldProps->dwWindowOrigin.Y;
props.nFont = oldProps->nFont;
props.nInputBufferSize = oldProps->nInputBufferSize;
props.uCursorSize = oldProps->uCursorSize;
props.bInsertMode = oldProps->bInsertMode;
props.bAutoPosition = oldProps->bAutoPosition;
props.uHistoryBufferSize = oldProps->uHistoryBufferSize;
props.uNumberOfHistoryBuffers = oldProps->uNumberOfHistoryBuffers;
props.bHistoryNoDup = oldProps->bHistoryNoDup;
props.ColorTable [0] = oldProps->ColorTable [0];
props.ColorTable [1] = oldProps->ColorTable [1];
props.ColorTable [2] = oldProps->ColorTable [2];
props.ColorTable [3] = oldProps->ColorTable [3];
props.ColorTable [4] = oldProps->ColorTable [4];
props.ColorTable [5] = oldProps->ColorTable [5];
props.ColorTable [6] = oldProps->ColorTable [6];
props.ColorTable [7] = oldProps->ColorTable [7];
props.ColorTable [8] = oldProps->ColorTable [8];
props.ColorTable [9] = oldProps->ColorTable [9];
props.ColorTable [10] = oldProps->ColorTable [10];
props.ColorTable [11] = oldProps->ColorTable [11];
props.ColorTable [12] = oldProps->ColorTable [12];
props.ColorTable [13] = oldProps->ColorTable [13];
props.ColorTable [14] = oldProps->ColorTable [14];
props.ColorTable [15] = oldProps->ColorTable [15];
psldl->lpVtbl->RemoveDataBlock (psldl, NT_CONSOLE_PROPS_SIG);
}
props.dwScreenBufferSize.X = (SHORT)ExtraData->xSize;
props.dwScreenBufferSize.Y = (SHORT)ExtraData->ySize;
props.dwWindowSize.X = (SHORT)ExtraData->xSize;
props.dwWindowSize.Y = (SHORT)ExtraData->ySize;
props.dwFontSize.X = (UINT)ExtraData->xFontSize;
props.dwFontSize.Y = (UINT)ExtraData->yFontSize;
props.uFontFamily = ExtraData->FontFamily;
props.uFontWeight = ExtraData->FontWeight;
faceNameW = ConvertAtoW (ExtraData->FontName);
StringCopyW (props.FaceName, faceNameW);
FreeConvertedStr (faceNameW);
props.bFullScreen = ExtraData->FullScreen;
props.bQuickEdit = ExtraData->QuickEdit;
comResult = psldl->lpVtbl->AddDataBlock (psldl, &props);
if (comResult != S_OK) {
DEBUGMSGA ((DBG_WARNING, "LINKEDIT: AddDataBlock failed for %s", FileName));
}
}
comResult = PersistFile->lpVtbl->Save (PersistFile, fileNameW, FALSE);
if (comResult != S_OK) {
DEBUGMSGA ((DBG_WARNING, "LINKEDIT: Save failed for %s", FileName));
}
comResult = PersistFile->lpVtbl->SaveCompleted (PersistFile, fileNameW);
if (comResult != S_OK) {
DEBUGMSGA ((DBG_WARNING, "LINKEDIT: SaveCompleted failed for %s", FileName));
}
FreeConvertedStr (fileNameW);
fileNameW = NULL;
}
__finally {
if (fileNameW) {
FreeConvertedStr (fileNameW);
fileNameW = NULL;
}
}
return TRUE;
}
BOOL
ModifyShellLinkFileW (
IN PCWSTR FileName,
IN PCWSTR Target, OPTIONAL
IN PCWSTR Params, OPTIONAL
IN PCWSTR WorkDir, OPTIONAL
IN PCWSTR IconPath, OPTIONAL
IN INT IconNumber,
IN WORD HotKey,
IN PLNK_EXTRA_DATAW ExtraData, OPTIONAL
IN IShellLinkW *ShellLink,
IN IPersistFile *PersistFile
)
{
HRESULT comResult;
__try {
if (!DoesFileExistW (FileName)) {
__leave;
}
if (((Target == NULL) || (Target [0] == 0)) &&
((Params == NULL) || (Params [0] == 0)) &&
((WorkDir == NULL) || (WorkDir [0] == 0)) &&
((IconPath == NULL) || (IconPath [0] == 0)) &&
(HotKey == 0) &&
(ExtraData == NULL)
) {
__leave;
}
comResult = PersistFile->lpVtbl->Load(PersistFile, FileName, STGM_READ);
if (comResult != S_OK) {
LOGW ((LOG_ERROR, "LINKEDIT: Load failed for %s", FileName));
__leave;
}
if (Target != NULL) {
comResult = ShellLink->lpVtbl->SetPath (ShellLink, Target);
if (comResult != S_OK) {
DEBUGMSGW ((DBG_WARNING, "LINKEDIT: SetPath failed for %s", FileName));
}
}
if (Params != NULL) {
comResult = ShellLink->lpVtbl->SetArguments (ShellLink, Params);
if (comResult != S_OK) {
DEBUGMSGW ((DBG_WARNING, "LINKEDIT: SetArguments failed for %s", FileName));
}
}
if (WorkDir != NULL) {
comResult = ShellLink->lpVtbl->SetWorkingDirectory (ShellLink, WorkDir);
if (comResult != S_OK) {
DEBUGMSGW ((DBG_WARNING, "LINKEDIT: SetWorkingDirectory failed for %s", FileName));
}
}
if (IconPath != NULL) {
comResult = ShellLink->lpVtbl->SetIconLocation (ShellLink, IconPath, IconNumber);
if (comResult != S_OK) {
DEBUGMSGW ((DBG_WARNING, "LINKEDIT: SetIconLocation failed for %s", FileName));
}
}
// NTRAID#NTBUG9-153303-2000/08/01-jimschm Add HotKey processing here
//
// add NT_CONSOLE_PROPS
//
if (ExtraData) {
HRESULT hres;
NT_CONSOLE_PROPS props;
NT_CONSOLE_PROPS *oldProps;
IShellLinkDataList *psldl;
//
// Get a pointer to the IShellLinkDataList interface.
//
hres = ShellLink->lpVtbl->QueryInterface (ShellLink, &IID_IShellLinkDataList, &psldl);
if (!SUCCEEDED (hres)) {
DEBUGMSGW ((DBG_WARNING, "Cannot get IShellLinkDataList interface"));
__leave;
}
ZeroMemory (&props, sizeof (NT_CONSOLE_PROPS));
props.cbSize = sizeof (NT_CONSOLE_PROPS);
props.dwSignature = NT_CONSOLE_PROPS_SIG;
//
// let's try to get the extra data
//
comResult = psldl->lpVtbl->CopyDataBlock (psldl, NT_CONSOLE_PROPS_SIG, &oldProps);
if ((comResult != S_OK) || (oldProps->cbSize != props.cbSize)) {
// no extra data exists. We need to fill some good data for this console
props.wFillAttribute = 0x0007;
props.wPopupFillAttribute = 0x00f5;
props.dwWindowOrigin.X = 0;
props.dwWindowOrigin.Y = 0;
props.nFont = 0;
props.nInputBufferSize = 0;
props.uCursorSize = 0x0019;
props.bInsertMode = FALSE;
props.bAutoPosition = TRUE;
props.uHistoryBufferSize = 0x0032;
props.uNumberOfHistoryBuffers = 0x0004;
props.bHistoryNoDup = FALSE;
props.ColorTable [0] = 0x00000000;
props.ColorTable [1] = 0x00800000;
props.ColorTable [2] = 0x00008000;
props.ColorTable [3] = 0x00808000;
props.ColorTable [4] = 0x00000080;
props.ColorTable [5] = 0x00800080;
props.ColorTable [6] = 0x00008080;
props.ColorTable [7] = 0x00c0c0c0;
props.ColorTable [8] = 0x00808080;
props.ColorTable [9] = 0x00ff0000;
props.ColorTable [10] = 0x0000ff00;
props.ColorTable [11] = 0x00ffff00;
props.ColorTable [12] = 0x000000ff;
props.ColorTable [13] = 0x00ff00ff;
props.ColorTable [14] = 0x0000ffff;
props.ColorTable [15] = 0x00ffffff;
} else {
props.wFillAttribute = oldProps->wFillAttribute;
props.wPopupFillAttribute = oldProps->wPopupFillAttribute;
props.dwWindowOrigin.X = oldProps->dwWindowOrigin.X;
props.dwWindowOrigin.Y = oldProps->dwWindowOrigin.Y;
props.nFont = oldProps->nFont;
props.nInputBufferSize = oldProps->nInputBufferSize;
props.uCursorSize = oldProps->uCursorSize;
props.bInsertMode = oldProps->bInsertMode;
props.bAutoPosition = oldProps->bAutoPosition;
props.uHistoryBufferSize = oldProps->uHistoryBufferSize;
props.uNumberOfHistoryBuffers = oldProps->uNumberOfHistoryBuffers;
props.bHistoryNoDup = oldProps->bHistoryNoDup;
props.ColorTable [0] = oldProps->ColorTable [0];
props.ColorTable [1] = oldProps->ColorTable [1];
props.ColorTable [2] = oldProps->ColorTable [2];
props.ColorTable [3] = oldProps->ColorTable [3];
props.ColorTable [4] = oldProps->ColorTable [4];
props.ColorTable [5] = oldProps->ColorTable [5];
props.ColorTable [6] = oldProps->ColorTable [6];
props.ColorTable [7] = oldProps->ColorTable [7];
props.ColorTable [8] = oldProps->ColorTable [8];
props.ColorTable [9] = oldProps->ColorTable [9];
props.ColorTable [10] = oldProps->ColorTable [10];
props.ColorTable [11] = oldProps->ColorTable [11];
props.ColorTable [12] = oldProps->ColorTable [12];
props.ColorTable [13] = oldProps->ColorTable [13];
props.ColorTable [14] = oldProps->ColorTable [14];
props.ColorTable [15] = oldProps->ColorTable [15];
psldl->lpVtbl->RemoveDataBlock (psldl, NT_CONSOLE_PROPS_SIG);
}
props.dwScreenBufferSize.X = (SHORT)ExtraData->xSize;
props.dwScreenBufferSize.Y = (SHORT)ExtraData->ySize;
props.dwWindowSize.X = (SHORT)ExtraData->xSize;
props.dwWindowSize.Y = (SHORT)ExtraData->ySize;
props.dwFontSize.X = (UINT)ExtraData->xFontSize;
props.dwFontSize.Y = (UINT)ExtraData->yFontSize;
props.uFontFamily = ExtraData->FontFamily;
props.uFontWeight = ExtraData->FontWeight;
StringCopyW (props.FaceName, ExtraData->FontName);
props.bFullScreen = ExtraData->FullScreen;
props.bQuickEdit = ExtraData->QuickEdit;
comResult = psldl->lpVtbl->AddDataBlock (psldl, &props);
if (comResult != S_OK) {
DEBUGMSGW ((DBG_WARNING, "LINKEDIT: AddDataBlock failed for %s", FileName));
}
}
comResult = PersistFile->lpVtbl->Save (PersistFile, FileName, FALSE);
if (comResult != S_OK) {
DEBUGMSGW ((DBG_WARNING, "LINKEDIT: Save failed for %s", FileName));
}
comResult = PersistFile->lpVtbl->SaveCompleted (PersistFile, FileName);
if (comResult != S_OK) {
DEBUGMSGW ((DBG_WARNING, "LINKEDIT: SaveCompleted failed for %s", FileName));
}
}
__finally {
}
return TRUE;
}
BOOL
ModifyPifFileA (
IN PCSTR FileName,
IN PCSTR Target, OPTIONAL
IN PCSTR Params, OPTIONAL
IN PCSTR WorkDir, OPTIONAL
IN PCSTR IconPath, OPTIONAL
IN INT IconNumber
)
{
PCSTR fileImage = NULL;
HANDLE mapHandle = NULL;
HANDLE fileHandle = INVALID_HANDLE_VALUE;
PSTDPIF stdPif;
PWENHPIF40 wenhPif40;
PW386PIF30 w386ext30;
__try {
fileImage = MapFileIntoMemoryExA (FileName, &fileHandle, &mapHandle, TRUE);
if (fileImage == NULL) {
__leave;
}
__try {
stdPif = (PSTDPIF) fileImage;
if (Target != NULL) {
strncpy (stdPif->startfile, Target, PIFSTARTLOCSIZE);
}
if (Params != NULL) {
strncpy (stdPif->params, Params, PIFPARAMSSIZE);
}
if (WorkDir != NULL) {
strncpy (stdPif->defpath, WorkDir, PIFDEFPATHSIZE);
}
if (IconPath != NULL) {
wenhPif40 = (PWENHPIF40) pFindEnhPifSignature ((PVOID)fileImage, WENHHDRSIG40);
if (wenhPif40 != NULL) {
strncpy (wenhPif40->achIconFileProp, IconPath, PIFDEFFILESIZE);
wenhPif40->wIconIndexProp = (WORD)IconNumber;
}
}
// in all cases we want to take off MSDOS mode otherwise NT won't start these PIFs
w386ext30 = pFindEnhPifSignature ((PVOID)fileImage, W386HDRSIG30);
if (w386ext30) {
w386ext30->PfW386Flags = w386ext30->PfW386Flags & (~fRealMode);
w386ext30->PfW386Flags = w386ext30->PfW386Flags & (~fRealModeSilent);
}
}
__except (1) {
// something went wrong when we tried to read or write PIF file,
// let's just do nothing and exit from here
DEBUGMSGW ((DBG_WARNING, "Exception thrown when processing %s", FileName));
}
}
__finally {
UnmapFile ((PVOID) fileImage, mapHandle, fileHandle);
}
return TRUE;
}
BOOL
ModifyPifFileW (
IN PCWSTR FileName,
IN PCWSTR Target, OPTIONAL
IN PCWSTR Params, OPTIONAL
IN PCWSTR WorkDir, OPTIONAL
IN PCWSTR IconPath, OPTIONAL
IN INT IconNumber
)
{
PCSTR fileImage = NULL;
HANDLE mapHandle = NULL;
HANDLE fileHandle = INVALID_HANDLE_VALUE;
PCSTR AnsiStr = NULL;
PSTDPIF stdPif;
PWENHPIF40 wenhPif40;
PW386PIF30 w386ext30;
__try {
fileImage = MapFileIntoMemoryExW (FileName, &fileHandle, &mapHandle, TRUE);
if (fileImage == NULL) {
__leave;
}
__try {
stdPif = (PSTDPIF) fileImage;
if (Target != NULL) {
AnsiStr = ConvertWtoA (Target);
strncpy (stdPif->startfile, AnsiStr, PIFSTARTLOCSIZE);
FreeConvertedStr (AnsiStr);
}
if (Params != NULL) {
AnsiStr = ConvertWtoA (Params);
strncpy (stdPif->params, AnsiStr, PIFPARAMSSIZE);
FreeConvertedStr (AnsiStr);
}
if (WorkDir != NULL) {
AnsiStr = ConvertWtoA (WorkDir);
strncpy (stdPif->defpath, AnsiStr, PIFDEFPATHSIZE);
FreeConvertedStr (AnsiStr);
}
if (IconPath != NULL) {
wenhPif40 = (PWENHPIF40) pFindEnhPifSignature ((PVOID)fileImage, WENHHDRSIG40);
if (wenhPif40 != NULL) {
AnsiStr = ConvertWtoA (IconPath);
strncpy (wenhPif40->achIconFileProp, AnsiStr, PIFDEFFILESIZE);
FreeConvertedStr (AnsiStr);
wenhPif40->wIconIndexProp = (WORD)IconNumber;
}
}
// in all cases we want to take off MSDOS mode otherwise NT won't start these PIFs
w386ext30 = pFindEnhPifSignature ((PVOID)fileImage, W386HDRSIG30);
if (w386ext30) {
w386ext30->PfW386Flags = w386ext30->PfW386Flags & (~fRealMode);
w386ext30->PfW386Flags = w386ext30->PfW386Flags & (~fRealModeSilent);
}
}
__except (1) {
// something went wrong when we tried to read or write PIF file,
// let's just do nothing and exit from here
DEBUGMSGW ((DBG_WARNING, "Exception thrown when processing %s", FileName));
}
}
__finally {
UnmapFile ((PVOID) fileImage, mapHandle, fileHandle);
}
return TRUE;
}
BOOL
ModifyShortcutFileExA (
IN PCSTR FileName,
IN PCSTR ForcedExtension, OPTIONAL
IN PCSTR Target, OPTIONAL
IN PCSTR Params, OPTIONAL
IN PCSTR WorkDir, OPTIONAL
IN PCSTR IconPath, OPTIONAL
IN INT IconNumber,
IN WORD HotKey,
IN PLNK_EXTRA_DATAA ExtraData, OPTIONAL
IN IShellLinkA *ShellLink,
IN IPersistFile *PersistFile
)
{
PCSTR shortcutExt;
__try {
shortcutExt = ForcedExtension;
if (!shortcutExt) {
shortcutExt = GetFileExtensionFromPathA (FileName);
}
if (shortcutExt) {
if (StringIMatchA (shortcutExt, "LNK")) {
return ModifyShellLinkFileA (
FileName,
Target,
Params,
WorkDir,
IconPath,
IconNumber,
HotKey,
NULL,
ShellLink,
PersistFile
);
} else if (StringIMatchA (shortcutExt, "PIF")) {
return ModifyPifFileA (
FileName,
Target,
Params,
WorkDir,
IconPath,
IconNumber
);
}
}
}
__except (1) {
LOGA ((LOG_ERROR, "Cannot process shortcut %s", FileName));
}
return TRUE;
}
BOOL
ModifyShortcutFileExW (
IN PCWSTR FileName,
IN PCWSTR ForcedExtension, OPTIONAL
IN PCWSTR Target, OPTIONAL
IN PCWSTR Params, OPTIONAL
IN PCWSTR WorkDir, OPTIONAL
IN PCWSTR IconPath, OPTIONAL
IN INT IconNumber,
IN WORD HotKey,
IN PLNK_EXTRA_DATAW ExtraData, OPTIONAL
IN IShellLinkW *ShellLink,
IN IPersistFile *PersistFile
)
{
PCWSTR shortcutExt;
__try {
shortcutExt = ForcedExtension;
if (!shortcutExt) {
shortcutExt = GetFileExtensionFromPathW (FileName);
}
if (shortcutExt) {
if (StringIMatchW (shortcutExt, L"LNK")) {
return ModifyShellLinkFileW (
FileName,
Target,
Params,
WorkDir,
IconPath,
IconNumber,
HotKey,
NULL,
ShellLink,
PersistFile
);
} else if (StringIMatchW (shortcutExt, L"PIF")) {
return ModifyPifFileW (
FileName,
Target,
Params,
WorkDir,
IconPath,
IconNumber
);
}
}
}
__except (1) {
LOGW ((LOG_ERROR, "Cannot process shortcut %s", FileName));
}
return TRUE;
}
Reference in a new issue Copy permalink