1 ///////////////////////////////////////////////////////////////////////////////
2 // Name: msw/registry.cpp
3 // Purpose: implementation of registry classes and functions
4 // Author: Vadim Zeitlin
8 // Copyright: (c) 1998 Vadim Zeitlin <zeitlin@dptmaths.ens-cachan.fr>
9 // Licence: wxWindows license
10 // TODO: - parsing of registry key names
11 // - support of other (than REG_SZ/REG_DWORD) registry types
12 // - add high level functions (RegisterOleServer, ...)
13 ///////////////////////////////////////////////////////////////////////////////
16 #pragma implementation "registry.h"
19 // for compilers that support precompilation, includes "wx.h".
20 #include "wx/wxprec.h"
26 // other wxWindows headers
27 #include "wx/string.h"
30 #include "wx/config.h" // for wxExpandEnvVars
37 #define WIN32_LEAN_AND_MEAN
43 #include <stdlib.h> // for _MAX_PATH
50 #define HKEY_DEFINED // already defined in windows.h
51 #include "wx/msw/registry.h"
53 // some registry functions don't like signed chars
54 typedef unsigned char *RegString
;
56 // ----------------------------------------------------------------------------
58 // ----------------------------------------------------------------------------
60 // the standard key names, short names and handles all bundled together for
66 const wxChar
*szShortName
;
70 { HKEY_CLASSES_ROOT
, _T("HKEY_CLASSES_ROOT"), _T("HKCR") },
72 { HKEY_CURRENT_USER
, _T("HKEY_CURRENT_USER"), _T("HKCU") },
73 { HKEY_LOCAL_MACHINE
, _T("HKEY_LOCAL_MACHINE"), _T("HKLM") },
74 { HKEY_USERS
, _T("HKEY_USERS"), _T("HKU") }, // short name?
75 { HKEY_PERFORMANCE_DATA
, _T("HKEY_PERFORMANCE_DATA"), _T("HKPD") },
77 { HKEY_CURRENT_CONFIG
, _T("HKEY_CURRENT_CONFIG"), _T("HKCC") },
79 { HKEY_DYN_DATA
, _T("HKEY_DYN_DATA"), _T("HKDD") }, // short name?
81 #endif //WINVER >= 4.0
85 // the registry name separator (perhaps one day MS will change it to '/' ;-)
86 #define REG_SEPARATOR _T('\\')
88 // useful for Windows programmers: makes somewhat more clear all these zeroes
89 // being passed to Windows APIs
90 #define RESERVED (NULL)
92 // ----------------------------------------------------------------------------
94 // ----------------------------------------------------------------------------
95 // @ const_cast<> is not yet supported by all compilers
96 #define CONST_CAST ((wxRegKey *)this)->
99 #define m_dwLastError CONST_CAST m_dwLastError
102 // ----------------------------------------------------------------------------
103 // non member functions
104 // ----------------------------------------------------------------------------
106 // removes the trailing backslash from the string if it has one
107 static inline void RemoveTrailingSeparator(wxString
& str
);
109 // returns TRUE if given registry key exists
110 static bool KeyExists(WXHKEY hRootKey
, const wxChar
*szKey
);
112 // combines value and key name (uses static buffer!)
113 static const wxChar
*GetFullName(const wxRegKey
*pKey
,
114 const wxChar
*szValue
= NULL
);
116 // ============================================================================
117 // implementation of wxRegKey class
118 // ============================================================================
120 // ----------------------------------------------------------------------------
121 // static functions and variables
122 // ----------------------------------------------------------------------------
124 const size_t wxRegKey::nStdKeys
= WXSIZEOF(aStdKeys
);
126 // @@ should take a `StdKey key', but as it's often going to be used in loops
127 // it would require casts in user code.
128 const wxChar
*wxRegKey::GetStdKeyName(size_t key
)
130 // return empty string if key is invalid
131 wxCHECK_MSG( key
< nStdKeys
, _T(""), _T("invalid key in wxRegKey::GetStdKeyName") );
133 return aStdKeys
[key
].szName
;
136 const wxChar
*wxRegKey::GetStdKeyShortName(size_t key
)
138 // return empty string if key is invalid
139 wxCHECK( key
< nStdKeys
, _T("") );
141 return aStdKeys
[key
].szShortName
;
144 wxRegKey::StdKey
wxRegKey::ExtractKeyName(wxString
& strKey
)
146 wxString strRoot
= strKey
.Left(REG_SEPARATOR
);
150 for ( ui
= 0; ui
< nStdKeys
; ui
++ ) {
151 if ( strRoot
.CmpNoCase(aStdKeys
[ui
].szName
) == 0 ||
152 strRoot
.CmpNoCase(aStdKeys
[ui
].szShortName
) == 0 ) {
153 hRootKey
= aStdKeys
[ui
].hkey
;
158 if ( ui
== nStdKeys
) {
159 wxFAIL_MSG(_T("invalid key prefix in wxRegKey::ExtractKeyName."));
161 hRootKey
= HKEY_CLASSES_ROOT
;
164 strKey
= strKey
.After(REG_SEPARATOR
);
165 if ( !strKey
.IsEmpty() && strKey
.Last() == REG_SEPARATOR
)
166 strKey
.Truncate(strKey
.Len() - 1);
169 return (wxRegKey::StdKey
)(int)hRootKey
;
172 wxRegKey::StdKey
wxRegKey::GetStdKeyFromHkey(WXHKEY hkey
)
174 for ( size_t ui
= 0; ui
< nStdKeys
; ui
++ ) {
175 if ( (int) aStdKeys
[ui
].hkey
== (int) hkey
)
179 wxFAIL_MSG(_T("non root hkey passed to wxRegKey::GetStdKeyFromHkey."));
184 // ----------------------------------------------------------------------------
186 // ----------------------------------------------------------------------------
191 m_hRootKey
= (WXHKEY
) aStdKeys
[HKCR
].hkey
;
195 wxRegKey::wxRegKey(const wxString
& strKey
) : m_strKey(strKey
)
197 m_hRootKey
= (WXHKEY
) aStdKeys
[ExtractKeyName(m_strKey
)].hkey
;
198 m_hKey
= (WXHKEY
) NULL
;
202 // parent is a predefined (and preopened) key
203 wxRegKey::wxRegKey(StdKey keyParent
, const wxString
& strKey
) : m_strKey(strKey
)
205 RemoveTrailingSeparator(m_strKey
);
206 m_hRootKey
= (WXHKEY
) aStdKeys
[keyParent
].hkey
;
207 m_hKey
= (WXHKEY
) NULL
;
211 // parent is a normal regkey
212 wxRegKey::wxRegKey(const wxRegKey
& keyParent
, const wxString
& strKey
)
213 : m_strKey(keyParent
.m_strKey
)
215 // combine our name with parent's to get the full name
216 if ( !m_strKey
.IsEmpty() &&
217 (strKey
.IsEmpty() || strKey
[0] != REG_SEPARATOR
) ) {
218 m_strKey
+= REG_SEPARATOR
;
222 RemoveTrailingSeparator(m_strKey
);
224 m_hRootKey
= keyParent
.m_hRootKey
;
225 m_hKey
= (WXHKEY
) NULL
;
229 // dtor closes the key releasing system resource
230 wxRegKey::~wxRegKey()
235 // ----------------------------------------------------------------------------
236 // change the key name/hkey
237 // ----------------------------------------------------------------------------
239 // set the full key name
240 void wxRegKey::SetName(const wxString
& strKey
)
245 m_hRootKey
= (WXHKEY
) aStdKeys
[ExtractKeyName(m_strKey
)].hkey
;
248 // the name is relative to the parent key
249 void wxRegKey::SetName(StdKey keyParent
, const wxString
& strKey
)
254 RemoveTrailingSeparator(m_strKey
);
255 m_hRootKey
= (WXHKEY
) aStdKeys
[keyParent
].hkey
;
258 // the name is relative to the parent key
259 void wxRegKey::SetName(const wxRegKey
& keyParent
, const wxString
& strKey
)
263 // combine our name with parent's to get the full name
264 m_strKey
= keyParent
.m_strKey
;
265 if ( !strKey
.IsEmpty() && strKey
[0] != REG_SEPARATOR
)
266 m_strKey
+= REG_SEPARATOR
;
269 RemoveTrailingSeparator(m_strKey
);
271 m_hRootKey
= keyParent
.m_hRootKey
;
274 // hKey should be opened and will be closed in wxRegKey dtor
275 void wxRegKey::SetHkey(WXHKEY hKey
)
282 // ----------------------------------------------------------------------------
283 // info about the key
284 // ----------------------------------------------------------------------------
286 // returns TRUE if the key exists
287 bool wxRegKey::Exists() const
289 // opened key has to exist, try to open it if not done yet
290 return IsOpened() ? TRUE
: KeyExists(m_hRootKey
, m_strKey
);
293 // returns the full name of the key (prefix is abbreviated if bShortPrefix)
294 wxString
wxRegKey::GetName(bool bShortPrefix
) const
296 StdKey key
= GetStdKeyFromHkey((StdKey
) m_hRootKey
);
297 wxString str
= bShortPrefix
? aStdKeys
[key
].szShortName
298 : aStdKeys
[key
].szName
;
299 if ( !m_strKey
.IsEmpty() )
300 str
<< "\\" << m_strKey
;
305 #if defined( __GNUWIN32__ ) && !defined(wxUSE_NORLANDER_HEADERS)
306 bool wxRegKey::GetKeyInfo(size_t* pnSubKeys
,
309 size_t* pnMaxValueLen
) const
311 bool wxRegKey::GetKeyInfo(ulong
*pnSubKeys
,
314 ulong
*pnMaxValueLen
) const
317 #if defined(__WIN32__) && !defined(__TWIN32__)
318 m_dwLastError
= ::RegQueryInfoKey
322 NULL
, // (ptr to) size of class name buffer
324 pnSubKeys
, // [out] number of subkeys
325 pnMaxKeyLen
, // [out] max length of a subkey name
326 NULL
, // longest subkey class name
327 pnValues
, // [out] number of values
328 pnMaxValueLen
, // [out] max length of a value name
329 NULL
, // longest value data
330 NULL
, // security descriptor
331 NULL
// time of last modification
334 if ( m_dwLastError
!= ERROR_SUCCESS
) {
335 wxLogSysError(m_dwLastError
, _("can't get info about registry key '%s'"),
342 wxFAIL_MSG("GetKeyInfo() not implemented");
348 // ----------------------------------------------------------------------------
350 // ----------------------------------------------------------------------------
352 // opens key (it's not an error to call Open() on an already opened key)
353 bool wxRegKey::Open()
359 m_dwLastError
= RegOpenKey((HKEY
) m_hRootKey
, m_strKey
, &tmpKey
);
360 if ( m_dwLastError
!= ERROR_SUCCESS
) {
361 wxLogSysError(m_dwLastError
, _("can't open registry key '%s'"),
367 m_hKey
= (WXHKEY
) tmpKey
;
372 // creates key, failing if it exists and !bOkIfExists
373 bool wxRegKey::Create(bool bOkIfExists
)
375 // check for existence only if asked (i.e. order is important!)
376 if ( !bOkIfExists
&& Exists() ) {
384 m_dwLastError
= RegCreateKey((HKEY
) m_hRootKey
, m_strKey
, &tmpKey
);
385 if ( m_dwLastError
!= ERROR_SUCCESS
) {
386 wxLogSysError(m_dwLastError
, _("can't create registry key '%s'"),
392 m_hKey
= (WXHKEY
) tmpKey
;
397 // close the key, it's not an error to call it when not opened
398 bool wxRegKey::Close()
401 m_dwLastError
= RegCloseKey((HKEY
) m_hKey
);
402 if ( m_dwLastError
!= ERROR_SUCCESS
) {
403 wxLogSysError(m_dwLastError
, _("can't close registry key '%s'"),
417 // ----------------------------------------------------------------------------
418 // delete keys/values
419 // ----------------------------------------------------------------------------
420 bool wxRegKey::DeleteSelf()
425 // it already doesn't exist - ok!
430 // prevent a buggy program from erasing one of the root registry keys or an
431 // immediate subkey (i.e. one which doesn't have '\\' inside) of any other
432 // key except HKCR (HKCR has some "deleteable" subkeys)
433 if ( m_strKey
.IsEmpty() || (m_hRootKey
!= HKCR
&&
434 m_strKey
.Find(REG_SEPARATOR
) == wxNOT_FOUND
) ) {
435 wxLogError(_("Registry key '%s' is needed for normal system operation,\n"
436 "deleting it will leave your system in unusable state:\n"
437 "operation aborted."), GetFullName(this));
442 // we can't delete keys while enumerating because it confuses GetNextKey, so
443 // we first save the key names and then delete them all
444 wxArrayString astrSubkeys
;
448 bool bCont
= GetFirstKey(strKey
, lIndex
);
450 astrSubkeys
.Add(strKey
);
452 bCont
= GetNextKey(strKey
, lIndex
);
455 size_t nKeyCount
= astrSubkeys
.Count();
456 for ( size_t nKey
= 0; nKey
< nKeyCount
; nKey
++ ) {
457 wxRegKey
key(*this, astrSubkeys
[nKey
]);
458 if ( !key
.DeleteSelf() )
462 // now delete this key itself
465 m_dwLastError
= RegDeleteKey((HKEY
) m_hRootKey
, m_strKey
);
466 if ( m_dwLastError
!= ERROR_SUCCESS
) {
467 wxLogSysError(m_dwLastError
, _("can't delete key '%s'"),
475 bool wxRegKey::DeleteKey(const wxChar
*szKey
)
480 wxRegKey
key(*this, szKey
);
481 return key
.DeleteSelf();
484 bool wxRegKey::DeleteValue(const wxChar
*szValue
)
489 #if defined(__WIN32__) && !defined(__TWIN32__)
490 m_dwLastError
= RegDeleteValue((HKEY
) m_hKey
, WXSTRINGCAST szValue
);
491 if ( m_dwLastError
!= ERROR_SUCCESS
) {
492 wxLogSysError(m_dwLastError
, _("can't delete value '%s' from key '%s'"),
493 szValue
, GetName().c_str());
497 // named registry values don't exist in Win16 world
498 wxASSERT( IsEmpty(szValue
) );
500 // just set the (default and unique) value of the key to ""
501 m_dwLastError
= RegSetValue((HKEY
) m_hKey
, NULL
, REG_SZ
, "", RESERVED
);
502 if ( m_dwLastError
!= ERROR_SUCCESS
) {
503 wxLogSysError(m_dwLastError
, _("can't delete value of key '%s'"),
512 // ----------------------------------------------------------------------------
513 // access to values and subkeys
514 // ----------------------------------------------------------------------------
516 // return TRUE if value exists
517 bool wxRegKey::HasValue(const wxChar
*szValue
) const
519 // this function should be silent, so suppress possible messages from Open()
523 if ( CONST_CAST
Open() ) {
524 return RegQueryValueEx((HKEY
) m_hKey
, WXSTRINGCAST szValue
, RESERVED
,
525 NULL
, NULL
, NULL
) == ERROR_SUCCESS
;
530 // only unnamed value exists
531 return IsEmpty(szValue
);
535 // returns TRUE if this key has any values
536 bool wxRegKey::HasValues() const
538 // suppress possible messages from GetFirstValue()
541 // just call GetFirstValue with dummy parameters
544 return CONST_CAST
GetFirstValue(str
, l
);
547 // returns TRUE if this key has any subkeys
548 bool wxRegKey::HasSubkeys() const
550 // suppress possible messages from GetFirstKey()
553 // just call GetFirstKey with dummy parameters
556 return CONST_CAST
GetFirstKey(str
, l
);
559 // returns TRUE if given subkey exists
560 bool wxRegKey::HasSubKey(const wxChar
*szKey
) const
562 // this function should be silent, so suppress possible messages from Open()
565 if ( CONST_CAST
Open() )
566 return KeyExists(m_hKey
, szKey
);
571 wxRegKey::ValueType
wxRegKey::GetValueType(const wxChar
*szValue
) const
574 if ( ! CONST_CAST
Open() )
578 m_dwLastError
= RegQueryValueEx((HKEY
) m_hKey
, WXSTRINGCAST szValue
, RESERVED
,
579 &dwType
, NULL
, NULL
);
580 if ( m_dwLastError
!= ERROR_SUCCESS
) {
581 wxLogSysError(m_dwLastError
, _("can't read value of key '%s'"),
586 return (ValueType
)dwType
;
588 return IsEmpty(szValue
) ? Type_String
: Type_None
;
593 bool wxRegKey::SetValue(const wxChar
*szValue
, long lValue
)
596 wxFAIL_MSG("RegSetValueEx not implemented by TWIN32");
599 if ( CONST_CAST
Open() ) {
600 m_dwLastError
= RegSetValueEx((HKEY
) m_hKey
, szValue
, (DWORD
) RESERVED
, REG_DWORD
,
601 (RegString
)&lValue
, sizeof(lValue
));
602 if ( m_dwLastError
== ERROR_SUCCESS
)
606 wxLogSysError(m_dwLastError
, _("can't set value of '%s'"),
607 GetFullName(this, szValue
));
612 bool wxRegKey::QueryValue(const wxChar
*szValue
, long *plValue
) const
614 if ( CONST_CAST
Open() ) {
615 DWORD dwType
, dwSize
= sizeof(DWORD
);
616 RegString pBuf
= (RegString
)plValue
;
617 m_dwLastError
= RegQueryValueEx((HKEY
) m_hKey
, WXSTRINGCAST szValue
, RESERVED
,
618 &dwType
, pBuf
, &dwSize
);
619 if ( m_dwLastError
!= ERROR_SUCCESS
) {
620 wxLogSysError(m_dwLastError
, _("can't read value of key '%s'"),
625 // check that we read the value of right type
626 wxASSERT_MSG( dwType
== REG_DWORD
,
627 _T("Type mismatch in wxRegKey::QueryValue().") );
638 bool wxRegKey::QueryValue(const wxChar
*szValue
, wxString
& strValue
) const
640 if ( CONST_CAST
Open() ) {
642 // first get the type and size of the data
643 DWORD dwType
, dwSize
;
644 m_dwLastError
= RegQueryValueEx((HKEY
) m_hKey
, WXSTRINGCAST szValue
, RESERVED
,
645 &dwType
, NULL
, &dwSize
);
646 if ( m_dwLastError
== ERROR_SUCCESS
) {
647 RegString pBuf
= (RegString
)strValue
.GetWriteBuf(dwSize
);
648 m_dwLastError
= RegQueryValueEx((HKEY
) m_hKey
, WXSTRINGCAST szValue
, RESERVED
,
649 &dwType
, pBuf
, &dwSize
);
650 strValue
.UngetWriteBuf();
651 if ( m_dwLastError
== ERROR_SUCCESS
) {
652 // check that it was the right type
653 wxASSERT_MSG( dwType
== REG_SZ
,
654 _T("Type mismatch in wxRegKey::QueryValue().") );
660 // named registry values don't exist in Win16
661 wxASSERT( IsEmpty(szValue
) );
663 m_dwLastError
= RegQueryValue((HKEY
) m_hKey
, 0, strValue
.GetWriteBuf(256), &l
);
664 strValue
.UngetWriteBuf();
665 if ( m_dwLastError
== ERROR_SUCCESS
)
670 wxLogSysError(m_dwLastError
, _("can't read value of '%s'"),
671 GetFullName(this, szValue
));
675 bool wxRegKey::SetValue(const wxChar
*szValue
, const wxString
& strValue
)
677 if ( CONST_CAST
Open() ) {
678 #if defined( __WIN32__) && !defined(__TWIN32__)
679 m_dwLastError
= RegSetValueEx((HKEY
) m_hKey
, szValue
, (DWORD
) RESERVED
, REG_SZ
,
680 (RegString
)strValue
.c_str(),
682 if ( m_dwLastError
== ERROR_SUCCESS
)
685 // named registry values don't exist in Win16
686 wxASSERT( IsEmpty(szValue
) );
688 m_dwLastError
= RegSetValue((HKEY
) m_hKey
, NULL
, REG_SZ
, strValue
, NULL
);
689 if ( m_dwLastError
== ERROR_SUCCESS
)
694 wxLogSysError(m_dwLastError
, _("can't set value of '%s'"),
695 GetFullName(this, szValue
));
699 wxRegKey::operator wxString() const
702 QueryValue(NULL
, str
);
706 // ----------------------------------------------------------------------------
708 // NB: all these functions require an index variable which allows to have
709 // several concurrently running indexations on the same key
710 // ----------------------------------------------------------------------------
712 bool wxRegKey::GetFirstValue(wxString
& strValueName
, long& lIndex
)
718 return GetNextValue(strValueName
, lIndex
);
721 bool wxRegKey::GetNextValue(wxString
& strValueName
, long& lIndex
) const
723 wxASSERT( IsOpened() );
725 // are we already at the end of enumeration?
729 #if defined( __WIN32__) && !defined(__TWIN32__)
730 wxChar szValueName
[1024]; // @@ use RegQueryInfoKey...
731 DWORD dwValueLen
= WXSIZEOF(szValueName
);
733 m_dwLastError
= RegEnumValue((HKEY
) m_hKey
, lIndex
++,
734 szValueName
, &dwValueLen
,
737 NULL
, // [out] buffer for value
738 NULL
); // [i/o] it's length
740 if ( m_dwLastError
!= ERROR_SUCCESS
) {
741 if ( m_dwLastError
== ERROR_NO_MORE_ITEMS
) {
742 m_dwLastError
= ERROR_SUCCESS
;
746 wxLogSysError(m_dwLastError
, _("can't enumerate values of key '%s'"),
753 strValueName
= szValueName
;
755 // only one unnamed value
756 wxASSERT( lIndex
== 0 );
759 strValueName
.Empty();
765 bool wxRegKey::GetFirstKey(wxString
& strKeyName
, long& lIndex
)
771 return GetNextKey(strKeyName
, lIndex
);
774 bool wxRegKey::GetNextKey(wxString
& strKeyName
, long& lIndex
) const
776 wxASSERT( IsOpened() );
778 // are we already at the end of enumeration?
782 wxChar szKeyName
[_MAX_PATH
+ 1];
783 m_dwLastError
= RegEnumKey((HKEY
) m_hKey
, lIndex
++, szKeyName
, WXSIZEOF(szKeyName
));
785 if ( m_dwLastError
!= ERROR_SUCCESS
) {
786 if ( m_dwLastError
== ERROR_NO_MORE_ITEMS
) {
787 m_dwLastError
= ERROR_SUCCESS
;
791 wxLogSysError(m_dwLastError
, _("can't enumerate subkeys of key '%s'"),
798 strKeyName
= szKeyName
;
802 // returns TRUE if the value contains a number (else it's some string)
803 bool wxRegKey::IsNumericValue(const wxChar
*szValue
) const
805 ValueType type
= GetValueType(szValue
);
808 case Type_Dword_little_endian
:
809 case Type_Dword_big_endian
:
817 // ============================================================================
818 // implementation of global private functions
819 // ============================================================================
820 bool KeyExists(WXHKEY hRootKey
, const wxChar
*szKey
)
823 if ( RegOpenKey( (HKEY
) hRootKey
, szKey
, &hkeyDummy
) == ERROR_SUCCESS
) {
824 RegCloseKey(hkeyDummy
);
831 const wxChar
*GetFullName(const wxRegKey
*pKey
, const wxChar
*szValue
)
833 static wxString s_str
;
834 s_str
= pKey
->GetName();
835 if ( !wxIsEmpty(szValue
) )
836 s_str
<< _T("\\") << szValue
;
838 return s_str
.c_str();
841 void RemoveTrailingSeparator(wxString
& str
)
843 if ( !str
.IsEmpty() && str
.Last() == REG_SEPARATOR
)
844 str
.Truncate(str
.Len() - 1);