167 lines
5.2 KiB
C
167 lines
5.2 KiB
C
|
//
|
||
|
|
// SIGVERIF.H
|
||
|
|
//
|
||
|
|
#include <windows.h>
|
||
|
|
#include <windowsx.h>
|
||
|
|
#include <shellapi.h>
|
||
|
|
#include <commctrl.h>
|
||
|
|
#include <setupapi.h>
|
||
|
|
#include <cfgmgr32.h>
|
||
|
|
#include <tchar.h>
|
||
|
|
#include <shlobj.h>
|
||
|
|
#include <shlwapi.h>
|
||
|
|
#include <winspool.h>
|
||
|
|
#include <imagehlp.h>
|
||
|
|
#include <capi.h>
|
||
|
|
#include <softpub.h>
|
||
|
|
#include <sfc.h>
|
||
|
|
#include "resource.h"
|
||
|
|
|
||
|
|
// Macros and pre-defined values
|
||
|
|
#define cbX(X) sizeof(X)
|
||
|
|
#define cA(a) (cbX(a)/cbX(a[0]))
|
||
|
|
#define MALLOC(x) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, (x))
|
||
|
|
#define FREE(x) if (x) { HeapFree(GetProcessHeap(), 0, (x)); x = NULL; }
|
||
|
|
#define EXIST(x) (GetFileAttributes(x) != 0xFFFFFFFF)
|
||
|
|
#define MAX_INT 0x7FFFFFFF
|
||
|
|
#define HASH_SIZE 100
|
||
|
|
#define NUM_PAGES 2
|
||
|
|
|
||
|
|
// Registry key/value names for storing previous settings
|
||
|
|
#define SIGVERIF_HKEY HKEY_CURRENT_USER
|
||
|
|
#define SIGVERIF_KEY TEXT("Software\\Microsoft\\Sigverif")
|
||
|
|
#define SIGVERIF_FLAGS TEXT("Flags")
|
||
|
|
#define SIGVERIF_LOGNAME TEXT("Logname")
|
||
|
|
|
||
|
|
#define SIGVERIF_PRINTER_ENV TEXT("All")
|
||
|
|
|
||
|
|
// This structure holds all the information for a specific file.
|
||
|
|
typedef struct tagFileNode
|
||
|
|
{
|
||
|
|
LPTSTR lpFileName;
|
||
|
|
LPTSTR lpDirName;
|
||
|
|
LPTSTR lpVersion;
|
||
|
|
LPTSTR lpCatalog;
|
||
|
|
LPTSTR lpSignedBy;
|
||
|
|
LPTSTR lpTypeName;
|
||
|
|
INT iIcon;
|
||
|
|
BOOL bSigned;
|
||
|
|
BOOL bScanned;
|
||
|
|
BOOL bValidateAgainstAnyOs;
|
||
|
|
DWORD LastError;
|
||
|
|
SYSTEMTIME LastModified;
|
||
|
|
struct tagFileNode *next;
|
||
|
|
} FILENODE, *LPFILENODE;
|
||
|
|
|
||
|
|
// This structure is used by walkpath to keep track of subdirectories
|
||
|
|
typedef struct tagDirNode {
|
||
|
|
TCHAR DirName[MAX_PATH];
|
||
|
|
struct tagDirNode *next;
|
||
|
|
} DIRNODE, *LPDIRNODE;
|
||
|
|
|
||
|
|
// This structure is used when we walk the devicemanager list.
|
||
|
|
typedef struct _DeviceTreeNode
|
||
|
|
{
|
||
|
|
struct _DeviceTreeNode *Child;
|
||
|
|
struct _DeviceTreeNode *Sibling;
|
||
|
|
DEVINST DevInst;
|
||
|
|
TCHAR Driver[MAX_PATH];
|
||
|
|
} DEVTREENODE, *PDEVTREENODE;
|
||
|
|
|
||
|
|
typedef struct _DeviceTreeData
|
||
|
|
{
|
||
|
|
HDEVINFO hDeviceInfo;
|
||
|
|
DEVTREENODE RootNode;
|
||
|
|
} DEVICETREE, *PDEVICETREE;
|
||
|
|
|
||
|
|
// This is our global data structure that hold our global variables.
|
||
|
|
typedef struct tagAppData
|
||
|
|
{
|
||
|
|
HWND hDlg;
|
||
|
|
HWND hLogging;
|
||
|
|
HWND hSearch;
|
||
|
|
HICON hIcon;
|
||
|
|
HINSTANCE hInstance;
|
||
|
|
TCHAR szScanPath[MAX_PATH];
|
||
|
|
TCHAR szScanPattern[MAX_PATH];
|
||
|
|
TCHAR szAppDir[MAX_PATH];
|
||
|
|
TCHAR szLogFile[MAX_PATH];
|
||
|
|
TCHAR szWinDir[MAX_PATH];
|
||
|
|
LPFILENODE lpFileList;
|
||
|
|
LPFILENODE lpFileLast;
|
||
|
|
HCATADMIN hCatAdmin;
|
||
|
|
DWORD dwFiles;
|
||
|
|
DWORD dwSigned;
|
||
|
|
DWORD dwUnsigned;
|
||
|
|
BOOL bNoBVT;
|
||
|
|
BOOL bNoDev;
|
||
|
|
BOOL bNoPRN;
|
||
|
|
BOOL bOverwrite;
|
||
|
|
BOOL bLoggingEnabled;
|
||
|
|
BOOL bLogToRoot;
|
||
|
|
BOOL bFullSystemScan;
|
||
|
|
BOOL bScanning;
|
||
|
|
BOOL bStopScan;
|
||
|
|
BOOL bUserScan;
|
||
|
|
BOOL bSubFolders;
|
||
|
|
DWORD LastError;
|
||
|
|
} GAPPDATA, *LPGAPPDATA;
|
||
|
|
|
||
|
|
// Global function prototypes
|
||
|
|
BOOL BrowseForFolder(HWND hwnd, LPTSTR lpszBuf);
|
||
|
|
void BuildFileList(LPTSTR lpPathName);
|
||
|
|
BOOL VerifyFileList(void);
|
||
|
|
BOOL VerifyFileNode(LPFILENODE lpFileNode);
|
||
|
|
void MyLoadString(LPTSTR lpString, UINT uId);
|
||
|
|
void MyMessageBox(LPTSTR lpString);
|
||
|
|
void MyErrorBox(LPTSTR lpString);
|
||
|
|
void MyErrorBoxId(UINT uId);
|
||
|
|
void MyMessageBoxId(UINT uId);
|
||
|
|
UINT MyGetWindowsDirectory(LPTSTR lpDirName, UINT uSize);
|
||
|
|
LPTSTR MyStrStr(LPTSTR lpString, LPTSTR lpSubString);
|
||
|
|
INT_PTR CALLBACK Details_DlgProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
|
||
|
|
INT_PTR CALLBACK ListView_DlgProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
|
||
|
|
INT_PTR CALLBACK LogFile_DlgProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
|
||
|
|
LPFILENODE CreateFileNode(LPTSTR lpDirectory, LPTSTR lpFileName);
|
||
|
|
BOOL IsFileAlreadyInList(LPTSTR lpDirName, LPTSTR lpFileName);
|
||
|
|
void SigVerif_Help(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL bContext);
|
||
|
|
void AdvancedPropertySheet(HWND hwnd);
|
||
|
|
void InsertFileNodeIntoList(LPFILENODE lpFileNode);
|
||
|
|
void DestroyFileList(BOOL bClear);
|
||
|
|
void DestroyFileNode(LPFILENODE lpFileNode);
|
||
|
|
void PrintFileList(void);
|
||
|
|
void Progress_InitRegisterClass(void);
|
||
|
|
void BuildDriverFileList(void);
|
||
|
|
void BuildPrinterFileList(void);
|
||
|
|
void BuildCoreFileList(void);
|
||
|
|
void MyGetFileInfo(LPFILENODE lpFileInfo);
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Context-Sensitive Help/Identifiers specific to SigVerif
|
||
|
|
//
|
||
|
|
#define SIGVERIF_HELPFILE TEXT("SIGVERIF.HLP")
|
||
|
|
#define WINDOWS_HELPFILE TEXT("WINDOWS.HLP")
|
||
|
|
#define IDH_SIGVERIF_SEARCH_CHECK_SYSTEM 1000
|
||
|
|
#define IDH_SIGVERIF_SEARCH_LOOK_FOR 1010
|
||
|
|
#define IDH_SIGVERIF_SEARCH_SCAN_FILES 1020
|
||
|
|
#define IDH_SIGVERIF_SEARCH_LOOK_IN_FOLDER 1030
|
||
|
|
#define IDH_SIGVERIF_SEARCH_INCLUDE_SUBFOLDERS 1040
|
||
|
|
#define IDH_SIGVERIF_LOGGING_ENABLE_LOGGING 1050
|
||
|
|
#define IDH_SIGVERIF_LOGGING_APPEND 1060
|
||
|
|
#define IDH_SIGVERIF_LOGGING_OVERWRITE 1070
|
||
|
|
#define IDH_SIGVERIF_LOGGING_FILENAME 1080
|
||
|
|
#define IDH_SIGVERIF_LOGGING_VIEW_LOG 1090
|
||
|
|
|
||
|
|
//
|
||
|
|
// Context-Sensitive Help Identifiers for Browse button
|
||
|
|
//
|
||
|
|
#define IDH_BROWSE 28496
|
||
|
|
|
||
|
|
//
|
||
|
|
// g_App is allocated in SIGVERIF.C, so everywhere else we want to make it extern
|
||
|
|
//
|
||
|
|
#ifndef SIGVERIF_DOT_C
|
||
|
|
extern GAPPDATA g_App;
|
||
|
|
#endif
|