CRegistryKey Class Reference

#include <RegistryKey.h>

Inheritance diagram for CRegistryKey:

Collaboration diagram for CRegistryKey:

Public Member Functions
STDMETHODIMP	QueryInterface (REFIID InterfaceId, PVOID *Interface)
	STDMETHODIMP_ (ULONG) AddRef()
	STDMETHODIMP_ (ULONG) Release()
	CRegistryKey (IUnknown *OuterUnknown, HANDLE hKey, BOOL CanDelete)
virtual	~CRegistryKey ()
	CRegistryKey ()
HRESULT	InitRoot (const TCHAR *pszMachineName=NULL)
HRESULT	Init (HKEY hKey, const TCHAR *pszPath, const TCHAR *pszKeyName, REGSAM CurrentAccess)
HRESULT	Uninit ()
virtual	~CRegistryKey ()
const TCHAR *	GetKeyName ()
BOOL	IsRoot ()
LONG	OpenSubkey (REGSAM samDesired, const TCHAR *pszSubkeyName, HKEY &rhKey)
LONG	OpenSubkey (REGSAM samDesired, const TCHAR *pszSubkeyName, CRegistryKey &rKey)
LONG	GetSubkeyNameMaxLength (DWORD &rdwMaxSubkeyNameLength)
void	InitSubkeyEnumeration (TCHAR *pchSubkeyNameBuffer, DWORD dwBufferSize)
LONG	GetNextSubkeyName (DWORD *pdwActualSize=NULL)
LONG	GetSubkeyCount (DWORD &rdwSubkeyCount)
LONG	GetMaxValueNameLength (DWORD &rdwMaxValueNameBufferSize)
LONG	GetMaxValueDataSize (DWORD &rdwMaxValueDataBufferSize)
void	InitValueEnumeration (TCHAR *pszValueNameBuffer, DWORD dwValueNameBufferSize, BYTE *pbValueDataBuffer, DWORD dwValueDataBufferSize, DWORD *pdwType)
LONG	GetNextValue (DWORD *pdwNameActualSize=NULL, DWORD *pdwDataActualSize=NULL)
LONG	GetValueCount (DWORD &rdwValueCount)
LONG	GetDefaultValue (DWORD *pdwType, BYTE *pbValueDataBuffer, DWORD dwValueDataBufferSize, DWORD *pdwValueDataActualSize)
DWORD	GetValue (TCHAR *pchValueName, DWORD *pdwType, LPBYTE lpValueDataBuffer, DWORD *pdwValueDataSize)
LONG	CreateSubkey (REGSAM samDesired, const TCHAR *pszKeyName, HKEY &rhKey, BOOL *pblnOpened=NULL, BOOL blnVolatile=FALSE)
LONG	GetLastWriteTime (SYSTEMTIME &st)
const TCHAR *	GetLastWriteTime ()
LONG	DeleteValue (const TCHAR *pszValueName)
LONG	DeleteSubkey (const TCHAR *pszPatternSubkeyName)
LONG	SetValue (LPCTSTR pszValueName, DWORD dwType, BYTE *lpData, DWORD dwDataSize)
TCHAR *	GetSubKeyNameByIndex (DWORD dwIndex)
LONG	GetSecurityDescriptor (SECURITY_INFORMATION SecurityInformation, PSECURITY_DESCRIPTOR pSecurityDescriptor, LPDWORD lpcbSecurityDescriptor)
LONG	GetSecurityDescriptorLength (DWORD *pdwSecurityDescriptor)
BOOL	IsPredefined ()
	operator HKEY ()

Static Public Member Functions
static BOOL	IsHive (HKEY hKey)
static const TCHAR *	GetValueTypeName (DWORD dwType)

Public Attributes
	IMP_IRegistryKey

Protected Attributes
HANDLE	m_hKey
BOOL	m_Deleted
BOOL	m_CanDelete
LONG	m_Ref

Private Attributes
DWORD	m_dwCurrentSubKeyIndex
TCHAR *	m_pchSubkeyNameBuffer
DWORD	m_dwSubkeyNameBufferSize
DWORD	m_dwCurrentValueIndex
TCHAR *	m_pszValueNameBuffer
DWORD	m_dwValueNameBufferSize
BYTE *	m_pbValueDataBuffer
DWORD	m_dwValueDataBufferSize
DWORD *	m_pdwType
HKEY	m_hKey
TCHAR *	m_pszKeyName
TCHAR *	m_pszMachineName
REGSAM	m_CurrentAccess

