/**************************************************************************************** * 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 #include #include #include #include #include #include #include #include #include #include #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 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; }