mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
761 lines
24 KiB
761 lines
24 KiB
//
|
|
// DEVNODE.C
|
|
//
|
|
#include "sigverif.h"
|
|
#include <initguid.h>
|
|
#include <devguid.h>
|
|
|
|
//
|
|
// Given the full path to a driver, add it to the file list.
|
|
//
|
|
LPFILENODE
|
|
AddDriverFileToList(
|
|
LPTSTR lpDirName,
|
|
LPTSTR lpFullPathName
|
|
)
|
|
{
|
|
LPFILENODE lpFileNode = NULL;
|
|
TCHAR szFullPath[MAX_PATH];
|
|
TCHAR szDirName[MAX_PATH];
|
|
TCHAR szFileName[MAX_PATH];
|
|
LPTSTR lpFilePart;
|
|
BOOL bRet;
|
|
|
|
*szFullPath = 0;
|
|
*szDirName = 0;
|
|
*szFileName = 0;
|
|
|
|
//
|
|
// If no directory is passed in, try to get the full path
|
|
//
|
|
if (!lpDirName || !*lpDirName) {
|
|
|
|
bRet = GetFullPathName(lpFullPathName, MAX_PATH, szDirName, &lpFilePart);
|
|
|
|
if (bRet) {
|
|
|
|
lstrcpy(szFullPath, szDirName);
|
|
|
|
if (lpFilePart && *lpFilePart) {
|
|
|
|
lstrcpy(szFileName, lpFilePart);
|
|
*lpFilePart = 0;
|
|
|
|
if (lstrlen(szDirName) > 3) {
|
|
|
|
*(lpFilePart - 1) = 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
} else {
|
|
|
|
//
|
|
// Use the directory and filename that was passed in to us
|
|
// Expand out lpDirName in case there are any ".." entries
|
|
//
|
|
if (!GetFullPathName(lpDirName, MAX_PATH, szDirName, NULL)) {
|
|
|
|
lstrcpy(szDirName, lpDirName);
|
|
}
|
|
|
|
lstrcpy(szFileName, lpFullPathName);
|
|
}
|
|
|
|
if (*szDirName && *szFileName && !IsFileAlreadyInList(szDirName, szFileName)) {
|
|
|
|
//
|
|
// Create a filenode, based on the directory and filename
|
|
//
|
|
lpFileNode = CreateFileNode(szDirName, szFileName);
|
|
|
|
if (lpFileNode) {
|
|
|
|
InsertFileNodeIntoList(lpFileNode);
|
|
|
|
// Increment the total number of files we've found that meet the search criteria.
|
|
g_App.dwFiles++;
|
|
}
|
|
}
|
|
|
|
return lpFileNode;
|
|
}
|
|
|
|
void
|
|
GetFilesFromInfSection(
|
|
HINF hInf,
|
|
LPTSTR lpFileName,
|
|
LPTSTR lpSectionName
|
|
)
|
|
{
|
|
TCHAR szTarget[MAX_PATH];
|
|
TCHAR szBuffer[MAX_PATH];
|
|
LPTSTR lpBuffer;
|
|
LPTSTR lpString, lpSeparator;
|
|
BOOL bRet;
|
|
DWORD dwRequiredSize;
|
|
INFCONTEXT iContext;
|
|
|
|
ZeroMemory(szTarget, sizeof(szTarget));
|
|
SetupGetTargetPath(hInf, NULL, lpSectionName, szTarget, sizeof(szTarget), NULL);
|
|
|
|
// HYDRA HACK!!!
|
|
//
|
|
// Check to see if the target is %WINDIR% and if %WINDIR% has been redirected, change it back!!
|
|
// We have the real %WINDIR% stored in g_App.szWinDir, so we can stuff that into szTarget.
|
|
// We just have to remember to put back whatever was at the end of szTarget.
|
|
//
|
|
if (GetWindowsDirectory(szBuffer, MAX_PATH) &&
|
|
!_tcsnicmp(szBuffer, szTarget, lstrlen(szBuffer)) &&
|
|
_tcsicmp(g_App.szWinDir, szBuffer)) {
|
|
|
|
lstrcpy(szBuffer, szTarget + lstrlen(szBuffer));
|
|
lstrcpy(szTarget, g_App.szWinDir);
|
|
lstrcat(szTarget, szBuffer);
|
|
}
|
|
|
|
ZeroMemory(&iContext, sizeof(INFCONTEXT));
|
|
bRet = SetupFindFirstLine(hInf, lpSectionName, NULL, &iContext);
|
|
|
|
while (bRet && !g_App.bStopScan) {
|
|
dwRequiredSize = 0;
|
|
bRet = SetupGetLineText(&iContext, NULL, NULL, NULL, NULL, 0, &dwRequiredSize);
|
|
if (dwRequiredSize) {
|
|
|
|
lpBuffer = MALLOC((dwRequiredSize + 1) * sizeof(TCHAR));
|
|
|
|
if (lpBuffer) {
|
|
|
|
bRet = SetupGetLineText(&iContext, NULL, NULL, NULL, lpBuffer, dwRequiredSize, NULL);
|
|
|
|
if (bRet) {
|
|
lpString = lpBuffer;
|
|
|
|
if (lpString && *lpString) {
|
|
// If there's a comma, then terminate the string at the comma
|
|
lpSeparator = _tcschr(lpString, TEXT(','));
|
|
if (lpSeparator) {
|
|
// Null terminate at the comma, so the first entry is a null-terminated string
|
|
*lpSeparator = 0;
|
|
}
|
|
|
|
// Make sure we didn't just terminate ourselves.
|
|
if (*lpString) {
|
|
AddDriverFileToList(szTarget, lpString);
|
|
}
|
|
}
|
|
}
|
|
|
|
FREE(lpBuffer);
|
|
}
|
|
}
|
|
|
|
bRet = SetupFindNextLine(&iContext, &iContext);
|
|
}
|
|
}
|
|
|
|
void
|
|
GetStuffFromInfSection(
|
|
LPTSTR lpFileName,
|
|
LPTSTR lpSectionName
|
|
)
|
|
{
|
|
TCHAR szFullPath[MAX_PATH];
|
|
TCHAR szTarget[MAX_PATH];
|
|
TCHAR szKeyName[MAX_PATH];
|
|
LPTSTR lpString = NULL;
|
|
LPTSTR lpSeparator = NULL;
|
|
LPTSTR lpBuffer = NULL;
|
|
DWORD dwRequiredSize;
|
|
HINF hInf;
|
|
BOOL bRet;
|
|
UINT uError;
|
|
|
|
szFullPath[0] = 0;
|
|
GetFullPathName(lpFileName, MAX_PATH, szFullPath, NULL);
|
|
|
|
//
|
|
// Try opening the INF in the usual INF directory
|
|
//
|
|
hInf = SetupOpenInfFile(szFullPath, NULL, INF_STYLE_WIN4, &uError);
|
|
if (hInf == INVALID_HANDLE_VALUE) {
|
|
|
|
//
|
|
// Add the INF to the file list so it shows up as unscanned.
|
|
//
|
|
AddDriverFileToList(NULL, lpFileName);
|
|
return;
|
|
|
|
} else {
|
|
|
|
//
|
|
// The INF must exist, so add it to the file list for verification!
|
|
//
|
|
AddDriverFileToList(NULL, szFullPath);
|
|
}
|
|
|
|
MyLoadString(szKeyName, IDS_COPYFILES);
|
|
dwRequiredSize = 0;
|
|
bRet = SetupGetLineText(NULL, hInf, lpSectionName, szKeyName, NULL, 0, &dwRequiredSize);
|
|
if (dwRequiredSize) {
|
|
|
|
lpBuffer = MALLOC((dwRequiredSize + 1) * sizeof(TCHAR));
|
|
|
|
if (lpBuffer) {
|
|
|
|
bRet = SetupGetLineText(NULL, hInf, lpSectionName, szKeyName, lpBuffer, dwRequiredSize, NULL);
|
|
if (!bRet) {
|
|
//MyMessageBox(TEXT("SetupGetLineText Failed!"));
|
|
FREE(lpBuffer);
|
|
return;
|
|
}
|
|
}
|
|
|
|
lpString = lpBuffer;
|
|
}
|
|
|
|
ZeroMemory(szTarget, sizeof(szTarget));
|
|
SetupGetTargetPath(hInf, NULL, lpSectionName, szTarget, sizeof(szTarget), NULL);
|
|
|
|
while (lpString && *lpString && !g_App.bStopScan) {
|
|
// If there's a comma, then bump lpSeparator to after the comma
|
|
lpSeparator = _tcschr(lpString, TEXT(','));
|
|
if (lpSeparator) {
|
|
// Null terminate at the comma, so the first entry is a null-terminated string
|
|
*lpSeparator = 0;
|
|
lpSeparator++;
|
|
}
|
|
|
|
//
|
|
// If the section has an '@' symbol, then it is directly referencing a filename
|
|
// Otherwise, it's a section and we need to process that via GetFilesFromInfSection()
|
|
//
|
|
if (*lpString == TEXT('@')) {
|
|
lpString++;
|
|
if (*lpString) {
|
|
AddDriverFileToList(szTarget, lpString);
|
|
}
|
|
} else GetFilesFromInfSection(hInf, lpFileName, lpString);
|
|
|
|
lpString = lpSeparator;
|
|
}
|
|
|
|
if (lpBuffer) {
|
|
|
|
FREE(lpBuffer);
|
|
}
|
|
|
|
SetupCloseInfFile(hInf);
|
|
}
|
|
|
|
UINT
|
|
ScanQueueCallback(
|
|
PVOID Context,
|
|
UINT Notification,
|
|
UINT_PTR Param1,
|
|
UINT_PTR Param2
|
|
)
|
|
{
|
|
LPFILENODE lpFileNode;
|
|
TCHAR szBuffer[MAX_PATH];
|
|
LPTSTR lpFilePart;
|
|
|
|
if ((Notification == SPFILENOTIFY_QUEUESCAN_SIGNERINFO) &&
|
|
Param1) {
|
|
|
|
//
|
|
// Special case for printers:
|
|
// After setupapi copies files from the file queue into their destination
|
|
// location, the printer class installer moves some of these files into
|
|
// other 'special' locations. This can lead to the callback Win32Error
|
|
// returning ERROR_FILE_NOT_FOUND or ERROR_PATH_NOT_FOUND since the file
|
|
// is not present in the location where setupapi put it. So, we will
|
|
// catch this case for printers and not add the file to our list of
|
|
// files to scan. These 'special' printer files will get added later
|
|
// when we call the spooler APIs.
|
|
// Also note that we can't just skip getting the list of files for printers
|
|
// altogether since the printer class installer only moves some of the
|
|
// files that setupapi copies and not all of them.
|
|
//
|
|
if (Context &&
|
|
(IsEqualGUID((LPGUID)Context, &GUID_DEVCLASS_PRINTER)) &&
|
|
((((PFILEPATHS_SIGNERINFO)Param1)->Win32Error == ERROR_FILE_NOT_FOUND) ||
|
|
(((PFILEPATHS_SIGNERINFO)Param1)->Win32Error == ERROR_PATH_NOT_FOUND))) {
|
|
//
|
|
// Assume this was a file moved by the printer class installer. Don't
|
|
// add it to the list of files to be scanned at this time.
|
|
//
|
|
return NO_ERROR;
|
|
}
|
|
|
|
lpFileNode = AddDriverFileToList(NULL,
|
|
(LPTSTR)((PFILEPATHS_SIGNERINFO)Param1)->Target);
|
|
|
|
//
|
|
// Fill in some information into the FILENODE structure since we already
|
|
// scanned the file.
|
|
//
|
|
if (lpFileNode) {
|
|
|
|
lpFileNode->bScanned = TRUE;
|
|
lpFileNode->bSigned = (((PFILEPATHS_SIGNERINFO)Param1)->Win32Error == NO_ERROR);
|
|
|
|
if (lpFileNode->bSigned) {
|
|
|
|
if (((PFILEPATHS_SIGNERINFO)Param1)->CatalogFile) {
|
|
|
|
GetFullPathName(((PFILEPATHS_SIGNERINFO)Param1)->CatalogFile, MAX_PATH, szBuffer, &lpFilePart);
|
|
|
|
lpFileNode->lpCatalog = MALLOC((lstrlen(lpFilePart) + 1) * sizeof(TCHAR));
|
|
|
|
if (lpFileNode->lpCatalog) {
|
|
|
|
lstrcpy(lpFileNode->lpCatalog, lpFilePart);
|
|
}
|
|
}
|
|
|
|
if (((PFILEPATHS_SIGNERINFO)Param1)->DigitalSigner) {
|
|
|
|
lpFileNode->lpSignedBy = MALLOC((lstrlen(((PFILEPATHS_SIGNERINFO)Param1)->DigitalSigner) + 1) * sizeof(TCHAR));
|
|
|
|
if (lpFileNode->lpSignedBy) {
|
|
|
|
lstrcpy(lpFileNode->lpSignedBy, ((PFILEPATHS_SIGNERINFO)Param1)->DigitalSigner);
|
|
}
|
|
}
|
|
|
|
if (((PFILEPATHS_SIGNERINFO)Param1)->Version) {
|
|
|
|
lpFileNode->lpVersion = MALLOC((lstrlen(((PFILEPATHS_SIGNERINFO)Param1)->Version) + 1) * sizeof(TCHAR));
|
|
|
|
if (lpFileNode->lpVersion) {
|
|
|
|
lstrcpy(lpFileNode->lpVersion, ((PFILEPATHS_SIGNERINFO)Param1)->Version);
|
|
}
|
|
}
|
|
|
|
} else {
|
|
//
|
|
// Get the icon (if the file isn't signed) so we can display it in the listview faster.
|
|
//
|
|
MyGetFileInfo(lpFileNode);
|
|
}
|
|
}
|
|
}
|
|
|
|
return NO_ERROR;
|
|
}
|
|
|
|
void
|
|
BuildDriverFileList(
|
|
void
|
|
)
|
|
{
|
|
HDEVINFO hDeviceInfo = INVALID_HANDLE_VALUE;
|
|
SP_DEVINFO_DATA DeviceInfoData;
|
|
SP_DRVINFO_DATA DriverInfoData;
|
|
SP_DEVINSTALL_PARAMS DeviceInstallParams;
|
|
DWORD DeviceMemberIndex;
|
|
HSPFILEQ hFileQueue;
|
|
DWORD ScanResult;
|
|
|
|
//
|
|
// Build up a list of all the devices currently present in the system.
|
|
//
|
|
hDeviceInfo = SetupDiGetClassDevs(NULL,
|
|
NULL,
|
|
NULL,
|
|
DIGCF_ALLCLASSES | DIGCF_PRESENT
|
|
);
|
|
|
|
if (hDeviceInfo == INVALID_HANDLE_VALUE) {
|
|
|
|
goto clean0;
|
|
}
|
|
|
|
DeviceInfoData.cbSize = sizeof(DeviceInfoData);
|
|
DeviceMemberIndex = 0;
|
|
|
|
//
|
|
// Enumerate through the list of present devices.
|
|
//
|
|
while (SetupDiEnumDeviceInfo(hDeviceInfo,
|
|
DeviceMemberIndex++,
|
|
&DeviceInfoData
|
|
) &&
|
|
!g_App.bStopScan) {
|
|
|
|
DeviceInstallParams.cbSize = sizeof(DeviceInstallParams);
|
|
|
|
//
|
|
// Before we call SetupDiBuildDriverInfoList to build up a list of drivers
|
|
// for this device we first need to set the DI_FLAGSEX_INSTALLEDDRIVER flag
|
|
// (which tells the API to only include the currently installed driver in
|
|
// the list) and the DI_FLAGSEX_ALLOWEXCLUDEDRVS (allow ExcludeFromSelect
|
|
// devices in the list).
|
|
//
|
|
if (SetupDiGetDeviceInstallParams(hDeviceInfo,
|
|
&DeviceInfoData,
|
|
&DeviceInstallParams
|
|
)) {
|
|
|
|
DeviceInstallParams.FlagsEx = (DI_FLAGSEX_INSTALLEDDRIVER |
|
|
DI_FLAGSEX_ALLOWEXCLUDEDDRVS);
|
|
|
|
if (SetupDiSetDeviceInstallParams(hDeviceInfo,
|
|
&DeviceInfoData,
|
|
&DeviceInstallParams
|
|
) &&
|
|
SetupDiBuildDriverInfoList(hDeviceInfo,
|
|
&DeviceInfoData,
|
|
SPDIT_CLASSDRIVER
|
|
)) {
|
|
|
|
//
|
|
// Now we will get the one driver node that is in the list that
|
|
// was just built and make it the selected driver node.
|
|
//
|
|
DriverInfoData.cbSize = sizeof(DriverInfoData);
|
|
|
|
if (SetupDiEnumDriverInfo(hDeviceInfo,
|
|
&DeviceInfoData,
|
|
SPDIT_CLASSDRIVER,
|
|
0,
|
|
&DriverInfoData
|
|
) &&
|
|
SetupDiSetSelectedDriver(hDeviceInfo,
|
|
&DeviceInfoData,
|
|
&DriverInfoData
|
|
)) {
|
|
|
|
hFileQueue = SetupOpenFileQueue();
|
|
|
|
if (hFileQueue != INVALID_HANDLE_VALUE) {
|
|
|
|
//
|
|
// Set the FileQueue parameter to the file queue we just
|
|
// created and set the DI_NOVCP flag.
|
|
//
|
|
// The call SetupDiCallClassInstaller with DIF_INSTALLDEVICEFILES
|
|
// to build up a queue of all the files that are copied for
|
|
// this driver node.
|
|
//
|
|
DeviceInstallParams.FileQueue = hFileQueue;
|
|
DeviceInstallParams.Flags |= DI_NOVCP;
|
|
|
|
if (SetupDiSetDeviceInstallParams(hDeviceInfo,
|
|
&DeviceInfoData,
|
|
&DeviceInstallParams
|
|
) &&
|
|
SetupDiCallClassInstaller(DIF_INSTALLDEVICEFILES,
|
|
hDeviceInfo,
|
|
&DeviceInfoData
|
|
)) {
|
|
|
|
//
|
|
// Scan the file queue and have it call our callback
|
|
// function for each file in the queue.
|
|
//
|
|
SetupScanFileQueue(hFileQueue,
|
|
SPQ_SCAN_USE_CALLBACK_SIGNERINFO,
|
|
NULL,
|
|
ScanQueueCallback,
|
|
(PVOID)&(DeviceInfoData.ClassGuid),
|
|
&ScanResult
|
|
);
|
|
|
|
|
|
//
|
|
// Dereference the file queue so we can close it.
|
|
//
|
|
DeviceInstallParams.FileQueue = NULL;
|
|
DeviceInstallParams.Flags &= ~DI_NOVCP;
|
|
SetupDiSetDeviceInstallParams(hDeviceInfo,
|
|
&DeviceInfoData,
|
|
&DeviceInstallParams
|
|
);
|
|
}
|
|
|
|
SetupCloseFileQueue(hFileQueue);
|
|
}
|
|
}
|
|
|
|
SetupDiDestroyDriverInfoList(hDeviceInfo,
|
|
&DeviceInfoData,
|
|
SPDIT_CLASSDRIVER
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
clean0:
|
|
if (hDeviceInfo != INVALID_HANDLE_VALUE) {
|
|
|
|
SetupDiDestroyDeviceInfoList(hDeviceInfo);
|
|
}
|
|
}
|
|
|
|
void
|
|
BuildPrinterFileList(
|
|
void
|
|
)
|
|
{
|
|
BOOL bRet;
|
|
DWORD dwBytesNeeded = 0;
|
|
DWORD dwDrivers = 0;
|
|
LPBYTE lpBuffer = NULL, lpTemp = NULL;
|
|
LPTSTR lpFileName;
|
|
DRIVER_INFO_3 DriverInfo;
|
|
PDRIVER_INFO_3 lpDriverInfo;
|
|
TCHAR szBuffer[MAX_PATH];
|
|
LPFILENODE lpFileNode = NULL;
|
|
|
|
ZeroMemory(&DriverInfo, sizeof(DRIVER_INFO_3));
|
|
bRet = EnumPrinterDrivers( NULL,
|
|
SIGVERIF_PRINTER_ENV,
|
|
3,
|
|
(LPBYTE) &DriverInfo,
|
|
sizeof(DRIVER_INFO_3),
|
|
&dwBytesNeeded,
|
|
&dwDrivers);
|
|
|
|
if (!bRet && dwBytesNeeded > 0) {
|
|
|
|
lpBuffer = MALLOC(dwBytesNeeded);
|
|
|
|
//
|
|
// If we can't get any memory then just bail out of this function
|
|
//
|
|
if (!lpBuffer) {
|
|
|
|
return;
|
|
}
|
|
|
|
bRet = EnumPrinterDrivers( NULL,
|
|
SIGVERIF_PRINTER_ENV,
|
|
3,
|
|
(LPBYTE) lpBuffer,
|
|
dwBytesNeeded,
|
|
&dwBytesNeeded,
|
|
&dwDrivers);
|
|
}
|
|
|
|
if (dwDrivers > 0) {
|
|
|
|
//
|
|
// By default, go into the System directory, since Win9x doesn't give full paths to drivers.
|
|
//
|
|
GetSystemDirectory(szBuffer, MAX_PATH);
|
|
SetCurrentDirectory(szBuffer);
|
|
|
|
for (lpTemp = lpBuffer; dwDrivers > 0; dwDrivers--) {
|
|
|
|
lpDriverInfo = (PDRIVER_INFO_3) lpTemp;
|
|
|
|
if (lpDriverInfo->pName) {
|
|
|
|
if (lpDriverInfo->pDriverPath && *lpDriverInfo->pDriverPath) {
|
|
lpFileNode = AddDriverFileToList(NULL, lpDriverInfo->pDriverPath);
|
|
|
|
if (lpFileNode) {
|
|
lpFileNode->bValidateAgainstAnyOs = TRUE;
|
|
}
|
|
}
|
|
|
|
if (lpDriverInfo->pDataFile && *lpDriverInfo->pDataFile) {
|
|
lpFileNode = AddDriverFileToList(NULL, lpDriverInfo->pDataFile);
|
|
|
|
if (lpFileNode) {
|
|
lpFileNode->bValidateAgainstAnyOs = TRUE;
|
|
}
|
|
}
|
|
|
|
if (lpDriverInfo->pConfigFile && *lpDriverInfo->pConfigFile) {
|
|
lpFileNode = AddDriverFileToList(NULL, lpDriverInfo->pConfigFile);
|
|
|
|
if (lpFileNode) {
|
|
lpFileNode->bValidateAgainstAnyOs = TRUE;
|
|
}
|
|
}
|
|
|
|
if (lpDriverInfo->pHelpFile && *lpDriverInfo->pHelpFile) {
|
|
lpFileNode = AddDriverFileToList(NULL, lpDriverInfo->pHelpFile);
|
|
|
|
if (lpFileNode) {
|
|
lpFileNode->bValidateAgainstAnyOs = TRUE;
|
|
}
|
|
}
|
|
|
|
lpFileName = lpDriverInfo->pDependentFiles;
|
|
|
|
while (lpFileName && *lpFileName) {
|
|
|
|
lpFileNode = AddDriverFileToList(NULL, lpFileName);
|
|
|
|
if (lpFileNode) {
|
|
lpFileNode->bValidateAgainstAnyOs = TRUE;
|
|
}
|
|
|
|
for (;*lpFileName;lpFileName++);
|
|
lpFileName++;
|
|
}
|
|
}
|
|
|
|
lpTemp += sizeof(DRIVER_INFO_3);
|
|
}
|
|
}
|
|
|
|
if (lpBuffer) {
|
|
|
|
FREE(lpBuffer);
|
|
}
|
|
}
|
|
|
|
BOOL
|
|
GetBVTFileList(
|
|
LPTSTR lpDirName,
|
|
LPTSTR lpFileName
|
|
)
|
|
{
|
|
TCHAR szBuffer[MAX_PATH];
|
|
LPTSTR lpString, lpFilePart;
|
|
|
|
//
|
|
// If the user specified the "/BVT:" switch, we want to grab the string after it
|
|
//
|
|
MyLoadString(szBuffer, IDS_BVT);
|
|
lpString = MyStrStr(GetCommandLine(), szBuffer);
|
|
|
|
if (lpString && *lpString) {
|
|
|
|
//
|
|
// Switch back into the original startup directory
|
|
//
|
|
SetCurrentDirectory(g_App.szAppDir);
|
|
|
|
lpString += lstrlen(szBuffer);
|
|
|
|
if (!EXIST(lpString)) {
|
|
return FALSE;
|
|
}
|
|
|
|
GetFullPathName(lpString, MAX_PATH, lpDirName, &lpFilePart);
|
|
lstrcpy(lpFileName, lpDirName);
|
|
*lpFilePart = 0;
|
|
SetCurrentDirectory(lpDirName);
|
|
return TRUE;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
BOOL
|
|
IsNTServer(
|
|
void
|
|
)
|
|
{
|
|
BOOL bRet = FALSE;
|
|
HKEY hKey;
|
|
LONG lRet;
|
|
DWORD dwSize, dwType;
|
|
TCHAR szBuffer[MAX_PATH];
|
|
TCHAR szRegBuffer[MAX_PATH];
|
|
|
|
// Open HKLM\System\CurrentControlSet\Control\ProductOptions
|
|
MyLoadString(szRegBuffer, IDS_REG_PRODUCTOPTIONS);
|
|
lRet = RegOpenKey(HKEY_LOCAL_MACHINE, szRegBuffer, &hKey);
|
|
if (lRet == ERROR_SUCCESS) {
|
|
// Now we need to query the ProductType value
|
|
ZeroMemory(szBuffer, sizeof(szBuffer));
|
|
dwSize = sizeof(szBuffer);
|
|
MyLoadString(szRegBuffer, IDS_REG_PRODUCTTYPE);
|
|
lRet = RegQueryValueEx(hKey, szRegBuffer, NULL, &dwType, (PVOID)szBuffer, &dwSize);
|
|
if (lRet == ERROR_SUCCESS) {
|
|
// Check to see if ProductType contained "ServerNT"
|
|
MyLoadString(szRegBuffer, IDS_PRODUCT_SERVER);
|
|
if (!_tcsnicmp(szRegBuffer, szBuffer, lstrlen(szRegBuffer))) {
|
|
// If so, then we want to return TRUE!
|
|
bRet = TRUE;
|
|
}
|
|
}
|
|
RegCloseKey(hKey);
|
|
}
|
|
|
|
return bRet;
|
|
}
|
|
|
|
BOOL
|
|
IsProcessorAlpha(
|
|
void
|
|
)
|
|
{
|
|
OSVERSIONINFO osinfo;
|
|
SYSTEM_INFO sysinfo;
|
|
|
|
ZeroMemory(&osinfo, sizeof(OSVERSIONINFO));
|
|
osinfo.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
|
|
GetVersionEx(&osinfo);
|
|
|
|
if (osinfo.dwPlatformId == VER_PLATFORM_WIN32_NT) {
|
|
ZeroMemory(&sysinfo, sizeof(SYSTEM_INFO));
|
|
GetSystemInfo(&sysinfo);
|
|
if (sysinfo.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_ALPHA) {
|
|
return TRUE;
|
|
}
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
void
|
|
BuildCoreFileList(
|
|
void
|
|
)
|
|
{
|
|
LPFILENODE lpFileNode;
|
|
TCHAR szInfName[MAX_PATH];
|
|
TCHAR szSectionName[MAX_PATH];
|
|
TCHAR szBuffer[MAX_PATH];
|
|
|
|
//
|
|
// Check to see if the "/BVT:" switch was specified for a custom file list
|
|
// If so, then don't verify that the file is digitally signed.
|
|
//
|
|
if (GetBVTFileList(szBuffer, szInfName)) {
|
|
|
|
GetCurrentDirectory(MAX_PATH, szBuffer);
|
|
|
|
MyLoadString(szSectionName, IDS_MASTERFILELIST);
|
|
GetStuffFromInfSection(szInfName, szSectionName);
|
|
|
|
if (IsNTServer()) {
|
|
|
|
SetCurrentDirectory(szBuffer);
|
|
MyLoadString(szSectionName, IDS_SERVERFILELIST);
|
|
GetStuffFromInfSection(szInfName, szSectionName);
|
|
}
|
|
|
|
if (!IsProcessorAlpha()) {
|
|
|
|
SetCurrentDirectory(szBuffer);
|
|
MyLoadString(szSectionName, IDS_X86FILELIST);
|
|
GetStuffFromInfSection(szInfName, szSectionName);
|
|
}
|
|
|
|
} else {
|
|
|
|
PROTECTED_FILE_DATA pfd;
|
|
|
|
//
|
|
// The user didn't use "/BVT:", so call SfcGetNextProtectedFile to get
|
|
// the list of files that SFC protects.
|
|
//
|
|
|
|
pfd.FileNumber = 0;
|
|
|
|
while (!g_App.bStopScan && SfcGetNextProtectedFile(NULL, &pfd)) {
|
|
|
|
AddDriverFileToList(NULL, pfd.FileName);
|
|
}
|
|
}
|
|
}
|