Detailed Description

Definition at line 17 of file registry.cpp.

Constructor & Destructor Documentation

CRegistryKey() [1/2]

CRegistryKey::CRegistryKey ( IUnknown *  OuterUnknown, HANDLE  hKey, BOOL  CanDelete  )

inline

Definition at line 40 of file registry.cpp.

: m_hKey(hKey), m_Deleted(FALSE), m_CanDelete(CanDelete), m_Ref(0){}

~CRegistryKey() [1/2]

CRegistryKey::~CRegistryKey ( )

virtual

Definition at line 51 of file registry.cpp.

{
    if (!m_Deleted)
    {
         // close key only when has not been deleted yet
         ZwClose(m_hKey);
    }
}

CRegistryKey() [2/2]

CRegistryKey::CRegistryKey

(

)

Definition at line 44 of file RegistryKey.cpp.

{
  m_pszKeyName = NULL;
  m_pszMachineName = NULL;
    m_CurrentAccess = 0;
    m_hKey = NULL;
}

~CRegistryKey() [2/2]

virtual CRegistryKey::~CRegistryKey ( )

virtual

Member Function Documentation

CreateSubkey()

LONG CRegistryKey::CreateSubkey	(	REGSAM	samDesired,
		const TCHAR *	pszKeyName,
		HKEY &	rhKey,
		BOOL *	pblnOpened = NULL,
		BOOL	blnVolatile = FALSE
	)

Definition at line 501 of file RegistryKey.cpp.

{
  DWORD dwDisposition;

    LONG nError = RegCreateKeyEx(
      m_hKey,
      pszSubkeyName,
      0,
      NULL,
      blnVolatile?REG_OPTION_VOLATILE:REG_OPTION_NON_VOLATILE,
      samDesired,
      NULL,
      &rhKey,
      &dwDisposition);

  if ((nError == ERROR_SUCCESS)&&(pblnOpened))
    *pblnOpened = dwDisposition == REG_OPENED_EXISTING_KEY;

    return nError;
}

Referenced by CRegistryTree::NewKey().

DeleteSubkey()

LONG CRegistryKey::DeleteSubkey

(

const TCHAR *

pszPatternSubkeyName

)

Definition at line 562 of file RegistryKey.cpp.

{
  return RegDeleteKey(m_hKey,pszSubkeyName);
}

Referenced by CRegistryTree::DeleteSubkeys().

DeleteValue()

LONG CRegistryKey::DeleteValue

(

const TCHAR *

pszValueName

)

Definition at line 572 of file RegistryKey.cpp.

{
    return RegDeleteValue(m_hKey,pszValueName);
}

GetDefaultValue()

LONG CRegistryKey::GetDefaultValue	(	DWORD *	pdwType,
		BYTE *	pbValueDataBuffer,
		DWORD	dwValueDataBufferSize,
		DWORD *	pdwValueDataActualSize
	)

Definition at line 450 of file RegistryKey.cpp.

{
  DWORD dwBufferSize = dwValueDataBufferSize;

    LONG nError = RegQueryValueEx(m_hKey,NULL,NULL,pdwType,pbValueDataBuffer,&dwBufferSize);

  if (pdwValueDataActualSize && (nError == ERROR_SUCCESS))
    *pdwValueDataActualSize = dwBufferSize;

  return nError;
}

GetKeyName()

const TCHAR * CRegistryKey::GetKeyName

(

)

Definition at line 156 of file RegistryKey.cpp.

{
    return m_pszKeyName?m_pszKeyName:(m_pszMachineName?m_pszMachineName:_T("\\"));
}

Referenced by CRegistryTree::DeleteSubkeys(), CRegistryTree::GetCurrentPath(), CRegistryTree::GetKey(), and OpenSubkey().

GetLastWriteTime() [1/2]

LONG CRegistryKey::GetLastWriteTime

(

SYSTEMTIME &

st

)

Definition at line 526 of file RegistryKey.cpp.

{
    FILETIME ftLocal,ft;
    LONG nError = RegQueryInfoKeyW(m_hKey,NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL,&ft);

  if (nError == ERROR_SUCCESS)
  {
    FileTimeToLocalFileTime(&ft,&ftLocal);
    FileTimeToSystemTime(&ftLocal,&st);
  }

  return nError;
}

GetLastWriteTime() [2/2]

const TCHAR * CRegistryKey::GetLastWriteTime

