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"
31 #include "wx/config.h" // for wxExpandEnvVars
36 #define WIN32_LEAN_AND_MEAN
42 #include <stdlib.h> // for _MAX_PATH
49 #define HKEY_DEFINED // already defined in windows.h
50 #include "wx/msw/registry.h"
52 // some registry functions don't like signed chars
53 typedef unsigned char *RegString
;
55 // ----------------------------------------------------------------------------
57 // ----------------------------------------------------------------------------
59 // the standard key names, short names and handles all bundled together for
65 const char *szShortName
;
69 { HKEY_CLASSES_ROOT
, "HKEY_CLASSES_ROOT", "HKCR" },
71 { HKEY_CURRENT_USER
, "HKEY_CURRENT_USER", "HKCU" },
72 { HKEY_LOCAL_MACHINE
, "HKEY_LOCAL_MACHINE", "HKLM" },
73 { HKEY_USERS
, "HKEY_USERS", "HKU" }, // short name?
74 { HKEY_PERFORMANCE_DATA
, "HKEY_PERFORMANCE_DATA", "HKPD" },
76 { HKEY_CURRENT_CONFIG
, "HKEY_CURRENT_CONFIG", "HKCC" },
78 { HKEY_DYN_DATA
, "HKEY_DYN_DATA", "HKDD" }, // short name?
80 #endif //WINVER >= 4.0
84 // the registry name separator (perhaps one day MS will change it to '/' ;-)
85 #define REG_SEPARATOR '\\'
87 // useful for Windows programmers: makes somewhat more clear all these zeroes
88 // being passed to Windows APIs
89 #define RESERVED (NULL)
91 // ----------------------------------------------------------------------------
93 // ----------------------------------------------------------------------------
94 // @ const_cast<> is not yet supported by all compilers
95 #define CONST_CAST ((wxRegKey *)this)->
98 #define m_dwLastError CONST_CAST m_dwLastError
101 // ----------------------------------------------------------------------------
102 // non member functions
103 // ----------------------------------------------------------------------------
105 // removes the trailing backslash from the string if it has one
106 static inline void RemoveTrailingSeparator(wxString
& str
);
108 // returns TRUE if given registry key exists
109 static bool KeyExists(WXHKEY hRootKey
, const char *szKey
);
111 // combines value and key name (uses static buffer!)
112 static const char *GetFullName(const wxRegKey
*pKey
,
113 const char *szValue
= NULL
);
115 // ============================================================================
116 // implementation of wxRegKey class
117 // ============================================================================
119 // ----------------------------------------------------------------------------
120 // static functions and variables
121 // ----------------------------------------------------------------------------
123 const size_t wxRegKey::nStdKeys
= WXSIZEOF(aStdKeys
);
125 // @@ should take a `StdKey key', but as it's often going to be used in loops
126 // it would require casts in user code.
127 const char *wxRegKey::GetStdKeyName(size_t key
)
129 // return empty string if key is invalid
130 wxCHECK_MSG( key
< nStdKeys
, "", "invalid key in wxRegKey::GetStdKeyName" );
132 return aStdKeys
[key
].szName
;
135 const char *wxRegKey::GetStdKeyShortName(size_t key
)
137 // return empty string if key is invalid
138 wxCHECK( key
< nStdKeys
, "" );
140 return aStdKeys
[key
].szShortName
;
143 wxRegKey::StdKey
wxRegKey::ExtractKeyName(wxString
& strKey
)
145 wxString strRoot
= strKey
.Left(REG_SEPARATOR
);
149 for ( ui
= 0; ui
< nStdKeys
; ui
++ ) {
150 if ( strRoot
.CmpNoCase(aStdKeys
[ui
].szName
) == 0 ||
151 strRoot
.CmpNoCase(aStdKeys
[ui
].szShortName
) == 0 ) {
152 hRootKey
= aStdKeys
[ui
].hkey
;
157 if ( ui
== nStdKeys
) {
158 wxFAIL_MSG("invalid key prefix in wxRegKey::ExtractKeyName.");
160 hRootKey
= HKEY_CLASSES_ROOT
;
163 strKey
= strKey
.After(REG_SEPARATOR
);
164 if ( !strKey
.IsEmpty() && strKey
.Last() == REG_SEPARATOR
)
165 strKey
.Truncate(strKey
.Len() - 1);
168 return (wxRegKey::StdKey
)(int)hRootKey
;
171 wxRegKey::StdKey
wxRegKey::GetStdKeyFromHkey(WXHKEY hkey
)
173 for ( size_t ui
= 0; ui
< nStdKeys
; ui
++ ) {
174 if ( (int) aStdKeys
[ui
].hkey
== (int) hkey
)
178 wxFAIL_MSG("non root hkey passed to wxRegKey::GetStdKeyFromHkey.");
183 // ----------------------------------------------------------------------------
185 // ----------------------------------------------------------------------------
190 m_hRootKey
= (WXHKEY
) aStdKeys
[HKCR
].hkey
;
194 wxRegKey::wxRegKey(const wxString
& strKey
) : m_strKey(strKey
)
196 m_hRootKey
= (WXHKEY
) aStdKeys
[ExtractKeyName(m_strKey
)].hkey
;
197 m_hKey
= (WXHKEY
) NULL
;
201 // parent is a predefined (and preopened) key
202 wxRegKey::wxRegKey(StdKey keyParent
, const wxString
& strKey
) : m_strKey(strKey
)
204 RemoveTrailingSeparator(m_strKey
);
205 m_hRootKey
= (WXHKEY
) aStdKeys
[keyParent
].hkey
;
206 m_hKey
= (WXHKEY
) NULL
;
210 // parent is a normal regkey
211 wxRegKey::wxRegKey(const wxRegKey
& keyParent
, const wxString
& strKey
)
212 : m_strKey(keyParent
.m_strKey
)
214 // combine our name with parent's to get the full name
215 if ( !m_strKey
.IsEmpty() &&
216 (strKey
.IsEmpty() || strKey
[0] != REG_SEPARATOR
) ) {
217 m_strKey
+= REG_SEPARATOR
;
221 RemoveTrailingSeparator(m_strKey
);
223 m_hRootKey
= keyParent
.m_hRootKey
;
224 m_hKey
= (WXHKEY
) NULL
;
228 // dtor closes the key releasing system resource
229 wxRegKey::~wxRegKey()
234 // ----------------------------------------------------------------------------
235 // change the key name/hkey
236 // ----------------------------------------------------------------------------
238 // set the full key name
239 void wxRegKey::SetName(const wxString
& strKey
)
244 m_hRootKey
= (WXHKEY
) aStdKeys
[ExtractKeyName(m_strKey
)].hkey
;
247 // the name is relative to the parent key
248 void wxRegKey::SetName(StdKey keyParent
, const wxString
& strKey
)
253 RemoveTrailingSeparator(m_strKey
);
254 m_hRootKey
= (WXHKEY
) aStdKeys
[keyParent
].hkey
;
257 // the name is relative to the parent key
258 void wxRegKey::SetName(const wxRegKey
& keyParent
, const wxString
& strKey
)
262 // combine our name with parent's to get the full name
263 m_strKey
= keyParent
.m_strKey
;
264 if ( !strKey
.IsEmpty() && strKey
[0] != REG_SEPARATOR
)
265 m_strKey
+= REG_SEPARATOR
;
268 RemoveTrailingSeparator(m_strKey
);
270 m_hRootKey
= keyParent
.m_hRootKey
;
273 // hKey should be opened and will be closed in wxRegKey dtor
274 void wxRegKey::SetHkey(WXHKEY hKey
)
281 // ----------------------------------------------------------------------------
282 // info about the key
283 // ----------------------------------------------------------------------------
285 // returns TRUE if the key exists
286 bool wxRegKey::Exists() const
288 // opened key has to exist, try to open it if not done yet
289 return IsOpened() ? TRUE
: KeyExists(m_hRootKey
, m_strKey
);
292 // returns the full name of the key (prefix is abbreviated if bShortPrefix)
293 wxString
wxRegKey::GetName(bool bShortPrefix
) const
295 StdKey key
= GetStdKeyFromHkey((StdKey
) m_hRootKey
);
296 wxString str
= bShortPrefix
? aStdKeys
[key
].szShortName
297 : aStdKeys
[key
].szName
;
298 if ( !m_strKey
.IsEmpty() )
299 str
<< "\\" << m_strKey
;
305 bool wxRegKey::GetKeyInfo(size_t* pnSubKeys
,
308 size_t* pnMaxValueLen
) const
310 bool wxRegKey::GetKeyInfo(ulong
*pnSubKeys
,
313 ulong
*pnMaxValueLen
) const
317 m_dwLastError
= ::RegQueryInfoKey
321 NULL
, // (ptr to) size of class name buffer
323 pnSubKeys
, // [out] number of subkeys
324 pnMaxKeyLen
, // [out] max length of a subkey name
325 NULL
, // longest subkey class name
326 pnValues
, // [out] number of values
327 pnMaxValueLen
, // [out] max length of a value name
328 NULL
, // longest value data
329 NULL
, // security descriptor
330 NULL
// time of last modification
333 if ( m_dwLastError
!= ERROR_SUCCESS
) {
334 wxLogSysError(m_dwLastError
, _("can't get info about registry key '%s'"),
341 wxFAIL_MSG("GetKeyInfo() not implemented");
347 // ----------------------------------------------------------------------------
349 // ----------------------------------------------------------------------------
351 // opens key (it's not an error to call Open() on an already opened key)
352 bool wxRegKey::Open()
358 m_dwLastError
= RegOpenKey((HKEY
) m_hRootKey
, m_strKey
, &tmpKey
);
359 if ( m_dwLastError
!= ERROR_SUCCESS
) {
360 wxLogSysError(m_dwLastError
, _("can't open registry key '%s'"),
366 m_hKey
= (WXHKEY
) tmpKey
;
371 // creates key, failing if it exists and !bOkIfExists
372 bool wxRegKey::Create(bool bOkIfExists
)
374 // check for existence only if asked (i.e. order is important!)
375 if ( !bOkIfExists
&& Exists() ) {
383 m_dwLastError
= RegCreateKey((HKEY
) m_hRootKey
, m_strKey
, &tmpKey
);
384 if ( m_dwLastError
!= ERROR_SUCCESS
) {
385 wxLogSysError(m_dwLastError
, _("can't create registry key '%s'"),
391 m_hKey
= (WXHKEY
) tmpKey
;
396 // close the key, it's not an error to call it when not opened
397 bool wxRegKey::Close()
400 m_dwLastError
= RegCloseKey((HKEY
) m_hKey
);
401 if ( m_dwLastError
!= ERROR_SUCCESS
) {
402 wxLogSysError(m_dwLastError
, _("can't close registry key '%s'"),
416 // ----------------------------------------------------------------------------
417 // delete keys/values
418 // ----------------------------------------------------------------------------
419 bool wxRegKey::DeleteSelf()
424 // it already doesn't exist - ok!
429 // we can't delete keys while enumerating because it confuses GetNextKey, so
430 // we first save the key names and then delete them all
431 wxArrayString astrSubkeys
;
435 bool bCont
= GetFirstKey(strKey
, lIndex
);
437 astrSubkeys
.Add(strKey
);
439 bCont
= GetNextKey(strKey
, lIndex
);
442 size_t nKeyCount
= astrSubkeys
.Count();
443 for ( size_t nKey
= 0; nKey
< nKeyCount
; nKey
++ ) {
444 wxRegKey
key(*this, astrSubkeys
[nKey
]);
445 if ( !key
.DeleteSelf() )
449 // now delete this key itself
452 m_dwLastError
= RegDeleteKey((HKEY
) m_hRootKey
, m_strKey
);
453 if ( m_dwLastError
!= ERROR_SUCCESS
) {
454 wxLogSysError(m_dwLastError
, _("can't delete key '%s'"),
462 bool wxRegKey::DeleteKey(const char *szKey
)
467 wxRegKey
key(*this, szKey
);
468 return key
.DeleteSelf();
471 bool wxRegKey::DeleteValue(const char *szValue
)
477 m_dwLastError
= RegDeleteValue((HKEY
) m_hKey
, szValue
);
478 if ( m_dwLastError
!= ERROR_SUCCESS
) {
479 wxLogSysError(m_dwLastError
, _("can't delete value '%s' from key '%s'"),
480 szValue
, GetName().c_str());
484 // named registry values don't exist in Win16 world
485 wxASSERT( IsEmpty(szValue
) );
487 // just set the (default and unique) value of the key to ""
488 m_dwLastError
= RegSetValue((HKEY
) m_hKey
, NULL
, REG_SZ
, "", RESERVED
);
489 if ( m_dwLastError
!= ERROR_SUCCESS
) {
490 wxLogSysError(m_dwLastError
, _("can't delete value of key '%s'"),
499 // ----------------------------------------------------------------------------
500 // access to values and subkeys
501 // ----------------------------------------------------------------------------
503 // return TRUE if value exists
504 bool wxRegKey::HasValue(const char *szValue
) const
506 // this function should be silent, so suppress possible messages from Open()
510 if ( CONST_CAST
Open() ) {
511 return RegQueryValueEx((HKEY
) m_hKey
, szValue
, RESERVED
,
512 NULL
, NULL
, NULL
) == ERROR_SUCCESS
;
517 // only unnamed value exists
518 return IsEmpty(szValue
);
522 // returns TRUE if this key has any subkeys
523 bool wxRegKey::HasSubkeys() const
525 // suppress possible messages from GetFirstKey()
528 // just call GetFirstKey with dummy parameters
531 return CONST_CAST
GetFirstKey(str
, l
);
534 // returns TRUE if given subkey exists
535 bool wxRegKey::HasSubKey(const char *szKey
) const
537 // this function should be silent, so suppress possible messages from Open()
540 if ( CONST_CAST
Open() )
541 return KeyExists(m_hKey
, szKey
);
546 wxRegKey::ValueType
wxRegKey::GetValueType(const char *szValue
)
553 m_dwLastError
= RegQueryValueEx((HKEY
) m_hKey
, szValue
, RESERVED
,
554 &dwType
, NULL
, NULL
);
555 if ( m_dwLastError
!= ERROR_SUCCESS
) {
556 wxLogSysError(m_dwLastError
, _("can't read value of key '%s'"),
561 return (ValueType
)dwType
;
563 return IsEmpty(szValue
) ? Type_String
: Type_None
;
568 bool wxRegKey::SetValue(const char *szValue
, long lValue
)
570 if ( CONST_CAST
Open() ) {
571 m_dwLastError
= RegSetValueEx((HKEY
) m_hKey
, szValue
, (DWORD
) RESERVED
, REG_DWORD
,
572 (RegString
)&lValue
, sizeof(lValue
));
573 if ( m_dwLastError
== ERROR_SUCCESS
)
577 wxLogSysError(m_dwLastError
, _("can't set value of '%s'"),
578 GetFullName(this, szValue
));
582 bool wxRegKey::QueryValue(const char *szValue
, long *plValue
) const
584 if ( CONST_CAST
Open() ) {
585 DWORD dwType
, dwSize
= sizeof(DWORD
);
586 RegString pBuf
= (RegString
)plValue
;
587 m_dwLastError
= RegQueryValueEx((HKEY
) m_hKey
, szValue
, RESERVED
,
588 &dwType
, pBuf
, &dwSize
);
589 if ( m_dwLastError
!= ERROR_SUCCESS
) {
590 wxLogSysError(m_dwLastError
, _("can't read value of key '%s'"),
595 // check that we read the value of right type
596 wxASSERT_MSG( dwType
== REG_DWORD
,
597 "Type mismatch in wxRegKey::QueryValue()." );
608 bool wxRegKey::QueryValue(const char *szValue
, wxString
& strValue
) const
610 if ( CONST_CAST
Open() ) {
612 // first get the type and size of the data
613 DWORD dwType
, dwSize
;
614 m_dwLastError
= RegQueryValueEx((HKEY
) m_hKey
, szValue
, RESERVED
,
615 &dwType
, NULL
, &dwSize
);
616 if ( m_dwLastError
== ERROR_SUCCESS
) {
617 RegString pBuf
= (RegString
)strValue
.GetWriteBuf(dwSize
);
618 m_dwLastError
= RegQueryValueEx((HKEY
) m_hKey
, szValue
, RESERVED
,
619 &dwType
, pBuf
, &dwSize
);
620 strValue
.UngetWriteBuf();
621 if ( m_dwLastError
== ERROR_SUCCESS
) {
622 // check that it was the right type
623 wxASSERT_MSG( dwType
== REG_SZ
,
624 "Type mismatch in wxRegKey::QueryValue()." );
630 // named registry values don't exist in Win16
631 wxASSERT( IsEmpty(szValue
) );
633 m_dwLastError
= RegQueryValue((HKEY
) m_hKey
, 0, strValue
.GetWriteBuf(256), &l
);
634 strValue
.UngetWriteBuf();
635 if ( m_dwLastError
== ERROR_SUCCESS
)
640 wxLogSysError(m_dwLastError
, _("can't read value of '%s'"),
641 GetFullName(this, szValue
));
645 bool wxRegKey::SetValue(const char *szValue
, const wxString
& strValue
)
647 if ( CONST_CAST
Open() ) {
649 m_dwLastError
= RegSetValueEx((HKEY
) m_hKey
, szValue
, (DWORD
) RESERVED
, REG_SZ
,
650 (RegString
)strValue
.c_str(),
652 if ( m_dwLastError
== ERROR_SUCCESS
)
655 // named registry values don't exist in Win16
656 wxASSERT( IsEmpty(szValue
) );
658 m_dwLastError
= RegSetValue((HKEY
) m_hKey
, NULL
, REG_SZ
, strValue
, NULL
);
659 if ( m_dwLastError
== ERROR_SUCCESS
)
664 wxLogSysError(m_dwLastError
, _("can't set value of '%s'"),
665 GetFullName(this, szValue
));
669 wxRegKey::operator wxString() const
672 QueryValue(NULL
, str
);
676 // ----------------------------------------------------------------------------
678 // NB: all these functions require an index variable which allows to have
679 // several concurrently running indexations on the same key
680 // ----------------------------------------------------------------------------
682 bool wxRegKey::GetFirstValue(wxString
& strValueName
, long& lIndex
)
688 return GetNextValue(strValueName
, lIndex
);
691 bool wxRegKey::GetNextValue(wxString
& strValueName
, long& lIndex
) const
693 wxASSERT( IsOpened() );
695 // are we already at the end of enumeration?
700 char szValueName
[1024]; // @@ use RegQueryInfoKey...
701 DWORD dwValueLen
= WXSIZEOF(szValueName
);
704 m_dwLastError
= RegEnumValue((HKEY
) m_hKey
, lIndex
,
705 szValueName
, &dwValueLen
,
708 NULL
, // [out] buffer for value
709 NULL
); // [i/o] it's length
711 if ( m_dwLastError
!= ERROR_SUCCESS
) {
712 if ( m_dwLastError
== ERROR_NO_MORE_ITEMS
) {
713 m_dwLastError
= ERROR_SUCCESS
;
717 wxLogSysError(m_dwLastError
, _("can't enumerate values of key '%s'"),
724 strValueName
= szValueName
;
726 // only one unnamed value
727 wxASSERT( lIndex
== 0 );
730 strValueName
.Empty();
736 bool wxRegKey::GetFirstKey(wxString
& strKeyName
, long& lIndex
)
742 return GetNextKey(strKeyName
, lIndex
);
745 bool wxRegKey::GetNextKey(wxString
& strKeyName
, long& lIndex
) const
747 wxASSERT( IsOpened() );
749 // are we already at the end of enumeration?
753 char szKeyName
[_MAX_PATH
+ 1];
754 m_dwLastError
= RegEnumKey((HKEY
) m_hKey
, lIndex
++, szKeyName
, WXSIZEOF(szKeyName
));
756 if ( m_dwLastError
!= ERROR_SUCCESS
) {
757 if ( m_dwLastError
== ERROR_NO_MORE_ITEMS
) {
758 m_dwLastError
= ERROR_SUCCESS
;
762 wxLogSysError(m_dwLastError
, _("can't enumerate subkeys of key '%s'"),
769 strKeyName
= szKeyName
;
773 // ============================================================================
774 // implementation of global private functions
775 // ============================================================================
776 bool KeyExists(WXHKEY hRootKey
, const char *szKey
)
779 if ( RegOpenKey( (HKEY
) hRootKey
, szKey
, &hkeyDummy
) == ERROR_SUCCESS
) {
780 RegCloseKey(hkeyDummy
);
787 const char *GetFullName(const wxRegKey
*pKey
, const char *szValue
)
789 static wxString s_str
;
790 s_str
= pKey
->GetName();
791 if ( !IsEmpty(szValue
) )
792 s_str
<< "\\" << szValue
;
794 return s_str
.c_str();
797 void RemoveTrailingSeparator(wxString
& str
)
799 if ( !str
.IsEmpty() && str
.Last() == REG_SEPARATOR
)
800 str
.Truncate(str
.Len() - 1);