413 lines
12 KiB
C++
413 lines
12 KiB
C++
/*++
|
|
|
|
Copyright (c) 2000 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
DeleteSpecifiedFiles.cpp
|
|
|
|
Abstract:
|
|
|
|
This SHIM renames the MFC42loc.dll that is installed by the APP in the
|
|
%windir%\system32 directory and sets up the temporary file for destruction.
|
|
|
|
Notes:
|
|
|
|
This app. places the MFC42loc.dll into the %windir%\system32 directory even on a English language
|
|
locale thereby forcing some APPS to use it and thereby some 'Dialogs' and 'Message boxes' are messed up.
|
|
|
|
History:
|
|
|
|
08/21/2000 prashkud Created
|
|
|
|
--*/
|
|
|
|
#include "precomp.h"
|
|
#include "CharVector.h"
|
|
#include "strsafe.h"
|
|
#include <new> // for inplace new
|
|
|
|
IMPLEMENT_SHIM_BEGIN(DeleteSpecifiedFiles)
|
|
#include "ShimHookMacro.h"
|
|
|
|
APIHOOK_ENUM_BEGIN
|
|
APIHOOK_ENUM_END
|
|
|
|
class FILENAME_PATH
|
|
{
|
|
public:
|
|
CString FileName;
|
|
CString FileVersion;
|
|
BOOL bPresent;
|
|
|
|
FILENAME_PATH()
|
|
{
|
|
bPresent = FALSE;
|
|
}
|
|
};
|
|
|
|
class FileNamePathList : public VectorT<FILENAME_PATH>
|
|
{
|
|
};
|
|
|
|
VectorT<FILENAME_PATH> *g_StFileNamePath = NULL;
|
|
|
|
|
|
/*++
|
|
|
|
This function checks the 'FileVersion' of the first variable 'szFileName'
|
|
matches that of the 2nd parameter 'szFileVersion' and if it matches returns
|
|
TRUE else returns FALSE.
|
|
|
|
--*/
|
|
|
|
BOOL
|
|
FileCheckVersion(
|
|
const CString & csFileName,
|
|
const CString & csFileVersion
|
|
)
|
|
{
|
|
DWORD dwDummy;
|
|
PVOID pVersionInfo;
|
|
UINT cbTranslate, i;
|
|
WCHAR SubBlock[100];
|
|
LPWSTR lpBuffer;
|
|
DWORD dwBytes;
|
|
WORD wLangID;
|
|
struct LANGANDCODEPAGE
|
|
{
|
|
WORD wLanguage;
|
|
WORD wCodePage;
|
|
} *lpTranslate;
|
|
DWORD dwVersionInfoSize;
|
|
|
|
//
|
|
// There is no File Version specified. So, no point in going ahead. Return TRUE.
|
|
//
|
|
if (csFileVersion.IsEmpty())
|
|
{
|
|
return TRUE;
|
|
}
|
|
|
|
dwVersionInfoSize = GetFileVersionInfoSizeW((LPWSTR)csFileName.Get(), &dwDummy);
|
|
|
|
if (dwVersionInfoSize > 0)
|
|
{
|
|
pVersionInfo = malloc(dwVersionInfoSize);
|
|
if (pVersionInfo)
|
|
{
|
|
if (0 != GetFileVersionInfoW(
|
|
(LPWSTR)csFileName.Get(),
|
|
0,
|
|
dwVersionInfoSize,
|
|
pVersionInfo
|
|
))
|
|
{
|
|
// Now, pVersionInfo contains the required version block.
|
|
// Use it with VerQueryValue to get the
|
|
// the language info that is needed
|
|
// Get System locale before and note down the language for the system
|
|
// Read the list of languages and code pages.
|
|
|
|
if (VerQueryValueW(
|
|
pVersionInfo,
|
|
L"\\VarFileInfo\\Translation",
|
|
(LPVOID*)&lpTranslate,
|
|
&cbTranslate
|
|
))
|
|
{
|
|
//
|
|
// Read the language string each language and code page.
|
|
//
|
|
|
|
for (i=0; i < (cbTranslate/sizeof(struct LANGANDCODEPAGE)); i++)
|
|
{
|
|
if (StringCbPrintfW(
|
|
SubBlock,
|
|
(sizeof(SubBlock) - 1),
|
|
L"\\StringFileInfo\\%04x%04x\\FileVersion",
|
|
lpTranslate[i].wLanguage,
|
|
lpTranslate[i].wCodePage
|
|
) < 0)
|
|
{
|
|
DPFN(
|
|
eDbgLevelError,
|
|
"Long string - Cannot handle");
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
SubBlock[ARRAYSIZE(SubBlock) - 1] = '\0';
|
|
}
|
|
|
|
//
|
|
// Retrieve FileVersion for language and code page "i" from the pVersionInfo.
|
|
//
|
|
if (VerQueryValueW(
|
|
pVersionInfo,
|
|
SubBlock,
|
|
(LPVOID*)&lpBuffer,
|
|
(UINT*)&dwBytes))
|
|
{
|
|
if (!(csFileVersion.Compare(lpBuffer)))
|
|
{
|
|
DPFN(
|
|
eDbgLevelInfo,
|
|
"Version string for current file is %S,%S",
|
|
lpBuffer, csFileVersion.Get());
|
|
|
|
free(pVersionInfo);
|
|
return TRUE;
|
|
}
|
|
}
|
|
} // for loop
|
|
}
|
|
|
|
}
|
|
free(pVersionInfo);
|
|
}
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
/*++
|
|
|
|
This function as the name suggests deletes the file or if it is in use moves
|
|
it to the 'Temp' folder.
|
|
|
|
--*/
|
|
|
|
VOID
|
|
DeleteFiles()
|
|
{
|
|
for (int i = 0; i < g_StFileNamePath->Size(); ++i)
|
|
{
|
|
const FILENAME_PATH & fnp = g_StFileNamePath->Get(i);
|
|
|
|
DPFN( eDbgLevelSpew, "DeleteFiles file(%S) version(%S)", fnp.FileName.Get(), fnp.FileVersion.Get());
|
|
|
|
if (!fnp.bPresent)
|
|
{
|
|
//
|
|
// CheckFileVersion
|
|
//
|
|
if (FileCheckVersion(fnp.FileName, fnp.FileVersion))
|
|
{
|
|
LOGN(eDbgLevelError,"Deleting file %S.", fnp.FileName.Get());
|
|
|
|
// Delete the file..
|
|
if (!DeleteFileW(fnp.FileName))
|
|
{
|
|
CString csTempDir;
|
|
CString csTempPath;
|
|
|
|
LOGN(eDbgLevelError,"Moving file %S.", fnp.FileName.Get());
|
|
//
|
|
// Could not delete.Retry by renaming it and then deleting it.
|
|
// New file is %windir%\Temp
|
|
//
|
|
csTempDir.GetTempPathW();
|
|
csTempPath.GetTempFileNameW(csTempDir, L"XXXX", 0);
|
|
|
|
if (MoveFileExW( fnp.FileName, csTempPath, MOVEFILE_REPLACE_EXISTING ))
|
|
{
|
|
SetFileAttributesW(
|
|
csTempPath,
|
|
FILE_ATTRIBUTE_ARCHIVE |
|
|
FILE_ATTRIBUTE_TEMPORARY);
|
|
|
|
DeleteFileW(csTempPath);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/*++
|
|
|
|
This function checks for the existence of the file specified on the commandline .
|
|
This is called during the DLL_PROCESS_ATTACH notification
|
|
|
|
--*/
|
|
|
|
BOOL
|
|
CheckFileExistence()
|
|
{
|
|
// If any among the list is not present, mark it as 'Not Present' and only those marked as
|
|
// 'Not present' will be deleted.
|
|
|
|
BOOL bFileDoesNotExist = FALSE;
|
|
WIN32_FIND_DATAW StWin32FileData;
|
|
|
|
for (int i = 0; i < g_StFileNamePath->Size(); ++i)
|
|
{
|
|
FILENAME_PATH & fnp = g_StFileNamePath->Get(i);
|
|
|
|
DPFN( eDbgLevelSpew, "CheckFileExistence file(%S) version(%S)", fnp.FileName.Get(), fnp.FileVersion.Get());
|
|
|
|
HANDLE hTempFile = FindFirstFileW(fnp.FileName, &StWin32FileData);
|
|
if (INVALID_HANDLE_VALUE != hTempFile)
|
|
{
|
|
FindClose(hTempFile);
|
|
|
|
//
|
|
// File is present. Check its version if given.
|
|
//
|
|
if (FileCheckVersion(fnp.FileName, fnp.FileVersion))
|
|
{
|
|
fnp.bPresent = TRUE;
|
|
}
|
|
else
|
|
{
|
|
bFileDoesNotExist = TRUE;
|
|
fnp.bPresent = FALSE;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
bFileDoesNotExist = TRUE;
|
|
fnp.bPresent = FALSE;
|
|
}
|
|
}
|
|
|
|
return bFileDoesNotExist;
|
|
}
|
|
|
|
/*++
|
|
|
|
The command line can contain FileName:Path:VersionString,FileName1:Path1:VersionString1 etc....
|
|
Eg. Ole2.dll:system:604.5768.94567,MFC42.dll:0:,Foo.dll:d:\program Files\DisneyInteractive etc..
|
|
'system' implies the %windir%. '0' implies that the filename itself is a fully qualified path
|
|
OR one has the option of giving the path seperately OR it can be left blank.
|
|
|
|
--*/
|
|
|
|
BOOL
|
|
ParseCommandLine(LPCSTR lpszCommandLine)
|
|
{
|
|
CSTRING_TRY
|
|
{
|
|
g_StFileNamePath = new VectorT<FILENAME_PATH>;
|
|
if (!g_StFileNamePath)
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
CStringToken csCommandLine(COMMAND_LINE, ":,;");
|
|
CString csTok;
|
|
CString csFilePart;
|
|
DWORD dwState = 0;
|
|
|
|
|
|
while (csCommandLine.GetToken(csTok))
|
|
{
|
|
FILENAME_PATH fnp;
|
|
|
|
switch(dwState)
|
|
{
|
|
case 0:
|
|
dwState++;
|
|
fnp.FileName = csTok;
|
|
break;
|
|
|
|
case 1:
|
|
// The file part needs to be stored.
|
|
csFilePart = fnp.FileName;
|
|
|
|
if (csTok.CompareNoCase(L"system") == 0)
|
|
{
|
|
fnp.FileName.GetSystemDirectoryW();
|
|
fnp.FileName.AppendPath(csFilePart);
|
|
}
|
|
else
|
|
{
|
|
fnp.FileName = csTok;
|
|
fnp.FileName.AppendPath(csFilePart);
|
|
}
|
|
dwState++;
|
|
break;
|
|
|
|
case 2:
|
|
dwState = 0;
|
|
fnp.FileVersion = csTok;
|
|
|
|
DPFN( eDbgLevelInfo, "ParseCommandLine file(%S) version(%S)", fnp.FileName.Get(), fnp.FileVersion.Get());
|
|
|
|
// Found all three states, add it to the list
|
|
if (!g_StFileNamePath->AppendConstruct(fnp))
|
|
{
|
|
// Append failed, stop parsing
|
|
return FALSE;
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
CSTRING_CATCH
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
/*++
|
|
|
|
Register hooked functions
|
|
|
|
--*/
|
|
|
|
BOOL
|
|
NOTIFY_FUNCTION(
|
|
DWORD fdwReason
|
|
)
|
|
{
|
|
static BOOL fDidNotExist = FALSE;
|
|
|
|
switch (fdwReason)
|
|
{
|
|
case SHIM_STATIC_DLLS_INITIALIZED:
|
|
{
|
|
if (ParseCommandLine(COMMAND_LINE))
|
|
{
|
|
//
|
|
// Ok...CommandLine is in place...Now Check for those Files...
|
|
// If any one file exists, then we are not responsible. We bail out..
|
|
//
|
|
fDidNotExist = CheckFileExistence();
|
|
}
|
|
break;
|
|
}
|
|
|
|
case DLL_PROCESS_DETACH:
|
|
{
|
|
//
|
|
// Check for the specified file at the specified location. If it existed prior to
|
|
// to this in PROCESS_DLL_ATTACH, it is not installed us...Just bail out !
|
|
// If the file did not exist before, it is our problem and we should remove them.
|
|
//
|
|
if (fDidNotExist)
|
|
{
|
|
DeleteFiles();
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
HOOK_BEGIN
|
|
|
|
CALL_NOTIFY_FUNCTION
|
|
|
|
HOOK_END
|
|
|
|
|
|
IMPLEMENT_SHIM_END
|
|
|
|
|