(

)

Definition at line 540 of file RegistryKey.cpp.

{
    SYSTEMTIME st;
    if (GetLastWriteTime(st) != ERROR_SUCCESS)
    return _T("(Cannot get time last write time)");

    static TCHAR Buffer[256];
    _stprintf(Buffer,_T("%d.%d.%d %02d:%02d:%02d"),st.wDay,st.wMonth,st.wYear,st.wHour,st.wMinute,st.wSecond);
    return Buffer;
}

GetMaxValueDataSize()

LONG CRegistryKey::GetMaxValueDataSize

(

DWORD &

rdwMaxValueDataBufferSize

)

Definition at line 380 of file RegistryKey.cpp.

{
    return RegQueryInfoKeyW(m_hKey,NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL,&rdwMaxValueDataBuferSize,NULL,NULL);
}

GetMaxValueNameLength()

LONG CRegistryKey::GetMaxValueNameLength

(

DWORD &

rdwMaxValueNameBufferSize

)

Definition at line 385 of file RegistryKey.cpp.

{
  rdwMaxValueNameBuferSize = 0;

  if (!m_hKey)
    return 0; // the root key abstraction has only subkeys (hives)

    LONG nError = RegQueryInfoKeyW(m_hKey,NULL,NULL,NULL,NULL,NULL,NULL,NULL,&rdwMaxValueNameBuferSize,NULL,NULL,NULL);

  rdwMaxValueNameBuferSize++;
  return nError;
}

GetNextSubkeyName()

LONG CRegistryKey::GetNextSubkeyName

(

DWORD *

pdwActualSize = NULL

)

Definition at line 333 of file RegistryKey.cpp.

{
  LONG nError;

  if (m_hKey == NULL)
  {
    if (m_dwCurrentSubKeyIndex < (DWORD)(m_pszMachineName?5:7))
    {
      DWORD dwIndex = m_pszMachineName?m_dwCurrentSubKeyIndex+2:m_dwCurrentSubKeyIndex;
      _tcsncpy(m_pchSubkeyNameBuffer,g_ppszHiveNames[dwIndex],m_dwSubkeyNameBufferSize);
      nError = ERROR_SUCCESS;
      if (pdwActualSize)
        *pdwActualSize = strlen(m_pchSubkeyNameBuffer);
    }
    else
    {
      nError = ERROR_NO_MORE_ITEMS;
    }
  }
  else
  {
    DWORD dwActualSize = m_dwSubkeyNameBufferSize;
    FILETIME ft;
    nError = RegEnumKeyEx(m_hKey,
                          m_dwCurrentSubKeyIndex,
                          m_pchSubkeyNameBuffer,
                          &dwActualSize,
                          NULL,
                          NULL,
                          NULL,
                          &ft);
    if (pdwActualSize)
      *pdwActualSize = dwActualSize;
  }

    m_dwCurrentSubKeyIndex++;

  if (pdwActualSize)
    *pdwActualSize = strlen(m_pchSubkeyNameBuffer);
  return nError;
}

Referenced by CRegistryTree::DeleteSubkeys(), and CRegistryTree::InternalGetSubkey().

GetNextValue()

LONG CRegistryKey::GetNextValue	(	DWORD *	pdwNameActualSize = NULL,
		DWORD *	pdwDataActualSize = NULL
	)

Definition at line 416 of file RegistryKey.cpp.

{
  if (!m_hKey)
    return ERROR_NO_MORE_ITEMS; // the root key abstraction has only subkeys (hives)

    DWORD dwValueNameBufferSize = m_dwValueNameBufferSize;
    DWORD dwValueDataBufferSize = m_dwValueDataBufferSize;
  LONG nError = RegEnumValue(m_hKey,
                            m_dwCurrentValueIndex,
                            m_pszValueNameBuffer,
                            &dwValueNameBufferSize,
                            NULL,
                            m_pdwType,
                            m_pbValueDataBuffer,
                            &dwValueDataBufferSize);

  if (pdwNameActualSize)
    *pdwNameActualSize = dwValueNameBufferSize;

  if (pdwDataActualSize)
    *pdwDataActualSize = dwValueDataBufferSize;

    m_dwCurrentValueIndex++;
    return nError;
}

GetSecurityDescriptor()

LONG CRegistryKey::GetSecurityDescriptor	(	SECURITY_INFORMATION	SecurityInformation,
		PSECURITY_DESCRIPTOR	pSecurityDescriptor,
		LPDWORD	lpcbSecurityDescriptor
	)

