/*++
Copyright (c) 1996-1999 Microsoft Corporation
Module Name:
rtlprop.c
Abstract:
Implements the management of properties.
Author:
Rod Gamache (rodga) 7-Jan-1996
Revision History:
David Potter (davidp) 12-Mar-1997
Moved to CLUSRTL.
--*/
#define UNICODE 1
#define _UNICODE 1
#include "clusrtlp.h"
#include "stdio.h"
#include "stdlib.h"
#include "RegistryValueName.h"
#define CLRTL_NULL_STRING L"\0"
//
// Data alignment notes
//
// All data (including embedded pointers) are aligned on 32b boundaries (64b
// platforms where not a consideration when this code was originally
// written). This makes some of the typecasting a bit tricky since the
// embedded pointers are really pointers to pointers. All double star pointers
// (i.e., LPBYTE *) have to use UNALIGNED since it is possible that the
// starting address of the pointer, i.e., the value contained in the variable,
// could end in a 4 (which on 64b. platforms is unaligned). (LPBYTE *) becomes
// (LPBYTE UNALIGNED *) or BYTE * UNALIGNED *.
//
// Consider these statements from below:
//
// LPWSTR UNALIGNED * ppszOutValue;
// ppszOutValue = (LPWSTR UNALIGNED *) &pOutParams[propertyItem->Offset];
//
// ppszOutValue is an automatic variable so its address is guaranteed to be
// aligned. pOutParams is essentially an array of DWORDS (though we make no
// effort to enforce this. If that changed, property lists would be broken in
// many different places). It is possible that when we take the address of
// pOutParams + Offset, the offset may be on a DWORD boundary and not a QUAD
// boundary. Therefore we have to treat ppszOutValue's value (a pointer to
// WCHAR) as unaligned (the data itself is properly aligned). You can read the
// typecast as "an aligned pointer (ppszOutValue) to an unaligned pointer
// (address of pOutParams[Offset]) to an aligned WCHAR."
//
// LARGE_INTEGER allows us to "cheat" in that we can leverage the internal
// struct definition of LARGER_INTEGER to do 2 DWORD copies instead of
// treating the data for worst-case alignment.
//
// ISSUE-01/03/16 charlwi CLUSPROP_BUFFER_HELPER might not be correctly aligned
//
// still unsure about this but seeing how the whole property thing is
// 32b. aligned and this structure is used copiously for casting (at DWORD
// boundaries), the potential exists to have a pointer that is not QUADWORD
// aligned. When we go to deref it, we get an alignment fault.
//
//
// Static function prototypes.
//
static
DWORD
WINAPI
ClRtlpSetDwordProperty(
IN HANDLE hXsaction,
IN PVOID hkey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyItem,
IN const CRegistryValueName & rrvnModifiedNames,
IN PCLUSPROP_DWORD pInDwordValue,
IN BOOL bForceWrite,
IN OUT OPTIONAL LPBYTE pOutParams
);
static
DWORD
WINAPI
ClRtlpSetLongProperty(
IN HANDLE hXsaction,
IN PVOID hkey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyItem,
IN const CRegistryValueName & rrvnModifiedNames,
IN PCLUSPROP_LONG pInLongValue,
IN BOOL bForceWrite,
IN OUT OPTIONAL LPBYTE pOutParams
);
static
DWORD
WINAPI
ClRtlpSetULargeIntegerProperty(
IN HANDLE hXsaction,
IN PVOID hkey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyItem,
IN const CRegistryValueName & rrvnModifiedNames,
IN PCLUSPROP_ULARGE_INTEGER pInULargeIntegerValue,
IN BOOL bForceWrite,
IN OUT OPTIONAL LPBYTE pOutParams
);
static
DWORD
WINAPI
ClRtlpSetLargeIntegerProperty(
IN HANDLE hXsaction,
IN PVOID hkey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyItem,
IN const CRegistryValueName & rrvnModifiedNames,
IN PCLUSPROP_LARGE_INTEGER pInLargeIntegerValue,
IN BOOL bForceWrite,
IN OUT OPTIONAL LPBYTE pOutParams
);
static
DWORD
WINAPI
ClRtlpSetStringProperty(
IN HANDLE hXsaction,
IN PVOID hkey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyItem,
IN const CRegistryValueName & rrvnModifiedNames,
IN PCLUSPROP_SZ pInStringValue,
IN BOOL bForceWrite,
IN OUT OPTIONAL LPBYTE pOutParams
);
static
DWORD
WINAPI
ClRtlpSetMultiStringProperty(
IN HANDLE hXsaction,
IN PVOID hkey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyItem,
IN const CRegistryValueName & rrvnModifiedNames,
IN PCLUSPROP_MULTI_SZ pInMultiStringValue,
IN BOOL bForceWrite,
IN OUT OPTIONAL LPBYTE pOutParams
);
static
DWORD
WINAPI
ClRtlpSetBinaryProperty(
IN HANDLE hXsaction,
IN PVOID hkey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyItem,
IN const CRegistryValueName & rrvnModifiedNames,
IN PCLUSPROP_BINARY pInBinaryValue,
IN BOOL bForceWrite,
IN OUT OPTIONAL LPBYTE pOutParams
);
DWORD
WINAPI
ClRtlEnumProperties(
IN const PRESUTIL_PROPERTY_ITEM pPropertyTable,
OUT LPWSTR pszOutProperties,
IN DWORD cbOutPropertiesSize,
OUT LPDWORD pcbBytesReturned,
OUT LPDWORD pcbRequired
)
/*++
Routine Description:
Enumerates the properties for a given object.
Arguments:
pPropertyTable - Pointer to the property table to process.
pszOutProperties - Supplies the output buffer.
cbOutPropertiesSize - Supplies the size of the output buffer.
pcbBytesReturned - The number of bytes returned in pszOutProperties.
pcbRequired - The required number of bytes if pszOutProperties is too small.
Return Value:
ERROR_SUCCESS - Operation was successful.
ERROR_BAD_ARGUMENTS - An argument passed to the function was bad.
A Win32 error code on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
DWORD totalBufferLength = 0;
PRESUTIL_PROPERTY_ITEM property;
LPWSTR psz = pszOutProperties;
*pcbBytesReturned = 0;
*pcbRequired = 0;
if ( pPropertyTable == NULL ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlEnumProperties: pPropertyTable == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
//
// Clear the output buffer
//
if ( pszOutProperties != NULL ) {
ZeroMemory( pszOutProperties, cbOutPropertiesSize );
}
//
// Get the size of all property names for this object.
//
for ( property = pPropertyTable ; property->Name != NULL ; property++ ) {
totalBufferLength += (lstrlenW( property->Name ) + 1) * sizeof(WCHAR);
}
totalBufferLength += sizeof(UNICODE_NULL);
//
// If the output buffer is big enough, copy the property names.
//
if ( totalBufferLength > cbOutPropertiesSize ) {
*pcbRequired = totalBufferLength;
totalBufferLength = 0;
if ( pszOutProperties == NULL ) {
status = ERROR_SUCCESS;
} else {
status = ERROR_MORE_DATA;
}
} else {
DWORD cchCurrentNameSize;
for ( property = pPropertyTable ; property->Name != NULL ; property++ ) {
lstrcpyW( psz, property->Name );
cchCurrentNameSize = lstrlenW( psz ) + 1;
*pcbBytesReturned += cchCurrentNameSize * sizeof(WCHAR);
psz += cchCurrentNameSize;
}
*psz = L'\0';
*pcbBytesReturned += sizeof(WCHAR);
}
return(status);
} // ClRtlEnumProperties
DWORD
WINAPI
ClRtlEnumPrivateProperties(
IN PVOID hkeyClusterKey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
OUT LPWSTR pszOutProperties,
IN DWORD cbOutPropertiesSize,
OUT LPDWORD pcbBytesReturned,
OUT LPDWORD pcbRequired
)
/*++
Routine Description:
Enumerates the properties for a given object.
Arguments:
hkeyClusterKey - Supplies the handle to the key in the cluster database
to read from.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pszOutProperties - Supplies the output buffer.
cbOutPropertiesSize - Supplies the size of the output buffer.
pcbBytesReturned - The number of bytes returned in pszOutProperties.
pcbRequired - The required number of bytes if pszOutProperties is too small.
Return Value:
ERROR_SUCCESS - Operation was successful.
ERROR_BAD_ARGUMENTS - An argument passed to the function was bad.
A Win32 error code on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
DWORD totalBufferLength = 0;
LPWSTR psz = pszOutProperties;
DWORD ival;
DWORD currentNameLength = 20;
DWORD nameLength;
DWORD dataLength;
DWORD type;
LPWSTR pszName;
*pcbBytesReturned = 0;
*pcbRequired = 0;
//
// Validate inputs
//
if ( (hkeyClusterKey == NULL) ||
(pClusterRegApis == NULL) ) {
ClRtlDbgPrint( LOG_CRITICAL,
"ClRtlEnumPrivateProperties: hkeyClusterKey or pClusterRegApis == NULL. "
"Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
//
// Clear the output buffer
//
if ( pszOutProperties != NULL ) {
ZeroMemory( pszOutProperties, cbOutPropertiesSize );
}
//
// Allocate a property name buffer.
//
pszName = static_cast< LPWSTR >( LocalAlloc( LMEM_FIXED, currentNameLength * sizeof(WCHAR) ) );
if ( pszName == NULL ) {
return(ERROR_NOT_ENOUGH_MEMORY);
}
//
// Enumerate all properties to find the total size.
//
ival = 0;
while ( TRUE ) {
//
// Read the next property.
//
nameLength = currentNameLength;
dataLength = 0;
status = (*pClusterRegApis->pfnEnumValue)( hkeyClusterKey,
ival,
pszName,
&nameLength,
&type,
NULL,
&dataLength );
if ( status == ERROR_NO_MORE_ITEMS ) {
status = ERROR_SUCCESS;
break;
} else if ( status == ERROR_MORE_DATA ) {
CL_ASSERT( (nameLength+1) > currentNameLength );
LocalFree( pszName );
currentNameLength = nameLength + 1; // returned value doesn't include terminating NULL
pszName = static_cast< LPWSTR >( LocalAlloc( LMEM_FIXED, currentNameLength * sizeof(WCHAR) ) );
if ( pszName == NULL ) {
status = ERROR_NOT_ENOUGH_MEMORY;
break;
}
continue; // retry
} else if ( status != ERROR_SUCCESS ) {
break;
}
totalBufferLength += (nameLength + 1) * sizeof(WCHAR);
++ival;
}
//
// Continue only if the operations so far have been successful.
//
if ( status == ERROR_SUCCESS ) {
if ( totalBufferLength != 0 ) {
totalBufferLength += sizeof(UNICODE_NULL);
}
//
// If the output buffer is big enough, copy the property names.
//
if ( totalBufferLength > cbOutPropertiesSize ) {
*pcbRequired = totalBufferLength;
totalBufferLength = 0;
if ( (pszOutProperties == NULL) ||
(cbOutPropertiesSize == 0) ) {
status = ERROR_SUCCESS;
} else {
status = ERROR_MORE_DATA;
}
} else if ( totalBufferLength != 0 ) {
//
// Enumerate all properties for copying
//
for ( ival = 0; ; ival++ ) {
//
// Read the next property.
//
nameLength = currentNameLength;
dataLength = 0;
status = (*pClusterRegApis->pfnEnumValue)( hkeyClusterKey,
ival,
pszName,
&nameLength,
&type,
NULL,
&dataLength );
if ( status == ERROR_NO_MORE_ITEMS ) {
status = ERROR_SUCCESS;
break;
} else if ( status == ERROR_MORE_DATA ) {
CL_ASSERT( 0 ); // THIS SHOULDN'T HAPPEN
} else if ( status != ERROR_SUCCESS ) {
break;
}
//CL_ASSERT( (DWORD)lstrlenW( name ) == nameLength );
lstrcpyW( psz, pszName );
psz += nameLength + 1;
*pcbBytesReturned += (nameLength + 1) * sizeof(WCHAR);
}
*psz = L'\0';
*pcbBytesReturned += sizeof(WCHAR);
}
}
LocalFree( pszName );
return(status);
} // ClRtlEnumPrivateProperties
DWORD
WINAPI
ClRtlGetProperties(
IN PVOID hkeyClusterKey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyTable,
OUT PVOID pOutPropertyList,
IN DWORD cbOutPropertyListSize,
OUT LPDWORD pcbBytesReturned,
OUT LPDWORD pcbRequired
)
/*++
Routine Description:
Gets the properties for a given object.
Arguments:
hkeyClusterKey - Supplies the handle to the key in the cluster database
to read from.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pPropertyTable - Pointer to the property list to process.
pOutPropertyList - Supplies the output buffer.
cbOutPropertyListSize - Supplies the size of the output buffer.
pcbBytesReturned - The number of bytes returned in pOutPropertyList.
pcbRequired - The required number of bytes if pOutPropertyList is too small.
Return Value:
ERROR_SUCCESS - Operation was successful.
ERROR_BAD_ARGUMENTS - An argument passed to the function was bad.
ERROR_NOT_ENOUGH_MEMORY - Error allocating memory.
A Win32 error code on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
DWORD itemCount = 0;
DWORD totalBufferLength = 0;
PVOID outBuffer = pOutPropertyList;
DWORD bufferLength = cbOutPropertyListSize;
PRESUTIL_PROPERTY_ITEM property;
*pcbBytesReturned = 0;
*pcbRequired = 0;
if ( (hkeyClusterKey == NULL) ||
(pClusterRegApis == NULL) ||
(pPropertyTable == NULL) ) {
ClRtlDbgPrint( LOG_CRITICAL,
"ClRtlGetProperties: hkeyClusterKey, pClusterRegApis, or "
"pPropertyTable == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
//
// Clear the output buffer
//
if ( pOutPropertyList != NULL ) {
ZeroMemory( pOutPropertyList, cbOutPropertyListSize );
}
//
// Get the size of all properties for this object.
//
property = pPropertyTable;
while ( property->Name != NULL ) {
status = ClRtlGetPropertySize( hkeyClusterKey,
pClusterRegApis,
property,
&totalBufferLength,
&itemCount );
if ( status != ERROR_SUCCESS ) {
break;
}
property++;
}
//
// Continue only if the operations so far have been successful.
//
if ( status == ERROR_SUCCESS ) {
//
// Count for item count at front of return data and endmark.
//
totalBufferLength += sizeof(DWORD) + sizeof(CLUSPROP_SYNTAX);
//
// Verify the size of all the properties
//
if ( totalBufferLength > cbOutPropertyListSize ) {
*pcbRequired = totalBufferLength;
totalBufferLength = 0;
if ( pOutPropertyList == NULL ) {
status = ERROR_SUCCESS;
} else {
status = ERROR_MORE_DATA;
}
}
else {
*(LPDWORD)outBuffer = itemCount;
outBuffer = (PVOID)( (PUCHAR)outBuffer + sizeof(itemCount) );
bufferLength -= sizeof(itemCount);
//
// Now fetch all of the properties.
//
property = pPropertyTable;
while ( property->Name != NULL ) {
status = ClRtlGetProperty( hkeyClusterKey,
pClusterRegApis,
property,
&outBuffer,
&bufferLength );
if ( status != ERROR_SUCCESS ) {
break;
}
property++;
}
// Don't forget the ENDMARK
*(LPDWORD)outBuffer = CLUSPROP_SYNTAX_ENDMARK;
if ( (status != ERROR_SUCCESS) &&
(status != ERROR_MORE_DATA) ) {
totalBufferLength = 0;
}
}
*pcbBytesReturned = totalBufferLength;
}
return(status);
} // ClRtlGetProperties
DWORD
WINAPI
ClRtlGetAllProperties(
IN PVOID hkeyClusterKey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyTable,
OUT PVOID pOutPropertyList,
IN DWORD cbOutPropertyListSize,
OUT LPDWORD pcbReturned,
OUT LPDWORD pcbRequired
)
/*++
Routine Description:
Gets the default and 'unknown' properties for a given object.
Arguments:
hkeyClusterKey - Supplies the handle to the key in the cluster database
to read from.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pPropertyTable - Pointer to the property table to process.
pOutPropertyList - Supplies the output buffer.
cbOutPropertyListSize - Supplies the size of the output buffer.
pcbBytesReturned - The number of bytes returned in pOutPropertyList.
pcbRequired - The required number of bytes if pOutPropertyList is too small.
Return Value:
ERROR_SUCCESS - Operation was successful.
ERROR_BAD_ARGUMENTS - An argument passed to the function was bad.
ERROR_NOT_ENOUGH_MEMORY - Error allocating memory.
A Win32 error code on failure.
--*/
{
CLUSPROP_BUFFER_HELPER cbhKnownPropBuffer;
CLUSPROP_BUFFER_HELPER cbhUnknownPropBuffer;
DWORD cbKnownPropBufferSize;
DWORD cbUnknownPropBufferSize;
DWORD sc = ERROR_SUCCESS;
DWORD scUnknown;
DWORD dwSavedData;
DWORD cbUnknownRequired = 0;
DWORD cbUnknownReturned = 0;
cbhKnownPropBuffer.pb = static_cast< LPBYTE >( pOutPropertyList );
cbKnownPropBufferSize = cbOutPropertyListSize;
//
// First get the 'known' properties.
//
sc = ClRtlGetProperties(
hkeyClusterKey,
pClusterRegApis,
pPropertyTable,
cbhKnownPropBuffer.pb,
cbKnownPropBufferSize,
pcbReturned,
pcbRequired
);
if ( sc != ERROR_SUCCESS )
{
*pcbReturned = 0;
if ( sc != ERROR_MORE_DATA )
{
*pcbRequired = 0;
return sc;
}
// We already know that there is insufficient space.
scUnknown = ClRtlGetUnknownProperties(
hkeyClusterKey,
pClusterRegApis,
pPropertyTable,
NULL,
0,
&cbUnknownReturned,
&cbUnknownRequired
);
if ( ( scUnknown != ERROR_SUCCESS ) &&
( scUnknown != ERROR_MORE_DATA ) )
{
*pcbRequired = 0;
return scUnknown;
}
//
// If both known and unknown properties exist, only one endmark
// is required. So, subtract endmark size from total required size.
//
if ( ( *pcbRequired > sizeof(DWORD) ) &&
( cbUnknownRequired > sizeof(DWORD) ) )
{
*pcbRequired -= sizeof(CLUSPROP_SYNTAX);
}
//
// Subtract off the size of the property count for the
// unknown property list.
//
*pcbRequired += cbUnknownRequired - sizeof(DWORD);
return sc;
} // if: call to ClRtlGetProperties failed
// If we are here then the call to ClRtlGetProperties succeeded.
//
// Calculate the position in the output buffer where unknown properties
// should be stored. Subtract off the size of the property count for
// the unknown property list. These calculations will cause the buffer
// pointer to overlap the known property list buffer by one DWORD.
//
cbhUnknownPropBuffer.pb = cbhKnownPropBuffer.pb + *pcbReturned - sizeof(DWORD);
cbUnknownPropBufferSize = cbKnownPropBufferSize - *pcbReturned + sizeof(DWORD);
// If there are known properties, move the unknown property list
// buffer pointer to overlap that as well.
if ( *pcbReturned > sizeof(DWORD) )
{
cbhUnknownPropBuffer.pb -= sizeof(CLUSPROP_SYNTAX);
cbUnknownPropBufferSize += sizeof(CLUSPROP_SYNTAX);
} // if: a nonzero number of properties has been returned.
//
// Save the DWORD we are about to overlap.
//
dwSavedData = *(cbhUnknownPropBuffer.pdw);
scUnknown = ClRtlGetUnknownProperties(
hkeyClusterKey,
pClusterRegApis,
pPropertyTable,
cbhUnknownPropBuffer.pb,
cbUnknownPropBufferSize,
&cbUnknownReturned,
&cbUnknownRequired
);
if ( scUnknown == ERROR_SUCCESS )
{
//
// The order of the next three statements is very important
// since the known and the unknown property buffers can overlap.
//
DWORD nUnknownPropCount = cbhUnknownPropBuffer.pList->nPropertyCount;
*(cbhUnknownPropBuffer.pdw) = dwSavedData;
cbhKnownPropBuffer.pList->nPropertyCount += nUnknownPropCount;
//
// If both known and unknown properties exist, only one endmark
// is required. So, subtract endmark size from total returned size.
//
if ( ( *pcbReturned > sizeof(DWORD) ) &&
( cbUnknownReturned > sizeof(DWORD) ) )
{
*pcbReturned -= sizeof(CLUSPROP_SYNTAX);
}
//
// Add in the size of the unknown property list minus the
// size of the unknown property list property count.
//
*pcbReturned += cbUnknownReturned - sizeof(DWORD);
*pcbRequired = 0;
} // if: call to ClRtlGetUnknownProperties succeeded
else
{
if ( scUnknown == ERROR_MORE_DATA )
{
*pcbRequired = *pcbReturned;
*pcbReturned = 0;
//
// Both known and unknown properties exist. Only one endmark
// is required. So, subtract endmark size from total required size.
//
if ( ( *pcbRequired > sizeof(DWORD) ) &&
( cbUnknownRequired > sizeof(DWORD) ) )
{
*pcbRequired -= sizeof(CLUSPROP_SYNTAX);
}
//
// Add in the size of the unknown property list minus the
// size of the unknown property list property count.
//
*pcbRequired += cbUnknownRequired - sizeof(DWORD);
} // if: ClRtlGetUnknownProperties returned ERROR_MORE_DATA
else
{
*pcbRequired = 0;
*pcbReturned = 0;
} // else: ClRtlGetUnknownProperties failed for some unknown reason.
} // else: Call to ClRtlGetUnknownProperties failed.
return scUnknown;
} //*** ClRtlGetAllProperties()
DWORD
WINAPI
ClRtlGetPropertiesToParameterBlock(
IN HKEY hkeyClusterKey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyTable,
IN OUT LPBYTE pOutParams,
IN BOOL bCheckForRequiredProperties,
OUT OPTIONAL LPWSTR * pszNameOfPropInError
)
/*++
Routine Description:
Read properties based on a property table.
Arguments:
hkeyClusterKey - Supplies the cluster key where the properties are stored.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pPropertyTable - Pointer to the property table to process.
pOutParams - Parameter block to read into.
bCheckForRequiredProperties - Boolean value specifying whether missing
required properties should cause an error.
pszNameOfPropInError - String pointer in which to return the name of the
property in error (optional).
Return Value:
ERROR_SUCCESS - Properties read successfully.
ERROR_INVALID_DATA - Required property not present.
ERROR_INVALID_PARAMETER -
A Win32 error code on failure.
--*/
{
PRESUTIL_PROPERTY_ITEM propertyItem = pPropertyTable;
HKEY key;
DWORD status = ERROR_SUCCESS;
DWORD valueType;
DWORD valueSize;
LPWSTR pszInValue;
LPBYTE pbInValue;
DWORD dwInValue;
LPWSTR UNALIGNED * ppszOutValue;
LPBYTE UNALIGNED * ppbOutValue;
DWORD * pdwOutValue;
LONG * plOutValue;
ULARGE_INTEGER * pullOutValue;
LARGE_INTEGER * pllOutValue;
CRegistryValueName rvn;
if ( pszNameOfPropInError != NULL ) {
*pszNameOfPropInError = NULL;
}
if ( (hkeyClusterKey == NULL) ||
(pPropertyTable == NULL) ||
(pOutParams == NULL) ||
(pClusterRegApis == NULL) ||
(pClusterRegApis->pfnOpenKey == NULL) ||
(pClusterRegApis->pfnCloseKey == NULL) ||
(pClusterRegApis->pfnQueryValue == NULL) ) {
ClRtlDbgPrint( LOG_CRITICAL,
"ClRtlGetPropertiesToParameterBlock: hkeyClusterKey, pPropertyTable, "
"pOutParams, pClusterRegApis, or required pfns == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
while ( propertyItem->Name != NULL ) {
//
// Use the wrapper class CRegistryValueName to parse value name to see if it
// contains a backslash.
//
status = rvn.ScInit( propertyItem->Name, propertyItem->KeyName );
if ( status != ERROR_SUCCESS ) {
break;
}
//
// If the value resides at a different location, open the key.
//
if ( rvn.PszKeyName() != NULL ) {
DWORD disposition;
status = (*pClusterRegApis->pfnOpenKey)(
hkeyClusterKey,
rvn.PszKeyName(),
KEY_ALL_ACCESS,
(void **) &key
);
// If key could not be opened, then we may need to put the default value
// for the property item.
if ( status != ERROR_SUCCESS ) {
status = ERROR_FILE_NOT_FOUND;
}
} else {
key = hkeyClusterKey;
}
switch ( propertyItem->Format ) {
case CLUSPROP_FORMAT_DWORD:
pdwOutValue = (DWORD *) &pOutParams[propertyItem->Offset];
valueSize = sizeof(DWORD);
// If OpenKey has succeeded.
if ( status == ERROR_SUCCESS ) {
status = (*pClusterRegApis->pfnQueryValue)( key,
rvn.PszName(),
&valueType,
(LPBYTE) pdwOutValue,
&valueSize );
}
if ( status == ERROR_SUCCESS ) {
if ( valueType != REG_DWORD ) {
ClRtlDbgPrint( LOG_CRITICAL,
"ClRtlGetPropertiesToParameterBlock: Property '%1!ls!' "
"expected to be REG_DWORD (%2!d!), was %3!d!.\n",
propertyItem->Name, REG_DWORD, valueType );
status = ERROR_INVALID_PARAMETER;
}
} else if ( (status == ERROR_FILE_NOT_FOUND) &&
(!(propertyItem->Flags & RESUTIL_PROPITEM_REQUIRED) ||
! bCheckForRequiredProperties) ) {
*pdwOutValue = propertyItem->Default;
status = ERROR_SUCCESS;
}
break;
case CLUSPROP_FORMAT_LONG:
plOutValue = (LONG *) &pOutParams[propertyItem->Offset];
valueSize = sizeof(LONG);
// If OpenKey has succeeded.
if ( status == ERROR_SUCCESS ) {
status = (*pClusterRegApis->pfnQueryValue)( key,
rvn.PszName(),
&valueType,
(LPBYTE) plOutValue,
&valueSize );
}
if ( status == ERROR_SUCCESS ) {
if ( valueType != REG_DWORD ) {
ClRtlDbgPrint( LOG_CRITICAL,
"ClRtlGetPropertiesToParameterBlock: Property '%1!ls!' "
"expected to be REG_DWORD (%2!d!), was %3!d!.\n",
propertyItem->Name, REG_DWORD, valueType );
status = ERROR_INVALID_PARAMETER;
}
} else if ( (status == ERROR_FILE_NOT_FOUND) &&
(!(propertyItem->Flags & RESUTIL_PROPITEM_REQUIRED) ||
!bCheckForRequiredProperties) ) {
*plOutValue = propertyItem->Default;
status = ERROR_SUCCESS;
}
break;
case CLUSPROP_FORMAT_ULARGE_INTEGER:
pullOutValue = (ULARGE_INTEGER *) &pOutParams[propertyItem->Offset];
valueSize = sizeof(ULARGE_INTEGER);
// If OpenKey has succeeded.
if ( status == ERROR_SUCCESS ) {
status = (*pClusterRegApis->pfnQueryValue)( key,
rvn.PszName(),
&valueType,
(LPBYTE) pullOutValue,
&valueSize );
}
if ( status == ERROR_SUCCESS ) {
if ( valueType != REG_QWORD ) {
ClRtlDbgPrint( LOG_CRITICAL,
"ClRtlGetPropertiesToParameterBlock: Property '%1!ls!' "
"expected to be REG_QWORD (%2!d!), was %3!d!.\n",
propertyItem->Name, REG_QWORD, valueType );
status = ERROR_INVALID_PARAMETER;
}
} else if ( (status == ERROR_FILE_NOT_FOUND) &&
(!(propertyItem->Flags & RESUTIL_PROPITEM_REQUIRED) ||
! bCheckForRequiredProperties) ) {
pullOutValue->u = propertyItem->ULargeIntData->Default.u;
status = ERROR_SUCCESS;
}
break;
case CLUSPROP_FORMAT_LARGE_INTEGER:
pllOutValue = (LARGE_INTEGER *) &pOutParams[propertyItem->Offset];
valueSize = sizeof(LARGE_INTEGER);
// If OpenKey has succeeded.
if ( status == ERROR_SUCCESS ) {
status = (*pClusterRegApis->pfnQueryValue)( key,
rvn.PszName(),
&valueType,
(LPBYTE) pllOutValue,
&valueSize );
}
if ( status == ERROR_SUCCESS ) {
if ( valueType != REG_QWORD ) {
ClRtlDbgPrint( LOG_CRITICAL,
"ClRtlGetPropertiesToParameterBlock: Property '%1!ls!' "
"expected to be REG_QWORD (%2!d!), was %3!d!.\n",
propertyItem->Name, REG_QWORD, valueType );
status = ERROR_INVALID_PARAMETER;
}
} else if ( (status == ERROR_FILE_NOT_FOUND) &&
(!(propertyItem->Flags & RESUTIL_PROPITEM_REQUIRED) ||
! bCheckForRequiredProperties) )
{
pllOutValue->u = propertyItem->LargeIntData->Default.u;
status = ERROR_SUCCESS;
}
break;
case CLUSPROP_FORMAT_SZ:
case CLUSPROP_FORMAT_EXPAND_SZ:
ppszOutValue = (LPWSTR UNALIGNED *) &pOutParams[propertyItem->Offset];
// If OpenKey has succeeded.
if ( status == ERROR_SUCCESS ) {
pszInValue = ClRtlGetSzValue( key,
rvn.PszName(),
pClusterRegApis );
}
else {
pszInValue = NULL;
SetLastError(status);
}
if ( pszInValue == NULL ) {
status = GetLastError();
if ( (status == ERROR_FILE_NOT_FOUND) &&
(!(propertyItem->Flags & RESUTIL_PROPITEM_REQUIRED) ||
! bCheckForRequiredProperties) ) {
status = ERROR_SUCCESS;
// Deallocate old value.
if ( *ppszOutValue != NULL ) {
LocalFree( *ppszOutValue );
*ppszOutValue = NULL;
}
// If a default is specified, copy it.
if ( propertyItem->lpDefault != NULL ) {
*ppszOutValue = (LPWSTR) LocalAlloc( LMEM_FIXED, (lstrlenW( (LPCWSTR) propertyItem->lpDefault ) + 1) * sizeof(WCHAR) );
if ( *ppszOutValue == NULL ) {
status = GetLastError();
} else {
lstrcpyW( *ppszOutValue, (LPCWSTR) propertyItem->lpDefault );
}
}
}
} else {
if ( *ppszOutValue != NULL ) {
LocalFree( *ppszOutValue );
}
*ppszOutValue = pszInValue;
}
break;
case CLUSPROP_FORMAT_MULTI_SZ:
case CLUSPROP_FORMAT_BINARY:
ppbOutValue = (LPBYTE UNALIGNED *) &pOutParams[propertyItem->Offset];
pdwOutValue = (DWORD *) &pOutParams[propertyItem->Offset+sizeof(LPBYTE*)];
// If OpenKey has succeeded.
if ( status == ERROR_SUCCESS ) {
status = ClRtlGetBinaryValue( key,
rvn.PszName(),
&pbInValue,
&dwInValue,
pClusterRegApis );
}
if ( status == ERROR_SUCCESS ) {
if ( *ppbOutValue != NULL ) {
LocalFree( *ppbOutValue );
}
*ppbOutValue = pbInValue;
*pdwOutValue = dwInValue;
} else if ( (status == ERROR_FILE_NOT_FOUND) &&
(!(propertyItem->Flags & RESUTIL_PROPITEM_REQUIRED) ||
! bCheckForRequiredProperties) ) {
status = ERROR_SUCCESS;
// Deallocate old value.
if ( *ppbOutValue != NULL ) {
LocalFree( *ppbOutValue );
*ppbOutValue = NULL;
}
*pdwOutValue = 0;
// If a default is specified, copy it.
if ( propertyItem->lpDefault != NULL ) {
*ppbOutValue = (LPBYTE) LocalAlloc( LMEM_FIXED, propertyItem->Minimum );
if ( *ppbOutValue == NULL ) {
status = GetLastError();
} else {
CopyMemory( *ppbOutValue, (const PVOID) propertyItem->lpDefault, propertyItem->Minimum );
*pdwOutValue = propertyItem->Minimum;
}
}
}
break;
}
//
// Close the key if we opened it.
//
if ( (rvn.PszKeyName() != NULL) &&
(key != NULL) ) {
(*pClusterRegApis->pfnCloseKey)( key );
}
//
// Handle any errors that occurred.
//
if ( status != ERROR_SUCCESS ) {
if ( pszNameOfPropInError != NULL ) {
*pszNameOfPropInError = propertyItem->Name;
}
if ( propertyItem->Flags & RESUTIL_PROPITEM_REQUIRED ) {
if ( status == ERROR_FILE_NOT_FOUND ) {
status = ERROR_INVALID_DATA;
}
break;
} else {
status = ERROR_SUCCESS;
}
}
//
// Advance to the next property.
//
propertyItem++;
}
return(status);
} // ClRtlGetPropertiesToParameterBlock
DWORD
WINAPI
ClRtlPropertyListFromParameterBlock(
IN const PRESUTIL_PROPERTY_ITEM pPropertyTable,
OUT PVOID pOutPropertyList,
IN OUT LPDWORD pcbOutPropertyListSize,
IN const LPBYTE pInParams,
OUT LPDWORD pcbBytesReturned,
OUT LPDWORD pcbRequired
)
/*++
Routine Description:
Constructs a property list from a parameter block.
Arguments:
pPropertyTable - Pointer to the property table to process.
pOutPropertyList - Supplies the output buffer.
pcbOutPropertyListSize - Supplies the size of the output buffer.
pInParams - Supplies the input parameter block.
pcbBytesReturned - The number of bytes returned in pOutPropertyList.
pcbRequired - The required number of bytes if pOutPropertyList is too small.
Return Value:
ERROR_SUCCESS - Operation was successful.
ERROR_BAD_ARGUMENTS - An argument passed to the function was bad.
ERROR_NOT_ENOUGH_MEMORY - Error allocating memory.
ERROR_MORE_DATA - Output buffer isn't big enough to build the property list.
A Win32 error code on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
DWORD bytesReturned;
DWORD required;
DWORD nameSize;
DWORD dataSize;
DWORD bufferIncrement;
DWORD totalBufferSize = 0;
LPDWORD ptrItemCount;
WORD propertyFormat;
BOOL copying = TRUE;
PRESUTIL_PROPERTY_ITEM propertyItem = pPropertyTable;
CLUSPROP_BUFFER_HELPER props;
LPWSTR UNALIGNED * ppszValue;
PBYTE UNALIGNED * ppbValue;
DWORD * pdwValue;
ULARGE_INTEGER * pullValue;
LPWSTR pszUnexpanded;
LPWSTR pszExpanded = NULL;
*pcbBytesReturned = 0;
*pcbRequired = 0;
if ( (pPropertyTable == NULL) ||
(pInParams == NULL) ||
(pcbOutPropertyListSize == NULL) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlPropertyListFromParameterBlock: pPropertyTable, pInParams, or pcbOutPropertyListSize == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
//
// Clear the output buffer.
//
if ( pOutPropertyList != NULL ) {
ZeroMemory( pOutPropertyList, *pcbOutPropertyListSize );
} else {
copying = FALSE;
}
//
// Need a DWORD of item count.
//
props.pb = (LPBYTE) pOutPropertyList;
ptrItemCount = props.pdw++;
totalBufferSize += sizeof(DWORD);
if ( totalBufferSize > *pcbOutPropertyListSize ) {
copying = FALSE;
}
while ( propertyItem->Name != NULL ) {
//
// Copy the property name.
//
nameSize = (lstrlenW( propertyItem->Name ) + 1) * sizeof(WCHAR);
bufferIncrement = sizeof(CLUSPROP_PROPERTY_NAME) + ALIGN_CLUSPROP( nameSize );
totalBufferSize += bufferIncrement;
if ( copying && (totalBufferSize <= *pcbOutPropertyListSize) ) {
props.pName->Syntax.dw = CLUSPROP_SYNTAX_NAME;
props.pName->cbLength = nameSize;
lstrcpyW( props.pName->sz, propertyItem->Name );
props.pb += bufferIncrement;
} else {
copying = FALSE;
}
//
// Copy the property value.
//
propertyFormat = (WORD) propertyItem->Format;
switch ( propertyItem->Format ) {
case CLUSPROP_FORMAT_DWORD:
case CLUSPROP_FORMAT_LONG:
pdwValue = (DWORD *) &pInParams[propertyItem->Offset];
bufferIncrement = sizeof(CLUSPROP_DWORD);
totalBufferSize += bufferIncrement;
if ( copying && (totalBufferSize <= *pcbOutPropertyListSize) ) {
props.pSyntax->wFormat = propertyFormat;
props.pSyntax->wType = CLUSPROP_TYPE_LIST_VALUE;
props.pDwordValue->cbLength = sizeof(DWORD);
props.pDwordValue->dw = *pdwValue;
props.pb += bufferIncrement;
} else {
copying = FALSE;
}
break;
case CLUSPROP_FORMAT_ULARGE_INTEGER:
case CLUSPROP_FORMAT_LARGE_INTEGER:
pullValue = (ULARGE_INTEGER *) &pInParams[propertyItem->Offset];
bufferIncrement = sizeof(CLUSPROP_ULARGE_INTEGER);
totalBufferSize += bufferIncrement;
if ( copying && (totalBufferSize <= *pcbOutPropertyListSize) ) {
props.pSyntax->wFormat = propertyFormat;
props.pSyntax->wType = CLUSPROP_TYPE_LIST_VALUE;
props.pULargeIntegerValue->cbLength = sizeof(ULARGE_INTEGER);
props.pULargeIntegerValue->li.u = pullValue->u;
props.pb += bufferIncrement;
} else {
copying = FALSE;
}
break;
case CLUSPROP_FORMAT_SZ:
case CLUSPROP_FORMAT_EXPAND_SZ:
ppszValue = (LPWSTR UNALIGNED *) &pInParams[propertyItem->Offset];
pszUnexpanded = *ppszValue;
if ( *ppszValue != NULL ) {
dataSize = (lstrlenW( *ppszValue ) + 1) * sizeof(WCHAR);
} else {
dataSize = sizeof(WCHAR);
}
bufferIncrement = sizeof(CLUSPROP_SZ) + ALIGN_CLUSPROP( dataSize );
totalBufferSize += bufferIncrement;
if ( copying && (totalBufferSize <= *pcbOutPropertyListSize) ) {
props.pSyntax->wFormat = propertyFormat;
props.pSyntax->wType = CLUSPROP_TYPE_LIST_VALUE;
props.pStringValue->cbLength = dataSize;
if ( *ppszValue != NULL ) {
lstrcpyW( props.pStringValue->sz, *ppszValue );
} else {
props.pStringValue->sz[0] = L'\0';
}
props.pb += bufferIncrement;
} else {
copying = FALSE;
}
//
// See if there is a different expanded string and, if so,
// return it as an additional value in the value list.
//
if ( pszUnexpanded != NULL ) {
pszExpanded = ClRtlExpandEnvironmentStrings( pszUnexpanded );
if ( pszExpanded == NULL ) {
status = GetLastError();
break;
}
if ( lstrcmpW( pszExpanded, pszUnexpanded ) != 0 ) {
dataSize = (lstrlenW( pszExpanded ) + 1) * sizeof( WCHAR );
bufferIncrement = sizeof( CLUSPROP_SZ ) + ALIGN_CLUSPROP( dataSize );
totalBufferSize += bufferIncrement;
if ( totalBufferSize <= *pcbOutPropertyListSize ) {
props.pSyntax->dw = CLUSPROP_SYNTAX_LIST_VALUE_EXPANDED_SZ;
props.pStringValue->cbLength = dataSize;
lstrcpyW( props.pStringValue->sz, pszExpanded );
props.pb += bufferIncrement;
} else {
copying = FALSE;
}
}
LocalFree( pszExpanded );
pszExpanded = NULL;
}
break;
case CLUSPROP_FORMAT_MULTI_SZ:
ppszValue = (LPWSTR UNALIGNED *) &pInParams[propertyItem->Offset];
pdwValue = (DWORD *) &pInParams[propertyItem->Offset+sizeof(LPWSTR*)];
if ( *ppszValue != NULL ) {
bufferIncrement = sizeof(CLUSPROP_SZ) + ALIGN_CLUSPROP( *pdwValue );
} else {
bufferIncrement = sizeof(CLUSPROP_SZ) + ALIGN_CLUSPROP( sizeof(WCHAR) );
}
totalBufferSize += bufferIncrement;
if ( copying && (totalBufferSize <= *pcbOutPropertyListSize) ) {
props.pSyntax->wFormat = propertyFormat;
props.pSyntax->wType = CLUSPROP_TYPE_LIST_VALUE;
if ( *ppszValue != NULL ) {
props.pStringValue->cbLength = *pdwValue;
CopyMemory( props.pStringValue->sz, *ppszValue, *pdwValue );
} else {
props.pStringValue->cbLength = sizeof(WCHAR);
props.pStringValue->sz[0] = L'\0';
}
props.pb += bufferIncrement;
} else {
copying = FALSE;
}
break;
case CLUSPROP_FORMAT_BINARY:
ppbValue = (PBYTE UNALIGNED *) &pInParams[propertyItem->Offset];
pdwValue = (DWORD *) &pInParams[propertyItem->Offset+sizeof(PBYTE*)];
if ( *ppbValue != NULL ) {
bufferIncrement = sizeof(CLUSPROP_BINARY) + ALIGN_CLUSPROP( *pdwValue );
} else {
bufferIncrement = sizeof(CLUSPROP_BINARY);
}
totalBufferSize += bufferIncrement;
if ( copying && (totalBufferSize <= *pcbOutPropertyListSize) ) {
props.pSyntax->wFormat = propertyFormat;
props.pSyntax->wType = CLUSPROP_TYPE_LIST_VALUE;
if ( *ppbValue != NULL ) {
props.pBinaryValue->cbLength = *pdwValue;
CopyMemory( props.pBinaryValue->rgb, *ppbValue, *pdwValue );
} else {
props.pBinaryValue->cbLength = 0;
}
props.pb += bufferIncrement;
} else {
copying = FALSE;
}
break;
default:
break;
}
if ( status != ERROR_SUCCESS ) {
break;
}
//
// Add the value-closing endmark.
//
bufferIncrement = sizeof(CLUSPROP_SYNTAX);
totalBufferSize += bufferIncrement;
if ( copying && (totalBufferSize <= *pcbOutPropertyListSize) ) {
props.pSyntax->dw = CLUSPROP_SYNTAX_ENDMARK;
props.pb += bufferIncrement;
(*ptrItemCount)++;
} else {
copying = FALSE;
}
//
// Advance to the next property.
//
propertyItem++;
}
if ( status == ERROR_SUCCESS ) {
// Don't forget the ENDMARK.
totalBufferSize += sizeof(CLUSPROP_SYNTAX);
if ( copying && (totalBufferSize <= *pcbOutPropertyListSize) ) {
props.pSyntax->dw = CLUSPROP_SYNTAX_ENDMARK;
} else {
copying = FALSE;
}
//
// Return size of data.
//
if ( copying == FALSE ) {
*pcbRequired = totalBufferSize;
*pcbBytesReturned = 0;
status = ERROR_MORE_DATA;
} else {
*pcbRequired = 0;
*pcbBytesReturned = totalBufferSize;
status = ERROR_SUCCESS;
}
}
return(status);
} // ClRtlPropertyListFromParameterBlock
DWORD
WINAPI
ClRtlGetPrivateProperties(
IN PVOID hkeyClusterKey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
OUT PVOID pOutPropertyList,
IN DWORD cbOutPropertyListSize,
OUT LPDWORD pcbBytesReturned,
OUT LPDWORD pcbRequired
)
/*++
Routine Description:
Gets the private properties for a given object.
Arguments:
hkeyClusterKey - Supplies the handle to the key in the cluster database
to read from.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pOutPropertyList - Supplies the output buffer.
cbOutPropertyListSize - Supplies the size of the output buffer.
pcbBytesReturned - The number of bytes returned in pOutPropertyList.
pcbRequired - The required number of bytes if pOutPropertyList is too small.
Return Value:
ERROR_SUCCESS - Operation was successful.
ERROR_BAD_ARGUMENTS - An argument passed to the function was bad.
ERROR_NOT_ENOUGH_MEMORY - Error allocating memory.
ERROR_MORE_DATA - Output buffer isn't big enough to build the property list.
A Win32 error code on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
DWORD ival;
DWORD currentNameLength = 80;
DWORD currentDataLength = 80;
DWORD nameLength;
DWORD dataLength;
DWORD dataLengthExpanded;
DWORD type;
LPWSTR name;
PUCHAR data;
LPDWORD ptrItemCount;
DWORD itemCount = 0;
BOOL copying = TRUE;
DWORD totalBufferSize = 0;
DWORD bufferIncrement;
DWORD bufferIncrementExpanded;
CLUSPROP_BUFFER_HELPER props;
LPWSTR pszExpanded = NULL;
*pcbBytesReturned = 0;
*pcbRequired = 0;
if ( (hkeyClusterKey == NULL) ||
(pClusterRegApis->pfnEnumValue == NULL) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlGetPrivateProperties: hkeyClusterKey or pClusterRegApis->pfnEnumValue == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
//
// Clear the output buffer
//
if ( pOutPropertyList != NULL ) {
ZeroMemory( pOutPropertyList, cbOutPropertyListSize );
} else {
copying = FALSE;
}
//
// Need a DWORD of item count.
//
props.pb = (LPBYTE) pOutPropertyList;
ptrItemCount = props.pdw++;
totalBufferSize += sizeof(DWORD);
if ( totalBufferSize > cbOutPropertyListSize ) {
copying = FALSE;
}
//
// Allocate a property name buffer.
//
name = (LPWSTR) LocalAlloc( LMEM_FIXED, currentNameLength * sizeof(WCHAR) );
if ( name == NULL ) {
return(ERROR_NOT_ENOUGH_MEMORY);
}
//
// Allocate a property value data buffer.
//
data = (PUCHAR) LocalAlloc( LMEM_FIXED, currentDataLength );
if ( data == NULL ) {
LocalFree( name );
return(ERROR_NOT_ENOUGH_MEMORY);
}
//
// Enumerate all properties and return them!
//
for ( ival = 0; ; ival++ ) {
retry:
//
// Read the next property.
//
nameLength = currentNameLength;
dataLength = currentDataLength;
status = (*pClusterRegApis->pfnEnumValue)( hkeyClusterKey,
ival,
name,
&nameLength,
&type,
data,
&dataLength );
if ( status == ERROR_MORE_DATA ) {
if ( (nameLength+1) > currentNameLength ) {
currentNameLength = nameLength+1;
LocalFree( name );
name = (LPWSTR) LocalAlloc( LMEM_FIXED, currentNameLength * sizeof(WCHAR) );
if ( name == NULL ) {
status = ERROR_NOT_ENOUGH_MEMORY;
break;
}
}
if ( dataLength > currentDataLength ) {
currentDataLength = dataLength;
LocalFree( data );
data = (PUCHAR) LocalAlloc( LMEM_FIXED, currentDataLength );
if ( data == NULL ) {
status = ERROR_NOT_ENOUGH_MEMORY;
break;
}
}
goto retry;
} else if ( status == ERROR_NO_MORE_ITEMS ) {
status = ERROR_SUCCESS;
break;
} else if ( status != ERROR_SUCCESS ) {
break;
}
//
// Skip this property if it isn't of a known type.
//
if ( (type != REG_SZ) &&
(type != REG_EXPAND_SZ) &&
(type != REG_MULTI_SZ) &&
(type != REG_BINARY) &&
(type != REG_DWORD) &&
(type != REG_QWORD) ) {
continue;
}
itemCount++;
//
// Copy the property name.
// Need a DWORD for the next name Syntax + DWORD for name byte count +
// the namelength (in bytes? + NULL?)... must be rounded!
//
bufferIncrement = sizeof(CLUSPROP_PROPERTY_NAME)
+ ALIGN_CLUSPROP( (nameLength + 1) * sizeof(WCHAR) );
totalBufferSize += bufferIncrement;
if ( copying && (totalBufferSize <= cbOutPropertyListSize) ) {
props.pName->Syntax.dw = CLUSPROP_SYNTAX_NAME;
props.pName->cbLength = (nameLength + 1) * sizeof(WCHAR);
lstrcpyW( props.pName->sz, name);
props.pb += bufferIncrement;
} else {
copying = FALSE;
}
switch ( type ) {
case REG_SZ:
case REG_EXPAND_SZ:
case REG_MULTI_SZ:
case REG_BINARY:
bufferIncrement = sizeof(CLUSPROP_BINARY)
+ ALIGN_CLUSPROP( dataLength );
totalBufferSize += bufferIncrement;
if ( ( type == REG_SZ )
|| ( type == REG_EXPAND_SZ ) ) {
pszExpanded = ClRtlExpandEnvironmentStrings( (LPCWSTR) data );
if ( pszExpanded == NULL ) {
status = GetLastError();
break;
}
if ( lstrcmpW( pszExpanded, (LPCWSTR) data ) != 0 ) {
dataLengthExpanded = (lstrlenW( pszExpanded ) + 1) * sizeof( WCHAR );
bufferIncrementExpanded = sizeof( CLUSPROP_SZ ) + ALIGN_CLUSPROP( dataLengthExpanded );
totalBufferSize += bufferIncrementExpanded;
}
}
if ( copying && (totalBufferSize <= cbOutPropertyListSize) ) {
if ( type == REG_SZ ) {
props.pSyntax->wFormat = CLUSPROP_FORMAT_SZ;
} else if ( type == REG_EXPAND_SZ ) {
props.pSyntax->wFormat = CLUSPROP_FORMAT_EXPAND_SZ;
} else if ( type == REG_MULTI_SZ ) {
props.pSyntax->wFormat = CLUSPROP_FORMAT_MULTI_SZ;
} else {
props.pSyntax->wFormat = CLUSPROP_FORMAT_BINARY;
}
props.pSyntax->wType = CLUSPROP_TYPE_LIST_VALUE;
props.pBinaryValue->cbLength = dataLength;
CopyMemory( props.pBinaryValue->rgb, data, dataLength );
props.pb += bufferIncrement;
//
// For SZ or EXPAND_SZ, see if there is a different
// expanded string and, if so, return it as an additional
// value in the value list.
//
if ( ( type == REG_SZ )
|| ( type == REG_EXPAND_SZ ) ) {
if ( lstrcmpW( pszExpanded, (LPCWSTR) data ) != 0 ) {
props.pSyntax->dw = CLUSPROP_SYNTAX_LIST_VALUE_EXPANDED_SZ;
props.pStringValue->cbLength = dataLengthExpanded;
lstrcpyW( props.pStringValue->sz, pszExpanded );
props.pb += bufferIncrementExpanded;
}
}
} else {
copying = FALSE;
}
if ( ( ( type == REG_SZ ) || ( type == REG_EXPAND_SZ ) )
&& ( pszExpanded != NULL ) )
{
LocalFree( pszExpanded );
pszExpanded = NULL;
}
break;
case REG_DWORD:
bufferIncrement = sizeof(CLUSPROP_DWORD);
totalBufferSize += bufferIncrement;
if ( copying && (totalBufferSize <= cbOutPropertyListSize) ) {
props.pSyntax->wFormat = CLUSPROP_FORMAT_DWORD;
props.pSyntax->wType = CLUSPROP_TYPE_LIST_VALUE;
props.pDwordValue->cbLength = sizeof(DWORD);
props.pDwordValue->dw = *(LPDWORD)data;
props.pb += bufferIncrement;
} else {
copying = FALSE;
}
break;
case REG_QWORD:
bufferIncrement = sizeof(CLUSPROP_ULARGE_INTEGER);
totalBufferSize += bufferIncrement;
if ( copying && (totalBufferSize <= cbOutPropertyListSize) ) {
props.pSyntax->wFormat = CLUSPROP_FORMAT_ULARGE_INTEGER;
props.pSyntax->wType = CLUSPROP_TYPE_LIST_VALUE;
props.pULargeIntegerValue->cbLength = sizeof(ULARGE_INTEGER);
props.pULargeIntegerValue->li.u = ((ULARGE_INTEGER *)data)->u;
props.pb += bufferIncrement;
} else {
copying = FALSE;
}
break;
default:
break;
}
if ( status != ERROR_SUCCESS ) {
break;
}
//
// Add the closing endmark.
//
bufferIncrement = sizeof(CLUSPROP_SYNTAX);
totalBufferSize += bufferIncrement;
if ( copying && (totalBufferSize <= cbOutPropertyListSize) ) {
props.pSyntax->dw = CLUSPROP_SYNTAX_ENDMARK;
props.pb += bufferIncrement;
} else {
copying = FALSE;
}
}
if ( status == ERROR_SUCCESS ) {
//
// Add the closing endmark.
//
bufferIncrement = sizeof(CLUSPROP_SYNTAX);
totalBufferSize += bufferIncrement;
if ( copying && (totalBufferSize <= cbOutPropertyListSize) ) {
props.pSyntax->dw = CLUSPROP_SYNTAX_ENDMARK;
props.pb += bufferIncrement;
} else {
copying = FALSE;
}
}
LocalFree( name );
LocalFree( data );
if ( status == ERROR_SUCCESS ) {
if ( !copying ) {
*pcbRequired = totalBufferSize;
status = ERROR_MORE_DATA;
} else {
*ptrItemCount = itemCount;
*pcbBytesReturned = totalBufferSize;
status = ERROR_SUCCESS;
}
}
return(status);
} // ClRtlGetPrivateProperties
DWORD
WINAPI
ClRtlGetUnknownProperties(
IN PVOID hkeyClusterKey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyTable,
OUT PVOID pOutPropertyList,
IN DWORD cbOutPropertyListSize,
OUT LPDWORD pcbBytesReturned,
OUT LPDWORD pcbRequired
)
/*++
Routine Description:
Gets the unknown properties for a given object.
Arguments:
hkeyClusterKey - Supplies the handle to the key in the cluster database
to read from.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pPropertyTable - Pointer to the property table to process.
pOutPropertyList - Supplies the output buffer.
cbOutPropertyListSize - Supplies the size of the output buffer.
pcbBytesReturned - The number of bytes returned in pOutPropertyList.
pcbRequired - The required number of bytes if pOutPropertyList is too small.
Return Value:
ERROR_SUCCESS - Operation was successful.
ERROR_BAD_ARGUMENTS - An argument passed to the function was bad.
ERROR_NOT_ENOUGH_MEMORY - Error allocating memory.
ERROR_MORE_DATA - Output buffer isn't big enough to build the property list.
A Win32 error code on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
DWORD ival;
DWORD currentNameLength = 80;
DWORD currentDataLength = 80;
DWORD nameLength;
DWORD dataLength;
DWORD dataLengthExpanded;
DWORD type;
LPWSTR name;
PUCHAR data;
LPDWORD ptrItemCount;
DWORD itemCount = 0;
BOOL copying = TRUE;
DWORD totalBufferSize = 0;
DWORD bufferIncrement;
DWORD bufferIncrementExpanded;
BOOL found;
LPWSTR pszExpanded = NULL;
CLUSPROP_BUFFER_HELPER props;
PRESUTIL_PROPERTY_ITEM property;
*pcbBytesReturned = 0;
*pcbRequired = 0;
if ( (hkeyClusterKey == NULL) ||
(pClusterRegApis->pfnEnumValue == NULL) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlGetPrivateProperties: hkeyClusterKey or pClusterRegApis->pfnEnumValue == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
//
// Clear the output buffer
//
if ( pOutPropertyList != NULL ) {
ZeroMemory( pOutPropertyList, cbOutPropertyListSize );
} else {
copying = FALSE;
}
//
// Need a DWORD of item count.
//
props.pb = (LPBYTE) pOutPropertyList;
ptrItemCount = props.pdw++;
totalBufferSize += sizeof(DWORD);
if ( totalBufferSize > cbOutPropertyListSize ) {
copying = FALSE;
}
//
// Allocate a property name buffer.
//
name = (LPWSTR) LocalAlloc( LMEM_FIXED, currentNameLength * sizeof(WCHAR) );
if ( name == NULL ) {
return(ERROR_NOT_ENOUGH_MEMORY);
}
//
// Allocate a property value data buffer.
//
data = (PUCHAR) LocalAlloc( LMEM_FIXED, currentDataLength );
if ( data == NULL ) {
LocalFree( name );
return(ERROR_NOT_ENOUGH_MEMORY);
}
//
// Enumerate all properties and return them!
//
for ( ival = 0; ; ival++ ) {
retry:
//
// Read the next property.
//
nameLength = currentNameLength;
dataLength = currentDataLength;
status = (*pClusterRegApis->pfnEnumValue)( hkeyClusterKey,
ival,
name,
&nameLength,
&type,
data,
&dataLength );
if ( status == ERROR_MORE_DATA ) {
if ( (nameLength+1) > currentNameLength ) {
currentNameLength = nameLength+1;
LocalFree( name );
name = (LPWSTR) LocalAlloc( LMEM_FIXED, currentNameLength * sizeof(WCHAR) );
if ( name == NULL ) {
status = ERROR_NOT_ENOUGH_MEMORY;
break;
}
}
if ( dataLength > currentDataLength ) {
currentDataLength = dataLength;
LocalFree( data );
data = (PUCHAR) LocalAlloc( LMEM_FIXED, currentDataLength );
if ( data == NULL ) {
status = ERROR_NOT_ENOUGH_MEMORY;
break;
}
}
goto retry;
} else if ( status == ERROR_NO_MORE_ITEMS ) {
status = ERROR_SUCCESS;
break;
} else if ( status != ERROR_SUCCESS ) {
break;
}
//
// Skip this property if it isn't of a known type.
//
if ( (type != REG_SZ) &&
(type != REG_EXPAND_SZ) &&
(type != REG_MULTI_SZ) &&
(type != REG_BINARY) &&
(type != REG_DWORD) &&
(type != REG_QWORD) ) {
continue;
}
//
// Check if this property item is 'known'. If so, continue.
//
found = FALSE;
property = pPropertyTable;
while ( property->Name != NULL ) {
if ( lstrcmpiW( property->Name, name ) == 0 ) {
found = TRUE;
break;
}
property++;
}
if ( found ) {
continue;
}
itemCount++;
//
// Copy the property name.
// Need a DWORD for the next name Syntax + DWORD for name byte count +
// the namelength (in bytes? + NULL?)... must be rounded!
//
bufferIncrement = sizeof(CLUSPROP_PROPERTY_NAME)
+ ALIGN_CLUSPROP( (nameLength + 1) * sizeof(WCHAR) );
totalBufferSize += bufferIncrement;
if ( copying && (totalBufferSize <= cbOutPropertyListSize) ) {
props.pName->Syntax.dw = CLUSPROP_SYNTAX_NAME;
props.pName->cbLength = (nameLength + 1) * sizeof(WCHAR);
lstrcpyW( props.pName->sz, name);
props.pb += bufferIncrement;
} else {
copying = FALSE;
}
switch ( type ) {
case REG_SZ:
case REG_EXPAND_SZ:
case REG_MULTI_SZ:
case REG_BINARY:
bufferIncrement = sizeof(CLUSPROP_BINARY)
+ ALIGN_CLUSPROP( dataLength );
totalBufferSize += bufferIncrement;
if ( type == REG_EXPAND_SZ ) {
pszExpanded = ClRtlExpandEnvironmentStrings( (LPCWSTR) data );
if ( pszExpanded == NULL ) {
status = GetLastError();
break;
}
if ( lstrcmpW( pszExpanded, (LPCWSTR) data ) != 0 ) {
dataLengthExpanded = (lstrlenW( pszExpanded ) + 1) * sizeof( WCHAR );
bufferIncrementExpanded = sizeof( CLUSPROP_SZ ) + ALIGN_CLUSPROP( dataLengthExpanded );
totalBufferSize += bufferIncrementExpanded;
}
}
if ( copying && (totalBufferSize <= cbOutPropertyListSize) ) {
if ( type == REG_SZ ) {
props.pSyntax->wFormat = CLUSPROP_FORMAT_SZ;
} else if ( type == REG_EXPAND_SZ ) {
props.pSyntax->wFormat = CLUSPROP_FORMAT_EXPAND_SZ;
} else if ( type == REG_MULTI_SZ ) {
props.pSyntax->wFormat = CLUSPROP_FORMAT_MULTI_SZ;
} else {
props.pSyntax->wFormat = CLUSPROP_FORMAT_BINARY;
}
props.pSyntax->wType = CLUSPROP_TYPE_LIST_VALUE;
props.pBinaryValue->cbLength = dataLength;
CopyMemory( props.pBinaryValue->rgb, data, dataLength );
props.pb += bufferIncrement;
//
// For SZ or EXPAND_SZ, see if there is a different
// expanded string and, if so, return it as an additional
// value in the value list.
//
if ( ( type == REG_SZ )
|| ( type == REG_EXPAND_SZ ) )
{
if ( pszExpanded != NULL ) {
if ( lstrcmpW( pszExpanded, (LPCWSTR) data ) != 0 ) {
props.pSyntax->dw = CLUSPROP_SYNTAX_LIST_VALUE_EXPANDED_SZ;
props.pStringValue->cbLength = dataLengthExpanded;
lstrcpyW( props.pStringValue->sz, pszExpanded );
props.pb += bufferIncrementExpanded;
}
}
}
} else {
copying = FALSE;
}
if ( type == REG_EXPAND_SZ ) {
LocalFree( pszExpanded );
pszExpanded = NULL;
}
break;
case REG_DWORD:
bufferIncrement = sizeof(CLUSPROP_DWORD);
totalBufferSize += bufferIncrement;
if ( copying && (totalBufferSize <= cbOutPropertyListSize) ) {
props.pSyntax->wFormat = CLUSPROP_FORMAT_DWORD;
props.pSyntax->wType = CLUSPROP_TYPE_LIST_VALUE;
props.pDwordValue->cbLength = sizeof(DWORD);
props.pDwordValue->dw = *(LPDWORD)data;
props.pb += bufferIncrement;
} else {
copying = FALSE;
}
break;
case REG_QWORD:
bufferIncrement = sizeof(CLUSPROP_ULARGE_INTEGER);
totalBufferSize += bufferIncrement;
if ( copying && (totalBufferSize <= cbOutPropertyListSize) ) {
props.pSyntax->wFormat = CLUSPROP_FORMAT_ULARGE_INTEGER;
props.pSyntax->wType = CLUSPROP_TYPE_LIST_VALUE;
props.pULargeIntegerValue->cbLength = sizeof(ULARGE_INTEGER);
props.pULargeIntegerValue->li.u = ((ULARGE_INTEGER *)data)->u;
props.pb += bufferIncrement;
} else {
copying = FALSE;
}
break;
default:
break;
}
//
// Add the closing endmark.
//
bufferIncrement = sizeof(CLUSPROP_SYNTAX);
totalBufferSize += bufferIncrement;
if ( copying && (totalBufferSize <= cbOutPropertyListSize) ) {
props.pSyntax->dw = CLUSPROP_SYNTAX_ENDMARK;
props.pb += bufferIncrement;
} else {
copying = FALSE;
}
}
LocalFree( name );
LocalFree( data );
if ( !copying ) {
*pcbRequired = totalBufferSize;
status = ERROR_MORE_DATA;
} else {
*ptrItemCount = itemCount;
*pcbBytesReturned = totalBufferSize;
status = ERROR_SUCCESS;
}
return(status);
} // ClRtlGetUnknownProperties
DWORD
WINAPI
ClRtlAddUnknownProperties(
IN PVOID hkeyClusterKey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyTable,
IN OUT PVOID pOutPropertyList,
IN DWORD cbOutPropertyListSize,
IN OUT LPDWORD pcbBytesReturned,
IN OUT LPDWORD pcbRequired
)
/*++
Routine Description:
Adds the unknown properties for a given object to the end of a property
list.
Arguments:
hkeyClusterKey - Supplies the handle to the key in the cluster database
to read from.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pPropertyTable - Pointer to the property table to process.
pOutPropertyList - Supplies the output buffer.
cbOutPropertyListSize - Supplies the size of the output buffer.
pcbBytesReturned - On input, contains the number of bytes in use in the
output buffer. On output, contains the total number of bytes in
pOutPropertyList.
pcbRequired - The required number of bytes if pOutPropertyList is too small.
Return Value:
ERROR_SUCCESS - Operation was successful.
ERROR_NOT_ENOUGH_MEMORY - Error allocating memory.
A Win32 error code on failure.
--*/
{
DWORD status;
CLUSPROP_BUFFER_HELPER copyBuffer;
CLUSPROP_BUFFER_HELPER outBuffer;
DWORD bufferLength;
DWORD bytesReturned;
DWORD required;
//
// Allocate a buffer for getting 'unknown' properties.
//
if ( (cbOutPropertyListSize > *pcbBytesReturned) &&
(*pcbRequired == 0) )
{
bufferLength = cbOutPropertyListSize + (2 * sizeof(DWORD)) - *pcbBytesReturned;
outBuffer.pb = (LPBYTE) LocalAlloc( LMEM_FIXED, bufferLength );
if ( outBuffer.pb == NULL ) {
*pcbBytesReturned = 0;
*pcbRequired = 0;
return(ERROR_NOT_ENOUGH_MEMORY);
}
} else {
bufferLength = 0;
outBuffer.pb = NULL;
}
//
// Get the 'unknown' properties.
//
status = ClRtlGetUnknownProperties( hkeyClusterKey,
pClusterRegApis,
pPropertyTable,
outBuffer.pb,
bufferLength,
&bytesReturned,
&required );
if ( status == ERROR_SUCCESS ) {
//
// Copy properties if any were found.
//
if ( bytesReturned > sizeof(DWORD) ) {
//
// Copy the unknown property data to the end of the property list.
//
CL_ASSERT( bytesReturned <= bufferLength );
copyBuffer.pb = (LPBYTE) pOutPropertyList;
copyBuffer.pList->nPropertyCount += outBuffer.pList->nPropertyCount;
copyBuffer.pb += *pcbBytesReturned - sizeof(CLUSPROP_SYNTAX);
CopyMemory( copyBuffer.pb, outBuffer.pb + sizeof(DWORD), bytesReturned - sizeof(DWORD) );
*pcbBytesReturned += bytesReturned - sizeof(DWORD) - sizeof(CLUSPROP_SYNTAX);
}
} else if ( ( status == ERROR_MORE_DATA )
&& ( required == sizeof(DWORD) ) ) {
required = 0;
status = ERROR_SUCCESS;
} else {
if ( *pcbRequired == 0 ) {
*pcbRequired = *pcbBytesReturned;
}
*pcbBytesReturned = 0;
}
//
// If there are any properties, the number of bytes required will include
// both a property count (DWORD) and an endmark (CLUSPROP_SYNTAX).
// Subtract these off because these appear in both lists.
//
if ( required > sizeof(DWORD) + sizeof(CLUSPROP_SYNTAX) ) {
required -= sizeof(DWORD) + sizeof(CLUSPROP_SYNTAX);
}
//
// Free the out buffer (which may be NULL)
//
LocalFree( outBuffer.pb );
//
// Adjust lengths
//
*pcbRequired += required;
return(status);
} // ClRtlAddUnknownProperties
DWORD
WINAPI
ClRtlGetPropertySize(
IN PVOID hkeyClusterKey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyTableItem,
IN OUT LPDWORD pcbOutPropertyListSize,
IN OUT LPDWORD pnPropertyCount
)
/*++
Routine Description:
Get the total number of bytes required for this property.
Arguments:
hkeyClusterKey - Supplies the handle to the key in the cluster database
to read from.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pPropertyTableItem - Supplies the property table item for the property
whose size is to be returned.
pcbOutPropertyListSize - Supplies the size of the output buffer
required to add this property to a property list.
pnPropertyCount - The count of properties is incremented.
Return Value:
ERROR_SUCCESS - Operation was successful.
ERROR_BAD_ARGUMENTS - An argument passed to the function was bad.
ERROR_INVALID_PARAMETER -
A Win32 error code on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
DWORD valueType;
DWORD bytesReturned;
DWORD headerLength;
PVOID key;
LPWSTR pszValue = NULL;
LPWSTR pszExpanded = NULL;
CRegistryValueName rvn;
if ( (hkeyClusterKey == NULL) ||
(pClusterRegApis->pfnQueryValue == NULL) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlGetPropertySize: hkeyClusterKey or pClusterRegApis->pfnQueryValue == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
//
// Use the wrapper class CRegistryValueName to parse value name to see if it
// contains a backslash.
//
status = rvn.ScInit( pPropertyTableItem->Name, pPropertyTableItem->KeyName );
if ( status != ERROR_SUCCESS )
{
return status;
}
//
// If the value resides at a different location, open the key.
//
if ( rvn.PszKeyName() != NULL ) {
if ( (pClusterRegApis->pfnOpenKey == NULL) ||
(pClusterRegApis->pfnCloseKey == NULL) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlGetPropertySize: pClusterRegApis->pfnOpenValue or pfnCloseKey == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
status = (*pClusterRegApis->pfnOpenKey)( hkeyClusterKey,
rvn.PszKeyName(),
KEY_READ,
&key);
} else {
key = hkeyClusterKey;
}
//
// Read the value size.
//
if ( status == ERROR_SUCCESS ) {
status = (*pClusterRegApis->pfnQueryValue)( key,
rvn.PszName(),
&valueType,
NULL,
&bytesReturned );
}
//
// If the value is not present, return the default value.
//
if ( status == ERROR_FILE_NOT_FOUND ) {
switch ( pPropertyTableItem->Format ) {
case CLUSPROP_FORMAT_DWORD:
case CLUSPROP_FORMAT_LONG:
status = ERROR_SUCCESS;
bytesReturned = sizeof(DWORD);
valueType = pPropertyTableItem->Format;
break;
case CLUSPROP_FORMAT_ULARGE_INTEGER:
case CLUSPROP_FORMAT_LARGE_INTEGER:
status = ERROR_SUCCESS;
bytesReturned = sizeof(ULARGE_INTEGER);
valueType = pPropertyTableItem->Format;
break;
case CLUSPROP_FORMAT_MULTI_SZ:
status = ERROR_SUCCESS;
if ( pPropertyTableItem->Default != 0 ) {
bytesReturned = pPropertyTableItem->Minimum;
} else {
bytesReturned = sizeof(WCHAR);
}
valueType = pPropertyTableItem->Format;
break;
case CLUSPROP_FORMAT_SZ:
case CLUSPROP_FORMAT_EXPAND_SZ:
status = ERROR_SUCCESS;
if ( pPropertyTableItem->Default != 0 ) {
bytesReturned = (lstrlenW((LPCWSTR)pPropertyTableItem->lpDefault) + 1) * sizeof(WCHAR);
} else {
bytesReturned = (lstrlenW(CLRTL_NULL_STRING) + 1) * sizeof(WCHAR);
}
valueType = pPropertyTableItem->Format;
break;
case CLUSPROP_FORMAT_BINARY:
status = ERROR_SUCCESS;
if ( pPropertyTableItem->Default != 0 ) {
bytesReturned = pPropertyTableItem->Minimum;
} else {
bytesReturned = 0;
}
valueType = pPropertyTableItem->Format;
break;
default:
valueType = CLUSPROP_FORMAT_UNKNOWN;
break;
}
} else if ( status == ERROR_SUCCESS ) {
switch ( valueType ) {
case REG_DWORD:
if ((pPropertyTableItem->Format == CLUSPROP_FORMAT_DWORD) ||
(pPropertyTableItem->Format == CLUSPROP_FORMAT_LONG))
{
valueType = pPropertyTableItem->Format;
} else {
valueType = CLUSPROP_FORMAT_UNKNOWN;
}
break;
case REG_QWORD:
if ((pPropertyTableItem->Format == CLUSPROP_FORMAT_ULARGE_INTEGER) ||
(pPropertyTableItem->Format == CLUSPROP_FORMAT_LARGE_INTEGER))
{
valueType = pPropertyTableItem->Format;
} else {
valueType = CLUSPROP_FORMAT_UNKNOWN;
}
break;
case REG_MULTI_SZ:
valueType = CLUSPROP_FORMAT_MULTI_SZ;
break;
case REG_SZ:
case REG_EXPAND_SZ:
//
// Include the size of the expanded string in both REG_SZ and REG_EXPAND_SZ
//
pszValue = ClRtlGetSzValue( (HKEY) key,
rvn.PszName(),
pClusterRegApis );
if ( pszValue != NULL ) {
pszExpanded = ClRtlExpandEnvironmentStrings( pszValue );
if ( pszExpanded == NULL ) {
status = GetLastError();
} else if ( lstrcmpW( pszValue, pszExpanded ) != 0 ) {
bytesReturned += ALIGN_CLUSPROP( (lstrlenW( pszExpanded ) + 1) * sizeof( WCHAR ) );
bytesReturned += sizeof(CLUSPROP_SZ);
}
LocalFree( pszValue );
LocalFree( pszExpanded );
}
if ( valueType == REG_SZ ) {
valueType = CLUSPROP_FORMAT_SZ;
}
else {
valueType = CLUSPROP_FORMAT_EXPAND_SZ;
}
break;
case REG_BINARY:
valueType = CLUSPROP_FORMAT_BINARY;
break;
default:
valueType = CLUSPROP_FORMAT_UNKNOWN;
break;
}
}
if ( status == ERROR_FILE_NOT_FOUND ) {
status = ERROR_SUCCESS;
} else if ( status == ERROR_SUCCESS ) {
if ( pPropertyTableItem->Format != valueType ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlGetPropertySize: Property '%1!ls!' format %2!d! expected, was %3!d!.\n", rvn.PszKeyName(), pPropertyTableItem->Format, valueType );
status = ERROR_INVALID_PARAMETER;
} else {
//assume that the size of dword and long
//is fixed to 32 bits
if (( valueType == CLUSPROP_FORMAT_DWORD ) ||
( valueType == CLUSPROP_FORMAT_LONG ))
{
headerLength = sizeof(CLUSPROP_PROPERTY_NAME)
+ ((lstrlenW( pPropertyTableItem->Name ) + 1) * sizeof(WCHAR))
+ sizeof(CLUSPROP_DWORD)
- 4 // CLUSPROP_DWORD.dw (specified by bytesReturned)
+ sizeof(CLUSPROP_SYNTAX); // for endmark
} else
if (( valueType == CLUSPROP_FORMAT_ULARGE_INTEGER ) ||
( valueType == CLUSPROP_FORMAT_LARGE_INTEGER ))
{
headerLength = sizeof(CLUSPROP_PROPERTY_NAME)
+ ((lstrlenW( pPropertyTableItem->Name ) + 1) * sizeof(WCHAR))
+ sizeof(CLUSPROP_ULARGE_INTEGER)
- 8 // CLUSPROP_ULARGE_INTEGER.li (specified by bytesReturned)
+ sizeof(CLUSPROP_SYNTAX); // for endmark
} else {
// NOTE: This assumes SZ, EXPAND_SZ, MULTI_SZ, and BINARY are the same size
headerLength = sizeof(CLUSPROP_PROPERTY_NAME)
+ ((lstrlenW( pPropertyTableItem->Name ) + 1) * sizeof(WCHAR))
+ sizeof(CLUSPROP_BINARY)
+ sizeof(CLUSPROP_SYNTAX); // for endmark
}
headerLength = ALIGN_CLUSPROP( headerLength );
bytesReturned = ALIGN_CLUSPROP( bytesReturned );
*pcbOutPropertyListSize += (bytesReturned + headerLength);
*pnPropertyCount += 1;
}
}
//
// Close the key if we opened it.
//
if ( ( rvn.PszKeyName() != NULL ) &&
( key != NULL ) ) {
(*pClusterRegApis->pfnCloseKey)( key );
}
return(status);
} // ClRtlGetPropertySize
DWORD
WINAPI
ClRtlGetProperty(
IN PVOID hkeyClusterKey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyTableItem,
OUT PVOID * pOutPropertyItem,
IN OUT LPDWORD pcbOutPropertyItemSize
)
/*++
Routine Description:
Arguments:
Return Value:
ERROR_SUCCESS -
ERROR_BAD_ARGUMENTS -
ERROR_MORE_DATA -
Notes:
The buffer size has already been determined to be large enough to hold
the return data.
--*/
{
DWORD status = ERROR_SUCCESS;
DWORD valueType;
DWORD bytesReturned;
DWORD bufferSize;
PVOID dataBuffer;
DWORD nameLength;
PVOID key = NULL;
CLUSTER_PROPERTY_FORMAT format;
CLUSPROP_BUFFER_HELPER props;
LPWSTR pszExpanded = NULL;
CRegistryValueName rvn;
if ( (hkeyClusterKey == NULL) ||
(pClusterRegApis->pfnQueryValue == NULL) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlGetProperty: hkeyClusterKey or pClusterRegApis->pfnQueryValue == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
//
// Use the wrapper class CRegistryValueName to parse value name to see if it
// contains a backslash.
//
status = rvn.ScInit( pPropertyTableItem->Name, pPropertyTableItem->KeyName );
if ( status != ERROR_SUCCESS )
{
return status;
}
//
// If the value resides at a different location, open the key.
//
if ( rvn.PszKeyName() != NULL ) {
if ( (pClusterRegApis->pfnOpenKey == NULL) ||
(pClusterRegApis->pfnCloseKey == NULL) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlGetProperty: pClusterRegApis->pfnOpenValue or pfnCloseKey == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
status = (*pClusterRegApis->pfnOpenKey)( hkeyClusterKey,
rvn.PszKeyName(),
KEY_READ,
&key);
} else {
key = hkeyClusterKey;
}
//
// Find out if this property is available
//
if ( status == ERROR_SUCCESS ) {
status = (*pClusterRegApis->pfnQueryValue)( key,
rvn.PszName(),
&valueType,
NULL,
&bytesReturned );
}
//
// If the value is not present, return the default value.
//
if ( status == ERROR_FILE_NOT_FOUND ) {
switch ( pPropertyTableItem->Format ) {
case CLUSPROP_FORMAT_DWORD:
case CLUSPROP_FORMAT_LONG:
status = ERROR_SUCCESS;
bytesReturned = sizeof(DWORD);
valueType = pPropertyTableItem->Format;
break;
case CLUSPROP_FORMAT_ULARGE_INTEGER:
case CLUSPROP_FORMAT_LARGE_INTEGER:
status = ERROR_SUCCESS;
bytesReturned = sizeof(ULARGE_INTEGER);
valueType = pPropertyTableItem->Format;
break;
case CLUSPROP_FORMAT_MULTI_SZ:
status = ERROR_SUCCESS;
if ( pPropertyTableItem->Default != 0 ) {
bytesReturned = pPropertyTableItem->Minimum;
} else {
bytesReturned = sizeof(WCHAR);
}
valueType = pPropertyTableItem->Format;
break;
case CLUSPROP_FORMAT_SZ:
case CLUSPROP_FORMAT_EXPAND_SZ:
status = ERROR_SUCCESS;
if ( pPropertyTableItem->Default != 0 ) {
bytesReturned = (lstrlenW((LPCWSTR)pPropertyTableItem->lpDefault) + 1) * sizeof(WCHAR);
} else {
bytesReturned = (lstrlenW(CLRTL_NULL_STRING) + 1) * sizeof(WCHAR);
}
valueType = pPropertyTableItem->Format;
break;
case CLUSPROP_FORMAT_BINARY:
status = ERROR_SUCCESS;
if ( pPropertyTableItem->Default != 0 ) {
bytesReturned = pPropertyTableItem->Minimum;
} else {
bytesReturned = 0;
}
valueType = pPropertyTableItem->Format;
break;
default:
valueType = CLUSPROP_FORMAT_UNKNOWN;
break;
}
}
if ( status == ERROR_SUCCESS ) {
//
// Get the property format
//
switch ( pPropertyTableItem->Format ) {
case CLUSPROP_FORMAT_BINARY:
case CLUSPROP_FORMAT_DWORD:
case CLUSPROP_FORMAT_LONG:
case CLUSPROP_FORMAT_ULARGE_INTEGER:
case CLUSPROP_FORMAT_LARGE_INTEGER:
case CLUSPROP_FORMAT_MULTI_SZ:
case CLUSPROP_FORMAT_SZ:
case CLUSPROP_FORMAT_EXPAND_SZ:
format = (enum CLUSTER_PROPERTY_FORMAT) pPropertyTableItem->Format;
break;
default:
format = CLUSPROP_FORMAT_UNKNOWN;
break;
}
props.pb = (LPBYTE) *pOutPropertyItem;
//
// Copy the property name, which includes its syntax and length.
//
nameLength = (lstrlenW( pPropertyTableItem->Name ) + 1) * sizeof(WCHAR);
props.pSyntax->dw = CLUSPROP_SYNTAX_NAME;
props.pName->cbLength = nameLength;
lstrcpyW( props.pName->sz, pPropertyTableItem->Name );
bytesReturned = sizeof(*props.pName) + ALIGN_CLUSPROP( nameLength );
*pcbOutPropertyItemSize -= bytesReturned;
props.pb += bytesReturned;
//
// Copy the property value header.
//
props.pSyntax->wFormat = (USHORT)format;
props.pSyntax->wType = CLUSPROP_TYPE_LIST_VALUE;
//
// Read the property value.
//
if ( pPropertyTableItem->Format == CLUSPROP_FORMAT_DWORD ||
pPropertyTableItem->Format == CLUSPROP_FORMAT_LONG )
{
bufferSize = *pcbOutPropertyItemSize
- (sizeof(*props.pDwordValue) - sizeof(props.pDwordValue->dw));
dataBuffer = &props.pDwordValue->dw;
} else
if ( pPropertyTableItem->Format == CLUSPROP_FORMAT_ULARGE_INTEGER ||
pPropertyTableItem->Format == CLUSPROP_FORMAT_ULARGE_INTEGER )
{
bufferSize = *pcbOutPropertyItemSize
- (sizeof(*props.pULargeIntegerValue) - sizeof(props.pULargeIntegerValue->li));
dataBuffer = &props.pULargeIntegerValue->li;
} else {
// NOTE: This assumes SZ, MULTI_SZ, and BINARY are the same size
bufferSize = *pcbOutPropertyItemSize - sizeof(*props.pBinaryValue);
dataBuffer = props.pBinaryValue->rgb;
}
bytesReturned = bufferSize;
if ( key == NULL ) {
status = ERROR_FILE_NOT_FOUND;
} else {
status = (*pClusterRegApis->pfnQueryValue)( key,
rvn.PszName(),
&valueType,
(LPBYTE) dataBuffer,
&bytesReturned );
}
//
// If the value is not present, return the default value.
//
if ( status == ERROR_FILE_NOT_FOUND ) {
switch ( pPropertyTableItem->Format ) {
case CLUSPROP_FORMAT_DWORD:
case CLUSPROP_FORMAT_LONG:
//assume size of dword and long is the same
status = ERROR_SUCCESS;
bytesReturned = sizeof(DWORD);
props.pDwordValue->dw = pPropertyTableItem->Default;
break;
case CLUSPROP_FORMAT_ULARGE_INTEGER:
case CLUSPROP_FORMAT_LARGE_INTEGER:
status = ERROR_SUCCESS;
bytesReturned = sizeof(ULARGE_INTEGER);
props.pULargeIntegerValue->li.u = pPropertyTableItem->ULargeIntData->Default.u;
break;
case CLUSPROP_FORMAT_MULTI_SZ:
status = ERROR_SUCCESS;
if ( pPropertyTableItem->Default != 0 ) {
bytesReturned = pPropertyTableItem->Minimum;
if ( bufferSize < bytesReturned ) {
CopyMemory( dataBuffer, (LPCWSTR)pPropertyTableItem->lpDefault, bytesReturned );
}
} else {
bytesReturned = sizeof(WCHAR);
}
break;
case CLUSPROP_FORMAT_SZ:
case CLUSPROP_FORMAT_EXPAND_SZ:
status = ERROR_SUCCESS;
if ( pPropertyTableItem->Default != 0 ) {
bytesReturned = (lstrlenW((LPCWSTR)pPropertyTableItem->lpDefault) + 1) * sizeof(WCHAR);
if ( bufferSize < bytesReturned ) {
lstrcpyW( (LPWSTR) dataBuffer, (LPCWSTR)pPropertyTableItem->lpDefault );
}
} else {
bytesReturned = (lstrlenW(CLRTL_NULL_STRING) + 1) * sizeof(WCHAR);
}
break;
case CLUSPROP_FORMAT_BINARY:
status = ERROR_SUCCESS;
if ( pPropertyTableItem->Default != 0 ) {
bytesReturned = pPropertyTableItem->Minimum;
if ( bufferSize < bytesReturned ) {
CopyMemory( dataBuffer, (LPBYTE)pPropertyTableItem->lpDefault, bytesReturned );
}
} else {
bytesReturned = 0;
}
break;
default:
break;
}
}
if ( bufferSize < bytesReturned ) {
status = ERROR_MORE_DATA;
} else if ( status == ERROR_SUCCESS ) {
props.pValue->cbLength = bytesReturned;
// Round the bytes used up to the next DWORD boundary.
bytesReturned = ALIGN_CLUSPROP( bytesReturned );
bytesReturned += sizeof(*props.pValue);
props.pb += bytesReturned;
//
// If this is an SZ or EXPAND_SZ, see if the expanded value should
// be added to the value list.
//
if ( ( pPropertyTableItem->Format == CLUSPROP_FORMAT_SZ )
|| ( pPropertyTableItem->Format == CLUSPROP_FORMAT_EXPAND_SZ ) ) {
pszExpanded = ClRtlExpandEnvironmentStrings( (LPCWSTR) dataBuffer );
if ( pszExpanded == NULL ) {
status = GetLastError();
} else {
if ( lstrcmpiW( pszExpanded, (LPCWSTR) dataBuffer ) != 0 ) {
props.pSyntax->dw = CLUSPROP_SYNTAX_LIST_VALUE_EXPANDED_SZ;
bufferSize = ALIGN_CLUSPROP( (lstrlenW( pszExpanded ) + 1) * sizeof( WCHAR ) );
props.pStringValue->cbLength = bufferSize;
lstrcpyW( props.pStringValue->sz, pszExpanded );
bytesReturned += sizeof( *props.pStringValue ) + bufferSize;
props.pb += sizeof( *props.pStringValue ) + bufferSize;
}
LocalFree( pszExpanded );
}
}
if ( status == ERROR_SUCCESS ) {
// Add the value list endmark.
props.pSyntax->dw = CLUSPROP_SYNTAX_ENDMARK;
props.pb += sizeof(*props.pSyntax);
bytesReturned += sizeof(*props.pSyntax);
*pcbOutPropertyItemSize -= bytesReturned;
*pOutPropertyItem = (PVOID)props.pb;
} // if: ERROR_SUCCESS
} // else if: ERROR_SUCCESS
} // if: ERROR_SUCCESS
if ( status == ERROR_FILE_NOT_FOUND ) {
status = ERROR_SUCCESS;
}
//
// Close the key if we opened it.
//
if ( (rvn.PszKeyName() != NULL) &&
(key != NULL) ) {
(*pClusterRegApis->pfnCloseKey)( key );
}
return(status);
} // ClRtlGetProperty
DWORD
WINAPI
ClRtlpSetPropertyTable(
IN HANDLE hXsaction,
IN PVOID hkeyClusterKey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyTable,
IN PVOID Reserved,
IN BOOL bAllowUnknownProperties,
IN const PVOID pInPropertyList,
IN DWORD cbInPropertyListSize,
IN BOOL bForceWrite,
IN OUT OPTIONAL LPBYTE pOutParams
)
/*++
Routine Description:
Arguments:
hkeyClusterKey - The opened cluster database key where properties are to
be written. If not specified, the property list will only be
validated.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pPropertyTable - Pointer to the property table to process.
Reserved - Reserved for future use.
bAllowUnknownProperties - Don't fail if unknown properties are found.
pInPropertyList - The input buffer.
cbInPropertyListSize - The input buffer size.
bForceWrite - TRUE = always write the properties to the cluster database.
FALSE = only write the properties if they changed.
pOutParams - Parameter block in which to return the data. If specified,
parameters will only be written if they are different between
the input buffer and the parameter block.
Return Value:
ERROR_SUCCESS if successful.
ERROR_BAD_ARGUMENTS - hkeyClusterKey is specified but proper cluster
registry APIs are not specified, or no property table is specified.
ERROR_INVALID_DATA - No property list is specified or the format of the
property list is invalid.
ERROR_INSUFFICIENT_BUFFER - The property list buffer isn't large enough to
contain all the data it indicates it should contain.
ERROR_INVALID_PARAMETER - The property list isn't formatted properly.
A Win32 Error on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
PRESUTIL_PROPERTY_ITEM propertyItem;
DWORD inBufferSize;
DWORD itemCount;
DWORD dataSize;
PVOID key;
CLUSPROP_BUFFER_HELPER buf;
PCLUSPROP_PROPERTY_NAME pName;
CRegistryValueName rvn;
if ( ( (hkeyClusterKey != NULL) &&
(pClusterRegApis->pfnSetValue == NULL) ) ||
( pPropertyTable == NULL ) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPropertyTable: hkeyClusterKey or pClusterRegApis->pfnSetValue == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
if ( pInPropertyList == NULL ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPropertyTable: pInPropertyList == NULL. Returning ERROR_INVALID_DATA\n" );
return(ERROR_INVALID_DATA);
}
buf.pb = (LPBYTE) pInPropertyList;
inBufferSize = cbInPropertyListSize;
//
// Get the number of items in this list
//
if ( inBufferSize < sizeof(DWORD) ) {
return(ERROR_INSUFFICIENT_BUFFER);
}
itemCount = buf.pList->nPropertyCount;
buf.pdw++;
inBufferSize -= sizeof(*buf.pdw);
//
// Parse the rest of the items in the buffer.
//
while ( itemCount-- ) {
//
// Verify that the buffer is big enough to contain the
// property name and a value.
//
pName = buf.pName;
if ( inBufferSize < sizeof(*pName) ) {
return(ERROR_INSUFFICIENT_BUFFER);
}
dataSize = sizeof(*pName) + ALIGN_CLUSPROP( pName->cbLength );
if ( inBufferSize < dataSize + sizeof(CLUSPROP_VALUE) ) {
return(ERROR_INSUFFICIENT_BUFFER);
}
//
// Verify that the syntax of the property name is correct.
//
if ( pName->Syntax.dw != CLUSPROP_SYNTAX_NAME ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPropertyTable: not a name syntax.\n" );
return(ERROR_INVALID_PARAMETER);
}
//
// Verify that the length is correct for the string.
//
if ( pName->cbLength != (lstrlenW( pName->sz ) + 1) * sizeof(WCHAR) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPropertyTable: name is not a valid C string.\n" );
return(ERROR_INVALID_DATA);
}
//
// Move the buffer pointer to the property value.
//
buf.pb += dataSize;
inBufferSize -= dataSize;
//
// Find the property name in the list of known properties.
//
propertyItem = pPropertyTable;
while ( propertyItem->Name != NULL ) {
if ( lstrcmpiW( pName->sz, propertyItem->Name ) == 0 ) {
//
// Verify that the buffer is big enough to contain the value.
//
dataSize = sizeof(*buf.pValue)
+ ALIGN_CLUSPROP( buf.pValue->cbLength )
+ sizeof(CLUSPROP_SYNTAX); // endmark
if ( inBufferSize < dataSize ) {
return(ERROR_INSUFFICIENT_BUFFER);
}
//
// Verify that the syntax type is LIST_VALUE.
//
if ( buf.pSyntax->wType != CLUSPROP_TYPE_LIST_VALUE ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPropertyTable: Property '%1!ls!' type CLUSPROP_TYPE_LIST_VALUE (%2!d!) expected, was %3!d!.\n", pName->sz, CLUSPROP_TYPE_LIST_VALUE, buf.pSyntax->wType );
return(ERROR_INVALID_PARAMETER);
}
//
// Verify that this property should be of this format.
//
if ( buf.pSyntax->wFormat != propertyItem->Format ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPropertyTable: Property '%1!ls!' format %2!d! expected, was %3!d!.\n", pName->sz, propertyItem->Format, buf.pSyntax->wType );
status = ERROR_INVALID_PARAMETER;
break;
}
//
// Make sure we are allowed to set this item.
//
if ( propertyItem->Flags & RESUTIL_PROPITEM_READ_ONLY ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPropertyTable: Property '%1!ls!' is non-writable.\n", pName->sz );
return(ERROR_INVALID_PARAMETER);
}
//
// Use the wrapper class CRegistryValueName to parse value name to see if it
// contains a backslash.
//
status = rvn.ScInit( propertyItem->Name, propertyItem->KeyName );
if ( status != ERROR_SUCCESS ) {
return status;
}
//
// If the value resides at a different location, create the key.
//
if ( (hkeyClusterKey != NULL) &&
(rvn.PszKeyName() != NULL) ) {
DWORD disposition;
if ( (pClusterRegApis->pfnCreateKey == NULL) ||
(pClusterRegApis->pfnCloseKey == NULL) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPropertyTable: pClusterRegApis->pfnCreateKey or pfnCloseKey == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
if ( hXsaction != NULL ) {
if ( pClusterRegApis->pfnLocalCreateKey == NULL ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPropertyTable: pClusterRegApis->pfnLocalCreateKey == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
status = (*pClusterRegApis->pfnLocalCreateKey)(hXsaction,
hkeyClusterKey,
rvn.PszKeyName(),
0,
KEY_ALL_ACCESS,
NULL,
&key,
&disposition );
}
else {
status = (*pClusterRegApis->pfnCreateKey)( hkeyClusterKey,
rvn.PszKeyName(),
0,
KEY_ALL_ACCESS,
NULL,
&key,
&disposition );
}
if ( status != ERROR_SUCCESS ) {
return(status);
}
} else {
key = hkeyClusterKey;
}
//
// Validate, write, and save the property data.
//
switch ( buf.pSyntax->wFormat ) {
case CLUSPROP_FORMAT_DWORD:
status = ClRtlpSetDwordProperty(
hXsaction,
key,
pClusterRegApis,
propertyItem,
rvn,
buf.pDwordValue,
bForceWrite,
pOutParams );
break;
case CLUSPROP_FORMAT_LONG:
status = ClRtlpSetLongProperty(
hXsaction,
key,
pClusterRegApis,
propertyItem,
rvn,
buf.pLongValue,
bForceWrite,
pOutParams );
break;
case CLUSPROP_FORMAT_ULARGE_INTEGER:
status = ClRtlpSetULargeIntegerProperty(
hXsaction,
key,
pClusterRegApis,
propertyItem,
rvn,
buf.pULargeIntegerValue,
bForceWrite,
pOutParams );
break;
case CLUSPROP_FORMAT_LARGE_INTEGER:
status = ClRtlpSetLargeIntegerProperty(
hXsaction,
key,
pClusterRegApis,
propertyItem,
rvn,
buf.pLargeIntegerValue,
bForceWrite,
pOutParams );
break;
case CLUSPROP_FORMAT_SZ:
case CLUSPROP_FORMAT_EXPAND_SZ:
status = ClRtlpSetStringProperty(
hXsaction,
key,
pClusterRegApis,
propertyItem,
rvn,
buf.pStringValue,
bForceWrite,
pOutParams );
break;
case CLUSPROP_FORMAT_MULTI_SZ:
status = ClRtlpSetMultiStringProperty(
hXsaction,
key,
pClusterRegApis,
propertyItem,
rvn,
buf.pStringValue,
bForceWrite,
pOutParams );
break;
case CLUSPROP_FORMAT_BINARY:
status = ClRtlpSetBinaryProperty(
hXsaction,
key,
pClusterRegApis,
propertyItem,
rvn,
buf.pBinaryValue,
bForceWrite,
pOutParams );
break;
default:
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPropertyTable: Property '%1!ls!' unknown format %2!d! specified.\n", pName->sz, buf.pSyntax->wFormat );
status = ERROR_INVALID_PARAMETER;
break;
} // switch: value data format
//
// Close the key if we opened it.
//
if ( (hkeyClusterKey != NULL) &&
(rvn.PszKeyName() != NULL) ) {
(*pClusterRegApis->pfnCloseKey)( key );
}
//
// If an error occurred processing the property, cleanup and return.
//
if ( status != ERROR_SUCCESS ) {
return(status);
}
//
// Move the buffer past the value.
//
buf.pb += dataSize;
inBufferSize -= dataSize;
break;
} else {
propertyItem++;
//
// If we reached the end of the list, then return failure
// if we do not allow unknown properties.
//
if ( (propertyItem->Name == NULL) &&
! bAllowUnknownProperties ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPropertyTable: Property '%1!ls!' not found.\n", pName->sz );
return(ERROR_INVALID_PARAMETER);
}
}
}
//
// If no property name was found, this is an invalid parameter if
// we don't allow unknown properties. Otherwise advance past the
// property value.
//
if ( propertyItem->Name == NULL) {
if ( ! bAllowUnknownProperties ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPropertyTable: Property '%1!ls!' not found.\n", pName->sz );
return(ERROR_INVALID_PARAMETER);
}
//
// Advance the buffer pointer past the value in the value list.
//
while ( (buf.pSyntax->dw != CLUSPROP_SYNTAX_ENDMARK) &&
(inBufferSize > 0) ) {
// ASSERT(inBufferSize > sizeof(*buf.pValue) + ALIGN_CLUSPROP(buf.pValue->cbLength));
buf.pb += sizeof(*buf.pValue) + ALIGN_CLUSPROP(buf.pValue->cbLength);
inBufferSize -= sizeof(*buf.pValue) + ALIGN_CLUSPROP(buf.pValue->cbLength);
} // while: more values in the list
//
// Advance the buffer pointer past the value list endmark.
//
// ASSERT(inBufferSize >= sizeof(*buf.pSyntax));
buf.pb += sizeof(*buf.pSyntax); // endmark
inBufferSize -= sizeof(*buf.pSyntax);
}
}
//
// Now find any parameters that are not represented in the property
// table. All of these extra properties will just be set without validation.
//
if ( (status == ERROR_SUCCESS) &&
(pInPropertyList != NULL) &&
bAllowUnknownProperties ) {
status = ClRtlpSetNonPropertyTable( hXsaction,
hkeyClusterKey,
pClusterRegApis,
pPropertyTable,
NULL,
pInPropertyList,
cbInPropertyListSize );
}
return(status);
} // ClRtlpSetPropertyTable
static
DWORD
WINAPI
ClRtlpSetDwordProperty(
IN HANDLE hXsaction,
IN PVOID hkey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyItem,
IN const CRegistryValueName & rrvnModifiedNames,
IN PCLUSPROP_DWORD pInDwordValue,
IN BOOL bForceWrite,
IN OUT OPTIONAL LPBYTE pOutParams
)
/*++
Routine Description:
Validate a DWORD property, write it to the cluster database (or delete it
if it is zero length), and save it in the specified parameter block.
Arguments:
hXsaction - Transaction handle.
hkey - The opened cluster database key where the property is to be written.
If not specified, the property will only be validated.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pPropertyItem - The property from a property table to set/validate.
rrvnModifiedNames - If the name of the property contains a backslash
this object contains the modified name and keyname.
pInDwordValue - The value from the property list to set/validate.
bForceWrite - TRUE = always write the properties to the cluster database.
FALSE = only write the properties if they changed.
pOutParams - Parameter block in which to return the data. If specified,
parameters will only be written if they are different between
the input data and the parameter block, unless bForceWrite == TRUE.
Return Value:
ERROR_SUCCESS if successful.
ERROR_INVALID_DATA - The format of the data is invalid for a property
list value.
A Win32 Error on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
BOOL bZeroLengthData;
DWORD * pdwValue;
// Loop to avoid goto's.
do
{
bZeroLengthData = ( pInDwordValue->cbLength == 0 );
//
// Validate the property data if not zero length.
//
if ( ! bZeroLengthData ) {
//
// Verify the length of the value.
//
if ( pInDwordValue->cbLength != sizeof(DWORD) ) {
ClRtlDbgPrint( LOG_UNUSUAL, "ClRtlpSetDwordProperty: Property '%1!ls!' length %2!d! not DWORD length.\n", rrvnModifiedNames.PszName(), pInDwordValue->cbLength );
status = ERROR_INVALID_DATA;
break;
} // if: data in value not size of DWORD
//
// Verify that the value is within the valid range.
//
if ( ( (pPropertyItem->Flags & RESUTIL_PROPITEM_SIGNED)
&& ((LONG) pInDwordValue->dw > (LONG) pPropertyItem->Maximum))
|| ( !(pPropertyItem->Flags & RESUTIL_PROPITEM_SIGNED)
&& (pInDwordValue->dw > pPropertyItem->Maximum)) ) {
ClRtlDbgPrint( LOG_UNUSUAL, "ClRtlpSetDwordProperty: Property '%1!ls!' value %2!u! too large.\n", rrvnModifiedNames.PszName(), pInDwordValue->dw );
status = ERROR_INVALID_DATA;
break;
} // if: value too high
if ( ( (pPropertyItem->Flags & RESUTIL_PROPITEM_SIGNED)
&& ((LONG) pInDwordValue->dw < (LONG) pPropertyItem->Minimum))
|| ( !(pPropertyItem->Flags & RESUTIL_PROPITEM_SIGNED)
&& (pInDwordValue->dw < pPropertyItem->Minimum)) ) {
ClRtlDbgPrint( LOG_UNUSUAL, "ClRtlpSetDwordProperty: Property '%1!ls!' value %2!u! too small.\n", rrvnModifiedNames.PszName(), pInDwordValue->dw );
status = ERROR_INVALID_DATA;
break;
} // if: value to low
} // if: non-zero length data
pdwValue = (DWORD *) &pOutParams[pPropertyItem->Offset];
//
// Write the value to the cluster database.
// If the data length is zero, delete the value.
//
if ( hkey != NULL ) {
if ( bZeroLengthData ) {
if ( hXsaction ) {
status = (*pClusterRegApis->pfnLocalDeleteValue)(
hXsaction,
hkey,
rrvnModifiedNames.PszName() );
} else {
status = (*pClusterRegApis->pfnDeleteValue)(
hkey,
rrvnModifiedNames.PszName() );
} // if/else: doing/not doing a transaction
//
// If the property doesn't exist in the
// cluster database, fix the status.
//
if ( status == ERROR_FILE_NOT_FOUND ) {
status = ERROR_SUCCESS;
} // if: property already doesn't exist
} else {
if ( hXsaction ) {
status = (*pClusterRegApis->pfnLocalSetValue)(
hXsaction,
hkey,
rrvnModifiedNames.PszName(),
REG_DWORD,
(CONST BYTE *) &pInDwordValue->dw,
sizeof(DWORD) );
} else {
status = (*pClusterRegApis->pfnSetValue)(
hkey,
rrvnModifiedNames.PszName(),
REG_DWORD,
(CONST BYTE *) &pInDwordValue->dw,
sizeof(DWORD) );
} // if/else: doing/not doing a transaction
} // if/else: zero length data
} // if: writing data
//
// Save the value to the output Parameter block.
// If the data length is zero, set to the default.
//
if ( (status == ERROR_SUCCESS)
&& (pOutParams != NULL) ) {
if ( bZeroLengthData ) {
*pdwValue = pPropertyItem->Default;
} else {
*pdwValue = pInDwordValue->dw;
} // if/else: zero length data
} // if: data written successfully and parameter block specified
} while ( 0 );
return status;
} // ClRtlpSetDwordProperty
static
DWORD
WINAPI
ClRtlpSetLongProperty(
IN HANDLE hXsaction,
IN PVOID hkey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyItem,
IN const CRegistryValueName & rrvnModifiedNames,
IN PCLUSPROP_LONG pInLongValue,
IN BOOL bForceWrite,
IN OUT OPTIONAL LPBYTE pOutParams
)
/*++
Routine Description:
Validate a LONG property, write it to the cluster database (or delete it
if it is zero length), and save it in the specified parameter block.
Arguments:
hXsaction - Transaction handle.
hkey - The opened cluster database key where the property is to be written.
If not specified, the property will only be validated.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pPropertyItem - The property from a property table to set/validate.
rrvnModifiedNames - If the name of the property contains a backslash
this object contains the modified name and keyname.
pInLongValue - The value from the property list to set/validate.
bForceWrite - TRUE = always write the properties to the cluster database.
FALSE = only write the properties if they changed.
pOutParams - Parameter block in which to return the data. If specified,
parameters will only be written if they are different between
the input data and the parameter block, unless bForceWrite == TRUE.
Return Value:
ERROR_SUCCESS if successful.
ERROR_INVALID_DATA - The format of the data is invalid for a property
list value.
A Win32 Error on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
BOOL bZeroLengthData;
LONG * plValue;
// Loop to avoid goto's.
do
{
bZeroLengthData = ( pInLongValue->cbLength == 0 );
//
// Validate the property data if not zero length.
//
if ( ! bZeroLengthData ) {
//
// Verify the length of the value.
//
if ( pInLongValue->cbLength != sizeof(LONG) ) {
ClRtlDbgPrint( LOG_UNUSUAL, "ClRtlpSetLongProperty: Property '%1!ls!' length %2!d! not LONG length.\n", rrvnModifiedNames.PszName(), pInLongValue->cbLength );
status = ERROR_INVALID_DATA;
break;
} // if: data in value not size of LONG
//
// Verify that the value is within the valid range.
//
if ( pInLongValue->l > (LONG) pPropertyItem->Maximum ) {
ClRtlDbgPrint( LOG_UNUSUAL, "ClRtlpSetLongProperty: Property '%1!ls!' value %2!d! too large.\n", rrvnModifiedNames.PszName(), pInLongValue->l );
status = ERROR_INVALID_DATA;
break;
} // if: value too high
if ( pInLongValue->l < (LONG) pPropertyItem->Minimum ) {
ClRtlDbgPrint( LOG_UNUSUAL, "ClRtlpSetLongProperty: Property '%1!ls!' value %2!d! too small.\n", rrvnModifiedNames.PszName(), pInLongValue->l );
status = ERROR_INVALID_DATA;
break;
} // if: value too small
} // if: non-zero length data
plValue = (LONG *) &pOutParams[pPropertyItem->Offset];
//
// Write the value to the cluster database.
// If the data length is zero, delete the value.
//
if ( hkey != NULL ) {
if ( bZeroLengthData ) {
if ( hXsaction ) {
status = (*pClusterRegApis->pfnLocalDeleteValue)(
hXsaction,
hkey,
rrvnModifiedNames.PszName() );
} else {
status = (*pClusterRegApis->pfnDeleteValue)(
hkey,
rrvnModifiedNames.PszName() );
} // if/else: doing/not doing a transaction
//
// If the property doesn't exist in the
// cluster database, fix the status.
//
if ( status == ERROR_FILE_NOT_FOUND ) {
status = ERROR_SUCCESS;
} // if: property already doesn't exist
} else {
if ( hXsaction ) {
status = (*pClusterRegApis->pfnLocalSetValue)(
hXsaction,
hkey,
rrvnModifiedNames.PszName(),
REG_DWORD,
(CONST BYTE *) &pInLongValue->l,
sizeof(LONG) );
} else {
status = (*pClusterRegApis->pfnSetValue)(
hkey,
rrvnModifiedNames.PszName(),
REG_DWORD,
(CONST BYTE *) &pInLongValue->l,
sizeof(LONG) );
} // if/else: doing/not doing a transaction
} // if/else: zero length data
} // if: writing data
//
// Save the value to the output Parameter block.
// If the data length is zero, set to the default.
//
if ( (status == ERROR_SUCCESS)
&& (pOutParams != NULL) ) {
if ( bZeroLengthData ) {
*plValue = (LONG) pPropertyItem->Default;
} else {
*plValue = pInLongValue->l;
} // if/else: zero length data
} // if: data written successfully and parameter block specified
} while ( 0 );
return status;
} // ClRtlpSetLongProperty
static
DWORD
WINAPI
ClRtlpSetULargeIntegerProperty(
IN HANDLE hXsaction,
IN PVOID hkey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyItem,
IN const CRegistryValueName & rrvnModifiedNames,
IN PCLUSPROP_ULARGE_INTEGER pInULargeIntegerValue,
IN BOOL bForceWrite,
IN OUT OPTIONAL LPBYTE pOutParams
)
/*++
Routine Description:
Validate a ULARGE_INTEGER property, write it to the cluster database (or
delete it if it is zero length), and save it in the specified parameter
block.
Arguments:
hXsaction - Transaction handle.
hkey - The opened cluster database key where the property is to be written.
If not specified, the property will only be validated.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pPropertyItem - The property from a property table to set/validate.
rrvnModifiedNames - If the name of the property contains a backslash
this object contains the modified name and keyname.
pInULargeIntegerValue - The value from the property list to set/validate.
bForceWrite - TRUE = always write the properties to the cluster database.
FALSE = only write the properties if they changed.
pOutParams - Parameter block in which to return the data. If specified,
parameters will only be written if they are different between
the input data and the parameter block, unless bForceWrite == TRUE.
Return Value:
ERROR_SUCCESS if successful.
ERROR_INVALID_DATA - The format of the data is invalid for a property
list value.
A Win32 Error on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
BOOL bZeroLengthData;
ULARGE_INTEGER * pullValue;
// Loop to avoid goto's.
do
{
bZeroLengthData = ( pInULargeIntegerValue->cbLength == 0 );
//
// Validate the property data if not zero length.
//
if ( ! bZeroLengthData ) {
//
// Verify the length of the value.
//
if ( pInULargeIntegerValue->cbLength != sizeof(ULARGE_INTEGER) ) {
ClRtlDbgPrint(LOG_UNUSUAL,
"ClRtlpSetULargeIntegerProperty: Property '%1!ls!' length %2!d! "
"not ULARGE_INTEGER length.\n",
rrvnModifiedNames.PszName(),
pInULargeIntegerValue->cbLength );
status = ERROR_INVALID_DATA;
break;
} // if: data in value not size of DWORD
//
// Verify that the value is within the valid range.
//
if ( ( (pPropertyItem->Flags & RESUTIL_PROPITEM_SIGNED)
&& ((LONGLONG)pInULargeIntegerValue->li.QuadPart > (LONGLONG)pPropertyItem->ULargeIntData->Maximum.QuadPart))
|| ( !(pPropertyItem->Flags & RESUTIL_PROPITEM_SIGNED)
&& (pInULargeIntegerValue->li.QuadPart > pPropertyItem->ULargeIntData->Maximum.QuadPart)) )
{
ClRtlDbgPrint(
LOG_UNUSUAL,
"ClRtlpSetULargeIntegerProperty: Property '%1!ls!' value %2!I64u! "
"too large.\n",
rrvnModifiedNames.PszName(),
pInULargeIntegerValue->li.QuadPart );
status = ERROR_INVALID_DATA;
break;
} // if: value too high
if ( ( (pPropertyItem->Flags & RESUTIL_PROPITEM_SIGNED)
&& ((LONGLONG)pInULargeIntegerValue->li.QuadPart < (LONGLONG)pPropertyItem->ULargeIntData->Minimum.QuadPart))
|| ( !(pPropertyItem->Flags & RESUTIL_PROPITEM_SIGNED)
&& (pInULargeIntegerValue->li.QuadPart < pPropertyItem->ULargeIntData->Minimum.QuadPart)) )
{
ClRtlDbgPrint(LOG_UNUSUAL,
"ClRtlpSetULargeIntegerProperty: Property '%1!ls!' value "
"%2!I64u! too small.\n",
rrvnModifiedNames.PszName(),
pInULargeIntegerValue->li.QuadPart );
status = ERROR_INVALID_DATA;
break;
} // if: value to low
} // if: non-zero length data
pullValue = (ULARGE_INTEGER *) &pOutParams[pPropertyItem->Offset];
//
// Write the value to the cluster database.
// If the data length is zero, delete the value.
//
if ( hkey != NULL ) {
if ( bZeroLengthData ) {
if ( hXsaction ) {
status = (*pClusterRegApis->pfnLocalDeleteValue)(
hXsaction,
hkey,
rrvnModifiedNames.PszName() );
} else {
status = (*pClusterRegApis->pfnDeleteValue)(
hkey,
rrvnModifiedNames.PszName() );
} // if/else: doing/not doing a transaction
//
// If the property doesn't exist in the
// cluster database, fix the status.
//
if ( status == ERROR_FILE_NOT_FOUND ) {
status = ERROR_SUCCESS;
} // if: property already doesn't exist
} else {
if ( hXsaction ) {
status = (*pClusterRegApis->pfnLocalSetValue)(
hXsaction,
hkey,
rrvnModifiedNames.PszName(),
REG_QWORD,
(CONST BYTE *) &pInULargeIntegerValue->li.QuadPart,
sizeof(ULARGE_INTEGER) );
} else {
status = (*pClusterRegApis->pfnSetValue)(
hkey,
rrvnModifiedNames.PszName(),
REG_QWORD,
(CONST BYTE *) &pInULargeIntegerValue->li.QuadPart,
sizeof(ULARGE_INTEGER) );
} // if/else: doing/not doing a transaction
} // if/else: zero length data
} // if: writing data
//
// Save the value to the output Parameter block.
// If the data length is zero, set to the default.
//
if ( (status == ERROR_SUCCESS) && (pOutParams != NULL) ) {
if ( bZeroLengthData ) {
pullValue->u = pPropertyItem->ULargeIntData->Default.u;
} else {
pullValue->u = pInULargeIntegerValue->li.u;
} // if/else: zero length data
} // if: data written successfully and parameter block specified
} while ( 0 );
return status;
} // ClRtlpSetULargeIntegerProperty
static
DWORD
WINAPI
ClRtlpSetLargeIntegerProperty(
IN HANDLE hXsaction,
IN PVOID hkey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyItem,
IN const CRegistryValueName & rrvnModifiedNames,
IN PCLUSPROP_LARGE_INTEGER pInLargeIntegerValue,
IN BOOL bForceWrite,
IN OUT OPTIONAL LPBYTE pOutParams
)
/*++
Routine Description:
Validate a LARGE_INTEGER property, write it to the cluster database (or
delete it if it is zero length), and save it in the specified parameter
block.
Arguments:
hXsaction - Transaction handle.
hkey - The opened cluster database key where the property is to be written.
If not specified, the property will only be validated.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pPropertyItem - The property from a property table to set/validate.
rrvnModifiedNames - If the name of the property contains a backslash
this object contains the modified name and keyname.
pInLargeIntegerValue - The value from the property list to set/validate.
bForceWrite - TRUE = always write the properties to the cluster database.
FALSE = only write the properties if they changed.
pOutParams - Parameter block in which to return the data. If specified,
parameters will only be written if they are different between
the input data and the parameter block, unless bForceWrite == TRUE.
Return Value:
ERROR_SUCCESS if successful.
ERROR_INVALID_DATA - The format of the data is invalid for a property
list value.
A Win32 Error on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
BOOL bZeroLengthData;
LARGE_INTEGER * pllValue;
// Loop to avoid goto's.
do
{
bZeroLengthData = ( pInLargeIntegerValue->cbLength == 0 );
//
// Validate the property data if not zero length.
//
if ( ! bZeroLengthData ) {
//
// Verify the length of the value.
//
if ( pInLargeIntegerValue->cbLength != sizeof(LARGE_INTEGER) ) {
ClRtlDbgPrint(LOG_UNUSUAL,
"ClRtlpSetLargeIntegerProperty: Property '%1!ls!' length %2!d! "
"not LARGE_INTEGER length.\n",
rrvnModifiedNames.PszName(),
pInLargeIntegerValue->cbLength );
status = ERROR_INVALID_DATA;
break;
} // if: data in value not size of DWORD
//
// Verify that the value is within the valid range.
//
if ( ( (pPropertyItem->Flags & RESUTIL_PROPITEM_SIGNED)
&& ((LONGLONG)pInLargeIntegerValue->li.QuadPart > (LONGLONG)pPropertyItem->LargeIntData->Maximum.QuadPart))
|| ( !(pPropertyItem->Flags & RESUTIL_PROPITEM_SIGNED)
&& (pInLargeIntegerValue->li.QuadPart > pPropertyItem->LargeIntData->Maximum.QuadPart)) )
{
ClRtlDbgPrint(LOG_UNUSUAL,
"ClRtlpSetLargeIntegerProperty: Property '%1!ls!' value %2!I64d! "
"too large.\n",
rrvnModifiedNames.PszName(),
pInLargeIntegerValue->li.QuadPart );
status = ERROR_INVALID_DATA;
break;
} // if: value too high
if ( ( (pPropertyItem->Flags & RESUTIL_PROPITEM_SIGNED)
&& ((LONGLONG)pInLargeIntegerValue->li.QuadPart < (LONGLONG)pPropertyItem->LargeIntData->Minimum.QuadPart))
|| ( !(pPropertyItem->Flags & RESUTIL_PROPITEM_SIGNED)
&& (pInLargeIntegerValue->li.QuadPart < pPropertyItem->LargeIntData->Minimum.QuadPart)) )
{
ClRtlDbgPrint(LOG_UNUSUAL,
"ClRtlpSetLargeIntegerProperty: Property '%1!ls!' value "
"%2!I64d! too small.\n",
rrvnModifiedNames.PszName(),
pInLargeIntegerValue->li.QuadPart );
status = ERROR_INVALID_DATA;
break;
} // if: value to low
} // if: non-zero length data
pllValue = (LARGE_INTEGER *) &pOutParams[pPropertyItem->Offset];
//
// Write the value to the cluster database.
// If the data length is zero, delete the value.
//
if ( hkey != NULL ) {
if ( bZeroLengthData ) {
if ( hXsaction ) {
status = (*pClusterRegApis->pfnLocalDeleteValue)(
hXsaction,
hkey,
rrvnModifiedNames.PszName() );
} else {
status = (*pClusterRegApis->pfnDeleteValue)(
hkey,
rrvnModifiedNames.PszName() );
} // if/else: doing/not doing a transaction
//
// If the property doesn't exist in the
// cluster database, fix the status.
//
if ( status == ERROR_FILE_NOT_FOUND ) {
status = ERROR_SUCCESS;
} // if: property already doesn't exist
} else {
if ( hXsaction ) {
status = (*pClusterRegApis->pfnLocalSetValue)(
hXsaction,
hkey,
rrvnModifiedNames.PszName(),
REG_QWORD,
(CONST BYTE *) &pInLargeIntegerValue->li.QuadPart,
sizeof(LARGE_INTEGER) );
} else {
status = (*pClusterRegApis->pfnSetValue)(
hkey,
rrvnModifiedNames.PszName(),
REG_QWORD,
(CONST BYTE *) &pInLargeIntegerValue->li.QuadPart,
sizeof(LARGE_INTEGER) );
} // if/else: doing/not doing a transaction
} // if/else: zero length data
} // if: writing data
//
// Save the value to the output Parameter block.
// If the data length is zero, set to the default.
//
if ( (status == ERROR_SUCCESS) && (pOutParams != NULL) ) {
if ( bZeroLengthData ) {
pllValue->u = pPropertyItem->LargeIntData->Default.u;
} else {
pllValue->u = pInLargeIntegerValue->li.u;
} // if/else: zero length data
} // if: data written successfully and parameter block specified
} while ( 0 );
return status;
} // ClRtlpSetLargeIntegerProperty
static
DWORD
WINAPI
ClRtlpSetStringProperty(
IN HANDLE hXsaction,
IN PVOID hkey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyItem,
IN const CRegistryValueName & rrvnModifiedNames,
IN PCLUSPROP_SZ pInStringValue,
IN BOOL bForceWrite,
IN OUT OPTIONAL LPBYTE pOutParams
)
/*++
Routine Description:
Validate a string property (SZ or EXPAND_SZ), write it to the cluster
database (or delete it if it is zero length), and save it in the
specified parameter block.
Arguments:
hXsaction - Transaction handle.
hkey - The opened cluster database key where the property is to be written.
If not specified, the property will only be validated.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pPropertyItem - The property from a property table to set/validate.
rrvnModifiedNames - If the name of the property contains a backslash
this object contains the modified name and keyname.
pInStringValue - The value from the property list to set/validate.
bForceWrite - TRUE = always write the properties to the cluster database.
FALSE = only write the properties if they changed.
pOutParams - Parameter block in which to return the data. If specified,
parameters will only be written if they are different between
the input data and the parameter block, unless bForceWrite == TRUE.
Return Value:
ERROR_SUCCESS if successful.
ERROR_INVALID_DATA - The format of the data is invalid for a property
list value.
A Win32 Error on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
BOOL bZeroLengthData;
LPWSTR UNALIGNED * ppszValue;
DWORD dwType;
// Loop to avoid goto's.
do
{
bZeroLengthData = ( pInStringValue->cbLength == 0 );
//
// Validate the property data if not zero length.
//
if ( ! bZeroLengthData ) {
//
// Verify the length of the value.
//
if ( pInStringValue->cbLength != (lstrlenW( pInStringValue->sz ) + 1) * sizeof(WCHAR) ) {
ClRtlDbgPrint( LOG_UNUSUAL, "ClRtlpSetStringProperty: Property '%1!ls!' length %2!d! doesn't match zero-term. length.\n", rrvnModifiedNames.PszName(), pInStringValue->cbLength );
status = ERROR_INVALID_DATA;
break;
} // if: string length doesn't match length in property
} // if: non-zero length data
ppszValue = (LPWSTR UNALIGNED *) &pOutParams[pPropertyItem->Offset];
//
// If the data changed, write it and save it.
// Do this even if only the case of the data changed.
//
if ( (pOutParams == NULL)
|| (*ppszValue == NULL)
|| bZeroLengthData
|| bForceWrite
|| (lstrcmpW( *ppszValue, pInStringValue->sz ) != 0) ) {
//
// Write the value to the cluster database.
// If the data length is zero, delete the value.
//
if ( hkey != NULL ) {
if ( bZeroLengthData ) {
if ( hXsaction ) {
status = (*pClusterRegApis->pfnLocalDeleteValue)(
hXsaction,
hkey,
rrvnModifiedNames.PszName() );
} else {
status = (*pClusterRegApis->pfnDeleteValue)(
hkey,
rrvnModifiedNames.PszName() );
} // if/else: doing/not doing a transaction
//
// If the property doesn't exist in the
// cluster database, fix the status.
//
if ( status == ERROR_FILE_NOT_FOUND ) {
status = ERROR_SUCCESS;
} // if: property already doesn't exist
} else {
if ( pPropertyItem->Format == CLUSPROP_FORMAT_EXPAND_SZ ) {
dwType = REG_EXPAND_SZ;
} else {
dwType = REG_SZ;
} // if/else: property format is EXPAND_SZ
if ( hXsaction ) {
status = (*pClusterRegApis->pfnLocalSetValue)(
hXsaction,
hkey,
rrvnModifiedNames.PszName(),
dwType,
(CONST BYTE *) &pInStringValue->sz,
pInStringValue->cbLength );
} else {
status = (*pClusterRegApis->pfnSetValue)(
hkey,
rrvnModifiedNames.PszName(),
dwType,
(CONST BYTE *) &pInStringValue->sz,
pInStringValue->cbLength );
} // if/else: doing/not doing a transaction
} // if/else: zero length data
} // if: writing data
//
// Save the value to the output Parameter block.
// If the data length is zero, set to the default.
//
if ( (status == ERROR_SUCCESS)
&& (pOutParams != NULL) ) {
if ( *ppszValue != NULL ) {
LocalFree( *ppszValue );
} // if: previous value in parameter block
if ( bZeroLengthData ) {
// If a default is specified, copy it.
if ( pPropertyItem->lpDefault != NULL ) {
*ppszValue = (LPWSTR) LocalAlloc(
LMEM_FIXED,
(lstrlenW( (LPCWSTR) pPropertyItem->lpDefault ) + 1) * sizeof(WCHAR)
);
if ( *ppszValue == NULL ) {
status = GetLastError();
ClRtlDbgPrint(
LOG_CRITICAL,
"ClRtlpSetStringProperty: error allocating memory for default "
"SZ value '%1!ls!' in parameter block for property '%2!ls!'.\n",
pPropertyItem->lpDefault,
rrvnModifiedNames.PszName() );
break;
} // if: error allocating memory
lstrcpyW( *ppszValue, (LPCWSTR) pPropertyItem->lpDefault );
} else {
*ppszValue = NULL;
} // if/else: default value specified
} else {
*ppszValue = (LPWSTR) LocalAlloc( LMEM_FIXED, pInStringValue->cbLength );
if ( *ppszValue == NULL ) {
status = GetLastError();
ClRtlDbgPrint(
LOG_CRITICAL,
"ClRtlpSetStringProperty: error allocating memory for SZ "
"value '%1!ls!' in parameter block for property '%2!ls!'.\n",
pInStringValue->cbLength,
rrvnModifiedNames.PszName() );
break;
} // if: error allocating memory
lstrcpyW( *ppszValue, pInStringValue->sz );
} // if/else: zero length data
} // if: data written successfully and parameter block specified
} // if: value changed or zero-length value
} while ( 0 );
return status;
} // ClRtlpSetStringProperty
static
DWORD
WINAPI
ClRtlpSetMultiStringProperty(
IN HANDLE hXsaction,
IN PVOID hkey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyItem,
IN const CRegistryValueName & rrvnModifiedNames,
IN PCLUSPROP_MULTI_SZ pInMultiStringValue,
IN BOOL bForceWrite,
IN OUT OPTIONAL LPBYTE pOutParams
)
/*++
Routine Description:
Validate a MULTI_SZ property, write it to the cluster database (or delete
it if it is zero length), and save it in the specified parameter block.
Arguments:
hXsaction - Transaction handle.
hkey - The opened cluster database key where the property is to be written.
If not specified, the property will only be validated.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pPropertyItem - The property from a property table to set/validate.
rrvnModifiedNames - If the name of the property contains a backslash
this object contains the modified name and keyname.
pInMultiStringValue - The value from the property list to set/validate.
bForceWrite - TRUE = always write the properties to the cluster database.
FALSE = only write the properties if they changed.
pOutParams - Parameter block in which to return the data. If specified,
parameters will only be written if they are different between
the input data and the parameter block, unless bForceWrite == TRUE.
Return Value:
ERROR_SUCCESS if successful.
ERROR_INVALID_DATA - The format of the data is invalid for a property
list value.
A Win32 Error on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
BOOL bZeroLengthData;
LPWSTR UNALIGNED * ppszValue;
DWORD * pdwValue;
DWORD dwType;
// Loop to avoid goto's.
do
{
bZeroLengthData = ( pInMultiStringValue->cbLength == 0 );
ppszValue = (LPWSTR UNALIGNED *) &pOutParams[pPropertyItem->Offset];
pdwValue = (DWORD *) &pOutParams[pPropertyItem->Offset + sizeof(LPWSTR *)];
//
// If the data changed, write it and save it.
// Do this even if only the case of the data changed.
//
if ( (pOutParams == NULL)
|| (*ppszValue == NULL)
|| (*pdwValue != pInMultiStringValue->cbLength)
|| bZeroLengthData
|| bForceWrite
|| (memcmp( *ppszValue, pInMultiStringValue->sz, *pdwValue ) != 0) ) {
//
// Write the value to the cluster database.
// If the data length is zero, delete the value.
//
if ( hkey != NULL ) {
if ( bZeroLengthData ) {
if ( hXsaction ) {
status = (*pClusterRegApis->pfnLocalDeleteValue)(
hXsaction,
hkey,
rrvnModifiedNames.PszName() );
} else {
status = (*pClusterRegApis->pfnDeleteValue)(
hkey,
rrvnModifiedNames.PszName() );
} // if/else: doing/not doing a transaction
//
// If the property doesn't exist in the
// cluster database, fix the status.
//
if ( status == ERROR_FILE_NOT_FOUND ) {
status = ERROR_SUCCESS;
} // if: property already doesn't exist
} else {
if ( pPropertyItem->Format == CLUSPROP_FORMAT_MULTI_SZ ) {
dwType = REG_MULTI_SZ;
} else {
dwType = REG_SZ;
} // if/else: property format is EXPAND_SZ
if ( hXsaction ) {
status = (*pClusterRegApis->pfnLocalSetValue)(
hXsaction,
hkey,
rrvnModifiedNames.PszName(),
dwType,
(CONST BYTE *) &pInMultiStringValue->sz,
pInMultiStringValue->cbLength );
} else {
status = (*pClusterRegApis->pfnSetValue)(
hkey,
rrvnModifiedNames.PszName(),
dwType,
(CONST BYTE *) &pInMultiStringValue->sz,
pInMultiStringValue->cbLength );
} // if/else: doing/not doing a transaction
} // if/else: zero length data
} // if: writing data
//
// Save the value to the output Parameter block.
// If the data length is zero, set to the default.
//
if ( (status == ERROR_SUCCESS)
&& (pOutParams != NULL) ) {
if ( *ppszValue != NULL ) {
LocalFree( *ppszValue );
} // if: previous value in parameter block
if ( bZeroLengthData ) {
// If a default is specified, copy it.
if ( pPropertyItem->lpDefault != NULL ) {
*ppszValue = (LPWSTR) LocalAlloc( LMEM_FIXED, pPropertyItem->Minimum );
if ( *ppszValue == NULL ) {
status = GetLastError();
*pdwValue = 0;
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetMultiStringProperty: error allocating memory for default MULTI_SZ value in parameter block for property '%1!ls!'.\n", rrvnModifiedNames.PszName() );
break;
} // if: error allocating memory
CopyMemory( *ppszValue, pPropertyItem->lpDefault, pPropertyItem->Minimum );
*pdwValue = pPropertyItem->Minimum;
} else {
*ppszValue = NULL;
*pdwValue = 0;
} // if/else: default value specified
} else {
*ppszValue = (LPWSTR) LocalAlloc( LMEM_FIXED, pInMultiStringValue->cbLength );
if ( *ppszValue == NULL ) {
status = GetLastError();
*pdwValue = 0;
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetMultiStringProperty: error allocating memory for MULTI_SZ value in parameter block for property '%1!ls!'.\n", rrvnModifiedNames.PszName() );
break;
} // if: error allocating memory
CopyMemory( *ppszValue, pInMultiStringValue->sz, pInMultiStringValue->cbLength );
*pdwValue = pInMultiStringValue->cbLength;
} // if/else: zero length data
} // if: data written successfully and parameter block specified
} // if: value changed or zero-length value
} while ( 0 );
return status;
} // ClRtlpSetMultiStringProperty
static
DWORD
WINAPI
ClRtlpSetBinaryProperty(
IN HANDLE hXsaction,
IN PVOID hkey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyItem,
IN const CRegistryValueName & rrvnModifiedNames,
IN PCLUSPROP_BINARY pInBinaryValue,
IN BOOL bForceWrite,
IN OUT OPTIONAL LPBYTE pOutParams
)
/*++
Routine Description:
Validate a BINARY property, write it to the cluster database (or delete
it if it is zero length), and save it in the specified parameter block.
Arguments:
hXsaction - Transaction handle.
hkey - The opened cluster database key where the property is to be written.
If not specified, the property will only be validated.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pPropertyItem - The property from a property table to set/validate.
rrvnModifiedNames - If the name of the property contains a backslash
this object contains the modified name and keyname.
pInBinaryValue - The value from the property list to set/validate.
bForceWrite - TRUE = always write the properties to the cluster database.
FALSE = only write the properties if they changed.
pOutParams - Parameter block in which to return the data. If specified,
parameters will only be written if they are different between
the input data and the parameter block, unless bForceWrite == TRUE.
Return Value:
ERROR_SUCCESS if successful.
ERROR_INVALID_DATA - The format of the data is invalid for a property
list value.
A Win32 Error on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
BOOL bZeroLengthData;
PBYTE UNALIGNED * ppbValue;
DWORD * pdwValue;
// Loop to avoid goto's.
do
{
bZeroLengthData = ( pInBinaryValue->cbLength == 0 );
ppbValue = (PBYTE UNALIGNED *) &pOutParams[pPropertyItem->Offset];
pdwValue = (DWORD *) &pOutParams[pPropertyItem->Offset + sizeof(PBYTE *)];
//
// If the data changed, write it and save it.
// Do this even if only the case of the data changed.
//
if ( (pOutParams == NULL)
|| (*ppbValue == NULL)
|| (*pdwValue != pInBinaryValue->cbLength)
|| bZeroLengthData
|| bForceWrite
|| (memcmp( *ppbValue, pInBinaryValue->rgb, *pdwValue ) != 0) ) {
//
// Write the value to the cluster database.
// If the data length is zero, delete the value.
//
if ( hkey != NULL ) {
if ( bZeroLengthData ) {
if ( hXsaction ) {
status = (*pClusterRegApis->pfnLocalDeleteValue)(
hXsaction,
hkey,
rrvnModifiedNames.PszName() );
} else {
status = (*pClusterRegApis->pfnDeleteValue)(
hkey,
rrvnModifiedNames.PszName() );
} // if/else: doing/not doing a transaction
//
// If the property doesn't exist in the
// cluster database, fix the status.
//
if ( status == ERROR_FILE_NOT_FOUND ) {
status = ERROR_SUCCESS;
} // if: property already doesn't exist
} else {
if ( hXsaction ) {
status = (*pClusterRegApis->pfnLocalSetValue)(
hXsaction,
hkey,
rrvnModifiedNames.PszName(),
REG_BINARY,
(CONST BYTE *) &pInBinaryValue->rgb,
pInBinaryValue->cbLength );
} else {
status = (*pClusterRegApis->pfnSetValue)(
hkey,
rrvnModifiedNames.PszName(),
REG_BINARY,
(CONST BYTE *) &pInBinaryValue->rgb,
pInBinaryValue->cbLength );
} // if/else: doing/not doing a transaction
} // if/else: zero length data
} // if: writing data
//
// Save the value to the output Parameter block.
// If the data length is zero, set to the default.
//
if ( (status == ERROR_SUCCESS)
&& (pOutParams != NULL) ) {
if ( *ppbValue != NULL ) {
LocalFree( *ppbValue );
} // if: previous value in parameter block
if ( bZeroLengthData ) {
// If a default is specified, copy it.
if ( pPropertyItem->lpDefault != NULL ) {
*ppbValue = (LPBYTE) LocalAlloc( LMEM_FIXED, pPropertyItem->Minimum );
if ( *ppbValue == NULL ) {
status = GetLastError();
*pdwValue = 0;
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetBinaryProperty: error allocating memory for default BINARY value in parameter block for property '%1!ls!'.\n", rrvnModifiedNames.PszName() );
break;
} // if: error allocating memory
CopyMemory( *ppbValue, pPropertyItem->lpDefault, pPropertyItem->Minimum );
*pdwValue = pPropertyItem->Minimum;
} else {
*ppbValue = NULL;
*pdwValue = 0;
} // if/else: default value specified
} else {
*ppbValue = (LPBYTE) LocalAlloc( LMEM_FIXED, pInBinaryValue->cbLength );
if ( *ppbValue == NULL ) {
status = GetLastError();
*pdwValue = 0;
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetBinaryProperty: error allocating memory for BINARY value in parameter block for property '%1!ls!'.\n", rrvnModifiedNames.PszName() );
break;
} // if: error allocating memory
CopyMemory( *ppbValue, pInBinaryValue->rgb, pInBinaryValue->cbLength );
*pdwValue = pInBinaryValue->cbLength;
} // if/else: zero length data
} // if: data written successfully and parameter block specified
} // if: value changed or zero-length value
} while ( 0 );
return status;
} // ClRtlpSetBinaryProperty
DWORD
WINAPI
ClRtlpSetNonPropertyTable(
IN HANDLE hXsaction,
IN PVOID hkeyClusterKey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyTable,
IN PVOID Reserved,
IN const PVOID pInPropertyList,
IN DWORD cbInPropertyListSize
)
/*++
Routine Description:
Set items that are not in the property table list.
Arguments:
hXsaction - Local Transaction handle.
hkeyClusterKey - The opened registry key for this object's parameters.
If not specified, the property list will only be validated.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pPropertyTable - Pointer to the property table to process.
pInPropertyList - The input buffer.
cbInPropertyListSize - The input buffer size.
Return Value:
ERROR_SUCCESS if successful.
A Win32 Error on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
PRESUTIL_PROPERTY_ITEM propertyItem;
DWORD inBufferSize;
DWORD itemCount;
DWORD dataSize;
CLUSPROP_BUFFER_HELPER buf;
PCLUSPROP_PROPERTY_NAME pName;
BOOL bZeroLengthData;
CRegistryValueName rvn;
//
// If hKeyClusterKey is present then 'normal' functions must be present.
//
if ( ( (hkeyClusterKey != NULL) &&
((pClusterRegApis->pfnSetValue == NULL) ||
(pClusterRegApis->pfnCreateKey == NULL) ||
(pClusterRegApis->pfnOpenKey == NULL) ||
(pClusterRegApis->pfnCloseKey == NULL)
)) ||
( pPropertyTable == NULL ) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetNonPropertyTable: pClusterRegApis->pfnSetValue, pfnCreateKey, pfnOpenKey, or pfnCloseKey == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
//
// If hKeyClusterKey and hXsaction are present
// then 'local' functions must be present.
//
if ( ((hkeyClusterKey != NULL) &&
(hXsaction != NULL )) &&
((pClusterRegApis->pfnLocalCreateKey == NULL) ||
(pClusterRegApis->pfnLocalDeleteValue == NULL)
) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetNonPropertyTable: pClusterRegApis->pfnpfnLocalCreateKey or pfnLocalDeleteValue == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
if ( pInPropertyList == NULL ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetNonPropertyTable: pInPropertyList == NULL. Returning ERROR_INVALID_DATA\n" );
return(ERROR_INVALID_DATA);
}
buf.pb = (LPBYTE) pInPropertyList;
inBufferSize = cbInPropertyListSize;
//
// Get the number of items in this list
//
if ( inBufferSize < sizeof(DWORD) ) {
return(ERROR_INSUFFICIENT_BUFFER);
}
itemCount = buf.pList->nPropertyCount;
buf.pdw++;
//
// Parse the rest of the items in the buffer.
//
while ( itemCount-- ) {
//
// Verify that the buffer is big enough to contain the
// property name and a value.
//
pName = buf.pName;
if ( inBufferSize < sizeof(*pName) ) {
return(ERROR_INSUFFICIENT_BUFFER);
}
dataSize = sizeof(*pName) + ALIGN_CLUSPROP( pName->cbLength );
if ( inBufferSize < dataSize + sizeof(CLUSPROP_VALUE) ) {
return(ERROR_INSUFFICIENT_BUFFER);
}
//
// Verify that the syntax of the property name is correct.
//
if ( pName->Syntax.dw != CLUSPROP_SYNTAX_NAME ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetNonPropertyTable: syntax %1!d! not a name syntax.\n", pName->Syntax.dw );
return(ERROR_INVALID_PARAMETER);
}
//
// Verify that the length is correct for the string.
//
if ( pName->cbLength != (lstrlenW( pName->sz ) + 1) * sizeof(WCHAR) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetNonPropertyTable: name is not a valid C string.\n" );
return(ERROR_INVALID_DATA);
}
//
// Move the buffer pointer to the property value.
//
buf.pb += dataSize;
inBufferSize -= dataSize;
//
// Find the property name in the list of known properties.
//
propertyItem = pPropertyTable;
while ( propertyItem->Name != NULL ) {
if ( lstrcmpiW( pName->sz, propertyItem->Name ) == 0 ) {
//
// Verify that the buffer is big enough to contain the value.
//
do {
dataSize = sizeof(*buf.pValue)
+ ALIGN_CLUSPROP( buf.pValue->cbLength );
if ( inBufferSize < dataSize ) {
return(ERROR_INSUFFICIENT_BUFFER);
}
//
// Skip this value.
//
buf.pb += dataSize;
inBufferSize -= dataSize;
} while ( buf.pSyntax->dw != CLUSPROP_SYNTAX_ENDMARK );
//
// Skip the endmark.
//
dataSize = sizeof( CLUSPROP_SYNTAX );
if ( inBufferSize < dataSize ) {
return(ERROR_INSUFFICIENT_BUFFER);
}
buf.pb += dataSize;
inBufferSize -= dataSize;
break;
} else {
propertyItem++;
}
}
//
// If no property name was found, just save this item.
//
if ( propertyItem->Name == NULL) {
//
// Verify that the buffer is big enough to contain the value.
//
dataSize = sizeof(*buf.pValue)
+ ALIGN_CLUSPROP( buf.pValue->cbLength );
if ( inBufferSize < dataSize + sizeof( CLUSPROP_SYNTAX ) ) {
return(ERROR_INSUFFICIENT_BUFFER);
}
//
// Verify that the syntax type is LIST_VALUE.
//
if ( buf.pSyntax->wType != CLUSPROP_TYPE_LIST_VALUE ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetNonPropertyTable: Property '%1!ls!' type CLUSPROP_TYPE_LIST_VALUE (%2!d!) expected, was %3!d!.\n", pName->sz, CLUSPROP_TYPE_LIST_VALUE, buf.pSyntax->wType );
return(ERROR_INVALID_PARAMETER);
}
//
// If the value is not specified, delete the property.
//
bZeroLengthData = ( buf.pValue->cbLength == 0 );
if ( bZeroLengthData ) {
if ( hkeyClusterKey != NULL ) {
PVOID key = NULL;
DWORD disposition;
//
// Use the wrapper class CRegistryValueName to parse value name to see if it
// contains a backslash.
//
status = rvn.ScInit( pName->sz, NULL );
if ( status != ERROR_SUCCESS ) {
break;
}
//
// If the value resides at a different location, open the key.
//
if ( rvn.PszKeyName() != NULL ) {
status = (*pClusterRegApis->pfnOpenKey)( hkeyClusterKey,
rvn.PszKeyName(),
KEY_ALL_ACCESS,
&key);
if ( status != ERROR_SUCCESS ) {
break;
}
} else {
key = hkeyClusterKey;
}
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalDeleteValue)(
hXsaction,
key,
rvn.PszName() );
}
else {
status = (*pClusterRegApis->pfnDeleteValue)(
key,
rvn.PszName() );
}
//
// If the property doesn't exist in the
// cluster database, fix the status.
//
if ( status == ERROR_FILE_NOT_FOUND ) {
status = ERROR_SUCCESS;
} // if: property already doesn't exist
//
// Close the key if we opened it.
//
if ( (rvn.PszKeyName() != NULL) &&
(key != NULL) ) {
(*pClusterRegApis->pfnCloseKey)( key );
}
} // if: key specified
} else {
PVOID key = NULL;
DWORD disposition;
if ( hkeyClusterKey != NULL ) {
//
// Use the wrapper class CRegistryValueName to parse value name to see if it
// contains a backslash.
//
status = rvn.ScInit( pName->sz, NULL );
if ( status != ERROR_SUCCESS ) {
break;
}
//
// If the value resides at a different location, open the key.
//
if ( rvn.PszKeyName() != NULL ) {
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalCreateKey)( hXsaction,
hkeyClusterKey,
rvn.PszKeyName(),
0,
KEY_ALL_ACCESS,
NULL,
&key,
&disposition);
}
else {
status = (*pClusterRegApis->pfnCreateKey)( hkeyClusterKey,
rvn.PszKeyName(),
0,
KEY_ALL_ACCESS,
NULL,
&key,
&disposition);
}
if ( status != ERROR_SUCCESS ) {
break;
}
} else {
key = hkeyClusterKey;
}
}
switch ( buf.pSyntax->wFormat ) {
case CLUSPROP_FORMAT_DWORD:
//
// Verify the length of the value.
//
if ( buf.pDwordValue->cbLength != sizeof(DWORD) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetNonPropertyTable: Property '%1!ls!' length %2!d! not DWORD length.\n", pName->sz, buf.pDwordValue->cbLength );
status = ERROR_INVALID_DATA;
break;
}
//
// Write the value to the cluster database.
//
if ( key != NULL ) {
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalSetValue)( hXsaction,
key,
rvn.PszName(),
REG_DWORD,
(CONST BYTE*)&buf.pDwordValue->dw,
sizeof(DWORD) );
}
else {
status = (*pClusterRegApis->pfnSetValue)( key,
rvn.PszName(),
REG_DWORD,
(CONST BYTE*)&buf.pDwordValue->dw,
sizeof(DWORD) );
}
}
break;
case CLUSPROP_FORMAT_LONG:
//
// Verify the length of the value.
//
if ( buf.pLongValue->cbLength != sizeof(LONG) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetNonPropertyTable: Property '%1!ls!' length %2!d! not LONG length.\n", pName->sz, buf.pLongValue->cbLength );
status = ERROR_INVALID_DATA;
break;
}
//
// Write the value to the cluster database.
//
if ( key != NULL ) {
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalSetValue)( hXsaction,
key,
rvn.PszName(),
REG_DWORD,
(CONST BYTE*)&buf.pLongValue->l,
sizeof(LONG) );
}
else {
status = (*pClusterRegApis->pfnSetValue)( key,
rvn.PszName(),
REG_DWORD,
(CONST BYTE*)&buf.pLongValue->l,
sizeof(LONG) );
}
}
break;
case CLUSPROP_FORMAT_ULARGE_INTEGER:
case CLUSPROP_FORMAT_LARGE_INTEGER:
//
// Write the value to the cluster database.
//
if ( key != NULL ) {
if ( hXsaction ) {
status = (*pClusterRegApis->pfnLocalSetValue)(
hXsaction,
key,
rvn.PszName(),
REG_QWORD,
(CONST BYTE*)&buf.pULargeIntegerValue->li.QuadPart,
sizeof(ULARGE_INTEGER));
} else {
status = (*pClusterRegApis->pfnSetValue)(
key,
rvn.PszName(),
REG_QWORD,
(CONST BYTE*)&buf.pULargeIntegerValue->li.QuadPart,
sizeof(ULARGE_INTEGER));
}
}
break;
case CLUSPROP_FORMAT_SZ:
//
// Verify the length of the value.
//
if ( buf.pStringValue->cbLength != (lstrlenW( buf.pStringValue->sz ) + 1) * sizeof(WCHAR) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetNonPropertyTable: Property '%1!ls!' length %2!d! doesn't match null-term. length.\n", pName->sz, buf.pStringValue->cbLength );
status = ERROR_INVALID_DATA;
break;
}
//
// Write the value to the cluster database.
//
if ( key != NULL ) {
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalSetValue)( hXsaction,
key,
rvn.PszName(),
REG_SZ,
(CONST BYTE*)buf.pStringValue->sz,
buf.pStringValue->cbLength );
}
else {
status = (*pClusterRegApis->pfnSetValue)( key,
rvn.PszName(),
REG_SZ,
(CONST BYTE*)buf.pStringValue->sz,
buf.pStringValue->cbLength );
}
}
break;
case CLUSPROP_FORMAT_EXPAND_SZ:
//
// Verify the length of the value.
//
if ( buf.pStringValue->cbLength != (lstrlenW( buf.pStringValue->sz ) + 1) * sizeof(WCHAR) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetNonPropertyTable: Property '%1!ls!' length %2!d! doesn't match null-term. length.\n", pName->sz, buf.pStringValue->cbLength );
status = ERROR_INVALID_DATA;
break;
}
//
// Write the value to the cluster database.
//
if ( key != NULL ) {
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalSetValue)( hXsaction,
key,
rvn.PszName(),
REG_EXPAND_SZ,
(CONST BYTE*)buf.pStringValue->sz,
buf.pStringValue->cbLength );
}
else {
status = (*pClusterRegApis->pfnSetValue)( key,
rvn.PszName(),
REG_EXPAND_SZ,
(CONST BYTE*)buf.pStringValue->sz,
buf.pStringValue->cbLength );
}
}
break;
case CLUSPROP_FORMAT_MULTI_SZ:
//
// Write the value to the cluster database.
//
if ( key != NULL ) {
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalSetValue)( hXsaction,
key,
rvn.PszName(),
REG_MULTI_SZ,
(CONST BYTE*)buf.pStringValue->sz,
buf.pStringValue->cbLength );
}
else {
status = (*pClusterRegApis->pfnSetValue)( key,
rvn.PszName(),
REG_MULTI_SZ,
(CONST BYTE*)buf.pStringValue->sz,
buf.pStringValue->cbLength );
}
}
break;
case CLUSPROP_FORMAT_BINARY:
//
// Write the value to the cluster database.
//
if ( key != NULL ) {
if ( hXsaction ) {
status = (*pClusterRegApis->pfnLocalSetValue)( hXsaction,
key,
rvn.PszName(),
REG_BINARY,
(CONST BYTE*)buf.pBinaryValue->rgb,
buf.pStringValue->cbLength );
} else {
status = (*pClusterRegApis->pfnSetValue)( key,
rvn.PszName(),
REG_BINARY,
(CONST BYTE*)buf.pBinaryValue->rgb,
buf.pStringValue->cbLength );
}
}
break;
default:
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetNonPropertyTable: Property '%1!ls!' unknown format %2!d! specified.\n", pName->sz, buf.pSyntax->wFormat );
status = ERROR_INVALID_PARAMETER;
break;
} // switch
//
// Close the key if we opened it.
//
if ( (rvn.PszKeyName() != NULL) &&
(key != NULL) ) {
(*pClusterRegApis->pfnCloseKey)( key );
}
} // if/else: zero length data
//
// Move the buffer past the value.
//
do {
dataSize = sizeof(*buf.pValue)
+ ALIGN_CLUSPROP( buf.pValue->cbLength );
buf.pb += dataSize;
inBufferSize -= dataSize;
} while ( buf.pSyntax->dw != CLUSPROP_SYNTAX_ENDMARK );
dataSize = sizeof( CLUSPROP_SYNTAX );
buf.pb += dataSize;
inBufferSize -= dataSize;
}
if ( status != ERROR_SUCCESS ) {
break;
}
}
return(status);
} // ClRtlpSetNonPropertyTable
DWORD
WINAPI
ClRtlSetPropertyParameterBlock(
IN HANDLE hXsaction,
IN PVOID hkeyClusterKey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PRESUTIL_PROPERTY_ITEM pPropertyTable,
IN PVOID Reserved,
IN const LPBYTE pInParams,
IN const PVOID pInPropertyList,
IN DWORD cbInPropertyListSize,
IN BOOL bForceWrite,
IN OUT OPTIONAL LPBYTE pOutParams
)
/*++
Routine Description:
Arguments:
hXsaction - Transaction key used when called from the cluster service.
hkeyClusterKey - The opened registry key for this object's parameters.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pPropertyTable - Pointer to the property table to process.
pInParams - Parameter block to set.
pInPropertyList - Full property list.
cbInPropertyListSize - Size of the input full property list.
bForceWrite - TRUE = always write the properties to the cluster database.
FALSE = only write the properties if they changed.
pOutParams - Parameters block to copy pInParams to. If specified,
parameters will only be written if they are different between
the two parameter blocks.
Return Value:
ERROR_SUCCESS if successful.
A Win32 Error on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
PRESUTIL_PROPERTY_ITEM propertyItem;
DWORD itemCount;
DWORD dataSize;
PVOID key;
LPWSTR UNALIGNED * ppszInValue;
LPWSTR UNALIGNED * ppszOutValue;
PBYTE UNALIGNED * ppbInValue;
PBYTE UNALIGNED * ppbOutValue;
DWORD * pdwInValue;
DWORD * pdwOutValue;
ULARGE_INTEGER * pullInValue;
ULARGE_INTEGER * pullOutValue;
CRegistryValueName rvn;
//
// If hKeyClusterKey is present then 'normal' functions must be present.
//
if ( (hkeyClusterKey == NULL) ||
(pClusterRegApis->pfnCreateKey == NULL) ||
(pClusterRegApis->pfnSetValue == NULL) ||
(pClusterRegApis->pfnCloseKey == NULL) ||
(pPropertyTable == NULL) ||
(pInParams == NULL) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlSetPropertyParameterBlock: hkeyClusterKey, pClusterRegApis->pfnCreateKey, pfnSetValue, or pfnCloseKey == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
//
// If hXsaction is present then 'local' functions must be present.
//
if ( (hXsaction != NULL ) &&
((pClusterRegApis->pfnLocalCreateKey == NULL) ||
(pClusterRegApis->pfnLocalSetValue == NULL) ) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlSetPropertyParameterBlock: pClusterRegApis->pfnLocalCreateKey or pfnLocalDeleteValue == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
//
// Parse the property table.
//
propertyItem = pPropertyTable;
while ( propertyItem->Name != NULL ) {
//
// Make sure we are allowed to set this item.
//
if ( propertyItem->Flags & RESUTIL_PROPITEM_READ_ONLY ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlSetPropertyParameterBlock: Property '%1!ls!' is non-writable.\n", propertyItem->Name );
return(ERROR_INVALID_PARAMETER);
}
//
// Use the wrapper class CRegistryValueName to parse value name to see if it
// contains a backslash.
//
status = rvn.ScInit( propertyItem->Name, propertyItem->KeyName );
if ( status != ERROR_SUCCESS ) {
break;
}
//
// If the value resides at a different location, create the key.
//
if ( rvn.PszKeyName() != NULL ) {
DWORD disposition;
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalCreateKey)( hXsaction,
hkeyClusterKey,
rvn.PszKeyName(),
0,
KEY_ALL_ACCESS,
NULL,
&key,
&disposition );
} else {
status = (*pClusterRegApis->pfnCreateKey)( hkeyClusterKey,
rvn.PszKeyName(),
0,
KEY_ALL_ACCESS,
NULL,
&key,
&disposition );
}
if ( status != ERROR_SUCCESS ) {
return(status);
}
} else {
key = hkeyClusterKey;
}
switch ( propertyItem->Format ) {
case CLUSPROP_FORMAT_DWORD:
case CLUSPROP_FORMAT_LONG:
pdwInValue = (DWORD *) &pInParams[propertyItem->Offset];
pdwOutValue = (DWORD *) &pOutParams[propertyItem->Offset];
//
// Write the value to the cluster database.
//
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalSetValue)( hXsaction,
key,
rvn.PszName(),
REG_DWORD,
(CONST BYTE*)pdwInValue,
sizeof(DWORD) );
} else {
status = (*pClusterRegApis->pfnSetValue)( key,
rvn.PszName(),
REG_DWORD,
(CONST BYTE*)pdwInValue,
sizeof(DWORD) );
}
//
// Save the value to the output Parameter block.
//
if ( (status == ERROR_SUCCESS) &&
(pOutParams != NULL) ) {
*pdwOutValue = *pdwInValue;
}
break;
case CLUSPROP_FORMAT_ULARGE_INTEGER:
case CLUSPROP_FORMAT_LARGE_INTEGER:
pullInValue = (ULARGE_INTEGER *) &pInParams[propertyItem->Offset];
pullOutValue = (ULARGE_INTEGER *) &pOutParams[propertyItem->Offset];
//
// Write the value to the cluster database.
//
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalSetValue)( hXsaction,
key,
rvn.PszName(),
REG_QWORD,
(CONST BYTE*)pullInValue,
sizeof(ULARGE_INTEGER) );
} else {
status = (*pClusterRegApis->pfnSetValue)( key,
rvn.PszName(),
REG_QWORD,
(CONST BYTE*)pullInValue,
sizeof(ULARGE_INTEGER) );
}
//
// Save the value to the output Parameter block.
//
if ( (status == ERROR_SUCCESS) && (pOutParams != NULL) ) {
pullOutValue->u = pullInValue->u;
}
break;
case CLUSPROP_FORMAT_SZ:
case CLUSPROP_FORMAT_EXPAND_SZ:
ppszInValue = (LPWSTR UNALIGNED *) &pInParams[propertyItem->Offset];
ppszOutValue = (LPWSTR UNALIGNED *) &pOutParams[propertyItem->Offset];
//
// If the data changed, write it and save it.
// Do this even if only the case of the data changed.
//
if ( bForceWrite ||
(pOutParams == NULL) ||
(*ppszOutValue == NULL) ||
(lstrcmpW( *ppszInValue, *ppszOutValue ) != 0) ) {
//
// Write the value to the cluster database.
//
if ( *ppszInValue != NULL ) {
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalSetValue)( hXsaction,
key,
rvn.PszName(),
(propertyItem->Format == CLUSPROP_FORMAT_EXPAND_SZ
? REG_EXPAND_SZ
: REG_SZ),
(CONST BYTE*)*ppszInValue,
(lstrlenW(*ppszInValue) + 1) * sizeof(WCHAR) );
} else {
status = (*pClusterRegApis->pfnSetValue)( key,
rvn.PszName(),
(propertyItem->Format == CLUSPROP_FORMAT_EXPAND_SZ
? REG_EXPAND_SZ
: REG_SZ),
(CONST BYTE*)*ppszInValue,
(lstrlenW(*ppszInValue) + 1) * sizeof(WCHAR) );
}
}
//
// Save the value to the output Parameter block.
//
if ( (status == ERROR_SUCCESS) &&
(pOutParams != NULL) ) {
if ( *ppszOutValue != NULL ) {
LocalFree( *ppszOutValue );
}
if ( *ppszInValue == NULL ) {
*ppszOutValue = NULL;
} else {
*ppszOutValue = (LPWSTR) LocalAlloc( LMEM_FIXED, (lstrlenW( *ppszInValue )+1) * sizeof(WCHAR) );
if ( *ppszOutValue == NULL ) {
status = GetLastError();
ClRtlDbgPrint(
LOG_CRITICAL,
"ClRtlSetPropertyParameterBlock: error allocating memory for "
"SZ value '%1!ls!' in parameter block for property '%2!ls!'.\n",
*ppszInValue,
propertyItem->Name );
break;
}
lstrcpyW( *ppszOutValue, *ppszInValue );
}
}
}
break;
case CLUSPROP_FORMAT_MULTI_SZ:
ppszInValue = (LPWSTR UNALIGNED *) &pInParams[propertyItem->Offset];
pdwInValue = (DWORD *) &pInParams[propertyItem->Offset+sizeof(LPWSTR*)];
ppszOutValue = (LPWSTR UNALIGNED *) &pOutParams[propertyItem->Offset];
pdwOutValue = (DWORD *) &pOutParams[propertyItem->Offset+sizeof(LPWSTR*)];
//
// If the data changed, write it and save it.
//
if ( bForceWrite ||
(pOutParams == NULL) ||
(*ppszOutValue == NULL) ||
(*pdwInValue != *pdwOutValue) ||
(memcmp( *ppszInValue, *ppszOutValue, *pdwInValue ) != 0) ) {
//
// Write the value to the cluster database.
//
if ( *ppszInValue != NULL ) {
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalSetValue)( hXsaction,
key,
rvn.PszName(),
REG_MULTI_SZ,
(CONST BYTE*)*ppszInValue,
*pdwInValue );
} else {
status = (*pClusterRegApis->pfnSetValue)( key,
rvn.PszName(),
REG_MULTI_SZ,
(CONST BYTE*)*ppszInValue,
*pdwInValue );
}
}
//
// Save the value to the output Parameter block.
//
if ( (status == ERROR_SUCCESS) &&
(pOutParams != NULL) ) {
if ( *ppszOutValue != NULL ) {
LocalFree( *ppszOutValue );
}
if ( *ppszInValue == NULL ) {
*ppszOutValue = NULL;
} else {
*ppszOutValue = (LPWSTR) LocalAlloc( LMEM_FIXED, *pdwInValue );
if ( *ppszOutValue == NULL ) {
status = GetLastError();
*pdwOutValue = 0;
ClRtlDbgPrint(
LOG_CRITICAL,
"ClRtlSetPropertyParameterBlock: error allocating memory for "
"MULTI_SZ value in parameter block for property '%1!ls!'.\n",
propertyItem->Name );
break;
}
CopyMemory( *ppszOutValue, *ppszInValue, *pdwInValue );
*pdwOutValue = *pdwInValue;
}
}
}
break;
case CLUSPROP_FORMAT_BINARY:
ppbInValue = (PBYTE UNALIGNED *) &pInParams[propertyItem->Offset];
pdwInValue = (DWORD *) &pInParams[propertyItem->Offset+sizeof(LPWSTR*)];
ppbOutValue = (PBYTE UNALIGNED *) &pOutParams[propertyItem->Offset];
pdwOutValue = (DWORD *) &pOutParams[propertyItem->Offset+sizeof(PBYTE*)];
//
// If the data changed, write it and save it.
//
if ( bForceWrite ||
(pOutParams == NULL) ||
(*ppbOutValue == NULL) ||
(*pdwInValue != *pdwOutValue) ||
(memcmp( *ppbInValue, *ppbOutValue, *pdwInValue ) != 0) ) {
//
// Write the value to the cluster database.
//
if ( *ppbInValue != NULL ) {
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalSetValue)( hXsaction,
key,
rvn.PszName(),
REG_BINARY,
(CONST BYTE*)*ppbInValue,
*pdwInValue );
}
else {
status = (*pClusterRegApis->pfnSetValue)( key,
rvn.PszName(),
REG_BINARY,
(CONST BYTE*)*ppbInValue,
*pdwInValue );
}
}
//
// Save the value to the output Parameter block.
//
if ( (status == ERROR_SUCCESS) &&
(pOutParams != NULL) ) {
if ( *ppbOutValue != NULL ) {
LocalFree( *ppbOutValue );
}
if ( *ppbInValue == NULL ) {
*ppbOutValue = NULL;
} else {
*ppbOutValue = (LPBYTE) LocalAlloc( LMEM_FIXED, *pdwInValue );
if ( *ppbOutValue == NULL ) {
status = GetLastError();
*pdwOutValue = 0;
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlSetPropertyParameterBlock: error allocating memory for BINARY value in parameter block for property '%1!ls!'.\n", propertyItem->Name );
break;
}
CopyMemory( *ppbOutValue, *ppbInValue, *pdwInValue );
*pdwOutValue = *pdwInValue;
}
}
}
break;
default:
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlSetPropertyParameterBlock: Property '%1!ls!' unknown format %2!d! specified.\n", propertyItem->Name, propertyItem->Format );
status = ERROR_INVALID_PARAMETER;
break;
}
//
// Close the key if we opened it.
//
if ( rvn.PszKeyName() != NULL ) {
(*pClusterRegApis->pfnCloseKey)( key );
}
//
// If an error occurred processing the property, cleanup and return.
//
if ( status != ERROR_SUCCESS ) {
return(status);
}
propertyItem++;
}
//
// Now find any parameters that are not represented in the property
// table. All of these extra properties will just be set without validation.
//
if ( (status == ERROR_SUCCESS) &&
(pInPropertyList != NULL) ) {
status = ClRtlpSetNonPropertyTable( hXsaction,
hkeyClusterKey,
pClusterRegApis,
pPropertyTable,
NULL,
pInPropertyList,
cbInPropertyListSize );
}
return(status);
} // ClRtlSetPropertyParameterBlock
DWORD
WINAPI
ClRtlpSetPrivatePropertyList(
IN HANDLE hXsaction,
IN PVOID hkeyClusterKey,
IN const PCLUSTER_REG_APIS pClusterRegApis,
IN const PVOID pInPropertyList,
IN DWORD cbInPropertyListSize
)
/*++
Routine Description:
Arguments:
hkeyClusterKey - The opened registry key for this resource's parameters.
If not specified, the property list will only be validated.
pClusterRegApis - Supplies a structure of function pointers for accessing
the cluster database.
pInPropertyList - The input buffer.
cbInPropertyListSize - The input buffer size.
Return Value:
ERROR_SUCCESS if successful.
A Win32 Error on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
DWORD inBufferSize;
DWORD itemCount;
DWORD dataSize;
DWORD valueSize;
CLUSPROP_BUFFER_HELPER bufSizeTest;
CLUSPROP_BUFFER_HELPER buf;
PCLUSPROP_PROPERTY_NAME pName;
BOOL bZeroLengthData;
CRegistryValueName rvn;
if ( (hkeyClusterKey != NULL) &&
( (pClusterRegApis->pfnSetValue == NULL) ||
(pClusterRegApis->pfnCreateKey == NULL) ||
(pClusterRegApis->pfnOpenKey == NULL) ||
(pClusterRegApis->pfnCloseKey == NULL) ||
(pClusterRegApis->pfnDeleteValue == NULL)
) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPrivatePropertyList: pClusterRegApis->pfnCreateKey, pfnOpenKey, pfnSetValue, pfnCloseKey, or pfnDeleteValue == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
if ( pInPropertyList == NULL ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPrivatePropertyList: pInPropertyList == NULL. Returning ERROR_INVALID_DATA\n" );
return(ERROR_INVALID_DATA);
}
//
// If hXsaction is present then 'local' functions must be present.
//
if ( (hXsaction != NULL ) &&
( (pClusterRegApis->pfnLocalCreateKey == NULL) ||
(pClusterRegApis->pfnLocalDeleteValue == NULL)
) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPrivatePropertyList: pClusterRegApis->pfnLocalCreateKey or pfnLocalDeleteValue == NULL. Returning ERROR_BAD_ARGUMENTS\n" );
return(ERROR_BAD_ARGUMENTS);
}
buf.pb = (LPBYTE) pInPropertyList;
inBufferSize = cbInPropertyListSize;
//
// Get the number of items in this list
//
if ( inBufferSize < sizeof(DWORD) ) {
return(ERROR_INSUFFICIENT_BUFFER);
}
itemCount = buf.pList->nPropertyCount;
buf.pdw++;
//
// Parse the rest of the items in the buffer.
//
while ( itemCount-- ) {
pName = buf.pName;
if ( inBufferSize < sizeof(*pName) ) {
return(ERROR_INSUFFICIENT_BUFFER);
}
dataSize = sizeof(*pName) + ALIGN_CLUSPROP( pName->cbLength );
if ( inBufferSize < dataSize + sizeof(CLUSPROP_VALUE) ) {
return(ERROR_INSUFFICIENT_BUFFER);
}
//
// Verify that the syntax of the property name is correct.
//
if ( pName->Syntax.dw != CLUSPROP_SYNTAX_NAME ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPrivatePropertyList: syntax %1!d! not a name syntax.\n", pName->Syntax.dw );
return(ERROR_INVALID_PARAMETER);
}
//
// Verify that the length is correct for the string.
//
if ( pName->cbLength != (lstrlenW( pName->sz ) + 1) * sizeof(WCHAR) ) {
ClRtlDbgPrint( LOG_CRITICAL, "SetPrivatePropertyList: name is not a valid C string.\n" );
return(ERROR_INVALID_DATA);
}
//
// Move the buffer pointer to the property value.
//
buf.pb += dataSize;
inBufferSize -= dataSize;
//
// Verify that the buffer is big enough to contain the value.
//
bufSizeTest.pb = buf.pb;
dataSize = 0;
do {
valueSize = sizeof( *bufSizeTest.pValue )
+ ALIGN_CLUSPROP( bufSizeTest.pValue->cbLength );
bufSizeTest.pb += valueSize;
dataSize += valueSize;
} while ( bufSizeTest.pSyntax->dw != CLUSPROP_SYNTAX_ENDMARK );
dataSize += sizeof( CLUSPROP_SYNTAX );
if ( inBufferSize < dataSize ) {
return(ERROR_INSUFFICIENT_BUFFER);
}
//
// Verify that the syntax type is SPECIAL.
//
if ( buf.pSyntax->wType != CLUSPROP_TYPE_LIST_VALUE ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPrivatePropertyList: Property '%1!ls!' type CLUSPROP_TYPE_LIST_VALUE (%2!d!) expected, was %3!d!.\n", pName->sz, CLUSPROP_TYPE_LIST_VALUE, buf.pSyntax->wType );
return(ERROR_INVALID_PARAMETER);
}
//
// If the value is not specified, delete the property.
//
bZeroLengthData = ( buf.pValue->cbLength == 0 );
if ( bZeroLengthData ) {
if ( hkeyClusterKey != NULL ) {
PVOID key = NULL;
//
// Use the wrapper class CRegistryValueName to parse value name to see if it
// contains a backslash.
//
status = rvn.ScInit( pName->sz, NULL );
if ( status != ERROR_SUCCESS ) {
break;
}
//
// If the value resides at a different location, open the key.
//
if ( rvn.PszKeyName() != NULL ) {
status = (*pClusterRegApis->pfnOpenKey)( hkeyClusterKey,
rvn.PszKeyName(),
KEY_ALL_ACCESS,
&key);
if ( status != ERROR_SUCCESS ) {
break;
}
} else {
key = hkeyClusterKey;
}
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalDeleteValue)(
hXsaction,
key,
rvn.PszName() );
}
else {
status = (*pClusterRegApis->pfnDeleteValue)(
key,
rvn.PszName() );
}
//
// If the property doesn't exist in the
// cluster database, fix the status.
//
if ( status == ERROR_FILE_NOT_FOUND ) {
status = ERROR_SUCCESS;
} // if: property already doesn't exist
//
// Close the key if we opened it.
//
if ( (rvn.PszKeyName() != NULL) &&
(key != NULL) ) {
(*pClusterRegApis->pfnCloseKey)( key );
}
} // if: key specified
} else {
PVOID key = NULL;
DWORD disposition;
if ( hkeyClusterKey != NULL ) {
//
// Use the wrapper class CRegistryValueName to parse value name to see if it
// contains a backslash.
//
status = rvn.ScInit( pName->sz, NULL );
if ( status != ERROR_SUCCESS ) {
break;
}
//
// If the value resides at a different location, open the key.
//
if ( rvn.PszKeyName() != NULL ) {
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalCreateKey)(
hXsaction,
hkeyClusterKey,
rvn.PszKeyName(),
0,
KEY_ALL_ACCESS,
NULL,
&key,
&disposition);
}
else {
status = (*pClusterRegApis->pfnCreateKey)( hkeyClusterKey,
rvn.PszKeyName(),
0,
KEY_ALL_ACCESS,
NULL,
&key,
&disposition);
}
if ( status != ERROR_SUCCESS ) {
break;
}
} else {
key = hkeyClusterKey;
}
}
//
// Parse the property and set it in the cluster database
//
switch ( buf.pSyntax->wFormat ) {
case CLUSPROP_FORMAT_DWORD:
case CLUSPROP_FORMAT_LONG:
//
// Verify the length of the value.
//
if ( buf.pDwordValue->cbLength != sizeof(DWORD) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPrivatePropertyList: Property '%1!ls!' length %2!d! not DWORD or LONG length.\n", pName->sz, buf.pDwordValue->cbLength );
status = ERROR_INVALID_DATA;
break;
}
//
// Write the value to the cluster database.
//
if ( key != NULL ) {
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalSetValue)( hXsaction,
key,
rvn.PszName(),
REG_DWORD,
(CONST BYTE*)&buf.pDwordValue->dw,
sizeof(DWORD) );
}
else {
status = (*pClusterRegApis->pfnSetValue)( key,
rvn.PszName(),
REG_DWORD,
(CONST BYTE*)&buf.pDwordValue->dw,
sizeof(DWORD));
}
}
break;
case CLUSPROP_FORMAT_ULARGE_INTEGER:
case CLUSPROP_FORMAT_LARGE_INTEGER:
//
// Verify the length of the value.
//
if ( buf.pULargeIntegerValue->cbLength != sizeof(ULARGE_INTEGER) ) {
ClRtlDbgPrint(LOG_CRITICAL,
"ClRtlpSetPrivatePropertyList: Property '%1!ls!' length "
"%2!d! not ULARGE_INTEGER or LARGE_INTEGER length.\n",
pName->sz,
buf.pULargeIntegerValue->cbLength );
status = ERROR_INVALID_DATA;
break;
}
//
// Write the value to the cluster database.
//
if ( key != NULL ) {
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalSetValue)(
hXsaction,
key,
rvn.PszName(),
REG_QWORD,
(CONST BYTE*)&buf.pULargeIntegerValue->li.QuadPart,
sizeof(ULARGE_INTEGER) );
}
else {
status = (*pClusterRegApis->pfnSetValue)(
key,
rvn.PszName(),
REG_QWORD,
(CONST BYTE*)&buf.pULargeIntegerValue->li.QuadPart,
sizeof(ULARGE_INTEGER));
}
}
break;
case CLUSPROP_FORMAT_SZ:
case CLUSPROP_FORMAT_EXPAND_SZ:
//
// Verify the length of the value.
//
if ( buf.pStringValue->cbLength != (lstrlenW( buf.pStringValue->sz ) + 1) * sizeof(WCHAR) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpSetPrivatePropertyList: Property '%1!ls!' length %2!d! doesn't match null-term. length.\n", pName->sz, buf.pStringValue->cbLength );
status = ERROR_INVALID_DATA;
break;
}
//
// Write the value to the cluster database.
//
if ( key != NULL ) {
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalSetValue)( hXsaction,
key,
rvn.PszName(),
(buf.pSyntax->wFormat == CLUSPROP_FORMAT_EXPAND_SZ
? REG_EXPAND_SZ
: REG_SZ),
(CONST BYTE*)buf.pStringValue->sz,
buf.pStringValue->cbLength);
}
else {
status = (*pClusterRegApis->pfnSetValue)( key,
rvn.PszName(),
(buf.pSyntax->wFormat == CLUSPROP_FORMAT_EXPAND_SZ
? REG_EXPAND_SZ
: REG_SZ),
(CONST BYTE*)buf.pStringValue->sz,
buf.pStringValue->cbLength);
}
}
break;
case CLUSPROP_FORMAT_MULTI_SZ:
//
// Write the value to the cluster database.
//
if ( key != NULL ) {
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalSetValue)( hXsaction,
key,
rvn.PszName(),
REG_MULTI_SZ,
(CONST BYTE*)buf.pStringValue->sz,
buf.pStringValue->cbLength );
}
else {
status = (*pClusterRegApis->pfnSetValue)( key,
rvn.PszName(),
REG_MULTI_SZ,
(CONST BYTE*)buf.pStringValue->sz,
buf.pStringValue->cbLength );
}
}
break;
case CLUSPROP_FORMAT_BINARY:
//
// Write the value to the cluster database.
//
if ( key != NULL ) {
if ( hXsaction != NULL ) {
status = (*pClusterRegApis->pfnLocalSetValue)( hXsaction,
key,
rvn.PszName(),
REG_BINARY,
(CONST BYTE*)buf.pBinaryValue->rgb,
buf.pBinaryValue->cbLength );
}
else {
status = (*pClusterRegApis->pfnSetValue)( key,
rvn.PszName(),
REG_BINARY,
(CONST BYTE*)buf.pBinaryValue->rgb,
buf.pBinaryValue->cbLength );
}
}
break;
default:
status = ERROR_INVALID_PARAMETER; // not tested
} // switch
//
// Close the key if we opened it.
//
if ( (rvn.PszKeyName() != NULL) &&
(key != NULL) ) {
(*pClusterRegApis->pfnCloseKey)( key );
}
} // if/else: zero length data
if ( status != ERROR_SUCCESS ) {
break;
}
//
// Move the buffer past the value.
//
buf.pb += dataSize;
inBufferSize -= dataSize;
}
return(status);
} // ClRtlpSetPrivatePropertyList
DWORD
WINAPI
ClRtlpFindSzProperty(
IN const PVOID pPropertyList,
IN DWORD cbPropertyListSize,
IN LPCWSTR pszPropertyName,
OUT LPWSTR * pszPropertyValue,
IN BOOL bReturnExpandedValue
)
/*++
Routine Description:
Finds the specified string property in the Property List buffer pointed at
by pPropertyList.
Arguments:
pPropertyList - a property list.
cbPropertyListSize - the size in bytes of the data in pPropertyList.
pszPropertyName - the property name to look for in the buffer.
pszPropertyValue - the matching string value found.
bReturnExpandedValue - TRUE = return expanded value if one is present,
FALSE = return the first value.
Return Value:
ERROR_SUCCESS if successful.
ERROR_INVALID_DATA -
ERROR_FILE_NOT_FOUND -
ERROR_NOT_ENOUGH_MEMORY -
A Win32 error code on failure.
--*/
{
CLUSPROP_BUFFER_HELPER props;
LPWSTR valueData;
LPWSTR listValueData;
DWORD listByteLength;
DWORD itemCount;
DWORD byteCount;
props.pb = (LPBYTE) pPropertyList;
itemCount = *(props.pdw++);
cbPropertyListSize -= sizeof(DWORD);
while ( itemCount-- &&
((LONG)cbPropertyListSize > 0) ) {
//
// If we found the specified property, validate the entry and return
// the value to the caller.
//
if ( (props.pName->Syntax.dw == CLUSPROP_SYNTAX_NAME) &&
(lstrcmpiW( props.pName->sz, pszPropertyName ) == 0) ) {
//
// Calculate the size of the name and move to the value.
//
byteCount = ALIGN_CLUSPROP(props.pName->cbLength);
props.pb += sizeof(*props.pValue) + byteCount;
//
// Make sure this is a string property.
//
if ( (props.pStringValue->Syntax.dw != CLUSPROP_SYNTAX_LIST_VALUE_SZ) &&
(props.pStringValue->Syntax.dw != CLUSPROP_SYNTAX_LIST_VALUE_EXPAND_SZ) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpFindSzProperty: Property '%1!ls!' syntax (%2!d!, %3!d!) not proper list string syntax.\n", pszPropertyName, props.pSyntax->wType, props.pSyntax->wFormat );
return(ERROR_INVALID_DATA);
}
//
// If caller wants the value, allocate a buffer for it
// and copy the value in.
//
if ( pszPropertyValue != NULL ) {
//
// If caller wants the expanded value, look at any
// additional values in the value list to see if one
// was returned.
//
listValueData = props.pStringValue->sz;
listByteLength = props.pStringValue->cbLength;
if ( bReturnExpandedValue ) {
//
// Skip past values in the value list looking for
// an expanded string value.
//
while ( (props.pSyntax->dw != CLUSPROP_SYNTAX_ENDMARK) &&
(cbPropertyListSize > 0) ) {
byteCount = sizeof(*props.pValue) + ALIGN_CLUSPROP(listByteLength);
cbPropertyListSize -= byteCount;
props.pb += byteCount;
if ( props.pSyntax->dw == CLUSPROP_SYNTAX_LIST_VALUE_EXPANDED_SZ ) {
listValueData = props.pStringValue->sz;
listByteLength = props.pStringValue->cbLength;
break;
}
}
}
//
// Allocate a buffer for the string value and
// copy the value from the property list.
//
valueData = (LPWSTR) LocalAlloc( LMEM_FIXED, listByteLength );
if ( valueData == NULL ) {
return(ERROR_NOT_ENOUGH_MEMORY);
}
CopyMemory( valueData, listValueData, listByteLength );
*pszPropertyValue = valueData;
}
//
// We found the property so return success.
//
return(ERROR_SUCCESS);
} else {
//
// Skip the name (value header + size of data).
//
byteCount = ALIGN_CLUSPROP(props.pValue->cbLength);
cbPropertyListSize -= sizeof(*props.pValue) + byteCount;
props.pb += sizeof(*props.pValue) + byteCount;
//
// Skip it's value list (one or more values + endmark).
//
while ( (props.pSyntax->dw != CLUSPROP_SYNTAX_ENDMARK) &&
(cbPropertyListSize > 0) ) {
byteCount = sizeof(*props.pValue) + ALIGN_CLUSPROP(props.pValue->cbLength);
cbPropertyListSize -= byteCount;
props.pb += byteCount;
}
cbPropertyListSize -= sizeof(*props.pSyntax);
props.pb += sizeof(*props.pSyntax);
}
}
return(ERROR_FILE_NOT_FOUND);
} // ClRtlpFindSzProperty
DWORD
WINAPI
ClRtlFindDwordProperty(
IN const PVOID pPropertyList,
IN DWORD cbPropertyListSize,
IN LPCWSTR pszPropertyName,
OUT LPDWORD pdwPropertyValue
)
/*++
Routine Description:
Finds the specified DWORD property in the Property List buffer pointed at
by pPropertyList.
Arguments:
pPropertyList - a property list.
cbPropertyListSize - the size in bytes of the data in pPropertyList.
pszPropertyName - the property name to look for in the buffer.
pdwPropertyValue - the matching DWORD value found.
Return Value:
ERROR_SUCCESS if successful.
ERROR_INVALID_DATA -
ERROR_FILE_NOT_FOUND -
A Win32 error code on failure.
--*/
{
CLUSPROP_BUFFER_HELPER props;
LPWSTR valueData;
DWORD itemCount;
DWORD byteCount;
props.pb = (LPBYTE) pPropertyList;
itemCount = *(props.pdw++);
cbPropertyListSize -= sizeof(DWORD);
while ( itemCount-- &&
((LONG)cbPropertyListSize > 0) ) {
//
// If we found the specified property, validate the entry and return
// the value to the caller.
//
if ( (props.pName->Syntax.dw == CLUSPROP_SYNTAX_NAME) &&
(lstrcmpiW( props.pName->sz, pszPropertyName ) == 0) ) {
//
// Calculate the size of the name and move to the value.
//
byteCount = ALIGN_CLUSPROP(props.pName->cbLength);
props.pb += sizeof(*props.pValue) + byteCount;
//
// Make sure this is a dword property.
//
if ( props.pDwordValue->Syntax.dw != CLUSPROP_SYNTAX_LIST_VALUE_DWORD ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlFindDwordProperty: Property '%1!ls!' syntax (%2!d!, %3!d!) not proper list DWORD syntax.\n", pszPropertyName, props.pSyntax->wType, props.pSyntax->wFormat );
return(ERROR_INVALID_DATA);
}
//
// If caller wants the value, allocate a buffer for it.
//
if ( pdwPropertyValue ) {
*pdwPropertyValue = props.pDwordValue->dw;
}
//
// We found the property so return success.
//
return(ERROR_SUCCESS);
} else {
//
// Skip the name (value header + size of data).
//
byteCount = ALIGN_CLUSPROP(props.pValue->cbLength);
cbPropertyListSize -= sizeof(*props.pValue) + byteCount;
props.pb += sizeof(*props.pValue) + byteCount;
//
// Skip it's value list and endmark.
//
while ( (props.pSyntax->dw != CLUSPROP_SYNTAX_ENDMARK) &&
(cbPropertyListSize > 0) ) {
byteCount = ALIGN_CLUSPROP(props.pValue->cbLength);
cbPropertyListSize -= sizeof(*props.pValue) + byteCount;
props.pb += sizeof(*props.pValue) + byteCount;
}
cbPropertyListSize -= sizeof(*props.pSyntax);
props.pb += sizeof(*props.pSyntax);
}
}
return(ERROR_FILE_NOT_FOUND);
} // ClRtlFindDwordProperty
DWORD
WINAPI
ClRtlFindLongProperty(
IN const PVOID pPropertyList,
IN DWORD cbPropertyListSize,
IN LPCWSTR pszPropertyName,
OUT LPLONG plPropertyValue
)
/*++
Routine Description:
Finds the specified dword in the Value List buffer pointed at by Buffer.
Arguments:
pPropertyList - a property list.
cbPropertyListSize - the size in bytes of the data in pPropertyList.
pszPropertyName - the property name to look for in the buffer.
plPropertyValue - the matching long value found.
Return Value:
ERROR_SUCCESS if successful.
ERROR_INVALID_DATA -
ERROR_FILE_NOT_FOUND -
A Win32 error code on failure.
--*/
{
CLUSPROP_BUFFER_HELPER props;
LPWSTR valueData;
DWORD itemCount;
DWORD byteCount;
props.pb = (LPBYTE) pPropertyList;
itemCount = *(props.pdw++);
cbPropertyListSize -= sizeof(DWORD);
while ( itemCount-- &&
(cbPropertyListSize > 0) ) {
//
// If we found the specified property, validate the entry and return
// the value to the caller.
//
if ( (props.pName->Syntax.dw == CLUSPROP_SYNTAX_NAME) &&
(lstrcmpiW( props.pName->sz, pszPropertyName ) == 0) ) {
//
// Calculate the size of the name and move to the value.
//
byteCount = ALIGN_CLUSPROP(props.pName->cbLength);
props.pb += sizeof(*props.pValue) + byteCount;
//
// Make sure this is a long property.
//
if ( props.pLongValue->Syntax.dw != CLUSPROP_SYNTAX_LIST_VALUE_DWORD ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlFindLongProperty: Property '%1!ls!' syntax (%2!d!, %3!d!) not proper list LONG syntax.\n", pszPropertyName, props.pSyntax->wType, props.pSyntax->wFormat );
return(ERROR_INVALID_DATA);
}
//
// If caller wants the value, allocate a buffer for it.
//
if ( plPropertyValue) {
*plPropertyValue = props.pLongValue->l;
}
//
// We found the property so return success.
//
return(ERROR_SUCCESS);
} else {
//
// Skip the name (value header + size of data).
//
byteCount = ALIGN_CLUSPROP(props.pValue->cbLength);
cbPropertyListSize -= sizeof(*props.pValue) + byteCount;
props.pb += sizeof(*props.pValue) + byteCount;
//
// Skip it's value list and endmark.
//
while ( (props.pSyntax->dw != CLUSPROP_SYNTAX_ENDMARK) &&
(cbPropertyListSize > 0) ) {
byteCount = ALIGN_CLUSPROP(props.pValue->cbLength);
cbPropertyListSize -= sizeof(*props.pValue) + byteCount;
props.pb += sizeof(*props.pValue) + byteCount;
}
cbPropertyListSize -= sizeof(*props.pSyntax);
props.pb += sizeof(*props.pSyntax);
}
}
return(ERROR_FILE_NOT_FOUND);
} // ClRtlFindLongProperty
DWORD
WINAPI
ClRtlFindULargeIntegerProperty(
IN const PVOID pPropertyList,
IN DWORD cbPropertyListSize,
IN LPCWSTR pszPropertyName,
OUT PULARGE_INTEGER pullPropertyValue
)
/*++
Routine Description:
Finds the specified ULARGE_INTEGER property in the Property List buffer
pointed at by pPropertyList.
Arguments:
pPropertyList - a property list.
cbPropertyListSize - the size in bytes of the data in pPropertyList.
pszPropertyName - the property name to look for in the buffer.
pullPropertyValue - the matching ULARGE_INTEGER value found.
Return Value:
ERROR_SUCCESS if successful.
ERROR_INVALID_DATA -
ERROR_FILE_NOT_FOUND -
A Win32 error code on failure.
--*/
{
CLUSPROP_BUFFER_HELPER props;
LPWSTR valueData;
DWORD itemCount;
DWORD byteCount;
props.pb = (LPBYTE) pPropertyList;
itemCount = *(props.pdw++);
cbPropertyListSize -= sizeof(DWORD);
while ( itemCount-- &&
((LONG)cbPropertyListSize > 0) ) {
//
// If we found the specified property, validate the entry and return
// the value to the caller.
//
if ( (props.pName->Syntax.dw == CLUSPROP_SYNTAX_NAME) &&
(lstrcmpiW( props.pName->sz, pszPropertyName ) == 0) ) {
//
// Calculate the size of the name and move to the value.
//
byteCount = ALIGN_CLUSPROP(props.pName->cbLength);
props.pb += sizeof(*props.pValue) + byteCount;
//
// Make sure this is a LARGE INT property.
//
if ( props.pULargeIntegerValue->Syntax.dw != CLUSPROP_SYNTAX_LIST_VALUE_LARGE_INTEGER ) {
ClRtlDbgPrint(LOG_CRITICAL,
"ClRtlFindULargeIntegerProperty: Property '%1!ls!' syntax "
"(%2!d!, %3!d!) not proper list ULARGE_INTEGER syntax.\n",
pszPropertyName,
props.pSyntax->wType,
props.pSyntax->wFormat );
return(ERROR_INVALID_DATA);
}
//
// If caller wants the value, allocate a buffer for it.
//
if ( pullPropertyValue ) {
pullPropertyValue->QuadPart = props.pULargeIntegerValue->li.QuadPart;
}
//
// We found the property so return success.
//
return(ERROR_SUCCESS);
} else {
//
// Skip the name (value header + size of data).
//
byteCount = ALIGN_CLUSPROP(props.pValue->cbLength);
cbPropertyListSize -= sizeof(*props.pValue) + byteCount;
props.pb += sizeof(*props.pValue) + byteCount;
//
// Skip it's value list and endmark.
//
while ( (props.pSyntax->dw != CLUSPROP_SYNTAX_ENDMARK) &&
(cbPropertyListSize > 0) )
{
byteCount = ALIGN_CLUSPROP(props.pValue->cbLength);
cbPropertyListSize -= sizeof(*props.pValue) + byteCount;
props.pb += sizeof(*props.pValue) + byteCount;
}
cbPropertyListSize -= sizeof(*props.pSyntax);
props.pb += sizeof(*props.pSyntax);
}
}
return(ERROR_FILE_NOT_FOUND);
} // ClRtlFindULargeIntegerProperty
DWORD
WINAPI
ClRtlFindLargeIntegerProperty(
IN const PVOID pPropertyList,
IN DWORD cbPropertyListSize,
IN LPCWSTR pszPropertyName,
OUT PLARGE_INTEGER pllPropertyValue
)
/*++
Routine Description:
Finds the specified LARGE_INTEGER property in the Property List buffer
pointed at by pPropertyList.
Arguments:
pPropertyList - a property list.
cbPropertyListSize - the size in bytes of the data in pPropertyList.
pszPropertyName - the property name to look for in the buffer.
pllPropertyValue - the matching ULARGE_INTEGER value found.
Return Value:
ERROR_SUCCESS if successful.
ERROR_INVALID_DATA -
ERROR_FILE_NOT_FOUND -
A Win32 error code on failure.
--*/
{
CLUSPROP_BUFFER_HELPER props;
LPWSTR valueData;
DWORD itemCount;
DWORD byteCount;
props.pb = (LPBYTE) pPropertyList;
itemCount = *(props.pdw++);
cbPropertyListSize -= sizeof(DWORD);
while ( itemCount-- &&
((LONG)cbPropertyListSize > 0) ) {
//
// If we found the specified property, validate the entry and return
// the value to the caller.
//
if ( (props.pName->Syntax.dw == CLUSPROP_SYNTAX_NAME) &&
(lstrcmpiW( props.pName->sz, pszPropertyName ) == 0) ) {
//
// Calculate the size of the name and move to the value.
//
byteCount = ALIGN_CLUSPROP(props.pName->cbLength);
props.pb += sizeof(*props.pValue) + byteCount;
//
// Make sure this is a large int property.
//
if ( props.pLargeIntegerValue->Syntax.dw != CLUSPROP_SYNTAX_LIST_VALUE_LARGE_INTEGER ) {
ClRtlDbgPrint(LOG_CRITICAL,
"ClRtlFindLargeIntegerProperty: Property '%1!ls!' syntax "
"(%2!d!, %3!d!) not proper list ULARGE_INTEGER syntax.\n",
pszPropertyName,
props.pSyntax->wType,
props.pSyntax->wFormat );
return(ERROR_INVALID_DATA);
}
//
// If caller wants the value, allocate a buffer for it.
//
if ( pllPropertyValue ) {
pllPropertyValue->QuadPart = props.pLargeIntegerValue->li.QuadPart;
}
//
// We found the property so return success.
//
return(ERROR_SUCCESS);
} else {
//
// Skip the name (value header + size of data).
//
byteCount = ALIGN_CLUSPROP(props.pValue->cbLength);
cbPropertyListSize -= sizeof(*props.pValue) + byteCount;
props.pb += sizeof(*props.pValue) + byteCount;
//
// Skip it's value list and endmark.
//
while ( (props.pSyntax->dw != CLUSPROP_SYNTAX_ENDMARK) &&
(cbPropertyListSize > 0) )
{
byteCount = ALIGN_CLUSPROP(props.pValue->cbLength);
cbPropertyListSize -= sizeof(*props.pValue) + byteCount;
props.pb += sizeof(*props.pValue) + byteCount;
}
cbPropertyListSize -= sizeof(*props.pSyntax);
props.pb += sizeof(*props.pSyntax);
}
}
return(ERROR_FILE_NOT_FOUND);
} // ClRtlFindLargeIntegerProperty
DWORD
WINAPI
ClRtlFindBinaryProperty(
IN const PVOID pPropertyList,
IN DWORD cbPropertyListSize,
IN LPCWSTR pszPropertyName,
OUT LPBYTE * pbPropertyValue,
OUT LPDWORD pcbPropertyValueSize
)
/*++
Routine Description:
Finds the specified binary property in the Property List buffer pointed at
by pPropertyList.
Arguments:
pPropertyList - a property list.
cbPropertyListSize - the size in bytes of the data in pPropertyList.
pszPropertyName - the property name to look for in the buffer.
pbPropertyValue - the matching binary value found.
pcbPropertyValueSize - the length of the matching binary value found.
Return Value:
ERROR_SUCCESS if successful.
ERROR_INVALID_DATA -
ERROR_NOT_ENOUGH_MEMORY -
ERROR_FILE_NOT_FOUND -
A Win32 error code on failure.
--*/
{
CLUSPROP_BUFFER_HELPER props;
PBYTE valueData;
DWORD itemCount;
DWORD byteCount;
props.pb = (LPBYTE) pPropertyList;
itemCount = *(props.pdw++);
cbPropertyListSize -= sizeof(DWORD);
while ( itemCount-- &&
((LONG)cbPropertyListSize > 0) ) {
//
// If we found the specified property, validate the entry and return
// the value to the caller.
//
if ( (props.pName->Syntax.dw == CLUSPROP_SYNTAX_NAME) &&
(lstrcmpiW( props.pName->sz, pszPropertyName ) == 0) ) {
//
// Calculate the size of the name and move to the value.
//
byteCount = ALIGN_CLUSPROP(props.pName->cbLength);
props.pb += sizeof(*props.pValue) + byteCount;
//
// Make sure this is a binary property.
//
if ( props.pStringValue->Syntax.dw != CLUSPROP_SYNTAX_LIST_VALUE_BINARY ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpFindBinaryProperty: Property '%1!ls!' syntax (%2!d!, %3!d!) not proper list binary syntax.\n", pszPropertyName, props.pSyntax->wType, props.pSyntax->wFormat );
return(ERROR_INVALID_DATA);
}
//
// If caller wants the value, allocate a buffer for it.
//
if ( pbPropertyValue ) {
valueData = (PBYTE) LocalAlloc( LMEM_FIXED, props.pBinaryValue->cbLength );
if ( !valueData ) {
return(ERROR_NOT_ENOUGH_MEMORY);
}
CopyMemory( valueData, props.pBinaryValue->rgb, props.pBinaryValue->cbLength );
*pbPropertyValue = valueData;
}
//
// If caller wants the value size
//
if ( pcbPropertyValueSize ) {
*pcbPropertyValueSize = props.pBinaryValue->cbLength;
}
//
// We found the property so return success.
//
return(ERROR_SUCCESS);
} else {
//
// Skip the name (value header + size of data).
//
byteCount = ALIGN_CLUSPROP(props.pValue->cbLength);
cbPropertyListSize -= sizeof(*props.pValue) + byteCount;
props.pb += sizeof(*props.pValue) + byteCount;
//
// Skip it's value list and endmark.
//
while ( (props.pSyntax->dw != CLUSPROP_SYNTAX_ENDMARK) &&
(cbPropertyListSize > 0) ) {
byteCount = ALIGN_CLUSPROP(props.pValue->cbLength);
cbPropertyListSize -= sizeof(*props.pValue) + byteCount;
props.pb += sizeof(*props.pValue) + byteCount;
}
cbPropertyListSize -= sizeof(*props.pSyntax);
props.pb += sizeof(*props.pSyntax);
}
}
return(ERROR_FILE_NOT_FOUND);
} // ClRtlFindBinaryProperty
DWORD
WINAPI
ClRtlFindMultiSzProperty(
IN const PVOID pPropertyList,
IN DWORD cbPropertyListSize,
IN LPCWSTR pszPropertyName,
OUT LPWSTR * pszPropertyValue,
OUT LPDWORD pcbPropertyValueSize
)
/*++
Routine Description:
Finds the specified multiple string property in the Property List buffer
pointed at by pPropertyList.
Arguments:
pPropertyList - a property list.
cbPropertyListSize - the size in bytes of the data in pPropertyList.
pszPropertyName - the property name to look for in the buffer.
pszPropertyValue - the matching multiple string value found.
pcbPropertyValueSize - the length of the matching multiple string value found.
Return Value:
ERROR_SUCCESS if successful.
ERROR_INVALID_DATA -
ERROR_NOT_ENOUGH_MEMORY -
ERROR_FILE_NOT_FOUND -
A Win32 error code on failure.
--*/
{
CLUSPROP_BUFFER_HELPER props;
LPWSTR valueData;
DWORD itemCount;
DWORD byteCount;
props.pb = (LPBYTE) pPropertyList;
itemCount = *(props.pdw++);
cbPropertyListSize -= sizeof(DWORD);
while ( itemCount-- &&
((LONG)cbPropertyListSize > 0) ) {
//
// If we found the specified property, validate the entry and return
// the value to the caller.
//
if ( (props.pName->Syntax.dw == CLUSPROP_SYNTAX_NAME) &&
(lstrcmpiW( props.pName->sz, pszPropertyName ) == 0) ) {
//
// Calculate the size of the name and move to the value.
//
byteCount = ALIGN_CLUSPROP(props.pName->cbLength);
props.pb += sizeof(*props.pValue) + byteCount;
//
// Make sure this is a multi-sz property.
//
if ( props.pStringValue->Syntax.dw != CLUSPROP_SYNTAX_LIST_VALUE_MULTI_SZ ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlpFindMultiSzProperty: Property '%1!ls!' syntax (%2!d!, %3!d!) not proper list MultiSz syntax.\n", pszPropertyName, props.pSyntax->wType, props.pSyntax->wFormat );
return(ERROR_INVALID_DATA);
}
//
// If caller wants the value, allocate a buffer for it.
//
if ( pszPropertyValue ) {
valueData = (LPWSTR) LocalAlloc( LMEM_FIXED, props.pMultiSzValue->cbLength );
if ( !valueData ) {
return(ERROR_NOT_ENOUGH_MEMORY);
}
CopyMemory( valueData, props.pBinaryValue->rgb, props.pMultiSzValue->cbLength );
*pszPropertyValue = valueData;
}
//
// If caller wants the value size
//
if ( pcbPropertyValueSize ) {
*pcbPropertyValueSize = props.pMultiSzValue->cbLength;
}
//
// We found the property so return success.
//
return(ERROR_SUCCESS);
} else {
//
// Skip the name (value header + size of data).
//
byteCount = ALIGN_CLUSPROP(props.pValue->cbLength);
cbPropertyListSize -= sizeof(*props.pValue) + byteCount;
props.pb += sizeof(*props.pValue) + byteCount;
//
// Skip it's value list and endmark.
//
while ( (props.pSyntax->dw != CLUSPROP_SYNTAX_ENDMARK) &&
(cbPropertyListSize > 0) ) {
byteCount = ALIGN_CLUSPROP(props.pValue->cbLength);
cbPropertyListSize -= sizeof(*props.pValue) + byteCount;
props.pb += sizeof(*props.pValue) + byteCount;
}
cbPropertyListSize -= sizeof(*props.pSyntax);
props.pb += sizeof(*props.pSyntax);
}
}
return(ERROR_FILE_NOT_FOUND);
} // ClRtlFindMultiSzProperty
DWORD
WINAPI
ClRtlGetBinaryValue(
IN HKEY hkeyClusterKey,
IN LPCWSTR pszValueName,
OUT LPBYTE * ppbOutValue,
OUT LPDWORD pcbOutValueSize,
IN const PCLUSTER_REG_APIS pClusterRegApis
)
/*++
Routine Description:
Queries a REG_BINARY or REG_MULTI_SZ value out of the cluster
database and allocates the necessary storage for it.
Arguments:
hkeyClusterKey - Supplies the cluster key where the value is stored
pszValueName - Supplies the name of the value.
ppbOutValue - Supplies the address of a pointer in which to return the value.
pcbOutValueSize - Supplies the address of a DWORD in which to return the
size of the value.
pfnQueryValue - Address of QueryValue function.
Return Value:
ERROR_SUCCESS - The value was read successfully.
ERROR_BAD_ARGUMENTS -
ERROR_NOT_ENOUGH_MEMORY - Error allocating memory for the value.
Win32 error code - The operation failed.
--*/
{
LPBYTE value = NULL;
DWORD valueSize;
DWORD valueType;
DWORD status;
PVOID key = NULL;
CRegistryValueName rvn;
//
// Initialize the output parameters.
//
*ppbOutValue = NULL;
*pcbOutValueSize = 0;
if ( (hkeyClusterKey == NULL) ||
(pClusterRegApis == NULL) ||
(pClusterRegApis->pfnQueryValue == NULL) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlGetBinaryValue: hkeyClusterKey, pClusterRegApis, or pClusterRegApis->pfnQueryValue == NULL. Returning ERROR_BAD_ARGUMENTS.\n" );
return(ERROR_BAD_ARGUMENTS);
}
//
// Use the wrapper class CRegistryValueName to parse value name to see if it
// contains a backslash.
//
status = rvn.ScInit( pszValueName, NULL );
if ( status != ERROR_SUCCESS )
{
return status;
}
//
// If the value resides at a different location, open the key.
//
if ( rvn.PszKeyName() != NULL ) {
if ( (pClusterRegApis->pfnOpenKey == NULL) ||
(pClusterRegApis->pfnCloseKey == NULL) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlGetBinaryValue: pClusterRegApis->pfnOpenKey or pfnCloseKey == NULL. Returning ERROR_BAD_ARGUMENTS.\n" );
return(ERROR_BAD_ARGUMENTS);
}
status = (*pClusterRegApis->pfnOpenKey)( hkeyClusterKey,
rvn.PszKeyName(),
KEY_READ,
&key);
} else {
key = hkeyClusterKey;
}
//
// Dummy do-while loop to avoid gotos.
//
do
{
//
// Get the size of the value so we know how much to allocate.
//
valueSize = 0;
status = (*pClusterRegApis->pfnQueryValue)( key,
rvn.PszName(),
&valueType,
NULL,
&valueSize );
if ( (status != ERROR_SUCCESS) &&
(status != ERROR_MORE_DATA) ) {
break;
}
//if the size is zero, just return
if (valueSize == 0)
{
break;
}
//
// Allocate a buffer to read the value into.
//
value = (LPBYTE) LocalAlloc( LMEM_FIXED, valueSize );
if ( value == NULL ) {
status = ERROR_NOT_ENOUGH_MEMORY;
break;
}
//
// Read the value from the cluster database.
//
status = (*pClusterRegApis->pfnQueryValue)( key,
pszValueName,
&valueType,
(LPBYTE)value,
&valueSize );
if ( status != ERROR_SUCCESS ) {
LocalFree( value );
} else {
*ppbOutValue = value;
*pcbOutValueSize = valueSize;
}
}
while (FALSE);
//
// Close the key if we opened it.
//
if ( (rvn.PszKeyName() != NULL) &&
(key != NULL) ) {
(*pClusterRegApis->pfnCloseKey)( key );
}
return(status);
} // ClRtlGetBinaryValue
LPWSTR
WINAPI
ClRtlGetSzValue(
IN HKEY hkeyClusterKey,
IN LPCWSTR pszValueName,
IN const PCLUSTER_REG_APIS pClusterRegApis
)
/*++
Routine Description:
Queries a REG_SZ or REG_EXPAND_SZ value out of the cluster database
and allocates the necessary storage for it.
Arguments:
hkeyClusterKey - Supplies the cluster key where the value is stored
pszValueName - Supplies the name of the value.
pfnQueryValue - Address of QueryValue function.
Return Value:
A pointer to a buffer containing the value if successful.
NULL if unsuccessful. Call GetLastError() to get more details.
--*/
{
PWSTR value;
DWORD valueSize;
DWORD valueType;
DWORD status;
PVOID key = NULL;
CRegistryValueName rvn;
if ( (hkeyClusterKey == NULL) ||
(pClusterRegApis == NULL) ||
(pClusterRegApis->pfnQueryValue == NULL) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlGetSzValue: hkeyClusterKey, pClusterRegApis, or pClusterRegApis->pfnQueryValue == NULL. Returning ERROR_BAD_ARGUMENTS.\n" );
SetLastError(ERROR_BAD_ARGUMENTS);
return(NULL);
}
//
// Use the wrapper class CRegistryValueName to parse value name to see if it
// contains a backslash.
//
status = rvn.ScInit( pszValueName, NULL );
if ( status != ERROR_SUCCESS )
{
SetLastError(status);
return(NULL);
}
//
// If the value resides at a different location, open the key.
//
if ( rvn.PszKeyName() != NULL ) {
if ( (pClusterRegApis->pfnOpenKey == NULL) ||
(pClusterRegApis->pfnCloseKey == NULL) ) {
ClRtlDbgPrint( LOG_CRITICAL, "ClRtlGetSzValue: pClusterRegApis->pfnOpenKey or pfnCloseKey == NULL. Returning ERROR_BAD_ARGUMENTS.\n" );
SetLastError(ERROR_BAD_ARGUMENTS);
return(NULL);
}
status = (*pClusterRegApis->pfnOpenKey)( hkeyClusterKey,
rvn.PszKeyName(),
KEY_READ,
&key);
} else {
key = hkeyClusterKey;
}
//
// Dummy do-while loop to avoid gotos.
//
do
{
//
// Get the size of the value so we know how much to allocate.
//
valueSize = 0;
status = (*pClusterRegApis->pfnQueryValue)( key,
rvn.PszName(),
&valueType,
NULL,
&valueSize );
if ( (status != ERROR_SUCCESS) &&
(status != ERROR_MORE_DATA) ) {
SetLastError( status );
value = NULL;
break;
}
//
// Add on the size of the null terminator.
//
valueSize += sizeof(UNICODE_NULL);
//
// Allocate a buffer to read the string into.
//
value = (PWSTR) LocalAlloc( LMEM_FIXED, valueSize );
if ( value == NULL ) {
SetLastError( ERROR_NOT_ENOUGH_MEMORY );
value = NULL;
break;
}
//
// Read the value from the cluster database.
//
status = (*pClusterRegApis->pfnQueryValue)( key,
rvn.PszName(),
&valueType,
(LPBYTE)value,
&valueSize );
if ( status != ERROR_SUCCESS ) {
LocalFree( value );
value = NULL;
} else if ( (valueType != REG_SZ) &&
(valueType != REG_EXPAND_SZ) &&
(valueType != REG_MULTI_SZ) ) {
status = ERROR_INVALID_PARAMETER;
LocalFree( value );
SetLastError( status );
value = NULL;
}
}
while (FALSE);
//
// Close the key if we opened it.
//
if ( (rvn.PszKeyName() != NULL) &&
(key != NULL) ) {
(*pClusterRegApis->pfnCloseKey)( key );
}
return(value);
} // ClRtlGetSzValue
DWORD
WINAPI
ClRtlDupParameterBlock(
OUT LPBYTE pOutParams,
IN const LPBYTE pInParams,
IN const PRESUTIL_PROPERTY_ITEM pPropertyTable
)
/*++
Routine Description:
Deallocates any buffers allocated for a parameter block that are
different than the buffers used for the input parameter block.
Arguments:
pOutParams - Parameter block to return.
pInParams - Reference parameter block.
pPropertyTable - Pointer to the property table to process.
Return Value:
ERROR_SUCCESS - Parameter block duplicated successfully.
--*/
{
DWORD status = ERROR_SUCCESS;
PRESUTIL_PROPERTY_ITEM propertyItem = pPropertyTable;
LPWSTR UNALIGNED * ppszInValue;
DWORD * pdwInValue;
ULARGE_INTEGER * pullInValue;
LPWSTR UNALIGNED * ppszOutValue;
DWORD * pdwOutValue;
ULARGE_INTEGER * pullOutValue;
while ( propertyItem->Name != NULL ) {
switch ( propertyItem->Format ) {
case CLUSPROP_FORMAT_DWORD:
case CLUSPROP_FORMAT_LONG:
pdwInValue = (DWORD *) &pInParams[propertyItem->Offset];
pdwOutValue = (DWORD *) &pOutParams[propertyItem->Offset];
*pdwOutValue = *pdwInValue;
break;
case CLUSPROP_FORMAT_ULARGE_INTEGER:
case CLUSPROP_FORMAT_LARGE_INTEGER:
pullInValue = (ULARGE_INTEGER *) &pInParams[propertyItem->Offset];
pullOutValue = (ULARGE_INTEGER *) &pOutParams[propertyItem->Offset];
pullOutValue->u = pullInValue->u;
break;
case CLUSPROP_FORMAT_SZ:
case CLUSPROP_FORMAT_EXPAND_SZ:
ppszInValue = (LPWSTR UNALIGNED *) &pInParams[propertyItem->Offset];
ppszOutValue = (LPWSTR UNALIGNED *) &pOutParams[propertyItem->Offset];
if ( *ppszInValue == NULL ) {
if ( propertyItem->lpDefault != NULL ) {
*ppszOutValue = (LPWSTR) LocalAlloc(
LMEM_FIXED,
(lstrlenW( (LPCWSTR) propertyItem->lpDefault ) + 1) * sizeof(WCHAR)
);
if ( *ppszOutValue == NULL ) {
status = GetLastError();
} else {
lstrcpyW( *ppszOutValue, (LPCWSTR) propertyItem->lpDefault );
}
} else {
*ppszOutValue = NULL;
}
} else {
*ppszOutValue = (LPWSTR) LocalAlloc( LMEM_FIXED, (lstrlenW( *ppszInValue ) + 1) * sizeof(WCHAR) );
if ( *ppszOutValue == NULL ) {
status = GetLastError();
} else {
lstrcpyW( *ppszOutValue, *ppszInValue );
}
}
break;
case CLUSPROP_FORMAT_MULTI_SZ:
case CLUSPROP_FORMAT_BINARY:
ppszInValue = (LPWSTR UNALIGNED *) &pInParams[propertyItem->Offset];
pdwInValue = (DWORD *) &pInParams[propertyItem->Offset + sizeof(LPCWSTR)];
ppszOutValue = (LPWSTR UNALIGNED *) &pOutParams[propertyItem->Offset];
pdwOutValue = (DWORD *) &pOutParams[propertyItem->Offset + sizeof(LPWSTR)];
if ( *ppszInValue == NULL ) {
if ( propertyItem->lpDefault != NULL ) {
*ppszOutValue = (LPWSTR) LocalAlloc( LMEM_FIXED, propertyItem->Minimum );
if ( *ppszOutValue == NULL ) {
status = GetLastError();
*pdwOutValue = 0;
} else {
*pdwOutValue = propertyItem->Minimum;
CopyMemory( *ppszOutValue, (const PVOID) propertyItem->lpDefault, *pdwOutValue );
}
} else {
*ppszOutValue = NULL;
*pdwOutValue = 0;
}
} else {
*ppszOutValue = (LPWSTR) LocalAlloc( LMEM_FIXED, *pdwInValue );
if ( *ppszOutValue == NULL ) {
status = GetLastError();
*pdwOutValue = 0;
} else {
CopyMemory( *ppszOutValue, *ppszInValue, *pdwInValue );
*pdwOutValue = *pdwInValue;
}
}
break;
}
propertyItem++;
}
//
// If an error occurred, make sure we don't leak memory.
//
if ( status != ERROR_SUCCESS ) {
ClRtlFreeParameterBlock( pOutParams, pInParams, pPropertyTable );
}
return(status);
} // ClRtlDupParameterBlock
void
WINAPI
ClRtlFreeParameterBlock(
IN OUT LPBYTE pOutParams,
IN const LPBYTE pInParams,
IN const PRESUTIL_PROPERTY_ITEM pPropertyTable
)
/*++
Routine Description:
Deallocates any buffers allocated for a parameter block that are
different than the buffers used for the input parameter block.
Arguments:
pOutParams - Parameter block to free.
pInParams - Reference parameter block.
pPropertyTable - Pointer to the property table to process.
Return Value:
None.
--*/
{
PRESUTIL_PROPERTY_ITEM propertyItem = pPropertyTable;
LPCWSTR UNALIGNED * ppszInValue;
LPWSTR UNALIGNED * ppszOutValue;
while ( propertyItem->Name != NULL ) {
switch ( propertyItem->Format ) {
case CLUSPROP_FORMAT_SZ:
case CLUSPROP_FORMAT_EXPAND_SZ:
case CLUSPROP_FORMAT_MULTI_SZ:
case CLUSPROP_FORMAT_BINARY:
ppszInValue = (LPCWSTR UNALIGNED *) &pInParams[propertyItem->Offset];
ppszOutValue = (LPWSTR UNALIGNED *) &pOutParams[propertyItem->Offset];
if ( (pInParams == NULL) || (*ppszInValue != *ppszOutValue) ) {
LocalFree( *ppszOutValue );
}
break;
}
propertyItem++;
}
} // ClRtlFreeParameterBlock
DWORD
WINAPI
ClRtlMarshallPropertyTable(
IN PRESUTIL_PROPERTY_ITEM pPropertyTable,
IN OUT DWORD dwSize,
IN OUT LPBYTE pBuffer,
OUT DWORD *Required
)
/*++
Routine Description
Rohit (rjain) : It marshalls the pPropertyTable into a buffer so that the
buffer can be passed as an argument to the NmUpdatePerformFixups2 handler
This function assumes that the value field in all the elements of the
table is 0.
Arguments
pPropertyTable - This table is converted to buffer
dwSize - size in bytes of the pbuffer supplied
pbuffer - byte array into which pPropertyTable is copied
Required - number of bytes required
Return Value
returns
ERROR_SUCCESS on success,
ERROR_MORE_DATA if the size of pbuffer is insufficient
++*/
{
DWORD dwPosition=sizeof(DWORD);
PRESUTIL_PROPERTY_ITEM pPropertyItem=pPropertyTable;
BOOL copying = TRUE;
DWORD items=0;
DWORD dwNameLength;
DWORD dwKeyLength;
DWORD status=ERROR_SUCCESS;
*Required=sizeof(DWORD); // first DWORD will contain the number of items in PropertyTable
while(pPropertyItem->Name != NULL)
{
items++;
dwNameLength=(lstrlenW(pPropertyItem->Name)+1)*sizeof(WCHAR);
if(pPropertyItem->KeyName==NULL)
dwKeyLength=0;
else
dwKeyLength=(lstrlenW(pPropertyItem->KeyName)+1)*sizeof(WCHAR);
*Required+=(dwNameLength+dwKeyLength+8*sizeof(DWORD));
// if pbufer is smaller than needed, copying is turned off
// and only the required size is calculated
if ((copying && (dwSize < *Required)))
copying=FALSE;
if(copying)
{
// copy length of name and then the name itself
CopyMemory(pBuffer+dwPosition,&dwNameLength,sizeof(DWORD));
dwPosition+=(sizeof(DWORD));
CopyMemory(pBuffer+dwPosition,pPropertyItem->Name,dwNameLength);
dwPosition+=dwNameLength;
//copy length of keyname and then the keyname itself
CopyMemory(pBuffer+dwPosition,&dwKeyLength,sizeof(DWORD));
dwPosition+=(sizeof(DWORD));
if(dwKeyLength!=0)
{
CopyMemory(pBuffer+dwPosition,pPropertyItem->KeyName,dwKeyLength);
dwPosition+=dwKeyLength;
}
//now copy remaining fields
CopyMemory(pBuffer+dwPosition,&(pPropertyItem->Format),sizeof(DWORD));
dwPosition+=(sizeof(DWORD));
// ISSUE-2000/11/21-charlwi
// this needs to be fixed for Large Int properties since they
// don't store their values in Default, Minimum, and Maximum
//IMP: the default value is always assumed to be a DWORD. This is
// because the values for properties in PropertyTable are stored
// in a seperate parameter list. See ClRtlSetPropertyTable
//
CopyMemory(pBuffer+dwPosition,&(pPropertyItem->Default),sizeof(DWORD));
dwPosition+=(sizeof(DWORD));
CopyMemory(pBuffer+dwPosition,&(pPropertyItem->Minimum),sizeof(DWORD));
dwPosition+=(sizeof(DWORD));
CopyMemory(pBuffer+dwPosition,&(pPropertyItem->Maximum),sizeof(DWORD));
dwPosition+=(sizeof(DWORD));
CopyMemory(pBuffer+dwPosition,&(pPropertyItem->Flags),sizeof(DWORD));
dwPosition+=(sizeof(DWORD));
CopyMemory(pBuffer+dwPosition,&(pPropertyItem->Offset),sizeof(DWORD));
dwPosition+=(sizeof(DWORD));
}
//
// Advance to the next property.
//
pPropertyItem++;
}
if(copying)
{
CopyMemory(pBuffer,&items,sizeof(DWORD));
status=ERROR_SUCCESS;
}
else
status=ERROR_MORE_DATA;
return status;
} // MarshallPropertyTable
DWORD
WINAPI
ClRtlUnmarshallPropertyTable(
IN OUT PRESUTIL_PROPERTY_ITEM *ppPropertyTable,
IN LPBYTE pBuffer
)
/*++
Routine Description
Rohit (rjain) : It unmarshalls the pBuffer into a RESUTIL_PROPERTY_ITEM table
Arguments
pPropertyTable - This is the resulting table
pbuffer - marshalled byte array
Return Value
returns
ERROR_SUCCESS on success,
Win32 error on error
++*/
{
PRESUTIL_PROPERTY_ITEM propertyItem;
DWORD items;
DWORD dwPosition=sizeof(DWORD);
DWORD dwLength;
DWORD i;
DWORD status=ERROR_SUCCESS;
if((pBuffer==NULL) ||(ppPropertyTable==NULL))
{
ClRtlDbgPrint( LOG_CRITICAL, "[ClRtl] Uncopy PropertyTable: Bad Argumnets\r\n");
return ERROR_BAD_ARGUMENTS;
}
CopyMemory(&items,pBuffer,sizeof(DWORD));
*ppPropertyTable=(PRESUTIL_PROPERTY_ITEM)LocalAlloc(LMEM_FIXED,(items+1)*sizeof(RESUTIL_PROPERTY_ITEM));
if(*ppPropertyTable == NULL)
{
status=GetLastError();
goto FnExit;
}
propertyItem=*ppPropertyTable;
for(i=0; i<items; i++)
{
CopyMemory(&dwLength,pBuffer+dwPosition,sizeof(DWORD));
dwPosition+=sizeof(DWORD);
propertyItem->Name = NULL;
propertyItem->Name=(LPWSTR)LocalAlloc(LMEM_FIXED,dwLength);
if(propertyItem->Name == NULL)
{
status=GetLastError();
goto FnExit;
}
CopyMemory(propertyItem->Name,pBuffer+dwPosition,dwLength);
dwPosition+=dwLength;
CopyMemory(&dwLength,pBuffer+dwPosition,sizeof(DWORD));
dwPosition+=sizeof(DWORD);
propertyItem->KeyName=NULL;
if (dwLength!=0)
{
propertyItem->KeyName=(LPWSTR)LocalAlloc(LMEM_FIXED,dwLength);
if(propertyItem->KeyName == NULL)
{
status=GetLastError();
goto FnExit;
}
CopyMemory(propertyItem->KeyName,pBuffer+dwPosition,dwLength);
dwPosition+=dwLength;
}
//now rest of the fields - all DWORDS
CopyMemory(&(propertyItem->Format),pBuffer+dwPosition,sizeof(DWORD));
dwPosition+=(sizeof(DWORD));
// ISSUE-2000/11/21-charlwi
// this needs to be fixed for Large Int properties since they don't
// store their values in Default, Minimum, and Maximum
// IMP: the default value is always passed as a DWORD. This is
// because the values for properties in PropertyTable are stored
// in a seperate parameter list so the value here won't be used.
// See ClRtlSetPropertyTable
//
CopyMemory(&(propertyItem->Default),pBuffer+dwPosition,sizeof(DWORD));
dwPosition+=(sizeof(DWORD));
CopyMemory(&(propertyItem->Minimum),pBuffer+dwPosition,sizeof(DWORD));
dwPosition+=(sizeof(DWORD));
CopyMemory(&(propertyItem->Maximum),pBuffer+dwPosition,sizeof(DWORD));
dwPosition+=(sizeof(DWORD));
CopyMemory(&(propertyItem->Flags), pBuffer+dwPosition,sizeof(DWORD));
dwPosition+=(sizeof(DWORD));
CopyMemory(&(propertyItem->Offset), pBuffer+dwPosition,sizeof(DWORD));
dwPosition+=(sizeof(DWORD));
propertyItem++;
}
// the last entry is marked NULL to indicate the end of table
propertyItem->Name=NULL;
FnExit:
return status;
} // UnmarshallPropertyTable
LPWSTR
WINAPI
ClRtlExpandEnvironmentStrings(
IN LPCWSTR pszSrc
)
/*++
Routine Description:
Expands environment strings and returns an allocated buffer containing
the result.
Arguments:
pszSrc - Source string to expand.
Return Value:
A pointer to a buffer containing the value if successful.
NULL if unsuccessful. Call GetLastError() to get more details.
--*/
{
DWORD status;
DWORD cchDst = 0;
LPWSTR pszDst = NULL;
//
// Get the required length of the output string.
//
cchDst = ExpandEnvironmentStrings( pszSrc, NULL, 0 );
if ( cchDst == 0 ) {
status = GetLastError();
} else {
//
// Allocate a buffer for the expanded string.
//
pszDst = (LPWSTR) LocalAlloc( LMEM_FIXED, cchDst * sizeof(WCHAR) );
if ( pszDst == NULL ) {
status = ERROR_NOT_ENOUGH_MEMORY;
} else {
//
// Get the expanded string.
//
cchDst = ExpandEnvironmentStrings( pszSrc, pszDst, cchDst );
if ( cchDst == 0 ) {
status = GetLastError();
LocalFree( pszDst );
pszDst = NULL;
} else {
status = ERROR_SUCCESS;
}
}
}
if ( status != ERROR_SUCCESS ) {
SetLastError( status );
}
return(pszDst);
} // ClRtlExpandEnvironmentStrings
DWORD
WINAPI
ClRtlGetPropertyFormatSize(
IN const PRESUTIL_PROPERTY_ITEM pPropertyTableItem,
IN OUT LPDWORD pcbOutPropertyListSize,
IN OUT LPDWORD pnPropertyCount
)
/*++
Routine Description:
Get the total number of bytes required for this property item format.
Arguments:
pPropertyTableItem - Supplies the property table item for the property
format whose size is to be returned.
pcbOutPropertyListSize - Supplies the size of the output buffer
required to add this property to a property list.
pnPropertyCount - The count of properties is incremented.
Return Value:
ERROR_SUCCESS - Operation was successful.
ERROR_BAD_ARGUMENTS - An argument passed to the function was bad.
ERROR_INVALID_PARAMETER -
A Win32 error code on failure.
--*/
{
WORD formatType;
DWORD valueLength;
DWORD nameLength;
PCLUSPROP_SYNTAX propSyntax;
//
// We will return a name, value pair.
// each of which must be aligned.
//
// Get the format type.
//propSyntax = (PCLUSPROP_SYNTAX) &pPropertyTableItem->Format;
//formatType = propSyntax->wFormat;
nameLength = sizeof(CLUSPROP_PROPERTY_NAME)
+ ((wcslen( pPropertyTableItem->Name ) + 1) * sizeof(WCHAR))
+ sizeof(CLUSPROP_SYNTAX); // for endmark
nameLength = ALIGN_CLUSPROP( nameLength );
valueLength = ALIGN_CLUSPROP( sizeof(CLUSPROP_WORD) );
*pcbOutPropertyListSize += (valueLength + nameLength);
*pnPropertyCount += 1;
return(ERROR_SUCCESS);
} // ClRtlGetPropertyFormatSize
DWORD
WINAPI
ClRtlGetPropertyFormat(
IN const PRESUTIL_PROPERTY_ITEM pPropertyTableItem,
IN OUT PVOID * pOutPropertyBuffer,
IN OUT LPDWORD pcbOutPropertyBufferSize
)
/*++
Routine Description:
Get the total number of bytes required for this property item format.
Arguments:
pPropertyTableItem - Supplies the property table item for the property
format whose size is to be returned.
pcbOutPropertyBuffer - Supplies the size of the output buffer
required to add this property to a property list.
pcbPropertyBufferSize - The size
Return Value:
ERROR_SUCCESS - Operation was successful.
ERROR_BAD_ARGUMENTS - An argument passed to the function was bad.
ERROR_INVALID_PARAMETER -
A Win32 error code on failure.
--*/
{
WORD formatType;
DWORD valueLength;
DWORD nameLength;
DWORD bytesReturned;
PCLUSPROP_SYNTAX propSyntax;
CLUSPROP_BUFFER_HELPER props;
props.pb = (LPBYTE) *pOutPropertyBuffer;
//
// We will return a name, value pair.
// each of which must be aligned.
//
// Get the format type.
propSyntax = (PCLUSPROP_SYNTAX) &pPropertyTableItem->Format;
formatType = propSyntax->wFormat;
//
// Copy the property name, which includes its syntax and length.
//
nameLength = (wcslen( pPropertyTableItem->Name ) + 1) * sizeof(WCHAR);
props.pName->Syntax.dw = CLUSPROP_SYNTAX_NAME;
props.pName->cbLength = nameLength;
wcscpy( props.pName->sz, pPropertyTableItem->Name );
bytesReturned = sizeof(*props.pName) + ALIGN_CLUSPROP( nameLength );
*pcbOutPropertyBufferSize -= bytesReturned;
props.pb += bytesReturned;
//
// Copy the property value, syntax, length, and ENDMARK
//
props.pWordValue->Syntax.wFormat = CLUSPROP_FORMAT_WORD;
props.pWordValue->Syntax.wType = CLUSPROP_TYPE_LIST_VALUE;
props.pName->cbLength = sizeof(WORD);
props.pWordValue->w = formatType;
bytesReturned = sizeof(*props.pWordValue) + sizeof(CLUSPROP_SYNTAX);
props.pb += sizeof(*props.pWordValue);
props.pSyntax->dw = CLUSPROP_SYNTAX_ENDMARK;
props.pb += sizeof(CLUSPROP_SYNTAX);
bytesReturned = sizeof(*props.pWordValue) + sizeof(CLUSPROP_SYNTAX);
*pcbOutPropertyBufferSize -= bytesReturned;
*pOutPropertyBuffer = props.pb;
return(ERROR_SUCCESS);
} // ClRtlGetPropertyFormat
DWORD
WINAPI
ClRtlGetPropertyFormats(
IN const PRESUTIL_PROPERTY_ITEM pPropertyTable,
OUT PVOID pOutPropertyFormatList,
IN DWORD cbOutPropertyFormatListSize,
OUT LPDWORD pcbBytesReturned,
OUT LPDWORD pcbRequired
)
/*++
Routine Description:
Gets the 'known' property formats for a given object - given its
property table.
Arguments:
pPropertyTable - Pointer to the property table to process.
pOutPropertyList - Supplies the output buffer.
cbOutPropertyListSize - Supplies the size of the output buffer.
pcbBytesReturned - The number of bytes returned in pOutPropertyList.
pcbRequired - The required number of bytes if pOutPropertyList is too small.
Return Value:
ERROR_SUCCESS - Operation was successful.
ERROR_BAD_ARGUMENTS - An argument passed to the function was bad.
ERROR_NOT_ENOUGH_MEMORY - Error allocating memory.
A Win32 error code on failure.
--*/
{
DWORD status = ERROR_SUCCESS;
DWORD itemCount = 0;
DWORD totalBufferLength = 0;
PVOID outBuffer = pOutPropertyFormatList;
DWORD bufferLength = cbOutPropertyFormatListSize;
PRESUTIL_PROPERTY_ITEM property;
*pcbBytesReturned = 0;
*pcbRequired = 0;
//
// Clear the output buffer
//
if ( pOutPropertyFormatList != NULL ) {
ZeroMemory( pOutPropertyFormatList, cbOutPropertyFormatListSize );
}
//
// Get the size of all properties for this object.
//
property = pPropertyTable;
while ( property->Name != NULL ) {
status = ClRtlGetPropertyFormatSize(
property,
&totalBufferLength,
&itemCount );
if ( status != ERROR_SUCCESS ) {
break;
}
property++;
}
//
// Continue only if the operations so far have been successful.
//
if ( status == ERROR_SUCCESS ) {
//
// Count for item count at front of return data and endmark.
//
totalBufferLength += sizeof(DWORD) + sizeof(CLUSPROP_SYNTAX);
//
// Verify the size of all the properties
//
if ( totalBufferLength > cbOutPropertyFormatListSize ) {
*pcbRequired = totalBufferLength;
totalBufferLength = 0;
if ( pOutPropertyFormatList == NULL ) {
status = ERROR_SUCCESS;
} else {
status = ERROR_MORE_DATA;
}
} else {
*(LPDWORD)outBuffer = itemCount;
outBuffer = (PVOID)( (PUCHAR)outBuffer + sizeof(itemCount) );
bufferLength -= sizeof(itemCount);
//
// Now fetch all of the property Formats.
//
property = pPropertyTable;
while ( property->Name != NULL ) {
status = ClRtlGetPropertyFormat(
property,
&outBuffer,
&itemCount );
if ( status != ERROR_SUCCESS ) {
break;
}
property++;
}
// Don't forget the ENDMARK
*(LPDWORD)outBuffer = CLUSPROP_SYNTAX_ENDMARK;
if ( (status != ERROR_SUCCESS) &&
(status != ERROR_MORE_DATA) ) {
totalBufferLength = 0;
}
}
*pcbBytesReturned = totalBufferLength;
}
return(status);
} // ClRtlGetPropertyFormats