windows-nt/Source/XPSP1/NT/admin/pchealth/sr/nttest/srdiag2/main.cxx

572 lines
14 KiB
C++
Raw Normal View History

2020-09-26 03:20:57 -05:00
//+---------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1999-2001.
//
// File: Main.cxx
//
// Contents: This the implementation for the main module of srdiag.exe.
//
// Classes: n/a
//
// Functions: DiaplayHelp
// ParseCmdLine
// Main
//
// Coupling:
//
// Notes:
//
// History: 20-04-2001 weiyouc Created
//
//----------------------------------------------------------------------------
//--------------------------------------------------------------------------
// Headers
//--------------------------------------------------------------------------
#include "SrHeader.hxx"
#ifdef ARRAYSIZE
#undef ARRAYSIZE
#endif
#include "stdafx.h"
#include "srdiag.h"
//--------------------------------------------------------------------------
// Local defines
//--------------------------------------------------------------------------
#define CAB_FILE_EXT ".cab"
//--------------------------------------------------------------------------
// Local function prototypes
//--------------------------------------------------------------------------
void DisplayHelp();
HRESULT ParseCmdLine(BOOL* pfDisplayHelp,
LPTSTR* pptszCabFileName,
LPTSTR* pptszCabLoc);
HRESULT GenerateCabFileName(LPTSTR* pptszCabFile);
HRESULT OpenCabFile(MPC::Cabinet* pCab,
LPTSTR ptszCabFileName,
LPTSTR ptszCabLoc);
HRESULT AppendFilesToCabFile(MPC::Cabinet * pCab);
//--------------------------------------------------------------------------
// Global Variables
//--------------------------------------------------------------------------
//
// List of files that we will collect at %windir% directory.
//
LPCTSTR g_tszWindirFileCollection[] =
{
TEXT("\\system32\\restore\\machineguid.txt"),
TEXT("\\system32\\restore\\filelist.xml"),
TEXT("\\system32\\restore\\rstrlog.dat")
};
//
// List of the files, that we will collect at the root of
// %systemdrv%\System Volume Information\_Restore{GUID} directory.
//
LPCTSTR g_tszSysVolFileCollection[] =
{
TEXT("_filelst.cfg"),
TEXT("drivetable.txt"),
TEXT("_driver.cfg"),
TEXT("fifo.log"),
};
//--------------------------------------------------------------------------
// Functions
//--------------------------------------------------------------------------
//+---------------------------------------------------------------------------
//
// Function: DisplayHelp
//
// Synopsis: Displays a help page for the user
//
// Arguments: (none)
//
// Returns: void
//
// History: 20-04-2001 weiyouc Created
//
// Notes:
//
//----------------------------------------------------------------------------
void DisplayHelp()
{
printf("\n");
printf("Microsoft Windows XP \n");
printf("Usage: SrDiag [/CabName:test.cab] [/CabLoc:\"c:\\temp\\\"] \n");
printf(" /CabName is the full name of the cab file that you wish to use. \n"
" If the cab file name is not specified, the system will \n"
" automatically generate one in the following format: \n"
" <machine_name>_mmddyy_hhss.cab \n");
printf(" /CabLoc points to the location to store the cab. \n"
" It must have a \\ on the end. \n"
" The default location is the current directory. \n");
}
//+---------------------------------------------------------------------------
//
// Function: ParseCmdLine
//
// Synopsis: Sets Globals based on Cmd and Enviroment settings
//
// Arguments: (none)
//
// Returns: HRESULT
//
// History: 20-04-2001 weiyouc Created
//
// Notes:
//
//----------------------------------------------------------------------------
HRESULT ParseCmdLine(BOOL* pfDisplayHelp,
LPTSTR* pptszCabFileName,
LPTSTR* pptszCabLoc)
{
HRESULT hr = S_OK;
//
// Do we want to generate full string for printed test cases?
//
*pfDisplayHelp = (GETPARAM_ISPRESENT("help") ||
GETPARAM_ISPRESENT("?"));
//
// Do we specify the cab file name or not
//
GETPARAM_ABORTONERROR("CabName:tstr", *pptszCabFileName);
//
// If cab file location is
//
GETPARAM_ABORTONERROR("CabLoc:tstr", *pptszCabLoc);
ErrReturn:
return hr;
} // ParseCmdLine
//+---------------------------------------------------------------------------
//
// Function: main
//
// Synopsis: Entry point for srdiag.exe
//
// Arguments: [argc] -- Command Line Arg Count
// [argv] -- Command Line Args
//
// Returns: VOID
//
// History: 20-04-2001 weiyouc Created
//
// Notes:
//
//----------------------------------------------------------------------------
void __cdecl main (int argc, char *argv[])
{
HRESULT hr = S_OK;
BOOL fDisplayHelp = FALSE;
LPTSTR ptszCabFileName = NULL;
LPTSTR ptszCabLoc = NULL;
MPC::Cabinet Cab;
hr = ParseCmdLine(&fDisplayHelp,
&ptszCabFileName,
&ptszCabLoc);
DH_HRCHECK_ABORT(hr, TEXT("ParseCmdLine"));
//
// Do we need to display help
//
if (fDisplayHelp)
{
DisplayHelp();
goto ErrReturn;
}
//
// Make sure that we always start from a clean environment
//
hr = CleanupFiles();
DH_HRCHECK_ABORT(hr, TEXT("CleanupFiles"));
//
// Open the cab file
//
hr = OpenCabFile(&Cab, ptszCabFileName, ptszCabLoc);
DH_HRCHECK_ABORT(hr, TEXT("OpenCabFile"));
//
// Now append the files to the cab file
//
hr = AppendFilesToCabFile(&Cab);
DH_HRCHECK_ABORT(hr, TEXT("AppendFilesToCabFile"));
//
// Now we create the cab file for real
//
hr = Cab.Compress();
DH_HRCHECK_ABORT(hr, TEXT("Cabinet::Compress"));
//
// Clean up the files
//
hr = CleanupFiles();
DH_HRCHECK_ABORT(hr, TEXT("CleanupFiles"));
ErrReturn:
CleanupTStr(&ptszCabFileName);
CleanupTStr(&ptszCabLoc);
return;
}
//+---------------------------------------------------------------------------
//
// Function: GenerateCabFileName
//
// Synopsis: Generate a cab file name = ComputerName + ddmmyy + hhmmss
//
// Arguments: [pptszCabFile] -- cab file name
//
// Returns: HRESULT
//
// History: 20-04-2001 weiyouc Created
//
// Notes:
//
//----------------------------------------------------------------------------
HRESULT GenerateCabFileName(LPTSTR* pptszCabFile)
{
HRESULT hr = S_OK;
time_t ltime = 0;
tm* ptmNow = NULL;
size_t stRet = 0;
TCHAR tszTmTemp[MAX_PATH];
TCHAR tszCabFile[MAX_PATH];
DH_VDATEPTROUT(pptszCabFile, LPTSTR);
//
// Copy Computer Name to CabFileName
//
_tcscpy(tszCabFile, _tgetenv(TEXT("COMPUTERNAME")));
DH_HRCHECK_ABORT(hr, TEXT("CopyString"));
//
// Append Undescore character to CabFileName
//
_tcscat(tszCabFile, TEXT("_"));
//
// Get System Time and Date
//
time(&ltime);
ptmNow = localtime(&ltime);
//
// Convert time/date to mmddyyhhmmss format (24hr)
//
stRet = _tcsftime(tszTmTemp,
MAX_PATH,
TEXT("%m%d%y_%H%M%S"),
ptmNow);
DH_ABORTIF(0 == stRet,
E_FAIL,
TEXT("_tcsftime"));
_tcscat(tszCabFile, tszTmTemp);
//
// Finally append on the extension and now we are set.
//
_tcscat(tszCabFile, TEXT(CAB_FILE_EXT));
hr = CopyString(tszCabFile, pptszCabFile);
DH_HRCHECK_ABORT(hr, TEXT("CopyString"));
ErrReturn:
return hr;
}
//+---------------------------------------------------------------------------
//
// Function: OpenCabFile
//
// Synopsis: Open the cab file
//
// Arguments: [pCab] -- pointer to the cab file
// [ptszCabFileName] -- cab file name
// [ptszCabLoc] -- cab file location
//
// Returns: HRESULT
//
// History: 20-04-2001 weiyouc Created
//
// Notes:
//
//----------------------------------------------------------------------------
HRESULT OpenCabFile(MPC::Cabinet* pCab,
LPTSTR ptszCabFileName,
LPTSTR ptszCabLoc)
{
HRESULT hr = S_OK;
LPTSTR ptszFullCabFileName = NULL;
DH_VDATEPTRIN(pCab, MPC::Cabinet);
//
// If the user does not specify the cab file name,
// we will generate one.
//
if (NULL == ptszCabFileName)
{
hr = GenerateCabFileName(&ptszCabFileName);
DH_HRCHECK_ABORT(hr, TEXT("GenerateCabFileName"));
}
//
// If the use specifies a location for the cab file,
// we need to glue it together with the cab file name
//
if (NULL != ptszCabLoc)
{
hr = SrTstTStrCat(ptszCabLoc,
ptszCabFileName,
&ptszFullCabFileName);
DH_HRCHECK_ABORT(hr, TEXT("SrTstStrCat"));
}
else
{
hr = CopyString(ptszCabFileName, &ptszFullCabFileName);
DH_HRCHECK_ABORT(hr, TEXT("CopyString"));
}
//
// Open the cabinet file
//
hr = pCab->put_CabinetFile(ptszFullCabFileName);
DH_HRCHECK_ABORT(hr, TEXT("Cabinet::put_CabinetFile"));
//
// Also set a flag to ignore the missing file
//
hr = pCab->put_IgnoreMissingFiles(TRUE);
DH_HRCHECK_ABORT(hr, TEXT("Cabinet::put_IgnoreMissingFiles"));
ErrReturn:
CleanupTStr(&ptszFullCabFileName);
return hr;
}
//+---------------------------------------------------------------------------
//
// Function: OpenCabFile
//
// Synopsis: Append files to the cab file. However it has not written
// anyting to the cab file yet
//
// Arguments: [pCab] -- pointer to the cab file
//
// Returns: HRESULT
//
// History: 20-04-2001 weiyouc Created
//
// Notes:
//
//----------------------------------------------------------------------------
HRESULT AppendFilesToCabFile(MPC::Cabinet* pCab)
{
HRESULT hr = S_OK;
int i = 0;
LPTSTR ptszFileName = NULL;
LPTSTR ptszDSOnSys = NULL;
DH_VDATEPTRIN(pCab, MPC::Cabinet);
//
// Add SR related registery settings
//
hr = GetSRRegInfo(TEXT("SR-Reg.TXT"));
DH_HRCHECK_ABORT(hr, TEXT("GetSRRegInfo"));
hr = pCab->AddFile(TEXT("SR-Reg.TXT"));
DH_HRCHECK_ABORT(hr, TEXT("MPC::Cabinet::AddFile()"));
//
// Add files based on WinDir relative root
//
for (i = 0; i < ARRAYSIZE(g_tszWindirFileCollection); i++)
{
hr = SrTstTStrCat(_tgetenv(TEXT("WINDIR")),
g_tszWindirFileCollection[i],
&ptszFileName);
DH_HRCHECK_ABORT(hr, TEXT("SrTstTStrCat"));
hr = pCab->AddFile(ptszFileName);
DH_HRCHECK_ABORT(hr, TEXT("MPC::Cabinet::AddFile"));
CleanupTStr(&ptszFileName);
}
//
// Parse the restore log and add it to the cab
//
hr = SrTstTStrCat(_tgetenv(TEXT("WINDIR")),
g_tszWindirFileCollection[2],
&ptszFileName);
DH_HRCHECK_ABORT(hr, TEXT("SrTstTStrCat"));
hr = ParseRstrLog(ptszFileName, TEXT("SR-RstrLog.TXT"));
DH_HRCHECK_ABORT(hr, TEXT("ParseRstrLog"));
hr = pCab->AddFile(TEXT("SR-RstrLog.TXT"));
DH_HRCHECK_ABORT(hr, TEXT("MPC::Cabinet::AddFile"));
CleanupTStr(&ptszFileName);
//
// Get the restore directory on the system drive,
// and then Add in critial files
//
hr = GetDSOnSysVol(&ptszDSOnSys);
DH_HRCHECK_ABORT(hr, TEXT("GetDSOnSysVol"));
//
// Add files Bases on System Volume Information relative root
//
for (i = 0; i < ARRAYSIZE(g_tszSysVolFileCollection); i++)
{
hr = SrTstTStrCat(ptszDSOnSys,
g_tszSysVolFileCollection[i],
&ptszFileName);
DH_HRCHECK_ABORT(hr, TEXT("SrTstTStrCat"));
hr = pCab->AddFile(ptszFileName);
DH_HRCHECK_ABORT(hr, TEXT("MPC::Cabinet::AddFile()"));
CleanupTStr(&ptszFileName);
}
//
// Get the Restore point enumeration, and then cab the file
//
hr = RPEnumDrives(pCab, TEXT("SR-RP.LOG"));
DH_HRCHECK_ABORT(hr, TEXT("RPEnumDrives"));
hr = pCab->AddFile(TEXT("SR-RP.LOG"));
DH_HRCHECK_ABORT(hr, TEXT("MPC::Cabinet::AddFile()"));
//
// Get change logs
//
hr = GetChgLogOnDrives(TEXT("SR-ChgLog.LOG"));
DH_HRCHECK_ABORT(hr, TEXT("GetChgLog"));
hr = pCab->AddFile(TEXT("SR-ChgLog.LOG"));
DH_HRCHECK_ABORT(hr, TEXT("MPC::Cabinet::AddFile()"));
//
// Get file versions
//
hr = GetSRFileInfo(TEXT("SR-FileList.LOG"));
DH_HRCHECK_ABORT(hr, TEXT("GetSRFileInfo"));
hr = pCab->AddFile(TEXT("SR-FileList.LOG"));
DH_HRCHECK_ABORT(hr, TEXT("MPC::Cabinet::AddFile()"));
//
// Get SR related event logs
//
hr = GetSREvents(TEXT("SR-EventLogs.TXT"));
DH_HRCHECK_ABORT(hr, TEXT("GetSREvents"));
hr = pCab->AddFile(TEXT("SR-EventLogs.TXT"));
DH_HRCHECK_ABORT(hr, TEXT("MPC::Cabinet::AddFile()"));
ErrReturn:
CleanupTStr(&ptszFileName);
CleanupTStr(&ptszDSOnSys);
return hr;
}
//+---------------------------------------------------------------------------
//
// Function: CleanupFiles
//
// Synopsis: Remove the temporary file used by srdiag
//
// Arguments: none
//
// Returns: HRESULT
//
// History: 20-04-2001 weiyouc Created
//
// Notes:
//
//----------------------------------------------------------------------------
HRESULT CleanupFiles()
{
HRESULT hr = S_OK;
DeleteFileW(L"SR-Reg.TXT");
DeleteFileW(L"SR-RstrLog.TXT");
DeleteFileW(L"SR-RP.LOG");
DeleteFileW(L"SR-ChgLog.LOG");
DeleteFileW(L"SR-FileList.LOG");
DeleteFileW(L"SR-EventLogs.TXT");
return hr;
}