Definition at line 557 of file RegistryKey.cpp.

{
    return RegGetKeySecurity(m_hKey,SecurityInformation,pSecurityDescriptor,lpcbSecurityDescriptor);
}

GetSecurityDescriptorLength()

LONG CRegistryKey::GetSecurityDescriptorLength

(

DWORD *

pdwSecurityDescriptor

)

Definition at line 551 of file RegistryKey.cpp.

{
    return RegQueryInfoKeyW(m_hKey,NULL,NULL,NULL,NULL,NULL,NULL,NULL,
        NULL,NULL,pdwSecurityDescriptor,NULL);
}

GetSubkeyCount()

LONG CRegistryKey::GetSubkeyCount

(

DWORD &

rdwSubkeyCount

)

Definition at line 375 of file RegistryKey.cpp.

{
    return RegQueryInfoKeyW(m_hKey,NULL,NULL,NULL,&rdwSubkeyCount,NULL,NULL,NULL,NULL,NULL,NULL,NULL);
}

GetSubKeyNameByIndex()

TCHAR* CRegistryKey::GetSubKeyNameByIndex

(

DWORD

dwIndex

)

GetSubkeyNameMaxLength()

LONG CRegistryKey::GetSubkeyNameMaxLength

(

DWORD &

rdwMaxSubkeyNameLength

)

Definition at line 300 of file RegistryKey.cpp.

{
  if (m_hKey == NULL)
  { // root key
    rdwMaxSubkeyNameLength = 0;
    for(int i = 0; i < 7 ; i++)
    {
      size_t l = _tcslen(g_ppszHiveNames[i]);
      if (rdwMaxSubkeyNameLength < l)
        rdwMaxSubkeyNameLength = l;
    }

    rdwMaxSubkeyNameLength++; // terminating null

    return ERROR_SUCCESS;
  }

  LONG nRet;

  nRet = RegQueryInfoKey(m_hKey,NULL,NULL,NULL,NULL,&rdwMaxSubkeyNameLength,NULL,NULL,NULL,NULL,NULL,NULL);

  rdwMaxSubkeyNameLength = (nRet == ERROR_SUCCESS)?(rdwMaxSubkeyNameLength+1):0;

  return nRet;
}

Referenced by CRegistryTree::DeleteSubkeys(), and CRegistryTree::InternalGetSubkey().

GetValue()

DWORD CRegistryKey::GetValue	(	TCHAR *	pchValueName,
		DWORD *	pdwType,
		LPBYTE	lpValueDataBuffer,
		DWORD *	pdwValueDataSize
	)

Definition at line 496 of file RegistryKey.cpp.

{
    return RegQueryValueEx(m_hKey,pchValueName,NULL,pdwType,lpValueDataBuffer,pdwValueDataSize);
}

GetValueCount()

LONG CRegistryKey::GetValueCount

(

DWORD &

rdwValueCount

)

Definition at line 442 of file RegistryKey.cpp.

{
  if (!m_hKey)
    return 0; // the root key abstraction has only subkeys (hives)

  return RegQueryInfoKeyW(m_hKey,NULL,NULL,NULL,NULL,NULL,NULL,&rdwValueCount,NULL,NULL,NULL,NULL);
}

GetValueTypeName()

const TCHAR * CRegistryKey::GetValueTypeName ( DWORD  dwType )

static

Definition at line 465 of file RegistryKey.cpp.

{
    switch(dwType)
    {
    case REG_NONE:
        return _T("REG_NONE");
    case REG_SZ:
        return _T("REG_SZ");
    case REG_EXPAND_SZ:
        return _T("REG_EXPAND_SZ");
    case REG_BINARY:
        return _T("REG_BINARY");
    case REG_DWORD_LITTLE_ENDIAN:
        return _T("REG_DWORD_LITTLE_ENDIAN");
    case REG_DWORD_BIG_ENDIAN:
        return _T("REG_DWORD_BIG_ENDIAN");
    case REG_LINK:
        return _T("REG_LINK");
    case REG_MULTI_SZ:
        return _T("REG_MULTI_SZ");
    case REG_RESOURCE_LIST:
        return _T("REG_RESOURCE_LIST");
    case REG_FULL_RESOURCE_DESCRIPTOR:
        return _T("REG_FULL_RESOURCE_DESCRIPTOR");
    case REG_RESOURCE_REQUIREMENTS_LIST:
        return _T("REG_RESOURCE_REQUIREMENTS_LIST");
    default:
        return _T("Unkown Type");
    }
}

