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.
1249 lines
52 KiB
1249 lines
52 KiB
/****************************************************************************************
|
|
* MSI will call these APIs to ask TS to propogate changes from .Default to the TS hive.
|
|
*
|
|
* NTSTATUS TermServPrepareAppInstallDueMSI()
|
|
*
|
|
* NTSTATUS TermServProcessAppIntallDueMSI( BOOLEAN cleanup )
|
|
*
|
|
* These API need not be called in the same boot cycles, many boot cycles could
|
|
* happen in between.
|
|
*
|
|
* Copyright (C) 1997-1999 Microsoft Corp.
|
|
****************************************************************************************/
|
|
|
|
#include <nt.h>
|
|
#include <ntrtl.h>
|
|
#include <nturtl.h>
|
|
#include <ntexapi.h>
|
|
#include <stdio.h>
|
|
#include <windows.h>
|
|
#include <winuser.h>
|
|
#include <stdlib.h>
|
|
#include <tsappcmp.h>
|
|
#include <stdio.h>
|
|
#include <fcntl.h>
|
|
|
|
#include "KeyNode.h"
|
|
#include "ValInfo.h"
|
|
|
|
// real externs!
|
|
extern "C" {
|
|
void TermsrvLogRegInstallTime(void);
|
|
}
|
|
|
|
// forward declaration
|
|
extern NTSTATUS DeleteReferenceHive(WCHAR *);
|
|
extern NTSTATUS CreateReferenceHive( WCHAR *, WCHAR *);
|
|
extern NTSTATUS DeltaDeleteKeys(WCHAR *, WCHAR *, WCHAR *);
|
|
extern NTSTATUS DeltaUpdateKeys(WCHAR *, WCHAR *, WCHAR *);
|
|
|
|
ULONG g_length_TERMSRV_USERREGISTRY_DEFAULT;
|
|
ULONG g_length_TERMSRV_INSTALL;
|
|
WCHAR g_debugFileName[MAX_PATH];
|
|
FILE *g_debugFilePointer=NULL;
|
|
BOOLEAN g_debugIO = FALSE;
|
|
BOOLEAN KeyNode::debug=FALSE; // init the static
|
|
|
|
#define TERMSRV_USERREGISTRY_DEFAULT TEXT("\\Registry\\USER\\.Default")
|
|
|
|
// This is for debug I/O, output looks better with it.
|
|
void Indent( ULONG indent)
|
|
{
|
|
|
|
for ( ULONG i = 1; i <indent ; i++ )
|
|
{
|
|
fwprintf( g_debugFilePointer, L" ");
|
|
}
|
|
}
|
|
|
|
// a key name is written to the log file based on the contect of pBasicInfo
|
|
void DebugKeyStamp( NTSTATUS status, KeyBasicInfo *pBasicInfo, int indent , WCHAR *pComments=L"" )
|
|
{
|
|
Indent(indent);
|
|
fwprintf( g_debugFilePointer,L"%ws, status=%lx, %ws\n",pBasicInfo->NameSz(), status , pComments);
|
|
fflush( g_debugFilePointer );
|
|
DbgPrint("%ws\n",pBasicInfo->NameSz());
|
|
}
|
|
|
|
// a debug stamp is written to the log file, including the line number where error happened
|
|
void DebugErrorStamp(NTSTATUS status , int lineNumber, ValueFullInfo *pValue=NULL)
|
|
{
|
|
fwprintf( g_debugFilePointer,
|
|
L"ERROR ?!? status = %lx, linenumber:%d\n", status, lineNumber);
|
|
if (pValue)
|
|
{
|
|
pValue->Print(g_debugFilePointer);
|
|
}
|
|
fflush(g_debugFilePointer);
|
|
}
|
|
|
|
// use this func to track the status value which is used to bail out in
|
|
// case of an error.
|
|
// this is only used in teh debug build, see below
|
|
BOOL NT_SUCCESS_OR_ERROR_STAMP( NTSTATUS status, ULONG lineNumber)
|
|
{
|
|
if ( g_debugIO )
|
|
{
|
|
if ( ( (ULONG)status) >=0xC0000000 )
|
|
{
|
|
DebugErrorStamp( status, lineNumber );
|
|
}
|
|
}
|
|
|
|
return ( (NTSTATUS)(status) >= 0 );
|
|
}
|
|
|
|
#ifdef DBG
|
|
#define NT_SUCCESS_EX(Status) NT_SUCCESS_OR_ERROR_STAMP( (Status), __LINE__ )
|
|
#else
|
|
#define NT_SUCCESS_EX(Status) NT_SUCCESS(Status)
|
|
#endif
|
|
|
|
/***************************************************************************
|
|
*
|
|
* All three branch-walker functions use this method to alter the status code, and
|
|
* if necessary, log an error message to the log file
|
|
*
|
|
***************************************************************************/
|
|
NTSTATUS AlterStatus( NTSTATUS status , int lineNumber )
|
|
{
|
|
switch( status )
|
|
{
|
|
case STATUS_ACCESS_DENIED:
|
|
// this should never happen since we run in the system context
|
|
if ( g_debugIO )
|
|
{
|
|
DebugErrorStamp( status, lineNumber );
|
|
}
|
|
status = STATUS_SUCCESS;
|
|
break;
|
|
|
|
case STATUS_SUCCESS:
|
|
break;
|
|
|
|
case STATUS_NO_MORE_ENTRIES:
|
|
status = STATUS_SUCCESS;
|
|
break;
|
|
|
|
default:
|
|
if ( g_debugIO )
|
|
{
|
|
DebugErrorStamp( status, lineNumber );
|
|
}
|
|
break;
|
|
|
|
|
|
}
|
|
return status;
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* Based on a special reg key/value init the debug flags and pointers which are
|
|
* used to log debug info into a log file. When called with start=TRUE, the
|
|
* relevant data structs are initialized. When called with start=FALSE, the log
|
|
* file is closed.
|
|
*
|
|
******************************************************************************/
|
|
void InitDebug( BOOLEAN start)
|
|
{
|
|
if ( start )
|
|
{
|
|
KeyNode tsHiveNode (NULL, KEY_READ, TERMSRV_INSTALL );
|
|
|
|
if ( NT_SUCCESS( tsHiveNode.Open() ) )
|
|
{
|
|
ValuePartialInfo debugValue( &tsHiveNode );
|
|
if ( NT_SUCCESS( debugValue.Status() ) && NT_SUCCESS( debugValue.Query(L"TS_MSI_DEBUG") ) )
|
|
{
|
|
g_debugIO = TRUE;
|
|
KeyNode::debug=TRUE;
|
|
for (ULONG i =0; i < debugValue.Ptr()->DataLength/sizeof(WCHAR); i++)
|
|
{
|
|
g_debugFileName[i] = ((WCHAR*)(debugValue.Ptr()->Data))[i];
|
|
}
|
|
g_debugFileName[i] = L'\0';
|
|
|
|
g_debugFilePointer = _wfopen( g_debugFileName, L"a+" );
|
|
fwprintf( g_debugFilePointer, L"----\n");
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if ( g_debugFilePointer )
|
|
{
|
|
fclose(g_debugFilePointer);
|
|
}
|
|
}
|
|
}
|
|
|
|
/***************************************************************************
|
|
*
|
|
* Function:
|
|
* TermServPrepareAppInstallDueMSI()
|
|
*
|
|
* Description:
|
|
* MSI service calls this function prior to starting an installation cycle.
|
|
* When called, this function blows away the RefHive (in case it was around
|
|
* with some stale data...), and then it creates a fresh copy of
|
|
* .Default\Software as the new RefHive.
|
|
*
|
|
* Return:
|
|
* NTSTATUS
|
|
*
|
|
***************************************************************************/
|
|
NTSTATUS TermServPrepareAppInstallDueMSI()
|
|
{
|
|
NTSTATUS status = STATUS_SUCCESS;
|
|
|
|
WCHAR sourceHive[MAX_PATH];
|
|
WCHAR referenceHive[MAX_PATH];
|
|
WCHAR destinationHive[MAX_PATH];
|
|
|
|
wcscpy(sourceHive, TERMSRV_USERREGISTRY_DEFAULT );
|
|
wcscat(sourceHive, L"\\Software");
|
|
g_length_TERMSRV_USERREGISTRY_DEFAULT = wcslen( TERMSRV_USERREGISTRY_DEFAULT );
|
|
|
|
wcscpy(referenceHive, TERMSRV_INSTALL );
|
|
wcscat(referenceHive, L"\\RefHive");
|
|
g_length_TERMSRV_INSTALL = wcslen( TERMSRV_INSTALL );
|
|
|
|
InitDebug( TRUE );
|
|
|
|
if ( g_debugIO)
|
|
{
|
|
fwprintf( g_debugFilePointer,L"In %ws\n",
|
|
L"TermServPrepareAppInstallDueMSI");
|
|
fflush( g_debugFilePointer );
|
|
}
|
|
|
|
// delete the existing hive (if any )
|
|
status = DeleteReferenceHive( referenceHive );
|
|
|
|
if ( NT_SUCCESS( status ) )
|
|
{
|
|
// 1-COPY
|
|
// copy all keys under .Default\Software into a special location
|
|
// under our TS hive, let's call it the RefHive
|
|
status = CreateReferenceHive(sourceHive, referenceHive);
|
|
|
|
}
|
|
|
|
InitDebug( FALSE );
|
|
|
|
return status;
|
|
}
|
|
|
|
/**********************************************************************************
|
|
*
|
|
* Function:
|
|
* TermServProcessAppInstallDueMSI
|
|
*
|
|
* Description:
|
|
* MSI service calls this function after calling TermServPrepareAppInstallDueMSI(),
|
|
* and after MSI finishing making an installation which updated the .Default
|
|
* hive (since MSI runs in the system context).
|
|
* This function will compare the content of .Default\SW to RefHive and then
|
|
* first it will create all new (missing) keys and values. Then it will
|
|
* compare any existing keys from .Default\SW with the equivalent RefHive, and
|
|
* if value is different, it will delete the equivalent value from our TS hive
|
|
* and then create a new value identical to what was found in .Default
|
|
*
|
|
* Return:
|
|
* NTSTATUS
|
|
*
|
|
**********************************************************************************/
|
|
NTSTATUS TermServProcessAppInstallDueMSI( BOOLEAN cleanup)
|
|
{
|
|
NTSTATUS status = STATUS_SUCCESS;
|
|
WCHAR sourceHive[MAX_PATH];
|
|
WCHAR referenceHive[MAX_PATH];
|
|
WCHAR destinationHive[MAX_PATH];
|
|
|
|
wcscpy(sourceHive, TERMSRV_USERREGISTRY_DEFAULT );
|
|
wcscat(sourceHive, L"\\Software");
|
|
g_length_TERMSRV_USERREGISTRY_DEFAULT = wcslen( TERMSRV_USERREGISTRY_DEFAULT );
|
|
|
|
wcscpy(referenceHive, TERMSRV_INSTALL );
|
|
wcscat(referenceHive, L"\\RefHive");
|
|
g_length_TERMSRV_INSTALL = wcslen( TERMSRV_INSTALL );
|
|
|
|
wcscpy(destinationHive, TERMSRV_INSTALL );
|
|
wcscat(destinationHive, L"\\Software");
|
|
|
|
InitDebug( TRUE );
|
|
|
|
if ( g_debugIO)
|
|
{
|
|
fwprintf( g_debugFilePointer,L"In %ws, cleanup=%lx\n",
|
|
L"TermServProcessAppIntallDueMSI", cleanup);
|
|
fflush( g_debugFilePointer );
|
|
}
|
|
|
|
if ( !cleanup )
|
|
{
|
|
// 2-DELETE
|
|
// compare .Dfeault keys to the equivalent keys in RefHive. If keys are
|
|
// missing from .Default, then delete the equivalent keys from our
|
|
// HKLM\...\TS\ hive
|
|
status = DeltaDeleteKeys(sourceHive, referenceHive, destinationHive);
|
|
|
|
if (NT_SUCCESS( status ) )
|
|
{
|
|
// Steps 3 and 4 are now combined.
|
|
// 3-CREATE
|
|
// compare .Default keys to the equivalent keys in RefHive, if keys are
|
|
// present in .Default that are missing from RefHive, then, add those keys
|
|
// to our HKLM\...\TS hive
|
|
// 4-CHANGE
|
|
// compare keys of .Default to RefHive. Those keys that are newer than
|
|
// RefHive, then, update the equivalent keys in HKLM\...\TS
|
|
|
|
status = DeltaUpdateKeys(sourceHive, referenceHive, destinationHive);
|
|
|
|
if (NT_SUCCESS( status ))
|
|
{
|
|
// update the time stamp in our hive since we want the standared TS reg key
|
|
// propogation to take place.
|
|
TermsrvLogRegInstallTime();
|
|
|
|
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// blow away the existing reference hive,
|
|
status = DeleteReferenceHive( referenceHive );
|
|
}
|
|
|
|
InitDebug( FALSE );
|
|
|
|
return status;
|
|
}
|
|
|
|
/*******************************************************************
|
|
*
|
|
* Function:
|
|
* EnumerateAndCreateRefHive
|
|
*
|
|
* Parameters:
|
|
* pSource points to the parent node, the branch we copy
|
|
* pref points to our RefHive which we are creating as a ref image
|
|
* pBasicInfo is a scratch pad passed around which is used to
|
|
* extract basic Key information
|
|
* pindextLevel is used to format the debug log output file
|
|
*
|
|
* Descritption:
|
|
* Create a copy of the .Default\Sofwtare as our RefHive
|
|
*
|
|
* Return:
|
|
* NTSTATUS
|
|
*******************************************************************/
|
|
NTSTATUS EnumerateAndCreateRefHive(
|
|
IN KeyNode *pSource,
|
|
IN KeyNode *pRef,
|
|
IN KeyBasicInfo *pBasicInfo,
|
|
IN ULONG *pIndentLevel
|
|
)
|
|
{
|
|
NTSTATUS status=STATUS_SUCCESS;
|
|
ULONG ulCount=0;
|
|
|
|
UNICODE_STRING UniString;
|
|
|
|
(*pIndentLevel)++;
|
|
|
|
while( NT_SUCCESS(status))
|
|
{
|
|
ULONG ultemp;
|
|
|
|
status = NtEnumerateKey( pSource->Key(),
|
|
ulCount++,
|
|
pBasicInfo->Type() , // keyInformationClass,
|
|
pBasicInfo->Ptr(), // pKeyInfo,
|
|
pBasicInfo->Size(), // keyInfoSize,
|
|
&ultemp);
|
|
|
|
if (NT_SUCCESS(status))
|
|
{
|
|
if ( g_debugIO)
|
|
{
|
|
DebugKeyStamp( status , pBasicInfo, *pIndentLevel );
|
|
}
|
|
|
|
// open a sub key
|
|
KeyNode SourceSubKey( pSource, pBasicInfo);
|
|
|
|
// create the Ref sub key
|
|
KeyNode RefSubKey( pRef, pBasicInfo);
|
|
|
|
if (NT_SUCCESS_EX( status = SourceSubKey.Open() ) )
|
|
{
|
|
if ( NT_SUCCESS_EX( status = RefSubKey.Create() ) )
|
|
{
|
|
NTSTATUS status3;
|
|
KEY_FULL_INFORMATION *ptrInfo;
|
|
ULONG size;
|
|
|
|
if (NT_SUCCESS(SourceSubKey.Query( &ptrInfo, &size )))
|
|
{
|
|
ValueFullInfo valueFullInfo( &SourceSubKey );
|
|
ValueFullInfo RefValue( &RefSubKey );
|
|
|
|
if ( NT_SUCCESS_EX( status = valueFullInfo.Status())
|
|
&& NT_SUCCESS_EX(status = RefValue.Status()) )
|
|
{
|
|
for (ULONG ulkey = 0; ulkey < ptrInfo->Values; ulkey++)
|
|
{
|
|
status = NtEnumerateValueKey(SourceSubKey.Key(),
|
|
ulkey,
|
|
valueFullInfo.Type(),
|
|
valueFullInfo.Ptr(),
|
|
valueFullInfo.Size(),
|
|
&ultemp);
|
|
|
|
if (NT_SUCCESS( status ))
|
|
{
|
|
status = RefValue.Create( &valueFullInfo );
|
|
// if status is not good, we bail out, since var "status" is set here
|
|
}
|
|
// else, no more entries left, we continue
|
|
}
|
|
}
|
|
// else, out of memory, status is set, we bail out.
|
|
}
|
|
// else, no values are present, continue with sub-key enums
|
|
|
|
if (NT_SUCCESS( status ) )
|
|
{
|
|
// enumerate sub key down.
|
|
status = EnumerateAndCreateRefHive(
|
|
&SourceSubKey,
|
|
&RefSubKey,
|
|
pBasicInfo,
|
|
pIndentLevel
|
|
);
|
|
}
|
|
}
|
|
// else, an error, status is set, so we bail out
|
|
|
|
}// else, open on source has failed, var-status is set, we bail out
|
|
|
|
status = AlterStatus( status, __LINE__ );
|
|
// else, an error, status is set, so we bail out
|
|
}
|
|
// else, no more left
|
|
|
|
}
|
|
|
|
(*pIndentLevel)--;
|
|
|
|
return( status );
|
|
}
|
|
|
|
/*******************************************************************
|
|
*
|
|
* Function:
|
|
* EnumerateAndDeltaDeleteKeys
|
|
*
|
|
* Parameters:
|
|
* pSource points to a node under .Dfeault
|
|
* pref points to a node under our RefHive
|
|
* pDestination is a node under our TS\install\SW hive
|
|
* pBasicInfo is a scratch pad passed around which is used to
|
|
* extract basic Key information
|
|
* pindextLevel is used to format the debug log output file
|
|
*
|
|
* Descritption:
|
|
* compare source to ref, if keys/values in source are deleted, then
|
|
* delete the equivalent key/value from destination
|
|
*
|
|
* Return:
|
|
* NTSTATUS
|
|
*******************************************************************/
|
|
NTSTATUS EnumerateAndDeltaDeleteKeys(
|
|
IN KeyNode *pSource, // this is under the latest updated .Default\SW hive
|
|
IN KeyNode *pRef, // this was a ref-copy of .Default\SW before the update
|
|
IN KeyNode *pDestination,// this is opur private TS-hive
|
|
IN KeyBasicInfo *pBasicInfo,
|
|
IN ULONG *pIndentLevel)
|
|
{
|
|
NTSTATUS status=STATUS_SUCCESS, st2;
|
|
ULONG ulCount=0;
|
|
|
|
UNICODE_STRING UniString;
|
|
ULONG size;
|
|
|
|
(*pIndentLevel)++;
|
|
|
|
while (NT_SUCCESS(status))
|
|
{
|
|
ULONG ultemp;
|
|
|
|
status = NtEnumerateKey( pRef->Key(),
|
|
ulCount++,
|
|
pBasicInfo->Type() , // keyInformationClass,
|
|
pBasicInfo->Ptr(), // pKeyInfo,
|
|
pBasicInfo->Size(), // keyInfoSize,
|
|
&ultemp);
|
|
|
|
// pBasicInfo was filled up thru NtEnumerateKey() above
|
|
|
|
if (NT_SUCCESS(status))
|
|
{
|
|
if ( g_debugIO)
|
|
{
|
|
DebugKeyStamp( status , pBasicInfo, *pIndentLevel );
|
|
}
|
|
|
|
KeyNode RefSubKey( pRef, pBasicInfo);
|
|
KeyNode SourceSubKey( pSource,pBasicInfo);
|
|
KeyNode DestinationSubKey( pDestination, pBasicInfo);
|
|
|
|
RefSubKey.Open();
|
|
SourceSubKey.Open();
|
|
DestinationSubKey.Open();
|
|
|
|
if (NT_SUCCESS( RefSubKey.Status() ) )
|
|
{
|
|
if ( ! NT_SUCCESS( SourceSubKey.Status () ) )
|
|
{
|
|
// key is missing from the .Default\SW hive, we should delete
|
|
// the same sub-tree from our TS\Install\SW hive
|
|
if ( NT_SUCCESS( DestinationSubKey.Status()) )
|
|
{
|
|
DestinationSubKey.DeleteSubKeys();
|
|
NTSTATUS st = DestinationSubKey.Delete();
|
|
|
|
if ( g_debugIO)
|
|
{
|
|
DebugKeyStamp( st, pBasicInfo, *pIndentLevel );
|
|
}
|
|
|
|
}
|
|
// else
|
|
// As long as the key is missing from
|
|
// Ts\install\Hive, we will regard this condition as acceptable.
|
|
}
|
|
else
|
|
{
|
|
// see if any values have been deleted
|
|
|
|
// don't bother unless the destination key exists, otherwise, no values
|
|
// will be there to delete...
|
|
if ( NT_SUCCESS( DestinationSubKey.Status() ) )
|
|
{
|
|
|
|
KEY_FULL_INFORMATION *ptrInfo;
|
|
ULONG size;
|
|
|
|
if (NT_SUCCESS_EX(status = RefSubKey.Query( &ptrInfo, &size )))
|
|
{
|
|
// from the key-full-information, create a key-value-full-information
|
|
|
|
ValueFullInfo refValueFullInfo( &RefSubKey );
|
|
ValueFullInfo sourceValue( &SourceSubKey );
|
|
|
|
// if no allocation errors, then...
|
|
if ( NT_SUCCESS_EX( status = refValueFullInfo.Status() )
|
|
&& NT_SUCCESS_EX( status = sourceValue.Status() ) )
|
|
{
|
|
for (ULONG ulkey = 0; ulkey < ptrInfo->Values; ulkey++)
|
|
{
|
|
if ( NT_SUCCESS_EX (
|
|
status = NtEnumerateValueKey(RefSubKey.Key(),
|
|
ulkey,
|
|
refValueFullInfo.Type(),
|
|
refValueFullInfo.Ptr(),
|
|
refValueFullInfo.Size(),
|
|
&ultemp) ) )
|
|
{
|
|
|
|
// for every value, see if the same value
|
|
// exists in the SourceSubKey. If it doesn't
|
|
// then delete the corresponding value from
|
|
// TS's hive
|
|
|
|
sourceValue.Query( refValueFullInfo.SzName() );
|
|
|
|
// if .Default\SW is missing a value, then delete the
|
|
// corresponding value from our TS\ hive
|
|
if ( sourceValue.Status() == STATUS_OBJECT_NAME_NOT_FOUND )
|
|
{
|
|
ValuePartialInfo destinationValue( &DestinationSubKey);
|
|
|
|
if (NT_SUCCESS_EX( status = destinationValue.Status () ) )
|
|
{
|
|
destinationValue.Delete( refValueFullInfo.SzName() );
|
|
}
|
|
// else, alloc error, status is set
|
|
}
|
|
else
|
|
{
|
|
if ( !NT_SUCCESS_EX ( status = sourceValue.Status() ) )
|
|
{
|
|
if ( g_debugIO )
|
|
{
|
|
DebugErrorStamp(status, __LINE__ );
|
|
}
|
|
// else, we will bail out here since var-status is set
|
|
}
|
|
// else, no error
|
|
}
|
|
// if-else
|
|
}
|
|
// else, no more entries
|
|
|
|
} // for loop
|
|
}
|
|
// else, we have an error due to no memory, var-status is set
|
|
}
|
|
// else, we have an error since we can not get info on this existing ref key, var-status is set
|
|
|
|
if ( NT_SUCCESS( status ) )
|
|
{
|
|
// we were able to open the source key, which means that
|
|
// key was not deleted from .default.
|
|
// so keep enuming away...
|
|
status = EnumerateAndDeltaDeleteKeys(
|
|
&SourceSubKey,
|
|
&RefSubKey,
|
|
&DestinationSubKey,
|
|
pBasicInfo ,
|
|
pIndentLevel);
|
|
|
|
}
|
|
//else, status is bad, no point to traverse, we are bailing out
|
|
}
|
|
//else, there is no destination sub key to bother with deletion
|
|
}
|
|
// if-else
|
|
}
|
|
// else, ref had no more sub-keys
|
|
|
|
status = AlterStatus( status, __LINE__ );
|
|
}
|
|
// else, no more entries
|
|
}
|
|
|
|
(*pIndentLevel)--;
|
|
|
|
// the typical status would be: STATUS_NO_MORE_ENTRIES
|
|
return status;
|
|
}
|
|
|
|
/*******************************************************************
|
|
*
|
|
* Function:
|
|
* EnumerateAndDeltaUpdateKeys
|
|
*
|
|
* Parameters:
|
|
* pSource points to a node under .Dfeault
|
|
* pref points to a node under our RefHive
|
|
* pDestination is a node under our TS\install\SW hive
|
|
* pBasicInfo is a scratch pad passed around which is used to
|
|
* extract basic Key information
|
|
* pindextLevel is used to format the debug log output file
|
|
*
|
|
* Descritption:
|
|
* compare source to ref, if new keys/values in source have been created
|
|
* then create the equivalent keys in our Ts\Install\SW branch (pDestination)
|
|
* Also, check all values in pSource to values in pRef, if not the same
|
|
* then delete the equivalent pDestination and create a new value
|
|
* identical to value from pSource
|
|
*
|
|
* Return:
|
|
* NTSTATUS
|
|
*******************************************************************/
|
|
NTSTATUS EnumerateAndDeltaUpdateKeys(
|
|
IN KeyNode *pSource, // this is under the latest updated .Default\SW hive
|
|
IN KeyNode *pRef, // this was a ref-copy of .Default\SW before the update
|
|
IN KeyNode *pDestination,// this is opur private TS-hive
|
|
IN KeyBasicInfo *pBasicInfo,
|
|
IN ULONG *pIndentLevel)
|
|
{
|
|
NTSTATUS status=STATUS_SUCCESS, st2;
|
|
ULONG ulCount=0;
|
|
|
|
UNICODE_STRING UniString;
|
|
ULONG size;
|
|
|
|
(*pIndentLevel)++;
|
|
|
|
while (NT_SUCCESS_EX(status))
|
|
{
|
|
ULONG ultemp;
|
|
|
|
status = NtEnumerateKey( pSource->Key(),
|
|
ulCount++,
|
|
pBasicInfo->Type() , // keyInformationClass,
|
|
pBasicInfo->Ptr(), // pKeyInfo,
|
|
pBasicInfo->Size(), // keyInfoSize,
|
|
&ultemp);
|
|
|
|
// pBasicInfo was filled up thru NtEnumerateKey() above
|
|
|
|
if (NT_SUCCESS_EX(status))
|
|
{
|
|
if ( g_debugIO)
|
|
{
|
|
DebugKeyStamp( status , pBasicInfo, *pIndentLevel );
|
|
}
|
|
|
|
KeyNode RefSubKey( pRef, pBasicInfo);
|
|
KeyNode SourceSubKey( pSource,pBasicInfo);
|
|
|
|
// calling Open() on this may fail, and we will need to delete and recreate it if required.
|
|
KeyNode *pDestinationSubKey = new KeyNode( pDestination, pBasicInfo);
|
|
|
|
RefSubKey.Open();
|
|
SourceSubKey.Open();
|
|
|
|
if ( pDestinationSubKey )
|
|
{
|
|
pDestinationSubKey->Open();
|
|
|
|
if (NT_SUCCESS_EX( status = SourceSubKey.Status() ) )
|
|
{
|
|
// key is missing from the ref-hive, we should add
|
|
// the same sub-tree into our TS\Install\SW hive
|
|
if ( RefSubKey.Status() == STATUS_OBJECT_NAME_NOT_FOUND
|
|
|| RefSubKey.Status() == STATUS_OBJECT_PATH_SYNTAX_BAD)
|
|
{
|
|
// @@@
|
|
// we expect the key not to exist, if it does, then what? delete it?
|
|
if ( !NT_SUCCESS( pDestinationSubKey->Status()) )
|
|
{
|
|
// here is what were are doing with the strings:
|
|
// 1) get the path below the "\Registry\User\.Default", which would be
|
|
// something like "\Software\SomeDir\SomeDirOther\etc", this is the sub-path
|
|
// 2) create a new node at the destination, which would be something like:
|
|
// \HKLM\SW\MS\Windows NT\CurrentVersion\TS\INstall + the sub path
|
|
// we got above.
|
|
|
|
PWCHAR pwch;
|
|
SourceSubKey.GetPath( &pwch );
|
|
|
|
// this is the trailing part of the key-path missing from our TS hive
|
|
PWCHAR pDestinationSubPath = &pwch[g_length_TERMSRV_USERREGISTRY_DEFAULT ];
|
|
PWCHAR pDestinationFullPath= new WCHAR [ g_length_TERMSRV_INSTALL +
|
|
wcslen( pDestinationSubPath) + sizeof(WCHAR )];
|
|
wcscpy( pDestinationFullPath, TERMSRV_INSTALL );
|
|
wcscat( pDestinationFullPath, pDestinationSubPath );
|
|
|
|
|
|
DELETE_AND_NULL( pDestinationSubKey );
|
|
// create a new KeyNode object where the root will be TERMSRV_INSTALL,
|
|
// below which we will create a sub-layer of nodes, or a single node.
|
|
pDestinationSubKey = new KeyNode( NULL , pDestination->Masks(), pDestinationFullPath);
|
|
|
|
// create the new key/branch/values
|
|
status = pDestinationSubKey->CreateEx() ;
|
|
|
|
if ( g_debugIO )
|
|
{
|
|
DebugKeyStamp( status, pBasicInfo, *pIndentLevel , L"[KEY WAS CREATED]");
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
else
|
|
{
|
|
// if we have anything but success, set status and bail out
|
|
if ( !NT_SUCCESS_EX( status = RefSubKey.Status()) )
|
|
{
|
|
if ( g_debugIO )
|
|
{
|
|
DebugErrorStamp(status, __LINE__ );
|
|
}
|
|
}
|
|
}
|
|
|
|
// Key (if it is NEW) is NOT missing from destination hive at this point
|
|
// either it did exist, or was created in the above block of code
|
|
|
|
// check if there are any new values in this node.
|
|
|
|
KEY_FULL_INFORMATION *ptrInfo;
|
|
ULONG size;
|
|
|
|
NTSTATUS st3 = SourceSubKey.Query( &ptrInfo, &size );
|
|
|
|
if (NT_SUCCESS( st3 ))
|
|
{
|
|
ValueFullInfo sourceValueFullInfo( &SourceSubKey );
|
|
|
|
if ( NT_SUCCESS_EX( status = sourceValueFullInfo.Status() ) )
|
|
{
|
|
for (ULONG ulkey = 0; ulkey < ptrInfo->Values; ulkey++)
|
|
{
|
|
status = NtEnumerateValueKey(SourceSubKey.Key(),
|
|
ulkey,
|
|
sourceValueFullInfo.Type(),
|
|
sourceValueFullInfo.Ptr(),
|
|
sourceValueFullInfo.Size(),
|
|
&ultemp);
|
|
|
|
// @@@
|
|
if ( ! NT_SUCCESS( status ))
|
|
{
|
|
DebugErrorStamp( status, __LINE__ );
|
|
}
|
|
|
|
// if the ref key is missing a value, then add
|
|
// value to the destination key.
|
|
|
|
KEY_VALUE_PARTIAL_INFORMATION *pValuePartialInfo;
|
|
ValuePartialInfo refValuePartialInfo( &RefSubKey );
|
|
|
|
if ( NT_SUCCESS_EX( status = refValuePartialInfo.Status() ) )
|
|
{
|
|
refValuePartialInfo.Query( sourceValueFullInfo.SzName() );
|
|
|
|
// if .Default\SW has a value that is missing from the ref hive, then add
|
|
// corresponding value into our TS\ hive
|
|
if ( !NT_SUCCESS( refValuePartialInfo.Status()) )
|
|
{
|
|
// make sure pDestinationSubKey exists, else, create the key first before we
|
|
// write a value. It is possible that even though the key did exists in the ref
|
|
// hive at the start, a new value was added for the first time, which means that the
|
|
// ts hive is getting the key and the value for the first time.
|
|
if ( !NT_SUCCESS( pDestinationSubKey->Status() ) )
|
|
{
|
|
// here is what were are doing with the strings:
|
|
// 1) get the path below the "\Registry\User\.Default", which would be
|
|
// something like "\Software\SomeDir\SomeDirOther\etc", this is the sub-path
|
|
// 2) create a new node at the destination, which would be something like:
|
|
// \HKLM\SW\MS\Windows NT\CurrentVersion\TS\INstall + the sub path
|
|
// we got above.
|
|
|
|
PWCHAR pwch;
|
|
SourceSubKey.GetPath( &pwch );
|
|
|
|
// this is the trailing part of the key-path missing from our TS hive
|
|
PWCHAR pDestinationSubPath = &pwch[g_length_TERMSRV_USERREGISTRY_DEFAULT ];
|
|
PWCHAR pDestinationFullPath= new WCHAR [ g_length_TERMSRV_INSTALL +
|
|
wcslen( pDestinationSubPath) + sizeof(WCHAR )];
|
|
wcscpy( pDestinationFullPath, TERMSRV_INSTALL );
|
|
wcscat( pDestinationFullPath, pDestinationSubPath );
|
|
|
|
|
|
DELETE_AND_NULL( pDestinationSubKey );
|
|
// create a new KeyNode object where the root will be TERMSRV_INSTALL,
|
|
// below which we will create a sub-layer of nodes, or a single node.
|
|
pDestinationSubKey = new KeyNode( NULL , pDestination->Masks(), pDestinationFullPath);
|
|
|
|
// create the new key/branch/values
|
|
status = pDestinationSubKey->CreateEx();
|
|
if ( g_debugIO )
|
|
{
|
|
DebugKeyStamp( status, pBasicInfo, *pIndentLevel, L"[KEY WAS CREATED]" );
|
|
}
|
|
|
|
}
|
|
//else, no problem, key did exist and we don't need to create it
|
|
|
|
// set value at the destination node
|
|
ValueFullInfo destinationValue( pDestinationSubKey );
|
|
if ( NT_SUCCESS_EX( status = destinationValue.Status()) )
|
|
{
|
|
status = destinationValue.Create( &sourceValueFullInfo );
|
|
|
|
NT_SUCCESS_EX( status );
|
|
// if status is error, we bail out.
|
|
}
|
|
//else, out of memory, var-status is set and we bail out.
|
|
|
|
}
|
|
else // values are not missing, see if they are the same
|
|
{
|
|
// compare the two data buffers, if the one from SourceSubKey is
|
|
// different than the one from the RefSubKey, then delete
|
|
// and create one in DestinationSubKey
|
|
|
|
ValueFullInfo sourceValue( &SourceSubKey);
|
|
ValueFullInfo refValue ( &RefSubKey );
|
|
|
|
if (NT_SUCCESS_EX( status = refValue.Status())
|
|
&& NT_SUCCESS_EX( status = sourceValue.Status())
|
|
)
|
|
{
|
|
sourceValue.Query( sourceValueFullInfo.SzName() );
|
|
refValue.Query ( sourceValueFullInfo.SzName() );
|
|
|
|
if (NT_SUCCESS( refValue.Status())
|
|
&& NT_SUCCESS( sourceValue.Status()))
|
|
{
|
|
BOOLEAN theSame = sourceValue.Compare( &refValue );
|
|
|
|
if (! theSame )
|
|
{
|
|
|
|
// make sure pDestinationSubKey exists, else, create the key first before we
|
|
// write a value. It is possible that even though the key did exists in the ref
|
|
// hive at the start, a new value was added for the first time, which means that the
|
|
// ts hive is getting the key and the value for the first time.
|
|
if ( !NT_SUCCESS( pDestinationSubKey->Status() ) )
|
|
{
|
|
// here is what were are doing with the strings:
|
|
// 1) get the path below the "\Registry\User\.Default", which would be
|
|
// something like "\Software\SomeDir\SomeDirOther\etc", this is the sub-path
|
|
// 2) create a new node at the destination, which would be something like:
|
|
// \HKLM\SW\MS\Windows NT\CurrentVersion\TS\INstall + the sub path
|
|
// we got above.
|
|
|
|
PWCHAR pwch;
|
|
SourceSubKey.GetPath( &pwch );
|
|
|
|
// this is the trailing part of the key-path missing from our TS hive
|
|
PWCHAR pDestinationSubPath = &pwch[g_length_TERMSRV_USERREGISTRY_DEFAULT ];
|
|
PWCHAR pDestinationFullPath= new WCHAR [ g_length_TERMSRV_INSTALL +
|
|
wcslen( pDestinationSubPath) + sizeof(WCHAR )];
|
|
wcscpy( pDestinationFullPath, TERMSRV_INSTALL );
|
|
wcscat( pDestinationFullPath, pDestinationSubPath );
|
|
|
|
|
|
DELETE_AND_NULL( pDestinationSubKey );
|
|
// create a new KeyNode object where the root will be TERMSRV_INSTALL,
|
|
// below which we will create a sub-layer of nodes, or a single node.
|
|
pDestinationSubKey = new KeyNode( NULL , pDestination->Masks(), pDestinationFullPath);
|
|
|
|
// create the new key/branch/values
|
|
status = pDestinationSubKey->CreateEx();
|
|
if ( g_debugIO )
|
|
{
|
|
DebugKeyStamp( status, pBasicInfo, *pIndentLevel , L"KEY WAS CREATED");
|
|
}
|
|
|
|
}
|
|
//else, no problem, key did exist and we don't need to create it
|
|
|
|
ValueFullInfo destinationValue( pDestinationSubKey );
|
|
if ( NT_SUCCESS( destinationValue.Status() ) )
|
|
{
|
|
// don't care if it exists or not, delete it first
|
|
destinationValue.Delete( sourceValueFullInfo.SzName() );
|
|
}
|
|
// else, there is no destination value to delete
|
|
|
|
// update/create item under destination
|
|
|
|
// Create a destination value identical to the source value
|
|
status = destinationValue.Create( &sourceValue );
|
|
|
|
// if status is error, we will bail out
|
|
if (!NT_SUCCESS_EX( status ))
|
|
{
|
|
if (g_debugIO)
|
|
{
|
|
DebugErrorStamp(status, __LINE__,
|
|
&sourceValue );
|
|
}
|
|
}
|
|
}
|
|
}
|
|
// else, values don't exits, doesn't make sense, maybe some dbug code here?
|
|
}
|
|
// else, var-status is set, we bail out.
|
|
}
|
|
//if-else
|
|
}
|
|
//else, out of memory, var-status is set, we bail out
|
|
}
|
|
// for-loop
|
|
}
|
|
//else, out of memory, var-status is set, we bail out
|
|
}
|
|
else
|
|
{
|
|
// this sbould not really happen, but for now...
|
|
if ( g_debugIO )
|
|
{
|
|
DebugErrorStamp( status, __LINE__ );
|
|
}
|
|
}
|
|
|
|
|
|
// by now, either both source and destination nodes exist, or
|
|
// a new destination node was just created above. In any case,
|
|
// we can continue the traversal.
|
|
if ( NT_SUCCESS( status ) )
|
|
{
|
|
status = EnumerateAndDeltaUpdateKeys(
|
|
&SourceSubKey,
|
|
&RefSubKey,
|
|
pDestinationSubKey,
|
|
pBasicInfo ,
|
|
pIndentLevel);
|
|
|
|
NT_SUCCESS_EX( status );
|
|
}
|
|
//else, we are bailing out
|
|
}
|
|
// else, var-status is set, we bail out.
|
|
|
|
// done with this sub key,
|
|
DELETE_AND_NULL( pDestinationSubKey );
|
|
}
|
|
else
|
|
{
|
|
status = STATUS_NO_MEMORY;
|
|
}
|
|
|
|
status = AlterStatus( status, __LINE__ );
|
|
}
|
|
// else, no more entries
|
|
}
|
|
// no more entries
|
|
|
|
|
|
(*pIndentLevel)--;
|
|
// the typical status would be: STATUS_NO_MORE_ENTRIES
|
|
return status;
|
|
}
|
|
|
|
// delete the ref-hive as specific by the uniRef string
|
|
NTSTATUS DeleteReferenceHive(WCHAR *uniRef)
|
|
{
|
|
if ( g_debugIO)
|
|
{
|
|
fwprintf( g_debugFilePointer,L"In %ws\n",
|
|
L"----DeleteReferenceHive");
|
|
fflush( g_debugFilePointer );
|
|
}
|
|
|
|
NTSTATUS status = STATUS_SUCCESS;
|
|
|
|
KeyNode Old( NULL, KEY_WRITE | KEY_READ | DELETE, uniRef );
|
|
if ( NT_SUCCESS( Old.Open() ) )
|
|
{
|
|
Old.DeleteSubKeys();
|
|
status = Old.Delete(); // delete the head of the branch
|
|
}
|
|
Old.Close();
|
|
|
|
return status;
|
|
}
|
|
|
|
/****************************************************************
|
|
*
|
|
* Function:
|
|
* CreateReferenceHive
|
|
*
|
|
* Parameters:
|
|
* uniSource (source ) string points to the node under .Default
|
|
* uniRef (ref ) string point to TS\Install\RefHive
|
|
* UniDest (Destination) string points to TS\Install\Software
|
|
*
|
|
* Description:
|
|
* from the .Default (source) hive, copy into TS\install\RefHive
|
|
* source hive is specified by the uniSoure string, and the
|
|
* ref-hive is specified by the uniRef string.
|
|
*
|
|
* Return:
|
|
* NTSTATUS, if successful, then STATUS_SUCCESS
|
|
*
|
|
****************************************************************/
|
|
NTSTATUS CreateReferenceHive(WCHAR *uniSource, WCHAR *uniRef)
|
|
{
|
|
if ( g_debugIO)
|
|
{
|
|
fwprintf( g_debugFilePointer,L"In %ws\n",
|
|
L"----CreateReferenceHive");
|
|
fflush( g_debugFilePointer );
|
|
}
|
|
|
|
// 1-COPY
|
|
// copy all keys under .Default\Software into a special location
|
|
// under our TS hive, let's call it the RefHive
|
|
// This will act as the reference hive
|
|
|
|
NTSTATUS status = STATUS_SUCCESS;
|
|
ULONG indentLevel=0;
|
|
|
|
// start creating our cache Ref hive
|
|
|
|
KeyNode Ref( NULL, MAXIMUM_ALLOWED, uniRef );
|
|
|
|
// if we were able to create our RefHive, then continue...
|
|
if ( NT_SUCCESS_EX( status = Ref.Create() ) )
|
|
{
|
|
KeyNode Source(NULL, KEY_READ, uniSource );
|
|
|
|
// open the source reg-key-path
|
|
if (NT_SUCCESS_EX( status = Source.Open() ))
|
|
{
|
|
KeyBasicInfo kBasicInfo;
|
|
|
|
if (NT_SUCCESS_EX( status = kBasicInfo.Status() ))
|
|
{
|
|
// this will be a recursive call, so we are saving allocation
|
|
// cycles by passing kBasicInfo as scratch pad.
|
|
status = EnumerateAndCreateRefHive( &Source,
|
|
&Ref,
|
|
&kBasicInfo,
|
|
&indentLevel);
|
|
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
if ( status == STATUS_NO_MORE_ENTRIES)
|
|
{
|
|
status = STATUS_SUCCESS;
|
|
}
|
|
|
|
return status;
|
|
}
|
|
|
|
/************************************************************************
|
|
*
|
|
* Function:
|
|
* DeltaDeleteKeys(WCHAR *uniSource, WCHAR *uniRef, WCHAR *uniDest)
|
|
*
|
|
* Parameters:
|
|
* uniSource (source ) string points to the node under .Default
|
|
* uniRef (ref ) string point to TS\Install\RefHive
|
|
* UniDest (Destination) string points to TS\Install\Software
|
|
*
|
|
* Description:
|
|
* compare .Dfeault keys to the equivalent keys in RefHive. If keys are
|
|
* missing from .Default, then delete the equivalent keys from our
|
|
* HKLM\...\TS\ hive
|
|
*
|
|
* Return:
|
|
* NTSTATUS, if successful, then STATS_SUCCESS
|
|
*
|
|
************************************************************************/
|
|
NTSTATUS DeltaDeleteKeys(WCHAR *uniSource, WCHAR *uniRef, WCHAR *uniDest)
|
|
{
|
|
if ( g_debugIO)
|
|
{
|
|
fwprintf( g_debugFilePointer,L"In %ws\n",
|
|
L"----DeltaDeleteKeys");
|
|
fflush( g_debugFilePointer );
|
|
}
|
|
|
|
// Step2-DELETE
|
|
// compare .Dfeault keys to the equivalent keys in RefHive. If keys are
|
|
// missing from .Default, then delete the equivalent keys from our
|
|
// HKLM\...\TS\ hive
|
|
|
|
KeyNode Source( NULL, KEY_READ, uniSource );
|
|
KeyNode Ref( NULL, MAXIMUM_ALLOWED, uniRef );
|
|
KeyNode Destination( NULL, MAXIMUM_ALLOWED, uniDest );
|
|
|
|
Source.Open();
|
|
Ref.Open();
|
|
Destination.Open();
|
|
|
|
ULONG indentLevel=0;
|
|
NTSTATUS status = STATUS_SUCCESS;
|
|
|
|
if ( NT_SUCCESS_EX( status = Source.Status() ) &&
|
|
NT_SUCCESS_EX( status = Ref.Status() ) &&
|
|
NT_SUCCESS_EX( status = Destination.Status() ) )
|
|
{
|
|
KeyBasicInfo basicInfo;
|
|
|
|
if( NT_SUCCESS_EX( status = basicInfo.Status() ) )
|
|
{
|
|
// walk and compare, if missing from Source, then delete from Destination
|
|
status = EnumerateAndDeltaDeleteKeys(
|
|
&Source,
|
|
&Ref,
|
|
&Destination,
|
|
&basicInfo,
|
|
&indentLevel);
|
|
}
|
|
}
|
|
|
|
if ( status == STATUS_NO_MORE_ENTRIES)
|
|
{
|
|
status = STATUS_SUCCESS;
|
|
}
|
|
|
|
return status;
|
|
}
|
|
|
|
/************************************************************************
|
|
*
|
|
* Function:
|
|
* DeltaUpdateKeys(WCHAR *uniSource, WCHAR *uniRef, WCHAR *uniDest)
|
|
*
|
|
* Parameters:
|
|
* uniSource (source ) string points to the node under .Default
|
|
* uniRef (ref ) string point to TS\Install\RefHive
|
|
* UniDest (Destination) string points to TS\Install\Software
|
|
*
|
|
* Description:
|
|
* Step-3 CREATE/Update keys and values
|
|
* compare .Default keys to the equivalent keys in RefHive, if keys are
|
|
* present in .Default that are missing from RefHive, then, add those keys
|
|
* to our HKLM\...\TS hive. Do the same for the values.
|
|
* Then, compare values from .Default to values in .Ref. If values have
|
|
* changed, then delete the value from our destination hive and create a
|
|
* new one with the appropriate data from .Default
|
|
*
|
|
* Return:
|
|
* NTSTATUS, if successful, then STATS_SUCCESS
|
|
*
|
|
************************************************************************/
|
|
NTSTATUS DeltaUpdateKeys (WCHAR *uniSource, WCHAR *uniRef, WCHAR *uniDest)
|
|
{
|
|
if ( g_debugIO)
|
|
{
|
|
fwprintf( g_debugFilePointer,L"In %ws\n",
|
|
L"----DeltaUpdateKeys");
|
|
fflush( g_debugFilePointer );
|
|
}
|
|
// 3-CREATE
|
|
// compare .Default keys to the equivalent keys in RefHive, if keys are
|
|
// present in .Default that are missing from RefHive, then, add those keys
|
|
// to our HKLM\...\TS hive
|
|
KeyNode Source( NULL, KEY_READ, uniSource );
|
|
KeyNode Ref( NULL, MAXIMUM_ALLOWED, uniRef );
|
|
KeyNode Destination( NULL, MAXIMUM_ALLOWED, uniDest );
|
|
|
|
Source.Open();
|
|
Ref.Open();
|
|
Destination.Open();
|
|
|
|
NTSTATUS status;
|
|
ULONG indentLevel=0;
|
|
|
|
if ( NT_SUCCESS_EX( status = Source.Status() ) &&
|
|
NT_SUCCESS_EX( status = Ref.Status() ) &&
|
|
NT_SUCCESS_EX( status = Destination.Status() ) )
|
|
{
|
|
|
|
KeyBasicInfo basicInfo;
|
|
|
|
// Constructor in KeyBasicInfo above allocates memory for pInfo
|
|
// check if memory allocation of pInfo succeeded
|
|
status = basicInfo.Status();
|
|
if (status != STATUS_SUCCESS) {
|
|
return status;
|
|
}
|
|
|
|
// walk and compare, if missing from Source, then delete from Destination
|
|
status = EnumerateAndDeltaUpdateKeys(
|
|
&Source,
|
|
&Ref,
|
|
&Destination,
|
|
&basicInfo,
|
|
&indentLevel);
|
|
}
|
|
|
|
if ( status == STATUS_NO_MORE_ENTRIES)
|
|
{
|
|
status = STATUS_SUCCESS;
|
|
}
|
|
|
|
return status;
|
|
}
|
|
|