826 lines
22 KiB
C++
826 lines
22 KiB
C++
|
//#pragma title( "TReg.cpp - NT registry class" )
|
||
|
|
/*
|
||
|
|
Copyright (c) 1995-1998, Mission Critical Software, Inc. All rights reserved.
|
||
|
|
===============================================================================
|
||
|
|
Module - TReg.cpp
|
||
|
|
System - Common
|
||
|
|
Author - Tom Bernhardt, Rich Denham
|
||
|
|
Created - 1995-09-01
|
||
|
|
Description - NT registry class.
|
||
|
|
Updates -
|
||
|
|
===============================================================================
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifdef USE_STDAFX
|
||
|
|
# include "stdafx.h"
|
||
|
|
#else
|
||
|
|
# include <windows.h>
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <time.h>
|
||
|
|
|
||
|
|
#include "Common.hpp"
|
||
|
|
#include "UString.hpp"
|
||
|
|
#include "Err.hpp"
|
||
|
|
#include "TNode.hpp"
|
||
|
|
|
||
|
|
#include "TReg.hpp"
|
||
|
|
|
||
|
|
// Short term solution
|
||
|
|
#define MAX_REG_NAMELEN 512
|
||
|
|
#define MAX_REG_VALUELEN 2048
|
||
|
|
|
||
|
|
|
||
|
|
// Destructor function was formerly inline.
|
||
|
|
// It is here to facilitate handle leak tracing.
|
||
|
|
|
||
|
|
TRegKey::~TRegKey()
|
||
|
|
{
|
||
|
|
Close();
|
||
|
|
};
|
||
|
|
|
||
|
|
// Close function was formerly inline.
|
||
|
|
// It is here to facilitate handle leak tracing.
|
||
|
|
void
|
||
|
|
TRegKey::Close()
|
||
|
|
{
|
||
|
|
if ( hKey != INVALID_HANDLE_VALUE )
|
||
|
|
{
|
||
|
|
RegCloseKey( hKey );
|
||
|
|
hKey = (HKEY) INVALID_HANDLE_VALUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
};
|
||
|
|
|
||
|
|
// open registry on remote computer
|
||
|
|
DWORD
|
||
|
|
TRegKey::Connect(
|
||
|
|
HKEY hPreDefined ,// in -must be HKEY_LOCAL_MACHINE or HKEY_USERS
|
||
|
|
TCHAR const * machineName // in -remote computer name
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DWORD rc; // return code
|
||
|
|
|
||
|
|
if ( hKey != INVALID_HANDLE_VALUE )
|
||
|
|
{
|
||
|
|
Close();
|
||
|
|
}
|
||
|
|
|
||
|
|
rc = RegConnectRegistry( const_cast<TCHAR *>(machineName), hPreDefined, &hKey );
|
||
|
|
|
||
|
|
if ( rc )
|
||
|
|
{
|
||
|
|
hKey = (HKEY) INVALID_HANDLE_VALUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
return rc;
|
||
|
|
}
|
||
|
|
|
||
|
|
// create new key
|
||
|
|
DWORD
|
||
|
|
TRegKey::Create(
|
||
|
|
TCHAR const * keyname ,// in -name/path of key to create/open
|
||
|
|
HKEY hParent ,// in -handle of parent key
|
||
|
|
DWORD * pDisp ,// out-disposition of create
|
||
|
|
DWORD access // in -security access mask for key
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DWORD disp;
|
||
|
|
DWORD rc;
|
||
|
|
|
||
|
|
if ( hKey != INVALID_HANDLE_VALUE )
|
||
|
|
{
|
||
|
|
Close();
|
||
|
|
}
|
||
|
|
|
||
|
|
rc = RegCreateKeyEx( hParent,
|
||
|
|
keyname,
|
||
|
|
0,
|
||
|
|
NULL,
|
||
|
|
REG_OPTION_NON_VOLATILE,
|
||
|
|
access,
|
||
|
|
NULL,
|
||
|
|
&hKey,
|
||
|
|
(pDisp!=NULL) ? pDisp : &disp );
|
||
|
|
if ( rc )
|
||
|
|
{
|
||
|
|
hKey = (HKEY) INVALID_HANDLE_VALUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
return rc;
|
||
|
|
}
|
||
|
|
|
||
|
|
// create new key (using backup/restore)
|
||
|
|
DWORD
|
||
|
|
TRegKey::CreateBR(
|
||
|
|
TCHAR const * keyname ,// in -name/path of key to create/open
|
||
|
|
HKEY hParent ,// in -handle of parent key
|
||
|
|
DWORD * pDisp ,// out-disposition of create
|
||
|
|
DWORD access // in -security access mask for key
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DWORD disp;
|
||
|
|
DWORD rc;
|
||
|
|
|
||
|
|
if ( hKey != INVALID_HANDLE_VALUE )
|
||
|
|
{
|
||
|
|
Close();
|
||
|
|
}
|
||
|
|
|
||
|
|
rc = RegCreateKeyEx( hParent,
|
||
|
|
keyname,
|
||
|
|
0,
|
||
|
|
NULL,
|
||
|
|
REG_OPTION_BACKUP_RESTORE,
|
||
|
|
access,
|
||
|
|
NULL,
|
||
|
|
&hKey,
|
||
|
|
(pDisp!=NULL) ? pDisp : &disp );
|
||
|
|
if ( rc )
|
||
|
|
{
|
||
|
|
hKey = (HKEY) INVALID_HANDLE_VALUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
return rc;
|
||
|
|
}
|
||
|
|
|
||
|
|
// open existing key
|
||
|
|
DWORD
|
||
|
|
TRegKey::Open(
|
||
|
|
TCHAR const * keyname ,// in -name/path of key to create/open
|
||
|
|
HKEY hParent ,// in -handle of parent key
|
||
|
|
DWORD access // in -security access mask for key
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DWORD rc;
|
||
|
|
|
||
|
|
if ( hKey != INVALID_HANDLE_VALUE )
|
||
|
|
{
|
||
|
|
Close();
|
||
|
|
}
|
||
|
|
|
||
|
|
rc = RegOpenKeyEx( hParent,
|
||
|
|
keyname,
|
||
|
|
0,
|
||
|
|
access,
|
||
|
|
&hKey );
|
||
|
|
if ( rc )
|
||
|
|
{
|
||
|
|
hKey = (HKEY) INVALID_HANDLE_VALUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
return rc;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Gets the subkey value of the specified index number
|
||
|
|
DWORD // ret-os return code
|
||
|
|
TRegKey::SubKeyEnum(
|
||
|
|
DWORD n ,// in -ordinal number of subkey
|
||
|
|
TCHAR * keyname ,// out-key name
|
||
|
|
DWORD keylen // in -max length of key name
|
||
|
|
) const
|
||
|
|
{
|
||
|
|
DWORD rc;
|
||
|
|
DWORD keyLen = keylen;
|
||
|
|
FILETIME lastWrite;
|
||
|
|
|
||
|
|
rc = RegEnumKeyEx( hKey,
|
||
|
|
n,
|
||
|
|
keyname,
|
||
|
|
&keyLen,
|
||
|
|
0,
|
||
|
|
NULL,
|
||
|
|
NULL,
|
||
|
|
&lastWrite );
|
||
|
|
|
||
|
|
return rc;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Enumerate value
|
||
|
|
// Note that "namelen" must be "sizeof name", not "DIM(name)"
|
||
|
|
// Same for "valuelen"
|
||
|
|
DWORD // ret-0 or error code
|
||
|
|
TRegKey::ValueEnum(
|
||
|
|
DWORD index ,// in -ordinal number of subkey
|
||
|
|
TCHAR * name ,// out-name
|
||
|
|
DWORD namelen ,// in -name length
|
||
|
|
void * value ,// out-value
|
||
|
|
DWORD * valuelen ,// i/o-value length
|
||
|
|
DWORD * type // out-value type code
|
||
|
|
) const
|
||
|
|
{
|
||
|
|
return RegEnumValue( hKey, index, name, &namelen, NULL, type, (BYTE *) value, valuelen );
|
||
|
|
}
|
||
|
|
|
||
|
|
// Get REG_DWORD value
|
||
|
|
DWORD // ret-OS return code
|
||
|
|
TRegKey::ValueGetDWORD(
|
||
|
|
TCHAR const * name ,// in -value name
|
||
|
|
DWORD * value // out-returned DWORD value
|
||
|
|
) const
|
||
|
|
{
|
||
|
|
DWORD osRc; // OS return code
|
||
|
|
DWORD type; // type of value
|
||
|
|
DWORD len = sizeof *value; // value length
|
||
|
|
|
||
|
|
osRc = RegQueryValueEx( hKey, name, NULL, &type, (BYTE *) value, &len );
|
||
|
|
|
||
|
|
if ( !osRc && (type != REG_DWORD) )
|
||
|
|
{
|
||
|
|
osRc = ERROR_FILE_NOT_FOUND;
|
||
|
|
}
|
||
|
|
|
||
|
|
return osRc;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Get REG_SZ value
|
||
|
|
DWORD // ret-OS return code
|
||
|
|
TRegKey::ValueGetStr(
|
||
|
|
TCHAR const * name ,// in -value name
|
||
|
|
TCHAR * value ,// out-value buffer
|
||
|
|
DWORD maxlen // in -sizeof value buffer
|
||
|
|
) const
|
||
|
|
{
|
||
|
|
DWORD osRc; // OS return code
|
||
|
|
DWORD type; // type of value
|
||
|
|
DWORD len; // value length
|
||
|
|
|
||
|
|
// force maxlen to an integral number of TEXT characters
|
||
|
|
maxlen = maxlen / (sizeof value[0]) * (sizeof value[0]);
|
||
|
|
|
||
|
|
if ( !maxlen )
|
||
|
|
{
|
||
|
|
osRc = ERROR_FILE_NOT_FOUND;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
len = maxlen;
|
||
|
|
osRc = RegQueryValueEx( hKey, name, NULL, &type, (BYTE *) value, &len );
|
||
|
|
len = len / (sizeof value[0]) * (sizeof value[0]);
|
||
|
|
if ( !osRc && (type != REG_SZ) )
|
||
|
|
{
|
||
|
|
osRc = ERROR_FILE_NOT_FOUND;
|
||
|
|
}
|
||
|
|
if ( osRc )
|
||
|
|
{
|
||
|
|
value[0] = TEXT('\0');
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{ // return of a null-terminated string is not guaranteed by API!
|
||
|
|
// force null-terminated string, truncate string if necessary.
|
||
|
|
if ( len >= maxlen )
|
||
|
|
{
|
||
|
|
len = maxlen - sizeof value[0];
|
||
|
|
}
|
||
|
|
value[len/(sizeof value[0])] = TEXT('\0');
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return osRc;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
TRegKey::ValueGet(
|
||
|
|
TCHAR const * name ,// in -name
|
||
|
|
void * value ,// out-value
|
||
|
|
DWORD * lenvalue ,// i/o-length of value
|
||
|
|
DWORD * typevalue // out-type of value
|
||
|
|
) const
|
||
|
|
{
|
||
|
|
return RegQueryValueEx( hKey, name, 0, typevalue, (UCHAR *) value, lenvalue );
|
||
|
|
}
|
||
|
|
|
||
|
|
// Set REG_SZ value
|
||
|
|
DWORD
|
||
|
|
TRegKey::ValueSetStr(
|
||
|
|
TCHAR const * name ,// in -value name
|
||
|
|
TCHAR const * value ,// out-value
|
||
|
|
DWORD type // in -value type
|
||
|
|
) const
|
||
|
|
{
|
||
|
|
return RegSetValueEx( hKey,
|
||
|
|
name,
|
||
|
|
NULL,
|
||
|
|
type,
|
||
|
|
(LPBYTE) value,
|
||
|
|
(UStrLen(value) + 1) * sizeof value[0] );
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
TRegKey::ValueSet(
|
||
|
|
TCHAR const * name ,// in -name
|
||
|
|
void const * value ,// in -value
|
||
|
|
DWORD lenvalue ,// in -length of value
|
||
|
|
DWORD typevalue // in -type of value
|
||
|
|
) const
|
||
|
|
{
|
||
|
|
return RegSetValueEx( hKey,
|
||
|
|
name,
|
||
|
|
0,
|
||
|
|
typevalue,
|
||
|
|
(UCHAR const *) value,
|
||
|
|
lenvalue );
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD // ret-0 or error code
|
||
|
|
TRegKey::ValueDel(
|
||
|
|
TCHAR const * name // in -value name
|
||
|
|
) const
|
||
|
|
{
|
||
|
|
DWORD rc;
|
||
|
|
|
||
|
|
rc = RegDeleteValue(hKey, name);
|
||
|
|
|
||
|
|
return rc;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD // ret-OS return code
|
||
|
|
TRegKey::HiveCopy(
|
||
|
|
TRegKey const * source // in -source hive
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DWORD retval=0; // returned value
|
||
|
|
DWORD index; // key/value index
|
||
|
|
TCHAR name[MAX_REG_NAMELEN]; // key name
|
||
|
|
TCHAR value[MAX_REG_VALUELEN]; // value name
|
||
|
|
DWORD valuelen; // value length
|
||
|
|
DWORD type; // value type
|
||
|
|
TRegKey srcNest; // nested source registry
|
||
|
|
TRegKey trgNest; // nested target registry
|
||
|
|
|
||
|
|
// process values at this level
|
||
|
|
for ( index = 0;
|
||
|
|
!retval;
|
||
|
|
index++ )
|
||
|
|
{
|
||
|
|
valuelen = sizeof value;
|
||
|
|
retval = source->ValueEnum( index, name, sizeof name, value, &valuelen, &type );
|
||
|
|
if ( !retval )
|
||
|
|
{
|
||
|
|
retval = this->ValueSet( name, value, valuelen, type );
|
||
|
|
}
|
||
|
|
else if ( retval == ERROR_MORE_DATA )
|
||
|
|
{
|
||
|
|
retval = 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( retval == ERROR_NO_MORE_ITEMS )
|
||
|
|
{
|
||
|
|
retval = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
// process keys at this level; for each key make a recursive call
|
||
|
|
for ( index = 0;
|
||
|
|
!retval;
|
||
|
|
index++ )
|
||
|
|
{
|
||
|
|
retval = source->SubKeyEnum( index, name, sizeof name );
|
||
|
|
if ( !retval )
|
||
|
|
{
|
||
|
|
retval = srcNest.Open( name, source );
|
||
|
|
if ( !retval )
|
||
|
|
{
|
||
|
|
retval = trgNest.Create( name, this );
|
||
|
|
if ( !retval )
|
||
|
|
{
|
||
|
|
retval = trgNest.HiveCopy( &srcNest );
|
||
|
|
trgNest.Close();
|
||
|
|
}
|
||
|
|
srcNest.Close();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( retval == ERROR_NO_MORE_ITEMS )
|
||
|
|
{
|
||
|
|
retval = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
return retval;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD // ret-OS return code
|
||
|
|
TRegKey::HiveDel()
|
||
|
|
{
|
||
|
|
DWORD retval = 0; // returned value
|
||
|
|
DWORD index; // value/key index
|
||
|
|
TCHAR name[300]; // name
|
||
|
|
DWORD namelen; // name length
|
||
|
|
BYTE value[300]; // value
|
||
|
|
DWORD valuelen; // value length
|
||
|
|
DWORD type; // value type code
|
||
|
|
TRegKey trgNest; // nested target registry
|
||
|
|
|
||
|
|
// delete values at this level
|
||
|
|
for ( index = 0;
|
||
|
|
!retval;
|
||
|
|
/* index++ */ ) // note that index remains at zero
|
||
|
|
{
|
||
|
|
namelen = sizeof name;
|
||
|
|
valuelen = sizeof value;
|
||
|
|
retval = ValueEnum( index, name, namelen, value, &valuelen, &type );
|
||
|
|
if ( retval == ERROR_MORE_DATA )
|
||
|
|
{
|
||
|
|
retval = 0;
|
||
|
|
}
|
||
|
|
if ( !retval )
|
||
|
|
{
|
||
|
|
retval = ValueDel( name );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( retval == ERROR_NO_MORE_ITEMS )
|
||
|
|
{
|
||
|
|
retval = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
// process keys at this level; for each key make a recursive call
|
||
|
|
for ( index = 0;
|
||
|
|
!retval;
|
||
|
|
/* index++ */ ) // note that index remains at zero
|
||
|
|
{
|
||
|
|
retval = SubKeyEnum( index, name, sizeof name );
|
||
|
|
if ( !retval )
|
||
|
|
{
|
||
|
|
retval = trgNest.Open( name, this );
|
||
|
|
if ( !retval )
|
||
|
|
{
|
||
|
|
retval = trgNest.HiveDel();
|
||
|
|
trgNest.Close();
|
||
|
|
}
|
||
|
|
retval = SubKeyDel( name );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( retval == ERROR_NO_MORE_ITEMS )
|
||
|
|
{
|
||
|
|
retval = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
return retval;
|
||
|
|
}
|
||
|
|
|
||
|
|
// These four classes are used only by TRegReplicate
|
||
|
|
// Class to represent one registry key
|
||
|
|
class RKey : public TNode
|
||
|
|
{
|
||
|
|
friend class RKeyList;
|
||
|
|
private:
|
||
|
|
TCHAR * name; // key name
|
||
|
|
protected:
|
||
|
|
public:
|
||
|
|
RKey() { name = NULL; };
|
||
|
|
~RKey() { if ( name ) delete name; };
|
||
|
|
BOOL New( TCHAR const * aname );
|
||
|
|
TCHAR const * GetName() const { return name; };
|
||
|
|
};
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
RKey::New(
|
||
|
|
TCHAR const * aname // in -key name
|
||
|
|
)
|
||
|
|
{
|
||
|
|
name = new TCHAR[UStrLen(aname)+1];
|
||
|
|
|
||
|
|
if ( name )
|
||
|
|
{
|
||
|
|
UStrCpy( name, aname );
|
||
|
|
}
|
||
|
|
|
||
|
|
return !!name;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Class to represent the set of registry keys at one level
|
||
|
|
class RKeyList : public TNodeListSortable
|
||
|
|
{
|
||
|
|
private:
|
||
|
|
static TNodeCompare( Compare ) { return UStrICmp(
|
||
|
|
((RKey const *) v1)->name,
|
||
|
|
((RKey const *) v2)->name ); }
|
||
|
|
protected:
|
||
|
|
public:
|
||
|
|
RKeyList() : TNodeListSortable( Compare ) {}
|
||
|
|
~RKeyList();
|
||
|
|
};
|
||
|
|
|
||
|
|
// RKeyList object destructor
|
||
|
|
RKeyList::~RKeyList()
|
||
|
|
{
|
||
|
|
DeleteAllListItems( RKey );
|
||
|
|
}
|
||
|
|
|
||
|
|
// Class to represent one registry value
|
||
|
|
class RValue : public TNode
|
||
|
|
{
|
||
|
|
friend class RValueList;
|
||
|
|
private:
|
||
|
|
TCHAR * name; // value's name
|
||
|
|
BYTE * value; // value's value
|
||
|
|
DWORD valuelen; // value's value length
|
||
|
|
DWORD type; // value's type
|
||
|
|
protected:
|
||
|
|
public:
|
||
|
|
RValue() { name = NULL; value = NULL; valuelen = type = 0; };
|
||
|
|
~RValue() { if ( name ) delete name;
|
||
|
|
if ( value ) delete value; };
|
||
|
|
BOOL New( TCHAR const * aname, BYTE const * avalue, DWORD valuelen, DWORD type );
|
||
|
|
TCHAR const * GetName() const { return name; };
|
||
|
|
BYTE const * GetValue() const { return value; };
|
||
|
|
DWORD GetValueLen() const { return valuelen; };
|
||
|
|
DWORD GetType() const { return type; };
|
||
|
|
};
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
RValue::New(
|
||
|
|
TCHAR const * aname ,// in -value's name
|
||
|
|
BYTE const * avalue ,// in -value's value
|
||
|
|
DWORD avaluelen ,// in -value's value length
|
||
|
|
DWORD atype // in -value's type
|
||
|
|
)
|
||
|
|
{
|
||
|
|
name = new TCHAR[UStrLen(aname)+1];
|
||
|
|
|
||
|
|
if ( name )
|
||
|
|
{
|
||
|
|
UStrCpy( name, aname );
|
||
|
|
}
|
||
|
|
|
||
|
|
value = new BYTE[avaluelen];
|
||
|
|
|
||
|
|
if ( value )
|
||
|
|
{
|
||
|
|
memcpy( value, avalue, avaluelen );
|
||
|
|
}
|
||
|
|
|
||
|
|
valuelen = avaluelen;
|
||
|
|
type = atype;
|
||
|
|
|
||
|
|
return name && value;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Class to represent the set of registry values at one level
|
||
|
|
class RValueList : public TNodeListSortable
|
||
|
|
{
|
||
|
|
private:
|
||
|
|
static TNodeCompare( Compare ) { return UStrICmp(
|
||
|
|
((RValue const *)v1)->name,
|
||
|
|
((RValue const *)v2)->name ); }
|
||
|
|
protected:
|
||
|
|
public:
|
||
|
|
RValueList() : TNodeListSortable( Compare ) {}
|
||
|
|
~RValueList();
|
||
|
|
};
|
||
|
|
|
||
|
|
// RValueList object destructor
|
||
|
|
RValueList::~RValueList()
|
||
|
|
{
|
||
|
|
DeleteAllListItems( RValue );
|
||
|
|
}
|
||
|
|
|
||
|
|
// Static subroutine used only by TRegReplicate
|
||
|
|
// collect all values at one registry level into a RValueList
|
||
|
|
DWORD static
|
||
|
|
CollectValues(
|
||
|
|
RValueList * pValueList ,// out-value list to be built
|
||
|
|
TRegKey const * pRegKey // in -registry key
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DWORD retval=0; // returned value
|
||
|
|
DWORD index; // value enum index
|
||
|
|
TCHAR name[MAX_REG_NAMELEN]; // value name
|
||
|
|
BYTE value[MAX_REG_VALUELEN]; // value value
|
||
|
|
DWORD valuelen; // value length
|
||
|
|
DWORD type; // value type
|
||
|
|
RValue * pValue; // new value
|
||
|
|
|
||
|
|
for ( index = 0;
|
||
|
|
!retval;
|
||
|
|
index++ )
|
||
|
|
{
|
||
|
|
valuelen = sizeof value;
|
||
|
|
retval = pRegKey->ValueEnum( index, name, sizeof name, value, &valuelen, &type );
|
||
|
|
if ( !retval )
|
||
|
|
{
|
||
|
|
pValue = new RValue;
|
||
|
|
if ( pValue )
|
||
|
|
{
|
||
|
|
if ( pValue->New( name, value, valuelen, type ) )
|
||
|
|
{
|
||
|
|
pValueList->Insert( pValue );
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
delete pValue;
|
||
|
|
pValue = NULL;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if ( !pValue )
|
||
|
|
{
|
||
|
|
retval = ERROR_NOT_ENOUGH_MEMORY;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if ( retval == ERROR_MORE_DATA )
|
||
|
|
{
|
||
|
|
retval = 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if ( retval == ERROR_NO_MORE_ITEMS )
|
||
|
|
{
|
||
|
|
retval = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
return retval;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Static subroutine used only by TRegReplicate
|
||
|
|
// collect all keys at one registry level into a RKeyList
|
||
|
|
DWORD static
|
||
|
|
CollectKeys(
|
||
|
|
RKeyList * pKeyList ,// out-key list to be built
|
||
|
|
TRegKey const * pRegKey // in -registry key
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DWORD retval=0; // returned value
|
||
|
|
DWORD index; // key enum index
|
||
|
|
TCHAR name[MAX_REG_NAMELEN]; // key name
|
||
|
|
RKey * pKey; // new key object
|
||
|
|
|
||
|
|
for ( index = 0;
|
||
|
|
!retval;
|
||
|
|
index++ )
|
||
|
|
{
|
||
|
|
retval = pRegKey->SubKeyEnum( index, name, sizeof name );
|
||
|
|
if ( !retval )
|
||
|
|
{
|
||
|
|
pKey = new RKey;
|
||
|
|
if ( pKey )
|
||
|
|
{
|
||
|
|
if ( pKey->New( name ) )
|
||
|
|
{
|
||
|
|
pKeyList->Insert( pKey );
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
delete pKey;
|
||
|
|
pKey = NULL;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if ( !pKey )
|
||
|
|
{
|
||
|
|
retval = ERROR_NOT_ENOUGH_MEMORY;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( retval == ERROR_NO_MORE_ITEMS )
|
||
|
|
{
|
||
|
|
retval = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
return retval;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Replicate registry hive
|
||
|
|
DWORD // ret-OS return code
|
||
|
|
TRegKey::HiveReplicate(
|
||
|
|
TRegKey const * source // in -source hive
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DWORD retval=0; // returned value
|
||
|
|
RValueList srcValues; // source values
|
||
|
|
RValueList trgValues; // target values
|
||
|
|
TNodeListEnum eSrcValue; // enumerate source values
|
||
|
|
RValue const * pSrcValue; // source value
|
||
|
|
TNodeListEnum eTrgValue; // enumerate target values
|
||
|
|
RValue const * pTrgValue; // target value
|
||
|
|
RKeyList srcKeys; // source keys
|
||
|
|
RKeyList trgKeys; // target keys
|
||
|
|
TNodeListEnum eSrcKey; // enumerate source keys
|
||
|
|
RKey const * pSrcKey; // source key
|
||
|
|
TNodeListEnum eTrgKey; // enumerate target keys
|
||
|
|
RKey const * pTrgKey; // target key
|
||
|
|
int cmpRc; // compare return code
|
||
|
|
TRegKey srcNest; // nested source registry
|
||
|
|
TRegKey trgNest; // nested target registry
|
||
|
|
|
||
|
|
// handle replication of values at this level
|
||
|
|
CollectValues( &srcValues, source );
|
||
|
|
CollectValues( &trgValues, this );
|
||
|
|
|
||
|
|
// now merge the values
|
||
|
|
pSrcValue = (RValue const *) eSrcValue.OpenFirst( &srcValues );
|
||
|
|
pTrgValue = (RValue const *) eTrgValue.OpenFirst( &trgValues );
|
||
|
|
while ( !retval && (pSrcValue || pTrgValue) )
|
||
|
|
{
|
||
|
|
if ( !pTrgValue )
|
||
|
|
{
|
||
|
|
cmpRc = -1;
|
||
|
|
}
|
||
|
|
else if ( !pSrcValue )
|
||
|
|
{
|
||
|
|
cmpRc = 1;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
cmpRc = UStrICmp( pSrcValue->GetName(), pTrgValue->GetName() );
|
||
|
|
}
|
||
|
|
if ( cmpRc < 0 )
|
||
|
|
{ // source value only (copy)
|
||
|
|
retval = this->ValueSet( pSrcValue->GetName(), pSrcValue->GetValue(),
|
||
|
|
pSrcValue->GetValueLen(), pSrcValue->GetType() );
|
||
|
|
pSrcValue = (RValue const *) eSrcValue.Next();
|
||
|
|
}
|
||
|
|
else if ( cmpRc > 0 )
|
||
|
|
{ // target value only (delete)
|
||
|
|
retval = this->ValueDel( pTrgValue->GetName() );
|
||
|
|
pTrgValue = (RValue const *) eTrgValue.Next();
|
||
|
|
}
|
||
|
|
else /* if ( cmpRc == 0 ) */
|
||
|
|
{ // identical value names (replicate)
|
||
|
|
retval = this->ValueSet( pSrcValue->GetName(), pSrcValue->GetValue(),
|
||
|
|
pSrcValue->GetValueLen(), pSrcValue->GetType() );
|
||
|
|
pSrcValue = (RValue const *) eSrcValue.Next();
|
||
|
|
pTrgValue = (RValue const *) eTrgValue.Next();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
eSrcValue.Close();
|
||
|
|
eTrgValue.Close();
|
||
|
|
|
||
|
|
// handle replication of keys at this level
|
||
|
|
CollectKeys( &srcKeys, source );
|
||
|
|
CollectKeys( &trgKeys, this );
|
||
|
|
|
||
|
|
// now merge the values
|
||
|
|
pSrcKey = (RKey const *) eSrcKey.OpenFirst( &srcKeys );
|
||
|
|
pTrgKey = (RKey const *) eTrgKey.OpenFirst( &trgKeys );
|
||
|
|
|
||
|
|
while ( !retval && (pSrcKey || pTrgKey) )
|
||
|
|
{
|
||
|
|
if ( !pTrgKey )
|
||
|
|
{
|
||
|
|
cmpRc = -1;
|
||
|
|
}
|
||
|
|
else if ( !pSrcKey )
|
||
|
|
{
|
||
|
|
cmpRc = 1;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
cmpRc = UStrICmp( pSrcKey->GetName(), pTrgKey->GetName() );
|
||
|
|
}
|
||
|
|
if ( cmpRc < 0 )
|
||
|
|
{ // source key only (copy hive)
|
||
|
|
retval = srcNest.Open( pSrcKey->GetName(), source );
|
||
|
|
if ( !retval )
|
||
|
|
{
|
||
|
|
retval = trgNest.Create( pSrcKey->GetName(), this );
|
||
|
|
if ( !retval )
|
||
|
|
{
|
||
|
|
retval = trgNest.HiveCopy( &srcNest );
|
||
|
|
trgNest.Close();
|
||
|
|
}
|
||
|
|
srcNest.Close();
|
||
|
|
}
|
||
|
|
pSrcKey = (RKey const *) eSrcKey.Next();
|
||
|
|
}
|
||
|
|
else if ( cmpRc > 0 )
|
||
|
|
{ // target key only (delete hive)
|
||
|
|
retval = trgNest.Open( pTrgKey->GetName(), this );
|
||
|
|
if ( !retval )
|
||
|
|
{
|
||
|
|
retval = trgNest.HiveDel();
|
||
|
|
trgNest.Close();
|
||
|
|
}
|
||
|
|
retval = SubKeyDel( pTrgKey->GetName() );
|
||
|
|
pTrgKey = (RKey const *) eTrgKey.Next();
|
||
|
|
}
|
||
|
|
else /* if ( cmpRc == 0 ) */
|
||
|
|
{ // identical keys (replicate hive)
|
||
|
|
retval = srcNest.Open( pSrcKey->GetName(), source );
|
||
|
|
if ( !retval )
|
||
|
|
{
|
||
|
|
retval = trgNest.Open( pSrcKey->GetName(), this );
|
||
|
|
if ( !retval )
|
||
|
|
{
|
||
|
|
retval = trgNest.HiveReplicate( &srcNest );
|
||
|
|
trgNest.Close();
|
||
|
|
}
|
||
|
|
srcNest.Close();
|
||
|
|
}
|
||
|
|
pSrcKey = (RKey const *) eSrcKey.Next();
|
||
|
|
pTrgKey = (RKey const *) eTrgKey.Next();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
eSrcKey.Close();
|
||
|
|
eTrgKey.Close();
|
||
|
|
|
||
|
|
return retval;
|
||
|
|
}
|
||
|
|
|
||
|
|
// TReg.cpp - end of file
|