Referenced by CShellCommandDir::Execute(), and CShellCommandValue::Execute().

Init()

HRESULT CRegistryKey::Init	(	HKEY	hKey,
		const TCHAR *	pszPath,
		const TCHAR *	pszKeyName,
		REGSAM	CurrentAccess
	)

Definition at line 91 of file RegistryKey.cpp.

{
  HRESULT hr = Uninit();
  if (FAILED(hr))
    return hr;

  if (!pszKeyName || !hKey)
    return E_INVALIDARG;

  // copy key name name
  size_t size = _tcslen(pszKeyName);
  if (pszPath)
    size += _tcslen(pszPath);

  m_pszKeyName = new (std::nothrow) TCHAR [size+2];
  if (!m_pszKeyName)
    return E_OUTOFMEMORY;
  _stprintf(m_pszKeyName,_T("%s%s\\"),pszPath?pszPath:_T(""),pszKeyName);

    m_CurrentAccess = CurrentAccess;
    m_hKey = hKey;
  ASSERT(m_hKey);

  return S_OK;
}

Referenced by CRegistryTree::InternalGetSubkey(), and OpenSubkey().

InitRoot()

HRESULT CRegistryKey::InitRoot

(

const TCHAR *

pszMachineName = NULL

)

Definition at line 52 of file RegistryKey.cpp.

{
  if ((pszMachineName)&&
      ((_tcslen(pszMachineName) < 3)||
       (pszMachineName[0] != _T('\\'))||
       (pszMachineName[1] != _T('\\'))
      )
    )
  {
    return E_INVALIDARG;
  }

  HRESULT hr = Uninit();
  if (FAILED(hr))
    return hr;

  if (pszMachineName)
  { // copy machine name
    size_t size = _tcslen(pszMachineName);

    m_pszMachineName = new (std::nothrow) TCHAR [size+2];
    if (!m_pszMachineName)
      return E_OUTOFMEMORY;
    _tcscpy(m_pszMachineName,pszMachineName);
    m_pszMachineName[size] = _T('\\');
    m_pszMachineName[size+1] = 0;
  }
  else
  {
    m_pszMachineName = NULL; // local registry
  }

  ASSERT(m_pszKeyName == NULL);
    m_CurrentAccess = 0;
    ASSERT(m_hKey == NULL);

  return S_OK;
}

Referenced by CRegistryTree::CRegistryTree(), CRegistryTree::GetKey(), and CRegistryTree::SetMachineName().

InitSubkeyEnumeration()

void CRegistryKey::InitSubkeyEnumeration	(	TCHAR *	pchSubkeyNameBuffer,
		DWORD	dwBufferSize
	)

Definition at line 326 of file RegistryKey.cpp.

{
  m_pchSubkeyNameBuffer = pszSubkeyNameBuffer;
  m_dwSubkeyNameBufferSize = dwBufferSize;
  m_dwCurrentSubKeyIndex = 0;
}

Referenced by CRegistryTree::DeleteSubkeys(), and CRegistryTree::InternalGetSubkey().

InitValueEnumeration()

void CRegistryKey::InitValueEnumeration	(	TCHAR *	pszValueNameBuffer,
		DWORD	dwValueNameBufferSize,
		BYTE *	pbValueDataBuffer,
		DWORD	dwValueDataBufferSize,
		DWORD *	pdwType
	)

Definition at line 398 of file RegistryKey.cpp.

{
  m_pszValueNameBuffer = pszValueNameBuffer;
  m_dwValueNameBufferSize = dwValueNameBufferSize;
  m_pbValueDataBuffer = pbValueDataBuffer;
  m_dwValueDataBufferSize = dwValueDataBufferSize;
  m_pdwType = pdwType;

    m_dwCurrentValueIndex = 0;
}

IsHive()

BOOL CRegistryKey::IsHive ( HKEY  hKey )

static

Definition at line 140 of file RegistryKey.cpp.

{
    return ((hKey == HKEY_CLASSES_ROOT)||
          (hKey == HKEY_CURRENT_USER)||
          (hKey == HKEY_LOCAL_MACHINE)||
          (hKey == HKEY_USERS)||
          (hKey == HKEY_PERFORMANCE_DATA)||
          (hKey == HKEY_CURRENT_CONFIG)||
          (hKey == HKEY_DYN_DATA));
}

Referenced by Uninit().

IsPredefined()

BOOL CRegistryKey::IsPredefined

(

)

IsRoot()

BOOL CRegistryKey::IsRoot

(

)

Definition at line 161 of file RegistryKey.cpp.

{
  return m_hKey == NULL;
}

Referenced by CRegistryTree::CRegistryTree(), CRegistryTree::GetKey(), CRegistryTree::GotoRoot(), CRegistryTree::InternalChangeCurrentKey(), CRegistryTree::IsCurrentRoot(), and CRegistryTree::~CRegistryTree().

OpenSubkey() [1/2]

LONG CRegistryKey::OpenSubkey	(	REGSAM	samDesired,
		const TCHAR *	pszSubkeyName,
		HKEY &	rhKey
	)

Definition at line 166 of file RegistryKey.cpp.

{
    if (m_hKey == NULL)
  { // subkey of root key is hive root key.
    if ((_tcsicmp(pszSubkeyName,_T("HKCR")) == 0)||
                (_tcsicmp(pszSubkeyName,_T("HKEY_CLASSES_ROOT")) == 0))
    {
      rhKey = HKEY_CLASSES_ROOT;

      if (m_pszMachineName)
        return ERROR_FILE_NOT_FOUND;

      return ERROR_SUCCESS;
    }
    else if ((_tcsicmp(pszSubkeyName,_T("HKCU")) == 0)||
             (_tcsicmp(pszSubkeyName,_T("HKEY_CURRENT_USER")) == 0))
    {
      rhKey = HKEY_CURRENT_USER;

      if (m_pszMachineName)
        return ERROR_FILE_NOT_FOUND;

      return ERROR_SUCCESS;
    }
    else if ((_tcsicmp(pszSubkeyName,_T("HKLM")) == 0)||
             (_tcsicmp(pszSubkeyName,_T("HKEY_LOCAL_MACHINE")) == 0))
    {
      rhKey = HKEY_LOCAL_MACHINE;

      if (m_pszMachineName)
        return RegConnectRegistry(m_pszMachineName,rhKey,&rhKey);

      return ERROR_SUCCESS;
    }
    else if ((_tcsicmp(pszSubkeyName,_T("HKU")) == 0)||
             (_tcsicmp(pszSubkeyName,_T("HKEY_USERS")) == 0))
    {
      rhKey = HKEY_USERS;

      if (m_pszMachineName)
        return RegConnectRegistry(m_pszMachineName,rhKey,&rhKey);

      return ERROR_SUCCESS;
    }
    else if ((_tcsicmp(pszSubkeyName,_T("HKPD")) == 0)||
             (_tcsicmp(pszSubkeyName,_T("HKEY_PERFORMANCE_DATA")) == 0))
    {
      rhKey = HKEY_PERFORMANCE_DATA;

      if (m_pszMachineName)
        return RegConnectRegistry(m_pszMachineName,rhKey,&rhKey);

      return ERROR_SUCCESS;
    }
    else if ((_tcsicmp(pszSubkeyName,_T("HKDD")) == 0)||
             (_tcsicmp(pszSubkeyName,_T("HKEY_DYN_DATA")) == 0))
    {
      rhKey = HKEY_DYN_DATA;

      if (m_pszMachineName)
        return RegConnectRegistry(m_pszMachineName,rhKey,&rhKey);

      return ERROR_SUCCESS;
    }
    else if ((_tcsicmp(pszSubkeyName,_T("HKCC")) == 0)||
             (_tcsicmp(pszSubkeyName,_T("HKEY_CURRENT_CONFIG")) == 0))
    {
      rhKey = HKEY_CURRENT_CONFIG;

      if (m_pszMachineName)
      {
        TCHAR *pch = m_pszMachineName;
        while (*pch)
          pch++;
        pch--;

        ASSERT(*pch == _T('\\'));
        if (*pch != _T('\\'))
          return ERROR_INTERNAL_ERROR;

        *pch = 0;

        LONG nError = RegConnectRegistry(m_pszMachineName,rhKey,&rhKey);

        *pch = _T('\\');

        return nError;
      }

      return ERROR_SUCCESS;
    }
    else
    {
      return ERROR_FILE_NOT_FOUND;
    }
  }

    return RegOpenKeyEx(m_hKey,pszSubkeyName,0,samDesired,&rhKey);
}

Referenced by CRegistryTree::DeleteSubkeys(), CRegistryTree::InternalGetSubkey(), and OpenSubkey().

OpenSubkey() [2/2]

LONG CRegistryKey::OpenSubkey	(	REGSAM	samDesired,
		const TCHAR *	pszSubkeyName,
		CRegistryKey &	rKey
	)

Definition at line 266 of file RegistryKey.cpp.

{
  HKEY hKey;
  LONG nError = OpenSubkey(samDesired, pszSubkeyName, hKey);

  if (nError == ERROR_SUCCESS)
  {
    const TCHAR *pszKeyName = GetKeyName();
    size_t size = _tcslen(pszKeyName) + _tcslen(pszSubkeyName) + 1;
    TCHAR *pszSubkeyFullName = new (std::nothrow) TCHAR [size];
    if (!pszSubkeyFullName)
    {
      nError = RegCloseKey(hKey);
      ASSERT(nError == ERROR_SUCCESS);
      return ERROR_OUTOFMEMORY;
    }
    _tcscpy(pszSubkeyFullName,pszKeyName);
    _tcscat(pszSubkeyFullName,pszSubkeyName);
    HRESULT hr = rKey.Init(hKey,GetKeyName(),pszSubkeyName,samDesired);
    delete[] pszSubkeyName;
    if (FAILED(hr))
    {
      nError = RegCloseKey(hKey);
      ASSERT(nError == ERROR_SUCCESS);
      if (hr == (HRESULT)E_OUTOFMEMORY)
        return ERROR_OUTOFMEMORY;
      else
        return ERROR_INTERNAL_ERROR;
    }
  }

  return nError;
}

operator HKEY()

CRegistryKey::operator HKEY ( )

inline

Definition at line 239 of file RegistryKey.h.

{return m_hKey;};

QueryInterface()

NTSTATUS NTAPI CRegistryKey::QueryInterface	(	REFIID	InterfaceId,
		PVOID *	Interface
	)

Definition at line 63 of file registry.cpp.

{
    UNICODE_STRING GuidString;

    DPRINT("CRegistryKey::QueryInterface entered\n");
    if (IsEqualGUIDAligned(refiid, IID_IRegistryKey) ||
        IsEqualGUIDAligned(refiid, IID_IUnknown))
    {
        *Output = PVOID(PREGISTRYKEY(this));
        PUNKNOWN(*Output)->AddRef();
        return STATUS_SUCCESS;
    }

    if (RtlStringFromGUID(refiid, &GuidString) == STATUS_SUCCESS)
    {
        DPRINT1("CRegistryKey::QueryInterface no interface!!! iface %S\n", GuidString.Buffer);
        RtlFreeUnicodeString(&GuidString);
    }

    return STATUS_UNSUCCESSFUL;
}

Referenced by PcNewRegistryKey().

SetValue()

LONG CRegistryKey::SetValue	(	LPCTSTR	pszValueName,
		DWORD	dwType,
		BYTE *	lpData,
		DWORD	dwDataSize
	)

Definition at line 567 of file RegistryKey.cpp.

{
    return RegSetValueEx(m_hKey,pszValueName,0,dwType,lpData,dwDataSize);
}

STDMETHODIMP_() [1/2]

CRegistryKey::STDMETHODIMP_ ( ULONG  )

inline

Definition at line 22 of file registry.cpp.

    {
        InterlockedIncrement(&m_Ref);
        return m_Ref;
    }

STDMETHODIMP_() [2/2]

CRegistryKey::STDMETHODIMP_ ( ULONG  )

inline

Definition at line 27 of file registry.cpp.

    {
        InterlockedDecrement(&m_Ref);

        if (!m_Ref)
        {
            delete this;
            return 0;
        }
        return m_Ref;
    }

Uninit()

HRESULT CRegistryKey::Uninit

(

)

Definition at line 117 of file RegistryKey.cpp.

{
    if (m_pszKeyName)
  {
    delete [] m_pszKeyName;
    m_pszKeyName = NULL;
  }

    if (m_pszMachineName)
  {
    delete [] m_pszMachineName;
    m_pszMachineName = NULL;
  }

  LONG nError = ERROR_SUCCESS;
    if((m_hKey != NULL)&&(!IsHive(m_hKey)))
        nError = RegCloseKey(m_hKey);

  m_hKey = NULL;

  return (nError == ERROR_SUCCESS)?S_OK:E_FAIL;
}

Referenced by Init(), and InitRoot().

Member Data Documentation

IMP_IRegistryKey

CRegistryKey::IMP_IRegistryKey

Definition at line 39 of file registry.cpp.

m_CanDelete

BOOL CRegistryKey::m_CanDelete

protected

Definition at line 47 of file registry.cpp.

m_CurrentAccess

REGSAM CRegistryKey::m_CurrentAccess

private

Definition at line 255 of file RegistryKey.h.

Referenced by CRegistryKey(), Init(), and InitRoot().

m_Deleted

BOOL CRegistryKey::m_Deleted

protected

Definition at line 46 of file registry.cpp.

Referenced by ~CRegistryKey().

m_dwCurrentSubKeyIndex

DWORD CRegistryKey::m_dwCurrentSubKeyIndex

private

Definition at line 239 of file RegistryKey.h.

Referenced by GetNextSubkeyName(), and InitSubkeyEnumeration().

m_dwCurrentValueIndex

DWORD CRegistryKey::m_dwCurrentValueIndex

private

Definition at line 245 of file RegistryKey.h.

Referenced by GetNextValue(), and InitValueEnumeration().

m_dwSubkeyNameBufferSize

DWORD CRegistryKey::m_dwSubkeyNameBufferSize

private

Definition at line 243 of file RegistryKey.h.

Referenced by GetNextSubkeyName(), and InitSubkeyEnumeration().

m_dwValueDataBufferSize

DWORD CRegistryKey::m_dwValueDataBufferSize

private

Definition at line 249 of file RegistryKey.h.

Referenced by GetNextValue(), and InitValueEnumeration().

m_dwValueNameBufferSize

DWORD CRegistryKey::m_dwValueNameBufferSize

private

Definition at line 247 of file RegistryKey.h.

Referenced by GetNextValue(), and InitValueEnumeration().

m_hKey [1/2]

HANDLE CRegistryKey::m_hKey

protected

Definition at line 45 of file registry.cpp.

Referenced by CreateSubkey(), CRegistryKey(), DeleteSubkey(), DeleteValue(), GetDefaultValue(), GetLastWriteTime(), GetMaxValueDataSize(), GetMaxValueNameLength(), GetNextSubkeyName(), GetNextValue(), GetSecurityDescriptor(), GetSecurityDescriptorLength(), GetSubkeyCount(), GetSubkeyNameMaxLength(), GetValue(), GetValueCount(), Init(), InitRoot(), IsRoot(), OpenSubkey(), operator HKEY(), SetValue(), Uninit(), and ~CRegistryKey().

m_hKey [2/2]

HKEY CRegistryKey::m_hKey

private

Definition at line 252 of file RegistryKey.h.

m_pbValueDataBuffer

BYTE* CRegistryKey::m_pbValueDataBuffer

private

Definition at line 248 of file RegistryKey.h.

Referenced by GetNextValue(), and InitValueEnumeration().

m_pchSubkeyNameBuffer

TCHAR* CRegistryKey::m_pchSubkeyNameBuffer

private

Definition at line 242 of file RegistryKey.h.

Referenced by GetNextSubkeyName(), and InitSubkeyEnumeration().

m_pdwType

DWORD* CRegistryKey::m_pdwType

private

Definition at line 250 of file RegistryKey.h.

Referenced by GetNextValue(), and InitValueEnumeration().

m_pszKeyName

TCHAR* CRegistryKey::m_pszKeyName

private

Definition at line 253 of file RegistryKey.h.

Referenced by CRegistryKey(), GetKeyName(), Init(), InitRoot(), and Uninit().

m_pszMachineName

TCHAR* CRegistryKey::m_pszMachineName

private

Definition at line 254 of file RegistryKey.h.

Referenced by CRegistryKey(), GetKeyName(), GetNextSubkeyName(), InitRoot(), OpenSubkey(), and Uninit().

m_pszValueNameBuffer

TCHAR* CRegistryKey::m_pszValueNameBuffer

private

Definition at line 246 of file RegistryKey.h.

Referenced by GetNextValue(), and InitValueEnumeration().

m_Ref

LONG CRegistryKey::m_Ref

protected

Definition at line 48 of file registry.cpp.

Referenced by STDMETHODIMP_().

---

The documentation for this class was generated from the following files:

• drivers/wdm/audio/backpln/portcls/registry.cpp
• modules/rosapps/applications/sysutils/regexpl/RegistryKey.h
• modules/rosapps/applications/sysutils/regexpl/RegistryKey.cpp